scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

Rename ollama prompt 'Function' word to 'Name'

Browse source
This commit is contained in:
Islam Nofl 2024-06-24 08:01:15 +03:00 • committed by Ishaan Jaff
parent a7efb9c332
commit cc261580ac
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/prompt_templates/factory.py
View file

@ -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: