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

30
llama_toolchain/agentic_system/client.py
View file

@ -104,34 +104,25 @@ async def run_main(host: str, port: int):
AgenticSystemToolDefinition(
tool_name=BuiltinTool.wolfram_alpha,
),
AgenticSystemToolDefinition(
tool_name=BuiltinTool.photogen,
),
AgenticSystemToolDefinition(
tool_name=BuiltinTool.code_interpreter,
),
]
tool_definitions += [
AgenticSystemToolDefinition(
tool_name="custom_tool",
description="a custom tool",
tool_name="get_boiling_point",
description="Get the boiling point of a imaginary liquids (eg. polyjuice)",
parameters={
"param1": ToolParamDefinition(
"liquid_name": ToolParamDefinition(
param_type="str",
description="a string parameter",
description="The name of the liquid",
required=True,
)
},
),
AgenticSystemToolDefinition(
tool_name="custom_tool_2",
description="a second custom tool",
parameters={
"param2": ToolParamDefinition(
),
"celcius": ToolParamDefinition(
param_type="str",
description="a string parameter",
required=True,
)
description="Whether to return the boiling point in Celcius",
required=False,
),
},
),
]
@ -163,7 +154,10 @@ async def run_main(host: str, port: int):
user_prompts = [
"Who are you?",
"what is the 100th prime number?",
"Search web for who was 44th President of USA?",
"Write code to check if a number is prime. Use that to check if 7 is prime",
"What is the boiling point of polyjuicepotion ?",
]
for content in user_prompts:
cprint(f"User> {content}", color="blue")