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] = {}