From b8a8b0847c9e68726e56674add4aaa2951482a99 Mon Sep 17 00:00:00 2001 From: corrm Date: Mon, 24 Jun 2024 05:56:56 +0300 Subject: [PATCH] Added improved function name handling in ollama_async_streaming --- litellm/llms/ollama_chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/ollama_chat.py b/litellm/llms/ollama_chat.py index a7439bbcc09..af6fd5b8060 100644 --- a/litellm/llms/ollama_chat.py +++ b/litellm/llms/ollama_chat.py @@ -434,7 +434,7 @@ async def ollama_async_streaming( { "id": f"call_{str(uuid.uuid4())}", "function": { - "name": function_call["name"], + "name": function_call.get("name", function_call.get("function", None)), "arguments": json.dumps(function_call["arguments"]), }, "type": "function",