scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

Merge pull request #4089 from paneru-rajan/ollama-func-calls

Browse source 
Fix: Output Structure of Ollama chat
This commit is contained in:
Krish Dholakia 2024-07-03 08:57:31 -07:00 • committed by GitHub
commit 2116dbcdc1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/ollama_chat.py
View file

@ -501,8 +501,10 @@ async def ollama_acompletion(
{
"id": f"call_{str(uuid.uuid4())}",
"function": {
"name": function_call["name"],
"arguments": json.dumps(function_call["arguments"]),
"name": function_call.get("name", function_name),
"arguments": json.dumps(
function_call.get("arguments", function_call)
),
},
"type": "function",
}