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

Merge pull request #1527 from TheDiscoMole/litellm_non_openai_tool_call_prompt

Browse source 
improving non-openai tool_call prompt
This commit is contained in:
Krish Dholakia 2024-03-18 18:29:42 -07:00 • committed by GitHub
commit 5edb703d78
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -957,7 +957,7 @@ def azure_text_pt(messages: list):
# Function call template
def function_call_prompt(messages: list, functions: list):
function_prompt = (
"Produce JSON OUTPUT ONLY! The following functions are available to you:"
"""Produce JSON OUTPUT ONLY! Adhere to this format {"name": "function_name", "arguments":{"argument_name": "argument_value"}} The following functions are available to you:"""
)
for function in functions:
function_prompt += f"""\n{function}\n"""
@ -965,7 +965,7 @@ def function_call_prompt(messages: list, functions: list):
function_added_to_prompt = False
for message in messages:
if "system" in message["role"]:
message["content"] += f"""{function_prompt}"""
message["content"] += f""" {function_prompt}"""
function_added_to_prompt = True
if function_added_to_prompt == False: