diff --git a/litellm/llms/ollama_chat.py b/litellm/llms/ollama_chat.py index af6fd5b8060..bb053f5e86d 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", }