diff --git a/llama_stack/providers/adapters/inference/together/__init__.py b/llama_stack/providers/adapters/inference/together/__init__.py index c964ddffb..05ea91e58 100644 --- a/llama_stack/providers/adapters/inference/together/__init__.py +++ b/llama_stack/providers/adapters/inference/together/__init__.py @@ -4,7 +4,7 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. -from .config import TogetherImplConfig, TogetherHeaderExtractor +from .config import TogetherImplConfig async def get_adapter_impl(config: TogetherImplConfig, _deps): diff --git a/llama_stack/providers/adapters/inference/together/config.py b/llama_stack/providers/adapters/inference/together/config.py index 378e85522..03ee047d2 100644 --- a/llama_stack/providers/adapters/inference/together/config.py +++ b/llama_stack/providers/adapters/inference/together/config.py @@ -8,10 +8,6 @@ from llama_models.schema_utils import json_schema_type from pydantic import BaseModel, Field -class TogetherHeaderExtractor(BaseModel): - together_api_key: str - - @json_schema_type class TogetherImplConfig(BaseModel): url: str = Field( diff --git a/llama_stack/providers/adapters/safety/together/together.py b/llama_stack/providers/adapters/safety/together/together.py index 6c243a724..940d02861 100644 --- a/llama_stack/providers/adapters/safety/together/together.py +++ b/llama_stack/providers/adapters/safety/together/together.py @@ -19,6 +19,7 @@ from .config import TogetherSafetyConfig SAFETY_SHIELD_TYPES = { "Llama-Guard-3-8B": "meta-llama/Meta-Llama-Guard-3-8B", + "Llama-Guard-3-11B-Vision": "meta-llama/Llama-Guard-3-11B-Vision-Turbo", }