diff --git a/llama_stack/providers/registry/inference.py b/llama_stack/providers/registry/inference.py index 8a3619118..18fe8274e 100644 --- a/llama_stack/providers/registry/inference.py +++ b/llama_stack/providers/registry/inference.py @@ -117,7 +117,7 @@ def available_providers() -> List[ProviderSpec]: ], module="llama_stack.providers.remote.inference.together", config_class="llama_stack.providers.remote.inference.together.TogetherImplConfig", - provider_data_validator="llama_stack.providers.remote.safety.together.TogetherProviderDataValidator", + provider_data_validator="llama_stack.providers.remote.inference.together.TogetherProviderDataValidator", ), ), remote_provider_spec( diff --git a/llama_stack/providers/remote/inference/together/__init__.py b/llama_stack/providers/remote/inference/together/__init__.py index 05ea91e58..2bbd9ed53 100644 --- a/llama_stack/providers/remote/inference/together/__init__.py +++ b/llama_stack/providers/remote/inference/together/__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 TogetherImplConfig +class TogetherProviderDataValidator(BaseModel): + together_api_key: str + + async def get_adapter_impl(config: TogetherImplConfig, _deps): from .together import TogetherInferenceAdapter