From e20d285edb7e9aa9a85350d0abc5acda5e447569 Mon Sep 17 00:00:00 2001
From: Matthew Farrellee <matt@cs.wisc.edu>
Date: Fri, 26 Sep 2025 16:13:06 -0400
Subject: [PATCH] fix: ensure ModelRegistryHelper init for together and
 fireworks

---
 llama_stack/providers/remote/inference/fireworks/fireworks.py | 1 +
 llama_stack/providers/remote/inference/together/together.py   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/llama_stack/providers/remote/inference/fireworks/fireworks.py b/llama_stack/providers/remote/inference/fireworks/fireworks.py
index cf7e93974..1025bfb53 100644
--- a/llama_stack/providers/remote/inference/fireworks/fireworks.py
+++ b/llama_stack/providers/remote/inference/fireworks/fireworks.py
@@ -64,6 +64,7 @@ class FireworksInferenceAdapter(OpenAIMixin, ModelRegistryHelper, Inference, Nee
     }
 
     def __init__(self, config: FireworksImplConfig) -> None:
+        ModelRegistryHelper.__init__(self)
         self.config = config
         self.allowed_models = config.allowed_models
 
diff --git a/llama_stack/providers/remote/inference/together/together.py b/llama_stack/providers/remote/inference/together/together.py
index 37973d635..c199677be 100644
--- a/llama_stack/providers/remote/inference/together/together.py
+++ b/llama_stack/providers/remote/inference/together/together.py
@@ -70,6 +70,7 @@ class TogetherInferenceAdapter(OpenAIMixin, ModelRegistryHelper, Inference, Need
     }
 
     def __init__(self, config: TogetherImplConfig) -> None:
+        ModelRegistryHelper.__init__(self)
         self.config = config
         self.allowed_models = config.allowed_models
         self._model_cache: dict[str, Model] = {}