Fix embed instruction in embed endpoint

julep-ai · Oct 22, 2024 · b68ab1d · b68ab1d
1 parent 4290875
commit b68ab1d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/agents-api/agents_api/routers/docs/embed.py b/agents-api/agents_api/routers/docs/embed.py
@@ -21,7 +21,8 @@ async def embed(
     text_to_embed: list[str] = (
         [text_to_embed] if isinstance(text_to_embed, str) else text_to_embed
     )
-
-    vectors = await litellm.aembedding(inputs=[data.embed_instruction] + text_to_embed)
+    vectors = await litellm.aembedding(
+        inputs=[data.embed_instruction + text for text in text_to_embed]
+    )
 
     return EmbedQueryResponse(vectors=vectors)