From cb77426fb5704da2060dadb997342265689a9262 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Tue, 12 Nov 2024 12:15:55 -0500 Subject: [PATCH] fix fireworks (#427) --- llama_stack/providers/registry/inference.py | 1 + .../providers/remote/inference/fireworks/__init__.py | 6 ++++++ 2 files changed, 7 insertions(+) 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