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

Fix: Output Structure of Ollama chat
2024-07-03 08:57:31 -07:00 · 2024-07-03 08:57:31 -07:00 · 2116dbcdc1
commit 2116dbcdc1
parent 67433a04a2 d5f6e3ac08
1 changed files with 4 additions and 2 deletions
--- 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",
                        }