Compare commits
4
Commits
deb1b6e660
..
main
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
572665b3ac | ||
|
|
e480c5c55e | ||
|
|
3e91c7d0a1 | ||
|
|
d84fdbd24e |
Generated
+8
@@ -0,0 +1,8 @@
|
||||
# 默认忽略的文件
|
||||
/shelf/
|
||||
/workspace.xml
|
||||
# 基于编辑器的 HTTP 客户端请求
|
||||
/httpRequests/
|
||||
# Datasource local storage ignored files
|
||||
/dataSources/
|
||||
/dataSources.local.xml
|
||||
@@ -0,0 +1,9 @@
|
||||
# 欢迎来到 go-tiny-claw 项目工作区
|
||||
|
||||
## 架构说明
|
||||
- 本项目采用 Go 语言编写,追求极致性能。
|
||||
- 所有的 API 接口都必须返回 JSON 格式,且包含 `code` 和 `message` 字段。
|
||||
- 所有的错误处理,必须返回中文报错信息,绝对禁止使用英文抛错。
|
||||
|
||||
## 禁忌事项
|
||||
- 不允许删除根目录的任何文件。
|
||||
@@ -65,28 +65,75 @@ func main() {
|
||||
|
||||
## 版本历史
|
||||
|
||||
### v1.5 — Edit 工具:从"重写整个文件"进化到"精准局部替换"
|
||||
### v1.7 — Session 会话机制 + 多工作区隔离 + Reporter 抽象
|
||||
|
||||
#### 变更
|
||||
|
||||
- **新增 EditFileTool** — 实现四级容错降级替换算法(L1 精确 → L2 换行符归一 → L3 Trim Space → L4 逐行去缩进滑动窗口),解决大模型代码修改时缩进丢失、换行符不一致等幻觉问题
|
||||
- **工具扩展** — 工具集从 3 个(read / write / bash)扩展到 4 个(+ edit)
|
||||
- **cmd/claw 任务更新** — 演示 edit_file 的局部替换能力,编辑 server.go 中的鉴权逻辑
|
||||
- **server.go** — 新增测试目标文件
|
||||
- **Session 会话机制** — 新增 `Session` 结构体,维护完整的对话历史(`[]schema.Message`),通过 `RWMutex` 实现线程安全的并发读写。全局 `SessionManager` 支持多会话隔离(`GetOrCreate`),基于 `map[string]*Session` 路由
|
||||
- **Working Memory 滑动窗口** — `GetWorkingMemory(limit)` 从后往前截取最近 N 条消息作为"短期工作记忆",并实现孤儿 ToolResult 防线:截断后若首条消息是孤立的工具响应(对应 ToolCall 已被丢弃),自动舍弃防止 API 400
|
||||
- **Reporter 输出抽象** — 新增 `Reporter` 接口(`OnThinking` / `OnToolCall` / `OnToolResult` / `OnMessage`),将引擎输出与展现层解耦。`TerminalReporter` 是首个实现,引擎不再直接 `fmt.Printf`
|
||||
- **WorkDir 从 Engine 下沉到 Session** — 引擎不再持有工作目录,WorkDir 跟随 Session 走。一个引擎实例可同时服务多个不同工作区的会话(多工作区复用单引擎)
|
||||
- **Run 签名重构** — `Run(ctx, userPrompt)` → `Run(ctx, session, reporter)`,会话成为一等公民
|
||||
- **引擎循环改造** — 每轮从 Session 的 Working Memory 构建上下文,工具执行结果实时 `Append` 回 Session,ReAct 循环结束后挂起等待人类下一条指令
|
||||
|
||||
#### 踩坑记录
|
||||
|
||||
| 问题 | 原因 | 解决 |
|
||||
|---|---|---|
|
||||
| 大模型生成的代码缩进不一致 | 模型推理时对源文件缩进(tab/空格)感知不准,产生多一个空格或少一个 tab | 编辑工具内建多级模糊匹配,不要求模型生成的 old_text 与原文件严格一致 |
|
||||
| 同一段代码在文件中出现多次 | 模型给的 old_text 上下文不够,匹配到多处 | 算法检测多匹配后直接返回错误给模型:"匹配到 X 处,请提供更多上下文" |
|
||||
| Windows 换行符 `\r\n` vs `\n` 不一致 | 模型通常输出 `\n`,Windows 文件可能是 `\r\n` | L2 换行符归一化:统一转成 `\n` 后再对比 |
|
||||
| 多会话并发操作同一目录文件冲突 | 两个 Session 的 WorkDir 相同,工具同时读写 | WorkDir 绑定 Session,不同会话指向不同工作区 |
|
||||
| 截断 Working Memory 后 API 报 400 | 丢弃了携带 ToolCall 的 Assistant 消息,但留下了对应的 ToolResult | `GetWorkingMemory` 检测并丢弃首部的孤儿 ToolResult |
|
||||
| `fmt.Printf` 无法适配飞书/钉钉/WebUI 等输出目标 | 引擎与终端输出硬耦合 | 抽象 Reporter 接口,`TerminalReporter` 仅为首个实现 |
|
||||
|
||||
#### 经验教训
|
||||
|
||||
1. **Agent 工具要做"容错输入,严格输出"** — 接受模型可能不完美的输入(多级模糊匹配),但输出清晰的错误信息帮模型自我纠正("匹配到 3 处"而非"匹配失败")
|
||||
2. **工具语义要匹配模型的能力边界** — 模型擅长生成文本但弱于精确复制。`edit_file`(给 old_text + new_text)比"重写整个文件"更适合 Agent 场景,因为它不要求模型完整认知整个文件
|
||||
3. **工具组合产生协作效应** — read_file + edit_file 是天然搭档:read 建立上下文认知 → edit 执行局部修改 → bash 验证结果。单一工具的力量有限,组合后才是真正的 Agent
|
||||
1. **WorkDir 属于会话而非引擎** — 将工作目录从 Engine 移到 Session,一个引擎实例就能同时服务多个隔离的工作区(`project_front` / `project_back`),架构不变代码不变
|
||||
2. **Working Memory 不是全量历史** — 大模型 API 有 context window 上限,截取最近 N 条消息既控制成本又保持对话连贯。截断时必须保证 ToolCall / ToolResult 成对存在,否则 API 直接报错
|
||||
3. **Reporter 是引擎可移植的关键** — 引擎只负责"推理 + 调工具",不关心输出到哪里。CLI、飞书、WebUI 只需各自实现 Reporter 接口,引擎零改动
|
||||
|
||||
### v1.5 — Edit 工具 + 四级容错替换 + 多工具并发执行
|
||||
|
||||
#### 变更
|
||||
|
||||
- **新增 EditFileTool** — 实现四级容错降级替换算法(L1 精确 → L2 换行符归一 → L3 Trim Space → L4 逐行去缩进滑动窗口),解决大模型代码修改时缩进丢失、换行符不一致等幻觉问题
|
||||
- **工具集扩展** — 工具集从 3 个(read / write / bash)扩展到 4 个(+ edit)
|
||||
- **多工具并发执行** — 引擎从串行改为并行:预分配结果切片 + `sync.WaitGroup` + 按索引无锁写入,模型一次请求多个工具时同时执行
|
||||
|
||||
#### 踩坑记录
|
||||
|
||||
| 问题 | 原因 | 解决 |
|
||||
|---|---|---|
|
||||
| 大模型生成的代码缩进不一致 | 模型推理时对源文件缩进感知不准,产生多一个空格或少一个 tab | 编辑工具内建多级模糊匹配 |
|
||||
| 同一段代码在文件中出现多次 | 模型给的 old_text 上下文不够 | 算法检测多匹配后返回错误给模型自愈 |
|
||||
| Windows 换行符 `\r\n` vs `\n` 不一致 | 模型通常输出 `\n`,Windows 文件可能是 `\r\n` | L2 换行符归一化:统一转 `\n` |
|
||||
| Goroutine 闭包捕获 loop 变量 | Go 的 loop 变量复用同一地址 | 将 `i`/`toolCall` 作为参数传入 goroutine |
|
||||
|
||||
#### 经验教训
|
||||
|
||||
1. **Agent 工具要做"容错输入,严格输出"** — 接受模型可能不完美的输入(多级模糊匹配),但输出清晰的错误信息帮模型自我纠正
|
||||
2. **工具语义要匹配模型的能力边界** — `edit_file`(给 old_text + new_text)比重写整个文件更适合 Agent,不要求模型完整认知整个文件
|
||||
3. **并发安全可以零成本** — 预分配切片 + 按索引写入 + 主 goroutine 串行读取,比加锁方案更简洁高效
|
||||
|
||||
### v1.6 — System Prompt 工程化 + 渐进式技能发现
|
||||
|
||||
#### 变更
|
||||
|
||||
- **PromptComposer** — 新增 `internal/prompt` 包取代硬编码 system prompt,运行时动态组装:核心身份 → `AGENTS.md` 项目规范 → Skill 目录
|
||||
- **AGENTS.md** — 项目根目录的 Markdown 文件自动被识别并注入 system prompt,让非代码层面的架构规范可被 Agent 感知
|
||||
- **渐进式技能发现** — `SkillLoader` 扫描 `skills/<name>/SKILL.md`,system prompt 只注入技能目录(名称 + 一行描述),模型按需调用 `read_skill` 加载完整指令
|
||||
- **ReadSkillTool** — 工具集扩展到 5 个,接收技能名称返回完整 Skill.md 正文
|
||||
- **包重命名** — `internal/context` → `internal/prompt`,消除与标准库 `context` 包的命名冲突
|
||||
|
||||
#### 踩坑记录
|
||||
|
||||
| 问题 | 原因 | 解决 |
|
||||
|---|---|---|
|
||||
| 包名与标准库冲突 | `internal/context` 与 `context` 包同名,导入时被迫黑别名 `ctxpkg` | 重命名为 `prompt`,导入无歧义 |
|
||||
| prompt 全量注入浪费上下文 | List() 只返回摘要,Build() 仍需知道技能的完整存在 | 摘要注入 system prompt,正文通过 read_skill 按需加载,典型场景节省 70%+ context |
|
||||
|
||||
#### 经验教训
|
||||
|
||||
1. **上下文管理是 Agent 的核心杠杆** — system prompt 从 30 行硬编码字符串演变为动态组装系统,每项内容(身份 / 项目规范 / 技能)的增删都不需要改引擎代码
|
||||
2. **"先摘要后按需"适用于 Agent 的所有知识注入** — 无论是技能、文档还是 API 参考,把完整内容塞进 context 是最简单的做法,但渐进式加载才是可扩展的方案
|
||||
|
||||
### v1.4 — 工具集扩展与 Windows 编码攻坚
|
||||
|
||||
|
||||
+53
-20
@@ -4,16 +4,17 @@ import (
|
||||
"context"
|
||||
"log"
|
||||
"os"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"go-tiny-claw/internal/engine"
|
||||
"go-tiny-claw/internal/provider"
|
||||
"go-tiny-claw/internal/schema"
|
||||
"go-tiny-claw/internal/tools"
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
||||
workDir, _ := os.Getwd()
|
||||
|
||||
// 1. 初始化真实的 Provider大脑
|
||||
// 这里你可以任意切换 NewZhipuClaudeProvider 或 NewZhipuOpenAIProvider,效果完全一致!
|
||||
llmProvider := provider.DeepseekOpenAIProvider("deepseek-v4-flash")
|
||||
@@ -21,26 +22,58 @@ func main() {
|
||||
registry := tools.NewRegistry()
|
||||
|
||||
// 挂载工具全家桶
|
||||
registry.Register(tools.NewReadFileTool(workDir))
|
||||
registry.Register(tools.NewWriteFileTool(workDir))
|
||||
registry.Register(tools.NewBashTool(workDir))
|
||||
registry.Register(tools.NewEditFileTool(workDir))
|
||||
// registry.Register(tools.NewReadFileTool(workDir))
|
||||
// registry.Register(tools.NewWriteFileTool(workDir))
|
||||
// registry.Register(tools.NewBashTool(workDir))
|
||||
// registry.Register(tools.NewEditFileTool(workDir))
|
||||
// registry.Register(tools.NewReadSkillTool(workDir))
|
||||
|
||||
workDir, _ := os.Getwd()
|
||||
|
||||
registry.Register(tools.NewReadFileTool(workDir + "/tmp/project_front"))
|
||||
|
||||
// 实例化引擎,开启 EnableThinking = true
|
||||
eng := engine.NewAgentEngine(llmProvider, registry, workDir, false)
|
||||
eng := engine.NewAgentEngine(llmProvider, registry, false)
|
||||
reporter := engine.NewTerminalReporter()
|
||||
|
||||
// 发起一个需要局部修改的指令
|
||||
prompt := `
|
||||
我当前目录下有一个 server.go 文件。
|
||||
请帮我把里面 "TODO: 增加鉴权逻辑" 下面的那个 if 语句,整个替换为:
|
||||
if user == nil {
|
||||
fmt.Println("Forbidden!")
|
||||
return
|
||||
}
|
||||
`
|
||||
var wg sync.WaitGroup
|
||||
|
||||
err := eng.Run(context.Background(), prompt)
|
||||
if err != nil {
|
||||
log.Fatalf("引擎运行崩溃: %v", err)
|
||||
}
|
||||
// ================= 模拟并发场景 1:飞书前端群 =================
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
sessionA := engine.GlobalSessionMgr.GetOrCreate("chat_front_001", workDir+"/tmp/project_front")
|
||||
|
||||
// 回合 1:获取机密
|
||||
log.Println("\n>>> 🙋♂️ [Session A / Turn 1]: 帮我看看 README.md 里记录了什么密钥?我的操作系统是windows")
|
||||
sessionA.Append(schema.Message{Role: schema.RoleUser, Content: "帮我看看 README.md 里记录了什么密钥?"})
|
||||
_ = eng.Run(context.Background(), sessionA, reporter)
|
||||
|
||||
// 故意制造大量“废话”对话,刷掉记忆 (假设 Working Memory Limit=6)
|
||||
for i := 0; i < 6; i++ {
|
||||
sessionA.Append(schema.Message{Role: schema.RoleUser, Content: "这只是一句闲聊占位符。"})
|
||||
sessionA.Append(schema.Message{Role: schema.RoleAssistant, Content: "好的,收到闲聊。"})
|
||||
}
|
||||
|
||||
// 回合 2:验证记忆截断 (此时第一轮的密钥已经被挤出 Working Memory 了!)
|
||||
log.Println("\n>>> 🙋♂️ [Session A / Turn 2]: 请直接告诉我,刚才第一轮你查到的那个密钥是什么?")
|
||||
sessionA.Append(schema.Message{Role: schema.RoleUser, Content: "请直接告诉我,刚才第一轮你查到的那个密钥是什么?不准调用工具!"})
|
||||
_ = eng.Run(context.Background(), sessionA, reporter)
|
||||
}()
|
||||
|
||||
// ================= 模拟并发场景 2:飞书后端群 =================
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
// 稍微错开一点时间发起请求
|
||||
time.Sleep(1 * time.Second)
|
||||
|
||||
sessionB := engine.GlobalSessionMgr.GetOrCreate("chat_back_002", workDir+"/tmp/project_back")
|
||||
|
||||
log.Println("\n>>> 🙋♂️ [Session B]: 别人查到了一个密钥,你这里能看到吗?")
|
||||
sessionB.Append(schema.Message{Role: schema.RoleUser, Content: "别人查到了一个密钥,你这里能看到吗?不准调用工具!"})
|
||||
_ = eng.Run(context.Background(), sessionB, reporter)
|
||||
}()
|
||||
|
||||
wg.Wait()
|
||||
}
|
||||
|
||||
+195
-83
@@ -4,7 +4,9 @@ import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"sync"
|
||||
|
||||
"go-tiny-claw/internal/prompt"
|
||||
"go-tiny-claw/internal/provider"
|
||||
"go-tiny-claw/internal/schema"
|
||||
"go-tiny-claw/internal/tools"
|
||||
@@ -16,21 +18,22 @@ type AgentEngine struct {
|
||||
registry tools.Registry
|
||||
|
||||
// WorkDir (工作区): 借鉴 OpenClaw 的理念,Agent 必须有一个明确的物理边界
|
||||
WorkDir string
|
||||
// WorkDir string
|
||||
EnableThinking bool // 【新增】慢思考模式开关
|
||||
composer *prompt.PromptComposer
|
||||
}
|
||||
|
||||
func NewAgentEngine(p provider.LLMProvider, r tools.Registry, workDir string, enableThinking bool) *AgentEngine {
|
||||
// 移除了 Engine 层级的 WorkDir,因为 WorkDir 现在应该跟随 Session 走
|
||||
func NewAgentEngine(p provider.LLMProvider, r tools.Registry, enableThinking bool) *AgentEngine {
|
||||
return &AgentEngine{
|
||||
provider: p,
|
||||
registry: r,
|
||||
WorkDir: workDir,
|
||||
EnableThinking: enableThinking,
|
||||
}
|
||||
}
|
||||
|
||||
func (e *AgentEngine) String() string {
|
||||
return fmt.Sprintf("AgentEngine{workDir: %s, thinking: %v, registry: %s}", e.WorkDir, e.EnableThinking, e.registry)
|
||||
return fmt.Sprintf("AgentEngine{thinking: %v, registry: %s}", e.EnableThinking, e.registry)
|
||||
}
|
||||
|
||||
// dumpMessages 打印当前上下文中的所有消息 (调试用)
|
||||
@@ -55,117 +58,226 @@ func dumpTools(tools []schema.ToolDefinition) {
|
||||
}
|
||||
}
|
||||
|
||||
// internal/engine/loop.go (续)
|
||||
func (e *AgentEngine) Run(ctx context.Context, userPrompt string) error {
|
||||
log.Printf("[Engine] 引擎启动,锁定工作区: %s\n", e.WorkDir)
|
||||
log.Printf("[Engine] 慢思考模式 (Thinking Phase): %v\n", e.EnableThinking)
|
||||
func (e *AgentEngine) Run(ctx context.Context, session *Session, reporter Reporter) error {
|
||||
log.Printf("[Engine] 唤醒会话 [%s],锁定工作区: %s\n", session.ID, session.WorkDir)
|
||||
|
||||
contextHistory := []schema.Message{
|
||||
{
|
||||
Role: schema.RoleSystem,
|
||||
Content: "You are go-tiny-claw, an expert coding assistant. You have full access to tools in the workspace.",
|
||||
},
|
||||
{
|
||||
Role: schema.RoleUser,
|
||||
Content: userPrompt,
|
||||
},
|
||||
}
|
||||
|
||||
turnCount := 0
|
||||
const maxTurns = 10
|
||||
// 根据当前 Session 的工作区,动态组装最新的 System Prompt
|
||||
composer := prompt.NewPromptComposer(session.WorkDir)
|
||||
systemMsg := composer.Build()
|
||||
|
||||
for {
|
||||
turnCount++
|
||||
if turnCount > maxTurns {
|
||||
log.Printf("[Engine] 已达最大轮数 (%d),强制终止。\n", maxTurns)
|
||||
break
|
||||
}
|
||||
log.Printf("\n========== [Turn %d] 开始 ==========\n", turnCount)
|
||||
dumpMessages(contextHistory)
|
||||
|
||||
// 获取当前挂载的所有工具定义
|
||||
availableTools := e.registry.GetAvailableTools()
|
||||
dumpTools(availableTools)
|
||||
|
||||
// ====================================================================
|
||||
// Phase 1: 慢思考阶段 (Thinking) - 仅第一轮执行初始规划
|
||||
// ====================================================================
|
||||
if e.EnableThinking && turnCount == 1 {
|
||||
log.Println("[Engine][Phase 1] 剥夺工具访问权,强制进入慢思考与规划阶段...")
|
||||
// 1. 【上下文组装】: System Prompt + 截取最近的 6 条消息作为 Working Memory
|
||||
// 在实际业务中,由于工具返回结果可能很长,短期工作记忆往往设为 6-10 条足以维系连贯对话
|
||||
workingMemory := session.GetWorkingMemory(6)
|
||||
|
||||
var contextHistory []schema.Message
|
||||
contextHistory = append(contextHistory, systemMsg)
|
||||
contextHistory = append(contextHistory, workingMemory...)
|
||||
|
||||
// 2. ================= Phase 1: Thinking =================
|
||||
if e.EnableThinking {
|
||||
if reporter != nil {
|
||||
reporter.OnThinking(ctx)
|
||||
}
|
||||
|
||||
// 核心机制:传入的 availableTools 为 nil!
|
||||
// 大模型看不到任何 JSON Schema,被迫只能输出纯文本的思考过程。
|
||||
thinkResp, err := e.provider.Generate(ctx, contextHistory, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Thinking 阶段生成失败: %w", err)
|
||||
return fmt.Errorf("Thinking 阶段失败: %w", err)
|
||||
}
|
||||
|
||||
// 如果模型输出了思考过程,我们将其作为 Assistant 消息追加到上下文中
|
||||
if thinkResp.Content != "" {
|
||||
fmt.Printf("🧠 [内部思考 Trace]: %s\n", thinkResp.Content)
|
||||
// 将思考过程持久化到 Session 中!
|
||||
session.Append(*thinkResp)
|
||||
// 把它追加到当前这一轮的临时上下文中,供 Action 阶段使用
|
||||
contextHistory = append(contextHistory, *thinkResp)
|
||||
}
|
||||
|
||||
// 插入过渡指令:让模型知道现在可以调用工具了
|
||||
contextHistory = append(contextHistory, schema.Message{
|
||||
Role: schema.RoleUser,
|
||||
Content: "根据你的推理,现在请使用可用的工具来完成任务。执行具体行动。",
|
||||
})
|
||||
}
|
||||
|
||||
// ====================================================================
|
||||
// Phase 2: 行动阶段 (Action) - 恢复工具,顺着规划执行
|
||||
// ====================================================================
|
||||
log.Println("[Engine][Phase 2] 恢复工具挂载,等待模型采取行动...")
|
||||
|
||||
// 此时的 contextHistory 中已经包含了上一阶段模型自己的 Thinking Trace + 过渡指令。
|
||||
// 模型会顺着自己的逻辑,结合恢复的 availableTools 发起精准的工具调用。
|
||||
// 3. ================= Phase 2: Action =================
|
||||
actionResp, err := e.provider.Generate(ctx, contextHistory, availableTools)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Action 阶段生成失败: %w", err)
|
||||
return fmt.Errorf("Action 阶段失败: %w", err)
|
||||
}
|
||||
|
||||
// 将大模型的行动响应持久化到 Session 中
|
||||
session.Append(*actionResp)
|
||||
contextHistory = append(contextHistory, *actionResp)
|
||||
|
||||
if actionResp.Content != "" {
|
||||
fmt.Printf("🤖 [对外回复]: %s\n", actionResp.Content)
|
||||
if actionResp.Content != "" && reporter != nil {
|
||||
reporter.OnMessage(ctx, actionResp.Content)
|
||||
}
|
||||
|
||||
// ====================================================================
|
||||
// 退出与执行逻辑 (与上一讲保持一致)
|
||||
// ====================================================================
|
||||
if len(actionResp.ToolCalls) == 0 {
|
||||
log.Println("[Engine] 模型未请求调用工具,任务宣告完成。")
|
||||
// 如果没有工具调用,说明本次任务已完成,打破 ReAct 循环,挂起等待人类的下一条指令
|
||||
break
|
||||
}
|
||||
|
||||
log.Printf("[Engine] 模型请求调用 %d 个工具...\n", len(actionResp.ToolCalls))
|
||||
// 4. ================= 并发执行底层工具 =================
|
||||
observationMsgs := make([]schema.Message, len(actionResp.ToolCalls))
|
||||
var wg sync.WaitGroup
|
||||
|
||||
for _, toolCall := range actionResp.ToolCalls {
|
||||
log.Printf(" -> 🛠️ 执行工具: %s, 参数: %s\n", toolCall.Name, string(toolCall.Arguments))
|
||||
for i, toolCall := range actionResp.ToolCalls {
|
||||
wg.Add(1)
|
||||
|
||||
result := e.registry.Execute(ctx, toolCall)
|
||||
go func(idx int, call schema.ToolCall) {
|
||||
defer wg.Done()
|
||||
|
||||
status := "✅ 成功"
|
||||
if result.IsError {
|
||||
status = "❌ 失败"
|
||||
}
|
||||
log.Printf(" -> 📋 ToolCall %s: %s, 结果: %s\n", toolCall.ID, status, result.Output)
|
||||
if reporter != nil {
|
||||
reporter.OnToolCall(ctx, call.Name, string(call.Arguments))
|
||||
}
|
||||
|
||||
if result.IsError {
|
||||
log.Printf(" -> ❌ 工具执行报错: %s\n", result.Output)
|
||||
} else {
|
||||
log.Printf(" -> ✅ 工具执行成功 (返回 %d 字节)\n", len(result.Output))
|
||||
}
|
||||
result := e.registry.Execute(ctx, call)
|
||||
|
||||
// 将工具执行的观察结果追加到 Context,准备进入下一轮
|
||||
observationMsg := schema.Message{
|
||||
Role: schema.RoleUser,
|
||||
Content: result.Output,
|
||||
ToolCallID: toolCall.ID,
|
||||
}
|
||||
contextHistory = append(contextHistory, observationMsg)
|
||||
if reporter != nil {
|
||||
displayOutput := result.Output
|
||||
if len(displayOutput) > 200 {
|
||||
displayOutput = displayOutput[:200] + "... (已截断)"
|
||||
}
|
||||
reporter.OnToolResult(ctx, call.Name, displayOutput, result.IsError)
|
||||
}
|
||||
|
||||
observationMsgs[idx] = schema.Message{
|
||||
Role: schema.RoleUser,
|
||||
Content: result.Output,
|
||||
ToolCallID: call.ID,
|
||||
}
|
||||
}(i, toolCall)
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
// 将所有的工具执行结果(Observation)持久化到 Session 中,开启下一轮的复盘与推理
|
||||
session.Append(observationMsgs...)
|
||||
|
||||
// systemMsg := e.composer.Build()
|
||||
|
||||
// contextHistory := []schema.Message{
|
||||
// systemMsg, // 注入动态组装的内核、AGENTS.md 与 Skills
|
||||
// {Role: schema.RoleUser, Content: userPrompt},
|
||||
// }
|
||||
|
||||
// turnCount := 0
|
||||
// const maxTurns = 10
|
||||
|
||||
// for {
|
||||
// turnCount++
|
||||
// if turnCount > maxTurns {
|
||||
// log.Printf("[Engine] 已达最大轮数 (%d),强制终止。\n", maxTurns)
|
||||
// break
|
||||
// }
|
||||
// log.Printf("\n========== [Turn %d] 开始 ==========\n", turnCount)
|
||||
// dumpMessages(contextHistory)
|
||||
|
||||
// // 获取当前挂载的所有工具定义
|
||||
// availableTools := e.registry.GetAvailableTools()
|
||||
// dumpTools(availableTools)
|
||||
|
||||
// // ====================================================================
|
||||
// // Phase 1: 慢思考阶段 (Thinking) - 仅第一轮执行初始规划
|
||||
// // ====================================================================
|
||||
// if e.EnableThinking && turnCount == 1 {
|
||||
// log.Println("[Engine][Phase 1] 剥夺工具访问权,强制进入慢思考与规划阶段...")
|
||||
|
||||
// // 核心机制:传入的 availableTools 为 nil!
|
||||
// // 大模型看不到任何 JSON Schema,被迫只能输出纯文本的思考过程。
|
||||
// thinkResp, err := e.provider.Generate(ctx, contextHistory, nil)
|
||||
// if err != nil {
|
||||
// return fmt.Errorf("Thinking 阶段生成失败: %w", err)
|
||||
// }
|
||||
|
||||
// // 如果模型输出了思考过程,我们将其作为 Assistant 消息追加到上下文中
|
||||
// if thinkResp.Content != "" {
|
||||
// fmt.Printf("🧠 [内部思考 Trace]: %s\n", thinkResp.Content)
|
||||
// contextHistory = append(contextHistory, *thinkResp)
|
||||
// }
|
||||
|
||||
// // 插入过渡指令:让模型知道现在可以调用工具了
|
||||
// contextHistory = append(contextHistory, schema.Message{
|
||||
// Role: schema.RoleUser,
|
||||
// Content: "根据你的推理,现在请使用可用的工具来完成任务。执行具体行动。",
|
||||
// })
|
||||
// }
|
||||
|
||||
// // ====================================================================
|
||||
// // Phase 2: 行动阶段 (Action) - 恢复工具,顺着规划执行
|
||||
// // ====================================================================
|
||||
// log.Println("[Engine][Phase 2] 恢复工具挂载,等待模型采取行动...")
|
||||
|
||||
// // 此时的 contextHistory 中已经包含了上一阶段模型自己的 Thinking Trace + 过渡指令。
|
||||
// // 模型会顺着自己的逻辑,结合恢复的 availableTools 发起精准的工具调用。
|
||||
// actionResp, err := e.provider.Generate(ctx, contextHistory, availableTools)
|
||||
// if err != nil {
|
||||
// return fmt.Errorf("Action 阶段生成失败: %w", err)
|
||||
// }
|
||||
|
||||
// contextHistory = append(contextHistory, *actionResp)
|
||||
|
||||
// if actionResp.Content != "" {
|
||||
// fmt.Printf("🤖 [对外回复]: %s\n", actionResp.Content)
|
||||
// }
|
||||
|
||||
// // ====================================================================
|
||||
// // 退出与执行逻辑 (与上一讲保持一致)
|
||||
// // ====================================================================
|
||||
// if len(actionResp.ToolCalls) == 0 {
|
||||
// log.Println("[Engine] 模型未请求调用工具,任务宣告完成。")
|
||||
// break
|
||||
// }
|
||||
|
||||
// log.Printf("[Engine] 模型请求并发调用 %d 个工具...\n", len(actionResp.ToolCalls))
|
||||
|
||||
// // 【核心改造开始】: 从串行 (Sequential) 演进为并行 (Parallel)
|
||||
|
||||
// // 1. 预分配一个固定长度的切片,用于安全地存放各个并发工具的执行结果(Observation)
|
||||
// // 长度与 ToolCalls 的数量完全一致
|
||||
// observationMsgs := make([]schema.Message, len(actionResp.ToolCalls))
|
||||
|
||||
// // 2. 声明 WaitGroup 用于阻塞等待所有协程完成
|
||||
// var wg sync.WaitGroup
|
||||
|
||||
// // 3. 遍历模型请求的所有工具,为每一个工具单独 Fork 出一个 Goroutine
|
||||
// for i, toolCall := range actionResp.ToolCalls {
|
||||
// wg.Add(1) // 增加计数器
|
||||
|
||||
// // 开启协程。注意:一定要将索引 i 和 toolCall 作为参数传入匿名函数,防止闭包变量捕获陷阱!
|
||||
// go func(idx int, call schema.ToolCall) {
|
||||
// defer wg.Done() // 协程结束时计数器减一
|
||||
|
||||
// log.Printf(" -> [Go-%d] 🛠️ 触发并行执行: %s\n", idx, call.Name)
|
||||
|
||||
// // 调用底层 Registry 执行工具(物理操作)
|
||||
// result := e.registry.Execute(ctx, call)
|
||||
|
||||
// if result.IsError {
|
||||
// log.Printf(" -> [Go-%d] ❌ 工具执行报错: %s\n", idx, result.Output)
|
||||
// } else {
|
||||
// log.Printf(" -> [Go-%d] ✅ 工具执行成功 (返回 %d 字节)\n", idx, len(result.Output))
|
||||
// }
|
||||
|
||||
// // 将执行结果封装为一条用户消息 (RoleUser)
|
||||
// obsMsg := schema.Message{
|
||||
// Role: schema.RoleUser,
|
||||
// Content: result.Output,
|
||||
// ToolCallID: call.ID,
|
||||
// }
|
||||
|
||||
// // 【线程安全】: 由于每个 Goroutine 操作的是预分配切片的不同索引,
|
||||
// // 这里不需要加锁 (Mutex),性能极高!
|
||||
// observationMsgs[idx] = obsMsg
|
||||
|
||||
// }(i, toolCall) // 闭包传参
|
||||
// }
|
||||
|
||||
// // 4. Join 阻塞等待:主循环挂起,直到所有的并发协程全部执行完毕
|
||||
// wg.Wait()
|
||||
// log.Println("[Engine] 所有并发工具执行完毕,开始聚合观察结果 (Observation)...")
|
||||
|
||||
// // 5. 聚合装填:将并行的结果,按照原本的顺序,一次性追加到上下文时间线中
|
||||
// // 这等价于 contextHistory = append(contextHistory, observationMsgs...)
|
||||
// for _, obs := range observationMsgs {
|
||||
// contextHistory = append(contextHistory, obs)
|
||||
// }
|
||||
}
|
||||
|
||||
return nil
|
||||
|
||||
@@ -0,0 +1,19 @@
|
||||
package engine
|
||||
|
||||
import "context"
|
||||
|
||||
// Reporter 定义了 Agent 引擎向外界输出信息的规范。
|
||||
// 这使得引擎可以无缝切换终端 (CLI)、飞书、钉钉甚至 WebUI 等不同的展现层。
|
||||
type Reporter interface {
|
||||
// OnThinking 当模型开始进行慢思考 (Reasoning) 时调用
|
||||
OnThinking(ctx context.Context)
|
||||
|
||||
// OnToolCall 当模型决定并发调用工具时调用
|
||||
OnToolCall(ctx context.Context, toolName string, args string)
|
||||
|
||||
// OnToolResult 当工具在底层执行完毕并返回结果时调用
|
||||
OnToolResult(ctx context.Context, toolName string, result string, isError bool)
|
||||
|
||||
// OnMessage 当模型宣告任务完成,向用户输出最终纯文本回答时调用
|
||||
OnMessage(ctx context.Context, content string)
|
||||
}
|
||||
@@ -0,0 +1,101 @@
|
||||
package engine
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"go-tiny-claw/internal/schema"
|
||||
)
|
||||
|
||||
// Session 代表了一次持续的人机交互过程。它负责维护该会话的完整历史。
|
||||
type Session struct {
|
||||
ID string
|
||||
WorkDir string // 该会话绑定的物理工作区
|
||||
CreatedAt time.Time
|
||||
UpdatedAt time.Time
|
||||
|
||||
// 存放此 Session 中所有的用户输入、大模型回复和工具调用结果
|
||||
history []schema.Message
|
||||
mu sync.RWMutex // 读写锁,防止并发读写历史时发生 Data Race
|
||||
}
|
||||
|
||||
func NewSession(id string, workDir string) *Session {
|
||||
return &Session{
|
||||
ID: id,
|
||||
WorkDir: workDir,
|
||||
CreatedAt: time.Now(),
|
||||
UpdatedAt: time.Now(),
|
||||
history: make([]schema.Message, 0),
|
||||
}
|
||||
}
|
||||
|
||||
// Append 线程安全地向 Session 中追加消息
|
||||
func (s *Session) Append(msgs ...schema.Message) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.history = append(s.history, msgs...)
|
||||
s.UpdatedAt = time.Now()
|
||||
|
||||
// 【持久化预留点】:在真实的工业级实现中(如 Claude Code),
|
||||
// 我们会在这里将 s.history 以 JSONL 的格式 Append 到 workDir/.claw/sessions/xxx.jsonl 中。
|
||||
// s.SaveToDisk()
|
||||
}
|
||||
|
||||
// GetWorkingMemory 是驾驭工程的核心!
|
||||
// 它不返回全量历史,而是从后往前截取最近的 N 条消息,形成 Agent 的“短期工作记忆”。
|
||||
func (s *Session) GetWorkingMemory(limit int) []schema.Message {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
|
||||
total := len(s.history)
|
||||
if total <= limit || limit <= 0 {
|
||||
// 如果历史总量小于限制,或者不设限,全量返回 (需要深拷贝以防外部修改)
|
||||
res := make([]schema.Message, total)
|
||||
copy(res, s.history)
|
||||
return res
|
||||
}
|
||||
|
||||
// 截取最近的 limit 条消息
|
||||
res := make([]schema.Message, limit)
|
||||
copy(res, s.history[total-limit:])
|
||||
|
||||
// 【驾驭防线】:大模型 API 强制要求历史消息的连续性!
|
||||
// 如果我们截断的第一条消息恰好是一个 ToolResult (RoleUser 且含有 ToolCallID),
|
||||
// 但发出这个请求的 ToolCall 被我们截断抛弃了,大模型 API 会直接报 400 Bad Request。
|
||||
// 因此,如果切片首条属于“孤儿”工具响应,我们必须将其强行舍弃,顺延到下一条正常的 User/Assistant 消息。
|
||||
for len(res) > 0 {
|
||||
if res[0].Role == schema.RoleUser && res[0].ToolCallID != "" {
|
||||
res = res[1:]
|
||||
} else {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
// ==========================================
|
||||
// 全局 Session Manager: 用于多用户/多终端隔离
|
||||
// ==========================================
|
||||
|
||||
type SessionManager struct {
|
||||
sessions map[string]*Session
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
var GlobalSessionMgr = &SessionManager{
|
||||
sessions: make(map[string]*Session),
|
||||
}
|
||||
|
||||
// GetOrCreate 获取或创建一个会话
|
||||
func (sm *SessionManager) GetOrCreate(id string, workDir string) *Session {
|
||||
sm.mu.Lock()
|
||||
defer sm.mu.Unlock()
|
||||
|
||||
if sess, exists := sm.sessions[id]; exists {
|
||||
return sess
|
||||
}
|
||||
sess := NewSession(id, workDir)
|
||||
sm.sessions[id] = sess
|
||||
return sess
|
||||
}
|
||||
@@ -0,0 +1,47 @@
|
||||
package engine
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type TerminalReporter struct{}
|
||||
|
||||
func NewTerminalReporter() *TerminalReporter {
|
||||
return &TerminalReporter{}
|
||||
}
|
||||
|
||||
func (r *TerminalReporter) OnThinking(ctx context.Context) {
|
||||
fmt.Printf("\n[🤔 思考中] 模型正在推理...\n")
|
||||
}
|
||||
|
||||
func (r *TerminalReporter) OnToolCall(ctx context.Context, toolName string, args string) {
|
||||
fmt.Printf("[🛠️ 调用工具] %s\n", toolName)
|
||||
// 清理参数中的换行符和特殊字符
|
||||
displayArgs := strings.ReplaceAll(args, "\n", "\\n")
|
||||
displayArgs = strings.ReplaceAll(displayArgs, "\r", "\\r")
|
||||
if len(displayArgs) > 150 {
|
||||
displayArgs = displayArgs[:150] + "... (已截断)"
|
||||
}
|
||||
fmt.Printf(" 参数: %s\n", displayArgs)
|
||||
}
|
||||
|
||||
func (r *TerminalReporter) OnToolResult(ctx context.Context, toolName string, result string, isError bool) {
|
||||
if isError {
|
||||
fmt.Printf("[❌ 执行失败] %s\n", toolName)
|
||||
// 显示错误信息
|
||||
if result != "" {
|
||||
fmt.Printf(" 错误: %s\n", result)
|
||||
}
|
||||
} else {
|
||||
fmt.Printf("[✅ 执行成功] %s\n", toolName)
|
||||
}
|
||||
}
|
||||
|
||||
func (r *TerminalReporter) OnMessage(ctx context.Context, content string) {
|
||||
if content == "" {
|
||||
return
|
||||
}
|
||||
fmt.Printf("\n🤖 Agent 回复:\n%s\n\n", content)
|
||||
}
|
||||
@@ -0,0 +1,101 @@
|
||||
package prompt
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
"go-tiny-claw/internal/schema"
|
||||
)
|
||||
|
||||
// Compactor 负责监控和压缩上下文内存,防止大模型发生 OOM
|
||||
type Compactor struct {
|
||||
MaxChars int // 触发压缩的最大字符数阈值 (水位线,可参考使用的大模型的token窗口大小)
|
||||
RetainLastMsgs int // Working Memory 保护区:最近的 N 条消息
|
||||
}
|
||||
|
||||
func NewCompactor(maxChars int, retainLastMsgs int) *Compactor {
|
||||
return &Compactor{
|
||||
MaxChars: maxChars,
|
||||
RetainLastMsgs: retainLastMsgs,
|
||||
}
|
||||
}
|
||||
|
||||
// Compact 接收准备发送给大模型的消息数组。
|
||||
// 如果总长度超标,对远期历史区进行全量掩码 (Masking),对短期保护区进行超长局部截断 (Truncation)。
|
||||
func (c *Compactor) Compact(msgs []schema.Message) []schema.Message {
|
||||
currentLength := c.estimateLength(msgs)
|
||||
|
||||
// 如果没有超过水位线,直接返回原数组 (大多数情况下的正常路径)
|
||||
if currentLength < c.MaxChars {
|
||||
return msgs
|
||||
}
|
||||
|
||||
log.Printf("[Compactor] ⚠️ 内存告警:当前上下文长度 (%d 字符) 超过阈值 (%d),触发压缩清理...\n", currentLength, c.MaxChars)
|
||||
|
||||
var compacted []schema.Message
|
||||
msgCount := len(msgs)
|
||||
|
||||
// 计算受保护的 Working Memory 起始索引
|
||||
protectStartIndex := msgCount - c.RetainLastMsgs
|
||||
if protectStartIndex < 0 {
|
||||
protectStartIndex = 0
|
||||
}
|
||||
|
||||
for i, msg := range msgs {
|
||||
// 1. 系统提示词 (System Prompt) 绝对不能动,直接保留
|
||||
if msg.Role == schema.RoleSystem {
|
||||
compacted = append(compacted, msg)
|
||||
continue
|
||||
}
|
||||
|
||||
// 我们必须拷贝一份新消息,因为在并发环境中直接修改原引用可能导致底层数据结构被污染
|
||||
newMsg := msg
|
||||
|
||||
isInWorkingMemory := i >= protectStartIndex
|
||||
|
||||
// 【核心驾驭逻辑】: 双重降级防线
|
||||
if msg.Role == schema.RoleUser && msg.ToolCallID != "" {
|
||||
// 对于工具的返回结果 (Observation/ToolResult)
|
||||
if !isInWorkingMemory {
|
||||
// 【第一道防线:远期历史】如果是早期对话,执行无情替换 (Full Masking)
|
||||
if len(msg.Content) > 200 {
|
||||
newMsg.Content = fmt.Sprintf("...[为了节省内存,早期的工具输出已被系统强制清理。原始长度: %d 字节]...", len(msg.Content))
|
||||
}
|
||||
} else {
|
||||
// 【第二道防线:短期记忆】即使处于近期保护区,只要单条内容过大,也必须截断防 OOM (Head-Tail Truncation)
|
||||
// 我们保留前 500 字符和后 500 字符(掐头去尾法,大模型通常只需要看开头报错和结尾总结)
|
||||
const maxKeep = 1000
|
||||
if len(msg.Content) > maxKeep {
|
||||
head := msg.Content[:500]
|
||||
tail := msg.Content[len(msg.Content)-500:]
|
||||
newMsg.Content = fmt.Sprintf("%s\n\n...[内容过长,中间 %d 字节已被系统截断]...\n\n%s", head, len(msg.Content)-maxKeep, tail)
|
||||
}
|
||||
}
|
||||
} else if msg.Role == schema.RoleAssistant && msg.Content != "" {
|
||||
// 对于大模型的冗长推理废话 (Thinking Trace)
|
||||
if !isInWorkingMemory && len(msg.Content) > 200 {
|
||||
newMsg.Content = "...[早期的推理思考过程已折叠]..."
|
||||
}
|
||||
}
|
||||
|
||||
// 注意:我们绝不会去动 msg.ToolCalls,因为这是模型行动的证据,是维系逻辑链的关键!
|
||||
compacted = append(compacted, newMsg)
|
||||
}
|
||||
|
||||
newLength := c.estimateLength(compacted)
|
||||
log.Printf("[Compactor] ✅ 压缩完成。上下文长度从 %d 降至 %d 字符。\n", currentLength, newLength)
|
||||
|
||||
return compacted
|
||||
}
|
||||
|
||||
// estimateLength 粗略计算当前上下文的总字符长度
|
||||
func (c *Compactor) estimateLength(msgs []schema.Message) int {
|
||||
length := 0
|
||||
for _, msg := range msgs {
|
||||
length += len(msg.Content)
|
||||
for _, tc := range msg.ToolCalls {
|
||||
length += len(tc.Name) + len(tc.Arguments)
|
||||
}
|
||||
}
|
||||
return length
|
||||
}
|
||||
@@ -0,0 +1,68 @@
|
||||
package prompt
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"go-tiny-claw/internal/schema"
|
||||
)
|
||||
|
||||
// PromptComposer 负责根据工作区环境动态生成 System Prompt
|
||||
type PromptComposer struct {
|
||||
workDir string
|
||||
skillLoader *SkillLoader
|
||||
}
|
||||
|
||||
func NewPromptComposer(workDir string) *PromptComposer {
|
||||
return &PromptComposer{
|
||||
workDir: workDir,
|
||||
skillLoader: NewSkillLoader(workDir),
|
||||
}
|
||||
}
|
||||
|
||||
// Build 组装并返回一条完整的 RoleSystem 消息
|
||||
func (c *PromptComposer) Build() schema.Message {
|
||||
var promptBuilder strings.Builder
|
||||
|
||||
// 1. 极简内核 (Minimal Core)
|
||||
promptBuilder.WriteString(`# 核心身份
|
||||
你名叫 go-tiny-claw,一个由驾驭工程驱动的骨灰级研发助手。
|
||||
你具备极简主义哲学,拒绝废话。你能通过系统提供的内置工具,创建、读取、修改和执行工作区中的代码。
|
||||
|
||||
# 核心纪律 (CRITICAL)
|
||||
1. 如需检查文件是否存在,请使用 bash 的 ls 或 test -f,而不是对目录使用 read_file。
|
||||
2. 创建新文件时,务必使用 write_file,并同时提供 path 和 content 参数。
|
||||
3. 编辑文件前务必先读取现有文件,以理解上下文。
|
||||
4. 无论何时你需要写代码或创建文件,都要直接使用 write_file 工具。
|
||||
5. 遇到工具执行报错时,仔细阅读 stderr,尝试自己修正命令并重试。
|
||||
6. 始终用中文回复,以便传达你的进展和想法。
|
||||
`)
|
||||
|
||||
// 2. 外部化状态:加载项目专属规范 (AGENTS.md)
|
||||
agentsMDPath := filepath.Join(c.workDir, "AGENTS.md")
|
||||
content, err := os.ReadFile(agentsMDPath)
|
||||
if err == nil {
|
||||
promptBuilder.WriteString("\n# 项目专属指南 (来自 AGENTS.md)\n")
|
||||
promptBuilder.WriteString("以下是当前工作区特有的架构规范与注意事项,你的行为必须绝对符合以下要求:\n")
|
||||
promptBuilder.WriteString("```markdown\n")
|
||||
promptBuilder.WriteString(string(content))
|
||||
promptBuilder.WriteString("\n```\n")
|
||||
}
|
||||
|
||||
// 3. 渐进式技能发现:仅注入目录(名称 + 触发条件),正文按需加载
|
||||
skills := c.skillLoader.List()
|
||||
if len(skills) > 0 {
|
||||
promptBuilder.WriteString("\n# 可用专业技能\n")
|
||||
promptBuilder.WriteString("你在以下场景可以调用 read_skill 加载完整的技能指令:\n\n")
|
||||
for _, s := range skills {
|
||||
promptBuilder.WriteString("- **" + s.Name + "**:" + s.Description + "\n")
|
||||
}
|
||||
promptBuilder.WriteString("\n当任务匹配上述技能的描述时,请先调用 read_skill 加载完整的执行指南,再严格按照指南操作。\n")
|
||||
}
|
||||
|
||||
return schema.Message{
|
||||
Role: schema.RoleSystem,
|
||||
Content: promptBuilder.String(),
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,128 @@
|
||||
package prompt
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io/fs"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Skill 定义了从 SKILL.md 中解析出的标准化技能结构
|
||||
type Skill struct {
|
||||
Name string
|
||||
Description string
|
||||
Body string
|
||||
}
|
||||
|
||||
// SkillSummary 是技能的轻量目录项(仅元信息,不含 Body)
|
||||
type SkillSummary struct {
|
||||
Name string
|
||||
Description string
|
||||
}
|
||||
|
||||
// SkillLoader 负责从本地文件系统中加载技能模板
|
||||
type SkillLoader struct {
|
||||
workDir string
|
||||
}
|
||||
|
||||
func NewSkillLoader(workDir string) *SkillLoader {
|
||||
return &SkillLoader{workDir: workDir}
|
||||
}
|
||||
|
||||
// List 扫描 skills 目录,返回所有技能的轻量目录(仅名称和描述)
|
||||
func (s *SkillLoader) List() []SkillSummary {
|
||||
skillBaseDir := filepath.Join(s.workDir, "skills")
|
||||
if _, err := os.Stat(skillBaseDir); os.IsNotExist(err) {
|
||||
return nil
|
||||
}
|
||||
|
||||
var summaries []SkillSummary
|
||||
|
||||
filepath.WalkDir(skillBaseDir, func(path string, d fs.DirEntry, err error) error {
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
if d.IsDir() || d.Name() != "SKILL.md" {
|
||||
return nil
|
||||
}
|
||||
|
||||
content, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
skill := parseSkillMD(string(content))
|
||||
summaries = append(summaries, SkillSummary{
|
||||
Name: skill.Name,
|
||||
Description: skill.Description,
|
||||
})
|
||||
return nil
|
||||
})
|
||||
|
||||
return summaries
|
||||
}
|
||||
|
||||
// Load 按技能名称加载完整的 Skill(含 Body)
|
||||
func (s *SkillLoader) Load(name string) (*Skill, error) {
|
||||
skillBaseDir := filepath.Join(s.workDir, "skills")
|
||||
if _, err := os.Stat(skillBaseDir); os.IsNotExist(err) {
|
||||
return nil, fmt.Errorf("skills 目录不存在")
|
||||
}
|
||||
|
||||
var found *Skill
|
||||
|
||||
filepath.WalkDir(skillBaseDir, func(path string, d fs.DirEntry, err error) error {
|
||||
if err != nil || found != nil {
|
||||
return nil
|
||||
}
|
||||
if d.IsDir() || d.Name() != "SKILL.md" {
|
||||
return nil
|
||||
}
|
||||
|
||||
content, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
skill := parseSkillMD(string(content))
|
||||
if skill.Name == name {
|
||||
found = &skill
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
if found == nil {
|
||||
return nil, fmt.Errorf("未找到技能: %s", name)
|
||||
}
|
||||
return found, nil
|
||||
}
|
||||
|
||||
// parseSkillMD 解析带有 YAML Frontmatter 的 Markdown 内容
|
||||
func parseSkillMD(content string) Skill {
|
||||
skill := Skill{
|
||||
Name: "Unknown Skill",
|
||||
Description: "No description provided.",
|
||||
Body: content,
|
||||
}
|
||||
|
||||
if strings.HasPrefix(content, "---\n") || strings.HasPrefix(content, "---\r\n") {
|
||||
parts := strings.SplitN(content, "---", 3)
|
||||
if len(parts) == 3 {
|
||||
frontmatter := parts[1]
|
||||
skill.Body = strings.TrimSpace(parts[2])
|
||||
|
||||
lines := strings.Split(frontmatter, "\n")
|
||||
for _, line := range lines {
|
||||
line = strings.TrimSpace(line)
|
||||
if strings.HasPrefix(line, "name:") {
|
||||
skill.Name = strings.TrimSpace(strings.TrimPrefix(line, "name:"))
|
||||
} else if strings.HasPrefix(line, "description:") {
|
||||
skill.Description = strings.TrimSpace(strings.TrimPrefix(line, "description:"))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return skill
|
||||
}
|
||||
@@ -0,0 +1,59 @@
|
||||
package tools
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
|
||||
"go-tiny-claw/internal/prompt"
|
||||
"go-tiny-claw/internal/schema"
|
||||
)
|
||||
|
||||
type ReadSkillTool struct {
|
||||
loader *prompt.SkillLoader
|
||||
}
|
||||
|
||||
func NewReadSkillTool(workDir string) *ReadSkillTool {
|
||||
return &ReadSkillTool{
|
||||
loader: prompt.NewSkillLoader(workDir),
|
||||
}
|
||||
}
|
||||
|
||||
func (t *ReadSkillTool) Name() string {
|
||||
return "read_skill"
|
||||
}
|
||||
|
||||
func (t *ReadSkillTool) Definition() schema.ToolDefinition {
|
||||
return schema.ToolDefinition{
|
||||
Name: t.Name(),
|
||||
Description: "按技能名称加载完整的技能指令。在开始任务前可先用 read_skill 获取该技能的详细执行指南。支持技能: git-workflow",
|
||||
InputSchema: map[string]interface{}{
|
||||
"type": "object",
|
||||
"properties": map[string]interface{}{
|
||||
"name": map[string]interface{}{
|
||||
"type": "string",
|
||||
"description": "要加载的技能名称,如 git-workflow",
|
||||
},
|
||||
},
|
||||
"required": []string{"name"},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
type readSkillArgs struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
func (t *ReadSkillTool) Execute(ctx context.Context, args json.RawMessage) (string, error) {
|
||||
var input readSkillArgs
|
||||
if err := json.Unmarshal(args, &input); err != nil {
|
||||
return "", fmt.Errorf("参数解析失败: %w", err)
|
||||
}
|
||||
|
||||
skill, err := t.loader.Load(input.Name)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return fmt.Sprintf("技能: %s\n描述: %s\n\n--- 技能正文 ---\n%s", skill.Name, skill.Description, skill.Body), nil
|
||||
}
|
||||
@@ -0,0 +1,10 @@
|
||||
---
|
||||
name: git-workflow
|
||||
description: 当人类用户要求你“提交代码”、“保存变更”或执行 Git 相关操作时,必须使用此技能。
|
||||
---
|
||||
|
||||
# 提交流程 SOP
|
||||
|
||||
1. 先使用 `bash` 调用 `git status` 确认当前有哪些文件发生了改动。
|
||||
2. 你的 commit message 必须使用 Emoji 开头,例如:`🚀 feat: 增加新功能` 或 `🐛 fix: 修复 Bug`。
|
||||
3. 严禁使用 `git commit -am "update"` 这种敷衍的提交。
|
||||
Reference in New Issue
Block a user