scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix: code quality on ollama chat

Browse source
This commit is contained in:
Ishaan Jaff 2025-05-12 16:47:55 -07:00
parent 2cc4a87861
commit 405b0c1ed7
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/ollama_chat.py
View file

@ -1,4 +1,3 @@
import inspect
import json
import time
import uuid
@ -208,9 +207,9 @@ class OllamaChatConfig(OpenAIGPTConfig):
litellm.add_function_to_prompt = (
True # so that main.py adds the function call to the prompt
)
optional_params[
"functions_unsupported_model"
] = non_default_params.get("functions")
optional_params["functions_unsupported_model"] = (
non_default_params.get("functions")
)
non_default_params.pop("tool_choice", None) # causes ollama requests to hang
non_default_params.pop("functions", None) # causes ollama requests to hang
return optional_params