From d5f6e3ac0898bc62afd03aacbb51feb4f6a5839c Mon Sep 17 00:00:00 2001
From: Edwin Jose George <52248484+edwinjosegeorge@users.noreply.github.com>
Date: Sun, 9 Jun 2024 16:37:58 +0930
Subject: [PATCH] refactor: black

---
 litellm/llms/ollama_chat.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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",
                         }