fix(vertex_httpx.py): support async completion calls

This commit is contained in:
Krrish Dholakia 2024-06-12 20:15:03 -07:00
parent 1dac2aa59f
commit afebf867f6
3 changed files with 142 additions and 22 deletions

View file

@ -329,6 +329,7 @@ async def acompletion(
or custom_llm_provider == "ollama_chat"
or custom_llm_provider == "replicate"
or custom_llm_provider == "vertex_ai"
or custom_llm_provider == "vertex_ai_beta"
or custom_llm_provider == "gemini"
or custom_llm_provider == "sagemaker"
or custom_llm_provider == "anthropic"