From e2af13550a29db7ea0757a313c3590105c925a74 Mon Sep 17 00:00:00 2001 From: Islam Nofl Date: Mon, 24 Jun 2024 08:01:15 +0300 Subject: [PATCH] Rename ollama prompt 'Function' word to 'Name' --- litellm/llms/prompt_templates/factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/prompt_templates/factory.py b/litellm/llms/prompt_templates/factory.py index 02ed93fae37..109c5b8d835 100644 --- a/litellm/llms/prompt_templates/factory.py +++ b/litellm/llms/prompt_templates/factory.py @@ -181,7 +181,7 @@ def ollama_pt( for call in message["tool_calls"]: function_name = call["function"]["name"] arguments = json.loads(call["function"]["arguments"]) - prompt += f"### Tool Call ({call["id"]}):\nFunction: {function_name}\nArguments: {json.dumps(arguments)}\n\n" + prompt += f"### Tool Call ({call["id"]}):\nName: {function_name}\nArguments: {json.dumps(arguments)}\n\n" elif "tool_call_id" in message: prompt += f"### Tool Call Result ({message["tool_call_id"]}):\n{message["content"]}\n\n" elif content: