|
|
|
|
|
from typing import List, Union |
|
|
|
|
|
import json |
|
|
|
|
|
from .base import BaseAgentTemplate |
|
|
|
|
|
|
|
|
class ToolBenchAgentTemplate(BaseAgentTemplate): |
|
|
|
|
|
def _format_tools(self, tools: List[Union[str, dict]], system: str, user_message=None) -> str: |
|
|
for i, tool in enumerate(tools): |
|
|
tools[i] = self.unwrap_tool(tool) |
|
|
tools = json.dumps(tools, ensure_ascii=False) |
|
|
return f"""You can use many tools(functions) to do the following task. |
|
|
First I will give you the task description, and your task start. |
|
|
At each step, you need to give your thought to analyze the status now and what to do next, \ |
|
|
with a function call to actually execute your step. Your output should follow this format: |
|
|
Thought: |
|
|
Action: |
|
|
Action Input: |
|
|
|
|
|
After the call, you will get the call result, and you are now in a new state. |
|
|
Then you will analyze your status now, then decide what to do next... |
|
|
After many (Thought-call) pairs, you finally perform the task, then you can give your final answer. |
|
|
Remember: |
|
|
1.the state change is irreversible, you can't go back to one of the former state, if you want to restart the task, \ |
|
|
say \"I give up and restart\". |
|
|
2.All the thought is short, at most in 5 sentence. |
|
|
3.You can do more then one try, so if your plan is to continuously try some conditions, \ |
|
|
you can do one of the conditions per try. |
|
|
Let's Begin! |
|
|
Task description: You should use functions to help handle the real time user queries. Remember: |
|
|
1.ALWAYS call \"Finish\" function at the end of the task. And the final answer should contain enough information \ |
|
|
to show to the user,If you can't handle the task, \ |
|
|
or you find that function calls always fail(the function is not valid now), \ |
|
|
use function Finish->give_up_and_restart. |
|
|
2.Do not use origin tool names, use only subfunctions' names. |
|
|
Specifically, you have access to the following APIs: {tools}""" |
|
|
|
