fix(vertex_ai.py): correctly parse optional params and pass vertex ai project

This commit is contained in:
Krrish Dholakia 2024-03-06 14:00:20 -08:00
parent 4e395bf244
commit cdb960eb34
5 changed files with 78 additions and 25 deletions

View file

@ -2617,6 +2617,7 @@ def embedding(
vertex_project=vertex_ai_project,
vertex_location=vertex_ai_location,
aembedding=aembedding,
print_verbose=print_verbose,
)
elif custom_llm_provider == "oobabooga":
response = oobabooga.embedding(