zhuyongxin
f75d0c49ae
将 ReAct 循环从单阶段升级为 Thinking + Action 双阶段架构： - Phase 1：剥夺工具访问权，强制模型先输出纯文本推理规划 - Phase 2：恢复工具挂载，模型按推理结果精准执行 - 新增 EnableThinking 开关，兼容旧模式
74 lines
2.2 KiB
Go
74 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
|
|
"go-tiny-claw/internal/engine"
|
|
"go-tiny-claw/internal/schema"
|
|
)
|
|
|
|
// 升级版 Mock Provider
|
|
type mockProvider struct {
|
|
turn int
|
|
}
|
|
|
|
func (m *mockProvider) Generate(ctx context.Context, msgs []schema.Message, tools []schema.ToolDefinition) (*schema.Message, error) {
|
|
// 如果工具列表为空，说明这是引擎发起的 Phase 1: Thinking 阶段
|
|
if len(tools) == 0 {
|
|
return &schema.Message{
|
|
Role: schema.RoleAssistant,
|
|
Content: "【推理中】目标是检查文件。我不能直接盲猜，我需要先调用 bash 工具执行 ls 命令，看看当前目录下有什么，然后再做定夺。",
|
|
}, nil
|
|
}
|
|
|
|
// 如果工具列表不为空，说明这是 Phase 2: Action 阶段
|
|
m.turn++
|
|
if m.turn == 1 {
|
|
// 第一轮 Action：顺着刚才的 Thinking，精准调用工具
|
|
return &schema.Message{
|
|
Role: schema.RoleAssistant,
|
|
Content: "我要执行我刚才计划的步骤了。",
|
|
ToolCalls: []schema.ToolCall{
|
|
{ID: "call_123", Name: "bash", Arguments: []byte(`{"command": "ls -la"}`)},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
// 第二轮 Action：直接总结退出
|
|
return &schema.Message{
|
|
Role: schema.RoleAssistant,
|
|
Content: "根据工具返回的结果，我看到了 main.go，任务圆满完成！",
|
|
}, nil
|
|
}
|
|
|
|
type mockRegistry struct{}
|
|
|
|
func (m *mockRegistry) GetAvailableTools() []schema.ToolDefinition {
|
|
// 为了让 Phase 2 能检测到工具，这里返回一个伪造的工具定义数组
|
|
return []schema.ToolDefinition{{Name: "bash"}}
|
|
}
|
|
|
|
func (m *mockRegistry) Execute(ctx context.Context, call schema.ToolCall) schema.ToolResult {
|
|
return schema.ToolResult{
|
|
ToolCallID: call.ID,
|
|
Output: "-rw-r--r-- 1 user group 234 Oct 24 10:00 main.go\n",
|
|
IsError: false,
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
workDir, _ := os.Getwd()
|
|
|
|
p := &mockProvider{}
|
|
r := &mockRegistry{}
|
|
|
|
// 实例化引擎，开启 EnableThinking = true
|
|
eng := engine.NewAgentEngine(p, r, workDir, true)
|
|
|
|
err := eng.Run(context.Background(), "帮我检查当前目录的文件")
|
|
if err != nil {
|
|
log.Fatalf("引擎崩溃: %v", err)
|
|
}
|
|
} |