diff --git a/llama_stack/providers/registry/inference.py b/llama_stack/providers/registry/inference.py index 440d475fe..54d55e60e 100644 --- a/llama_stack/providers/registry/inference.py +++ b/llama_stack/providers/registry/inference.py @@ -115,6 +115,7 @@ def available_providers() -> List[ProviderSpec]: ], module="llama_stack.providers.remote.inference.fireworks", config_class="llama_stack.providers.remote.inference.fireworks.FireworksImplConfig", + provider_data_validator="llama_stack.providers.remote.inference.fireworks.FireworksProviderDataValidator", ), ), remote_provider_spec( diff --git a/llama_stack/providers/remote/inference/fireworks/__init__.py b/llama_stack/providers/remote/inference/fireworks/__init__.py index a3f5a0bd4..8ae10e8a7 100644 --- a/llama_stack/providers/remote/inference/fireworks/__init__.py +++ b/llama_stack/providers/remote/inference/fireworks/__init__.py @@ -4,9 +4,15 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. +from pydantic import BaseModel + from .config import FireworksImplConfig +class FireworksProviderDataValidator(BaseModel): + fireworks_api_key: str + + async def get_adapter_impl(config: FireworksImplConfig, _deps): from .fireworks import FireworksInferenceAdapter