zhuyongxin
daa89531f5
- 新增 OpenAIProvider / ClaudeProvider，接入 Deepseek 真实 API - 修复 Deepseek thinking 模式 reasoning_content 回传问题 - 修复 Thinking 阶段过渡指令每轮重复插入导致的死循环 - 新增 maxTurns 保护、dumpMessages/dumpTools 调试输出
60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
|
|
"go-tiny-claw/internal/engine"
|
|
"go-tiny-claw/internal/schema"
|
|
)
|
|
|
|
// mockProvider 模拟大模型推理
|
|
type mockProvider struct {
|
|
turn int
|
|
}
|
|
|
|
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
|
|
}
|
|
return &schema.Message{
|
|
Role: schema.RoleAssistant,
|
|
Content: "我看到了文件列表，里面包含 main.go，任务完成！",
|
|
}, nil
|
|
}
|
|
|
|
// mockRegistry 模拟工具执行
|
|
type mockRegistry struct{}
|
|
|
|
func (m *mockRegistry) GetAvailableTools() []schema.ToolDefinition { return nil }
|
|
|
|
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{}
|
|
|
|
eng := engine.NewAgentEngine(p, r, workDir, true)
|
|
|
|
err := eng.Run(context.Background(), "帮我检查当前目录的文件")
|
|
if err != nil {
|
|
log.Fatalf("引擎崩溃: %v", err)
|
|
}
|
|
}
|