春江花月夜

1
# 工具列表
2
tools = [read_file, write_to_file, run_terminal_command]
3

4
# 返回 agent 对象（工具列表，模型，存放项目位置）
5
agent = ReActAgent(tools=tools, model="openai/gpt-4o", project_directory=project_dir)
6

7
task = input("请输入任务：")
8
res = agent.run(task)

1
class ReActAgent:
2
    def __init__(self, tools: List[Callable], model: str, project_directory: str):
3
        self.tools = { func.__name__: func for func in tools }
4
        self.model = model
5
        self.project_directory = project_directory
6
        self.client = OpenAI(
7
            base_url="https://openrouter.ai/api/v1",
8
            api_key=ReActAgent.get_api_key(),
9
        )
10

11
    def run(self, user_input: str):
12
      # 系统提示词 + 用户问题
13
        messages = [
14
            {"role": "system", "content": self.render_system_prompt(react_system_prompt_template)},
15
            {"role": "user", "content": f"<question>{user_input}</question>"}
16
        ]
17

18
        while True:（重复执行，直到得到结果）
19

20
            # 请求模型
21
            content = self.call_model(messages)
22

23
            # 检测 Thought
24
            thought_match = re.search(r"<thought>(.*?)</thought>", content, re.DOTALL)
25
            if thought_match:
26
                thought = thought_match.group(1)
27
                print(f"\n\n💭 Thought: {thought}")
28

29
            # 检测模型是否输出 Final Answer，如果是，直接返回
30
            if "<final_answer>" in content:
31
                final_answer = re.search(r"<final_answer>(.*?)</final_answer>", content, re.DOTALL)
32
                return final_answer.group(1)
33

34
            # 检测 Action
35
            action_match = re.search(r"<action>(.*?)</action>", content, re.DOTALL)
36
            if not action_match:
37
                raise RuntimeError("模型未输出 <action>")
38
            action = action_match.group(1)
39
            tool_name, args = self.parse_action(action)
40

41
            print(f"\n\n🔧 Action: {tool_name}({', '.join(args)})")
42

43
            # 提取信息
44
            try:
45
                observation = self.tools[tool_name](*args)
46
            except Exception as e:
47
                observation = f"工具执行错误：{str(e)}"
48
            print(f"\n\n🔍 Observation：{observation}")
49
            obs_msg = f"<observation>{observation}</observation>"
50
            messages.append({"role": "user", "content": obs_msg})