zhuyongxin
b94460083a
- Registry 从纯接口升级为完整实现：Register/Execute/路由查找/错误自愈 - 新增 BaseTool 接口，统一工具契约 - 新增 ReadFileTool，首个真实工具实现（路径防穿越 + 截断保护） - 新增 String() 方法，Registry/Engine 调试可视化 - 新增 ToolCall 执行状态日志 - cmd/claw 切换为真实 Registry + ReadFileTool
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
|
|
"go-tiny-claw/internal/engine"
|
|
"go-tiny-claw/internal/schema"
|
|
"go-tiny-claw/internal/tools"
|
|
)
|
|
|
|
// 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) Register(tool tools.BaseTool) {}
|
|
|
|
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)
|
|
}
|
|
}
|