zhuyongxin
daa89531f5
- 新增 OpenAIProvider / ClaudeProvider，接入 Deepseek 真实 API - 修复 Deepseek thinking 模式 reasoning_content 回传问题 - 修复 Thinking 阶段过渡指令每轮重复插入导致的死循环 - 新增 maxTurns 保护、dumpMessages/dumpTools 调试输出
66 lines
1.9 KiB
Go
66 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
"fmt"
|
|
|
|
"go-tiny-claw/internal/engine"
|
|
"go-tiny-claw/internal/schema"
|
|
"go-tiny-claw/internal/provider"
|
|
)
|
|
|
|
|
|
// 伪造的工具注册表 (用于测试 Provider 的工具提取能力)
|
|
type mockRegistry struct{}
|
|
|
|
func (m *mockRegistry) GetAvailableTools() []schema.ToolDefinition {
|
|
return []schema.ToolDefinition{
|
|
{
|
|
Name: "get_weather",
|
|
Description: "获取指定城市的当前天气情况。",
|
|
InputSchema: map[string]interface{}{
|
|
"type": "object",
|
|
"properties": map[string]interface{}{
|
|
"city": map[string]interface{}{
|
|
"type": "string",
|
|
},
|
|
},
|
|
"required": []string{"city"},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (m *mockRegistry) Execute(ctx context.Context, call schema.ToolCall) schema.ToolResult {
|
|
log.Printf(" -> [Mock 工具执行] 获取 %s 的天气中...\n", call.Name)
|
|
return schema.ToolResult{
|
|
ToolCallID: call.ID,
|
|
Output: "API 返回：今天是晴天，气温 25 度。",
|
|
IsError: false,
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
|
|
fmt.Printf("11111")
|
|
workDir, _ := os.Getwd()
|
|
|
|
// 1. 初始化真实的 Provider大脑 (指向智谱 GLM-4.5)
|
|
// 这里你可以任意切换 NewZhipuClaudeProvider 或 NewZhipuOpenAIProvider，效果完全一致！
|
|
llmProvider := provider.DeepseekOpenAIProvider("deepseek-v4-flash")
|
|
// 2. 注入伪造的工具注册表
|
|
registry := &mockRegistry{}
|
|
|
|
// 3. 实例化并运行引擎，开启 EnableThinking = true (开启慢思考阶段！)
|
|
eng := engine.NewAgentEngine(llmProvider, registry, workDir, true)
|
|
|
|
// 设定测试任务
|
|
prompt := "我想去北京跑步，帮我查查天气适合吗？"
|
|
|
|
err := eng.Run(context.Background(), prompt)
|
|
if err != nil {
|
|
log.Fatalf("引擎运行崩溃: %v", err)
|
|
}
|
|
} |