apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update extract_tools_from_system_prompt.py

Browse source
This commit is contained in:
Shuguang Chen 2025-03-21 09:53:28 -07:00
parent 581aebf79b
commit 665b9f411d
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/rest/utils/extract_tools_from_system_prompt.py
View file

@ -7,7 +7,6 @@ def extract_tools(system_prompt):
if l != -1 and r != -1:
tool_content = system_prompt[l + len("<tools>") : r]
print(tool_content.split("\n"))
tools = [json.loads(tool) for tool in tool_content.split("\n") if tool]
return tools
else:
@ -15,7 +14,7 @@ def extract_tools(system_prompt):
if __name__ == "__main__":
system_prompt = 'You are a helpful assistant designed to assist with the user query by making one or more function calls if needed.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>\n{"name": "verify_address", "description": "Verify the address", "parameters": {"type": "object", "property_address": {"type": "str", "description": "Complete address of the property", "format": "Street address, City, State Country"}, "required": ["property_address"]}}\n{"name": "track_shipment", "description": "Track the shipment", "parameters": {"type": "object", "properties": {"tracking_number": {"type": "str", "description": "The tracking number"}}, "required": ["tracking_number"]}}\n{"name": "generate_label", "description": "Generate the shipping label", "parameters": {"type": "object", "properties": {"sender": {"type": "str", "description": "The address to ship from"}, "recipient": {"type": "str", "description": "The address to ship to"}, "weight": {"type": "int", "description": "The weight of the package"}}, "required": ["sender", "recipient", "weight"]}}\n{"name": "calculate_shipping_rate", "description": "Calculate the shipping rate", "parameters": {"type": "object", "properties": {"sender": {"type": "str", "description": "The address to ship from"}, "recipient": {"type": "str", "description": "The address to ship to"}, "weight": {"type": "int", "description": "The weight of the package"}}, "required": ["sender", "recipient", "weight"]}}\n</tools>\n\nYour task is to decide which functions are needed and collect missing parameters if necessary.\n\nBased on your analysis, provide your response in one of the following JSON formats:\n1. If no functions are needed:\n```\n{"response": "Your response text here"}\n```\n2. If functions are needed but some required parameters are missing:\n```\n{"required_functions": ["func_name1", "func_name2", ...], "clarification": "Text asking for missing parameters"}\n```\n3. If functions are needed and all required parameters are available:\n```\n{"tool_calls": [{"name": "func_name1", "arguments": {"argument1": "value1", "argument2": "value2"}},... (more tool calls as required)]}\n```'
system_prompt = 'You are a helpful assistant designed to assist with the user query by making one or more function calls if needed.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>\n{"id": "get_new_releases", "type": "function", "function": {"name": "get_new_releases", "description": "Get a list of new album releases featured in Spotify (shown, for example, on a Spotify player\\u2019s \'Browse\' tab).", "parameters": {"type": "object", "properties": {"country": {"type": "str", "description": "The country where the album is released", "in_path": true}, "limit": {"type": "integer", "description": "The maximum number of results to return", "default": 5}}, "required": ["country"]}}}\n{"id": "search_for_item", "type": "function", "function": {"name": "search_for_item", "description": "Get information about albums, artists, playlists, tracks, shows, episodes, or audiobooks. You can search for an item by its name, creator, or topic.", "parameters": {"type": "object", "properties": {"q": {"type": "str", "description": "Your search query, which can include keywords related to the item name, its creator, or its topic."}, "type": {"type": "str", "description": "The type of the item to search for (e.g., album, artist, playlist, track, show, episode, audiobook).", "enum": ["album", "artist", "playlist", "track", "show", "episode", "audiobook"]}, "market": {"type": "str", "description": "A country code", "default": "US"}, "limit": {"type": "integer", "description": "The maximum number of results to return", "default": 5}}, "required": ["q", "type"]}}}\n</tools>\n\nYour task is to decide which functions are needed and collect missing parameters if necessary.\n\nBased on your analysis, provide your response in one of the following JSON formats:\n1. If no functions are needed:\n```\n{"response": "Your response text here"}\n```\n2. If functions are needed but some required parameters are missing:\n```\n{"required_functions": ["func_name1", "func_name2", ...], "clarification": "Text asking for missing parameters"}\n```\n3. If functions are needed and all required parameters are available:\n```\n{"tool_calls": [{"name": "func_name1", "arguments": {"argument1": "value1", "argument2": "value2"}},... (more tool calls as required)]}\n```'
tools = extract_tools(system_prompt)
print(json.dumps(tools, indent=4))