phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(ollama_chat.py): explicitly state if ollama call is streaming or not

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-06 07:43:47 -08:00
parent 9b2a2e6c8b
commit 50fb54883b
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/ollama_chat.py
View file

@ -146,7 +146,12 @@ def get_ollama_response(
optional_params[k] = v
stream = optional_params.pop("stream", False)
data = {"model": model, "messages": messages, "options": optional_params}
data = {
"model": model,
"messages": messages,
"options": optional_params,
"stream": stream,
}
## LOGGING
logging_obj.pre_call(
input=None,