Fix text embedding

ls1intum · Feb 28, 2025 · 9ac4624 · 9ac4624
1 parent 8624512
commit 9ac4624
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/app/llm/external/openai_embeddings.py b/app/llm/external/openai_embeddings.py
@@ -27,12 +27,7 @@ def embed(self, text: str) -> list[float]:
 
         for attempt in range(retries):
             try:
-                response = self._client.embeddings.create(
-                    model=self.model,
-                    input=text,
-                    encoding_format="float",
-                )
-                return response.data[0].embedding
+                return self._client.embed_query(text)
             except (
                 APIError,
                 APITimeoutError,