phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) support ollama_chat for acompletion

Browse source
This commit is contained in:
ishaan-jaff 2023-12-26 19:56:32 +05:30
parent 43a345ca13
commit 751d57379d
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/main.py
View file

@ -196,6 +196,7 @@ async def acompletion(*args, **kwargs):
or custom_llm_provider == "text-completion-openai"
or custom_llm_provider == "huggingface"
or custom_llm_provider == "ollama"
or custom_llm_provider == "ollama_chat"
or custom_llm_provider == "vertex_ai"
): # currently implemented aiohttp calls for just azure and openai, soon all.
if kwargs.get("stream", False):