phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 20:14:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

function_tag system prompt is also added as a user message

Browse source
This commit is contained in:
Hardik Shah 2024-08-15 12:11:35 -07:00
parent 8e526b9536
commit 00053b5bb0
2 changed files with 15 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_toolchain/agentic_system/meta_reference/system_prompt.py
View file

@ -48,15 +48,13 @@ def get_agentic_prefix_messages(
Cutting Knowledge Date: December 2023
Today Date: {formatted_date}\n"""
content += date_str
messages.append(SystemMessage(content=content))
if custom_tools:
if tool_prompt_format == ToolPromptFormat.function_tag:
custom_message = prompt_for_function_tag(custom_tools)
content += custom_message
messages.append(SystemMessage(content=content))
text = prompt_for_function_tag(custom_tools)
messages.append(UserMessage(content=text))
elif tool_prompt_format == ToolPromptFormat.json:
messages.append(SystemMessage(content=content))
# json is added as a user prompt
text = prompt_for_json(custom_tools)
messages.append(UserMessage(content=text))
else: