Skip to content

Commit

Permalink
Fix embed instruction in embed endpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
HamadaSalhab committed Oct 22, 2024
1 parent 66c4361 commit 014c5cb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion agents-api/agents_api/routers/docs/embed.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,6 @@ async def embed(
[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_to_embed)

return EmbedQueryResponse(vectors=vectors)

0 comments on commit 014c5cb

Please sign in to comment.