chore: misc fixes

julep-ai · Jan 8, 2025 · 5d77d81 · 5d77d81
1 parent 253c732
commit 5d77d81
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/agents-api/agents_api/clients/litellm.py b/agents-api/agents_api/clients/litellm.py
@@ -78,9 +78,8 @@ async def aembedding(
     custom_api_key: str | None = None,
     **settings,
 ) -> list[list[float]]:
-    # Temporarily commented out (causes errors when using voyage/voyage-3)
-    # if not custom_api_key:
-    # model = f"openai/{model}"  # FIXME: Is this still needed for litellm?
+    if not custom_api_key:
+        model = f"openai/{model}"  # This is needed for litellm
 
     input = (
         [inputs]