diff --git a/llama_stack/providers/utils/inference/openai_mixin.py b/llama_stack/providers/utils/inference/openai_mixin.py index c57f65bca..2fe343f63 100644 --- a/llama_stack/providers/utils/inference/openai_mixin.py +++ b/llama_stack/providers/utils/inference/openai_mixin.py @@ -296,7 +296,7 @@ class OpenAIMixin(ABC): return OpenAIEmbeddingsResponse( data=data, - model=response.model, + model=model, usage=usage, ) diff --git a/tests/integration/suites.py b/tests/integration/suites.py index fb2c44308..354dc1f4c 100644 --- a/tests/integration/suites.py +++ b/tests/integration/suites.py @@ -87,7 +87,7 @@ SETUP_DEFINITIONS: dict[str, Setup] = { description="OpenAI GPT models for high-quality responses and tool calling", defaults={ "text_model": "openai/gpt-4o", - "embedding_model": "sentence-transformers/all-MiniLM-L6-v2", + "embedding_model": "openai/text-embedding-3-small", }, ), "tgi": Setup(