diff --git a/litellm/llms/ollama_chat.py b/litellm/llms/ollama_chat.py index a7439bbcc09..2428e418d48 100644 --- a/litellm/llms/ollama_chat.py +++ b/litellm/llms/ollama_chat.py @@ -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", }