scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

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 3db9830d4b
commit 80eb8d0eae
1 changed files with 6 additions and 1 deletions
Show stats 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,