实战三：写一个代码助手 Agent

// 读文件
type readFileArgs struct {
    Path string `json:"path" jsonschema:"description=文件路径（相对于项目根）"`
}
readFile := agents.NewFunctionTool("read_file", "读取文件内容", func(ctx, tc, args readFileArgs) (string, error) {
    // 用 os.Root 限制在项目目录内（第 31 章）
    data, err := projectRoot.ReadFile(args.Path)
    return string(data), err
})

// 改文件（str_replace 风格，像 Codex）
type editFileArgs struct {
    Path       string `json:"path"`
    OldString  string `json:"old_string" jsonschema:"description=要替换的原文"`
    NewString  string `json:"new_string" jsonschema:"description=替换成的新文"`
}
editFile := agents.NewFunctionTool("edit_file", "精确替换文件内容", func(ctx, tc, args editFileArgs) (string, error) {
    content, _ := projectRoot.ReadFile(args.Path)
    newContent := strings.Replace(content, args.OldString, args.NewString, 1)
    if newContent == content {
        return "未找到要替换的内容", nil  // 没替换成功，回喂模型
    }
    projectRoot.WriteFile(args.Path, []byte(newContent), 0644)
    return "修改成功", nil
})

// 跑命令（测试、构建）
type runCmdArgs struct {
    Command string   `json:"command" jsonschema:"description=要执行的命令"`
}
runCmd := agents.NewFunctionTool("run_command", "执行 shell 命令", func(ctx, tc, args runCmdArgs) (string, error) {
    // 用 Sandbox 执行（第 32 章），安全隔离
    result, err := sandbox.Exec(ctx, sandbox.ExecRequest{
        Cmd:     strings.Fields(args.Command),
        Timeout: 60 * time.Second,
    })
    if err != nil { return "", err }
    // 代码错误（exit≠0）回喂模型，让它看报错修代码
    return fmt.Sprintf("exit=%d\nstdout:\n%s\nstderr:\n%s",
        result.ExitCode, result.Stdout, result.Stderr), nil
})

codeAgent := &agents.Agent{
    Name: "代码助手",
    Instructions: agents.StaticInstructions(`你是代码助手。
- 修改前先 read_file 理解现有代码
- 用 edit_file 做精确修改（不要全量重写）
- 改完用 run_command 跑测试
- 测试失败就看报错，继续修
- 不要改不相关的文件`),
    Model: "openai/gpt-4o",
    Tools: []agents.Tool{readFile, editFile, runCmd},
}

用户："修复 src/auth.go 里的空指针 bug"

Agent: read_file("src/auth.go")
  → 看到代码，发现第 42 行 user.Profile 可能为 nil

Agent: edit_file("src/auth.go",
    old: "return user.Profile.Name",
    new: "if user.Profile != nil { return user.Profile.Name }\nreturn \"\"")
  → "修改成功"

Agent: run_command("go test ./src/auth/...")
  → exit=1, stderr: "undefined: Profile"

Agent Thought: 哦，应该是 user.Profile 改成 userProfile 变量名
Agent: edit_file(...)  // 修正
Agent: run_command("go test ./src/auth/...")
  → exit=0, "PASS"

Agent: "已修复空指针 bug，测试通过。"