wehub-resource-sync
39 lines
1.9 KiB
Python
39 lines
1.9 KiB
Python
# Copyright (c) ModelScope Contributors. All rights reserved.
|
|
import json
|
|
from typing import List, Optional, Union
|
|
|
|
from .base import BaseAgentTemplate
|
|
|
|
|
|
class ToolBenchAgentTemplate(BaseAgentTemplate):
|
|
|
|
def _format_tools(self, tools: List[Union[str, dict]], system: Optional[str] = None, 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}"""
|