01 · 第一天：10 分钟造出你的第一个 Agent

# -*- coding: utf-8 -*-
"""一个最小但完整的 Agent。MockModel 让你不用 API Key 就能跑通全部机制。"""
import json, os

# ============ 第 1 步：定义工具 ============
# 工具 = 一个函数 + 一段给模型看的说明书
def calculator(expression: str) -> str:
    """计算数学表达式"""
    try:
        return str(eval(expression, {"__builtins__": {}}, {}))
    except Exception as e:
        return f"计算出错：{e}。请检查表达式，只支持 + - * / ** ()"

def list_files(path: str = ".") -> str:
    """列出目录下的文件"""
    try:
        items = os.listdir(path)[:20]
        return "\n".join(items) if items else "（空目录）"
    except Exception as e:
        return f"读取失败：{e}"

TOOLS = {"calculator": calculator, "list_files": list_files}

TOOL_DESCRIPTIONS = """你可以使用以下工具：
- calculator(expression): 计算数学表达式，如 "23*7+1"
- list_files(path): 列出目录下的文件

需要用工具时，只输出一行 JSON：{"tool": "工具名", "args": {参数}}
不需要工具时，直接输出答案。"""

# ============ 第 2 步：模型（先用假的） ============
class MockModel:
    """按剧本扮演一个真模型：先查文件、再算数、最后总结。
    换成真模型时，只需要替换这个类（见文末）。"""
    def __init__(self):
        self.script = [
            '{"tool": "list_files", "args": {"path": "."}}',
            '{"tool": "calculator", "args": {"expression": "365 * 24"}}',
            "根据工具结果：当前目录的文件已列出，一年有 8760 小时。任务完成。",
        ]
        self.step = 0

    def chat(self, messages):
        reply = self.script[min(self.step, len(self.script) - 1)]
        self.step += 1
        return reply

# ============ 第 3 步：那个循环 ============
def run_agent(model, user_task, max_steps=10):
    messages = [
        {"role": "system", "content": TOOL_DESCRIPTIONS},
        {"role": "user", "content": user_task},
    ]
    for step in range(max_steps):
        reply = model.chat(messages)
        messages.append({"role": "assistant", "content": reply})
        print(f"\n【第 {step+1} 轮】模型说：{reply}")

        # 判断：是工具调用，还是最终答案？
        try:
            call = json.loads(reply)
            assert "tool" in call
        except (json.JSONDecodeError, AssertionError):
            print("\n✅ 模型给出最终答案，循环结束。")
            return reply

        # 执行工具，把结果塞回对话
        tool_fn = TOOLS.get(call["tool"])
        result = tool_fn(**call["args"]) if tool_fn else f"错误：没有 {call['tool']} 这个工具"
        print(f"       工具返回：{result[:80]}")
        messages.append({"role": "user", "content": f"工具返回结果：\n{result}"})

    return "达到最大步数，强制停止。"  # ⭐ 护栏：永远要有最大步数

# ============ 跑起来 ============
run_agent(MockModel(), "看看当前目录有什么文件，然后告诉我一年有多少小时")

# pip install anthropic
import anthropic

class ClaudeModel:
    def __init__(self):
        self.client = anthropic.Anthropic()  # 读环境变量 ANTHROPIC_API_KEY

    def chat(self, messages):
        system = next(m["content"] for m in messages if m["role"] == "system")
        rest = [m for m in messages if m["role"] != "system"]
        resp = self.client.messages.create(
            model="claude-sonnet-5",
            max_tokens=1024,
            system=system,
            messages=rest,
        )
        return resp.content[0].text

run_agent(ClaudeModel(), "当前目录里有几个 py 文件？")