# tinyharness.py — 模型封装(可运行)
import json, anthropic
client = anthropic.Anthropic()   # 读环境变量 ANTHROPIC_API_KEY

def call_model(messages, tools):
    return client.messages.create(
        model="claude-sonnet-5",
        max_tokens=2048,
        messages=messages,
        tools=tools,          # 工具 schema 列表,下一步定义
    )

# tinyharness.py — 工具定义(可运行)
# 给模型看的 schema
TOOLS = [{
    "name": "read_file",
    "description": "读取一个文本文件的内容",
    "input_schema": {
        "type": "object",
        "properties": {"path": {"type": "string"}},
        "required": ["path"],
    },
}]

# 真正执行的函数(名字与 schema 的 name 对应)
def run_read_file(path):
    try:
        with open(path) as f:
            return f.read()
    except Exception as e:
        return f"错误:{e}"      # 错误也要可读地回给模型 → ACI 原则,CH4

TOOL_IMPL = {"read_file": run_read_file}

# tinyharness.py — 循环(可运行)
def loop(user_task):
    messages = [{"role": "user", "content": user_task}]
    while True:                                     # ← 何时停:见下
        resp = call_model(messages, TOOLS)          # 看+想
        messages.append({"role": "assistant", "content": resp.content})

        tool_uses = [b for b in resp.content if b.type == "tool_use"]
        if not tool_uses:                            # 无工具 → 自然终止
            return "".join(b.text for b in resp.content if b.type == "text")

        results = []
        for tu in tool_uses:                         # 做
            out = TOOL_IMPL[tu.name](**tu.input)
            results.append({"type": "tool_result",
                            "tool_use_id": tu.id, "content": out})
        messages.append({"role": "user", "content": results})  # 记
        # 回到 while 顶部,带着工具结果再问模型

if __name__ == "__main__":
    import sys
    print(loop(sys.argv[1]))