diff --git a/src/embedding_client.py b/src/embedding_client.py index e163cc66..d6c8b46d 100644 --- a/src/embedding_client.py +++ b/src/embedding_client.py @@ -99,7 +99,7 @@ class _EmbeddingClient: return self._validate_embedding_dimensions(response.embeddings[0].values) else: # openai response = await self.client.embeddings.create( - model=self.model, input=query + model=self.model, input=[query] ) return self._validate_embedding_dimensions(response.data[0].embedding) diff --git a/tests/llm/test_embedding_client.py b/tests/llm/test_embedding_client.py index 463b3dd6..14def176 100644 --- a/tests/llm/test_embedding_client.py +++ b/tests/llm/test_embedding_client.py @@ -51,7 +51,7 @@ async def test_openai_embedding_client_uses_configured_model_and_dimensions( assert embedding == [0.1] * 8 assert fake_embeddings.calls == [ - {"model": "text-embedding-3-small", "input": "hello world"} + {"model": "text-embedding-3-small", "input": ["hello world"]} ]