Skip to content

Commit

Permalink
Fix embedding model litellm error
Browse files Browse the repository at this point in the history
  • Loading branch information
HamadaSalhab committed Nov 22, 2024
1 parent 26a3315 commit 3d429be
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions agents-api/agents_api/clients/litellm.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,8 +58,9 @@ async def aembedding(
custom_api_key: None | str = None,
**settings,
) -> list[list[float]]:
if not custom_api_key:
model = f"openai/{model}" # FIXME: This is for litellm
# Temporarily commented out (causes errors when using voyage/voyage-3)
# if not custom_api_key:
# model = f"openai/{model}" # FIXME: This is for litellm

if isinstance(inputs, str):
input = [inputs]
Expand Down

0 comments on commit 3d429be

Please sign in to comment.