From 1c3e3f8816eb2bd4657af6f90c4dbf47923b2bd6 Mon Sep 17 00:00:00 2001 From: qxxaa Date: Mon, 20 Apr 2026 21:35:13 +0100 Subject: [PATCH] fix: embed() sends string input instead of array, breaking OpenAI-compatible providers (#586) * fix: wrap single embed() input in array for OpenAI-compatible provider compatibility * Fix input format in embedding test assertion --- src/embedding_client.py | 2 +- tests/llm/test_embedding_client.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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"]} ]