v1.1 慢思考模式 (Thinking Phase)

将 ReAct 循环从单阶段升级为 Thinking + Action 双阶段架构:
- Phase 1:剥夺工具访问权,强制模型先输出纯文本推理规划
- Phase 2:恢复工具挂载,模型按推理结果精准执行
- 新增 EnableThinking 开关,兼容旧模式
This commit is contained in:
zhuyongxin
2026-05-14 15:47:41 +08:00
parent b8459ef36c
commit f75d0c49ae
4 changed files with 143 additions and 109 deletions
+47 -45
View File
@@ -9,64 +9,66 @@ import (
"go-tiny-claw/internal/schema"
)
// ==========================================
// 1. 伪造的大模型 Provider
// ==========================================
// 升级版 Mock Provider
type mockProvider struct {
turn int
turn int
}
// 模拟大模型的响应:第一轮请求执行 bash,第二轮输出最终结果
func (m *mockProvider) Generate(ctx context.Context, msgs []schema.Message, _ []schema.ToolDefinition) (*schema.Message, error) {
m.turn++
if m.turn == 1 {
return &schema.Message{
Role: schema.RoleAssistant,
Content: "让我来看看当前目录下有什么文件。",
ToolCalls: []schema.ToolCall{
{ID: "call_123", Name: "bash", Arguments: []byte(`{"command": "ls -la"}`)},
},
}, nil
}
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
}
return &schema.Message{
Role: schema.RoleAssistant,
Content: "我看到了文件列表,里面包含 main.go,任务完成!",
}, 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
}
// ==========================================
// 2. 伪造的 Tool Registry
// ==========================================
type mockRegistry struct{}
func (m *mockRegistry) GetAvailableTools() []schema.ToolDefinition { return nil }
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,
}
return schema.ToolResult{
ToolCallID: call.ID,
Output: "-rw-r--r-- 1 user group 234 Oct 24 10:00 main.go\n",
IsError: false,
}
}
// ==========================================
// 3. 组装运行
// ==========================================
func main() {
// 获取当前执行目录作为 WorkDir 物理边界
workDir, _ := os.Getwd()
workDir, _ := os.Getwd()
p := &mockProvider{}
r := &mockRegistry{}
p := &mockProvider{}
r := &mockRegistry{}
// 实例化核心引擎
eng := engine.NewAgentEngine(p, r, workDir)
// 实例化引擎,开启 EnableThinking = true
eng := engine.NewAgentEngine(p, r, workDir, true)
// 发起任务指令
err := eng.Run(context.Background(), "帮我检查当前目录的文件")
if err != nil {
log.Fatalf("引擎崩溃: %v", err)
}
}
err := eng.Run(context.Background(), "帮我检查当前目录的文件")
if err != nil {
log.Fatalf("引擎崩溃: %v", err)
}
}