Commit Graph

137 Commits

Author SHA1 Message Date
kurihada 026429cb2f refactor(server): 将 Core 模块从动态导入改为静态导入
- 移除 adapter.ts 中约 160 行冗余接口定义
- 简化 initCore 函数,改为初始化检查逻辑
- 简化 getOrCreateAgent,直接使用 ConfigurationError 类
- 更新缓存类型注解使用 Core 导出的类型
- 简化事件订阅代码,直接使用 agentEventEmitter
- 在 Core index.ts 中添加 agentEventEmitter 导出
- 更新测试文件适配静态导入模式
2025-12-16 19:54:20 +08:00
kurihada 08d481483c feat(ui): 添加子 Agent 进度显示功能
当 build agent 调用 guide/explore 等子 agent 时,
用户可以在 web 页面实时看到子 agent 的执行进度

实现方案:
- Core: 使用 EventEmitter 模式发射子 agent 事件
- Server: 订阅事件并转发到 WebSocket
- UI: 处理事件并渲染 SubagentProgress 组件

新增文件:
- packages/core/src/agent/events.ts
- packages/ui/src/components/SubagentProgress.tsx

修改文件:
- core: executor.ts, manager.ts, types.ts, task.ts
- server: adapter.ts, types.ts
- ui: useChat.ts, types.ts
- web: Chat.tsx
2025-12-16 19:38:36 +08:00
kurihada f0ff887129 feat(core): 添加 guide agent 并使用模板变量
- 新增 guide agent 用于回答工具使用问题
- 更新 build agent 提示词使用模板变量
- 添加 GUIDE_SUBAGENT_TYPE 和 EXPLORE_SUBAGENT_TYPE 变量
2025-12-16 18:26:22 +08:00
kurihada 87edd6e35b refactor(core): 移除 BASH_TOOL_EXTRA_NOTES 和 GIT_COMMIT_AND_PR_CREATION_INSTRUCTION
- 从模板引擎中移除这两个函数的注册
- 从 bash 工具描述中移除对应的模板变量
- 删除不再需要的 git-instructions.txt 文件
- 更新单元测试,移除相关测试用例
2025-12-16 17:00:13 +08:00
kurihada 09dc046f8f feat(core): 扩展模板引擎支持函数调用和算术运算
- 新增模板函数调用语法 ${FUNC_NAME()}
- 新增除法算术运算 ${value/divisor}
- 添加 bash 工具配置常量 (CUSTOM_TIMEOUT_MS 等)
- 更新 bash 工具描述使用动态模板格式
- 添加 Git 操作指令文本 (commit/PR 说明)
- 添加模板渲染器单元测试 (31 个测试用例)
2025-12-16 16:52:46 +08:00
kurihada 7706af2c79 feat(core): 更新工具描述支持动态模板变量
- grep.txt: 使用 ${GREP_TOOL_NAME}、${BASH_TOOL_NAME}、${TASK_TOOL_NAME} 变量
- ask_user_question: 将 description 移至独立 txt 文件
- load_description.ts: 添加 ask_user_question -> plan 分类映射
2025-12-16 16:26:27 +08:00
kurihada 1d380d0bcb refactor(core): 统一模板引擎到 src/template/ 目录
- 将 agent/prompt-template/ 目录合并到 src/template/
- 新增通用模板函数 renderTemplate、render
- 新增 Agent 特定函数 renderPromptTemplate、renderPrompt
- 新增 createToolDescriptionContext 支持工具描述模板变量
- 支持 ${GREP_TOOL_NAME} 等 Claude Code 风格变量
- 更新所有相关导入路径
2025-12-16 16:05:10 +08:00
kurihada 30f35a6562 refactor(core): 重命名 grep_content 工具为 grep
- 重命名 grep_content.ts 为 grep.ts
- 更新工具名称从 grep_content 改为 grep
- 更新所有引用该工具的代码和测试文件
- 更新描述文件名称
2025-12-16 15:03:33 +08:00
kurihada 452f19e2a9 refactor(core): 移除未使用的 prompt-template templates 目录 2025-12-16 14:19:36 +08:00
kurihada 3baf3c0171 fix(core): 在 Agent.resolveSystemPrompt 中集成动态模板渲染 2025-12-16 14:16:52 +08:00
kurihada 58f1bc8718 feat(core): 实现动态提示词模板系统
- 新增 prompt-template 模块,支持运行时变量替换
- 支持 ${variable}、${obj.prop}、${cond ? "a" : "b"} 语法
- AgentInfo 新增 promptTemplate 字段标记动态模板
- Plan Agent 提示词改用模板语法,支持动态工具名和计划文件路径
- AgentExecutor.buildSystemPrompt 集成模板渲染
- 新增 27 个单元测试验证模板功能
2025-12-16 14:15:10 +08:00
kurihada a32c83480d feat(core): 新增 glob 工具,支持文件模式匹配
- 新增 glob 工具,支持 **/*.ts 等 glob 模式匹配文件
- 结果按修改时间排序,限制返回 100 个结果
- 自动忽略 node_modules、.git 等目录
- 更新 Plan Agent 提示词使用 glob 替代 search_files
- 在 Plan Agent 工具列表中启用 glob
2025-12-16 14:01:42 +08:00
kurihada cd0c2bdbfb feat(core): 重构 Plan 模式工具,新增 allowedWritePaths 路径限制
Plan 工具重构:
- 移除 plan_mode_respond 工具
- 新增 ask_user_question 工具:向用户提问并获取回复
- 新增 enter_plan_mode 工具:进入计划模式
- 新增 exit_plan_mode 工具:退出计划模式

allowedWritePaths 功能:
- AgentFilePermission 新增 allowedWritePaths 字段
- permission-merger 添加 isPathInAllowedWritePaths 检查函数
- executor 在写入操作时检查路径限制
2025-12-16 13:49:45 +08:00
kurihada f7b934a69e refactor(core): 简化 Agent 提示词系统,移除 provider 动态加载
- 直接在 build.ts 和 plan.ts 中定义 prompt 字段
- 简化 system-prompt.ts,移除 provider/forProvider/plan 函数
- 简化 agent.ts 中的 resolveSystemPrompt 方法
- 删除不再需要的提示词文件 (anthropic/beast/gemini/qwen/plan.txt)
- 更新 package.json build 脚本,移除复制 prompts 目录
2025-12-16 13:48:56 +08:00
kurihada 7f51d33319 fix(core): 初始化 Task 工具上下文,修复子 Agent 无法启动的问题 2025-12-16 11:56:12 +08:00
kurihada 6817aebfc4 refactor(core): 简化 Plan Agent bash 权限规则,移除 deny 改用 ask 2025-12-16 11:42:48 +08:00
kurihada f116b8a14a fix(core): 添加 Agent 级别的 bash 权限检查
在 getVercelTools 中添加 Agent 权限检查层,确保 Plan Agent 的 bash 规则
(如 touch * -> deny)在工具执行前生效,而不是仅依赖全局权限管理器。
2025-12-16 11:28:33 +08:00
kurihada 70394ed06c fix(core): 修复 Plan Agent 通过 heredoc 绕过写入限制的安全漏洞
- 重排 bash 权限规则顺序,deny 规则置于 allow 规则之前
- 添加 heredoc 重定向检测规则 (* << *)
- 新增 checkRedirectInRawCommand 预检函数,在 tree-sitter 解析前检测重定向
- 禁用 Plan Agent 的 tool_search 工具,防止动态发现写入工具
- 添加更多危险命令: ln, install, truncate, dd, tee
2025-12-16 11:23:02 +08:00
kurihada e698ec2a64 feat(ui): display agent name in message header 2025-12-16 10:43:02 +08:00
kurihada a6c1e792fa refactor(core,server): 简化配置加载,移除 workdir 参数
Provider 和 Agent 配置统一从全局目录加载,无需传递 workdir:
- ProviderRegistry.init() 不再需要 workdir 参数
- AgentRegistry.init() 不再需要 workdir 参数
- 配置文件路径统一使用 ~/.ai-terminal-assistant/
2025-12-16 10:23:28 +08:00
kurihada 6a28c98789 refactor(core): 统一存储路径到 ~/.ai-terminal-assistant/
- 新增 constants/paths.ts 模块统一管理存储路径
- Session 数据从 ~/.local/share/ai-assist/ 迁移到 ~/.ai-terminal-assistant/data/
- Checkpoint 从 ~/.ai-assist/checkpoints/ 迁移到 ~/.ai-terminal-assistant/checkpoints/
- MCP 配置从 ~/.ai-assist/config.* 迁移到 ~/.ai-terminal-assistant/mcp.*
- Agent 配置保持在 ~/.ai-terminal-assistant/agents.json
2025-12-16 10:09:21 +08:00
kurihada 9376887995 refactor(core): 统一配置系统,移除 config.json
- 移除 config.json,所有配置统一从 agents.json 和 providers.json 读取
- config-loader.ts 从全局目录 ~/.ai-terminal-assistant/ 加载配置
- loadConfig() 从 agentRegistry.getGlobalConfig() 获取 defaults.model
- 添加 loadVisionConfig() 支持 Vision 模型配置
- Tavily API Key 仅从环境变量读取
- UI AgentDefaultsEditor 添加 Vision 模型配置界面
- 更新相关测试
2025-12-16 00:33:29 +08:00
kurihada 76b1ae1573 fix(ui): CodeBlock 添加防抖优化,减少流式输出时的高亮闪烁
- 使用 ref 跟踪上一次高亮的代码,避免重复高亮
- 添加 150ms 防抖延迟,减少流式输出时的高亮次数
- 优化渲染逻辑,未高亮时显示纯文本
2025-12-15 23:18:07 +08:00
kurihada 92619df026 feat(web): 移除 Settings 按钮,在 Header 直接显示工作目录
- 移除 ConfigPanel 及其入口按钮
- 在 Chat 页面 Header 左侧显示当前工作目录
- 初始化时并行获取会话列表和工作目录
2025-12-15 23:17:27 +08:00
kurihada f3995d158c feat(core): 实现动态系统提示词加载
- 创建 SystemPrompt 模块,根据模型类型动态加载对应的提示词
- 从 OpenCode 拷贝提示词文件 (anthropic/beast/gemini/qwen/plan.txt)
- 修改 Build/Plan Agent 预设,移除内联 prompt
- Agent 类新增 resolveSystemPrompt 方法处理提示词加载逻辑
- 构建脚本自动复制 prompts 目录到 dist
2025-12-15 23:05:08 +08:00
kurihada 013e9a2712 feat(core): 移除内置 Agent 的 maxSteps 限制,默认使用 Build 模式
- 移除 build/plan/general/explore 的 maxSteps 配置
- 内置 Agent 不再限制步数(与 OpenCode 保持一致)
- 用户自定义 Agent 仍可按需配置 maxSteps
- 新建 Agent 默认使用 Build 模式
2025-12-15 22:27:42 +08:00
kurihada df811395f8 feat(core): 增强 General 和 Explore Agent 预设
General Agent:
- 添加详细的 prompt 引导子代理行为
- 禁用 todo_read/todo_write(避免与父 Agent 冲突)
- 增加 maxSteps 到 20

Explore Agent:
- 支持 thoroughness 参数 (quick/medium/thorough)
- 启用只读 bash 命令(ls, tree, find, git log 等)
- 禁用写入工具和 Todo 工具
- bash 默认策略改为 deny(比 Plan 更严格)
- 添加详细的探索深度指导 prompt

Plan Agent:
- 添加 web: 'ask' 权限配置
2025-12-15 22:01:53 +08:00
kurihada 35d87a04fb feat(core): 增强 Plan Agent 支持细粒度 bash 权限和模式切换
- Plan Agent 细粒度 bash 权限:允许只读命令 (ls, grep, git log 等),
  禁止危险操作 (rm, mv, git commit 等)
- 新增 plan_mode_respond 工具:结构化输出计划、进度和探索状态
- Agent.switchMode() 方法:支持 Build ↔ Plan 模式切换保留对话历史
- WebSocket mode_switch 消息:支持运行时动态切换模式
- 更新 Plan Agent prompt 引导使用 plan_mode_respond 工具
2025-12-15 20:51:57 +08:00
kurihada f238368f87 feat(ui): 支持 Auto Edit 运行时切换立即生效
- 添加 config_update WebSocket 消息类型
- setAutoApprove 切换时立即发送配置更新到服务端
- 服务端实时更新会话的 auto-approve 配置
2025-12-15 19:52:06 +08:00
kurihada ec3c7bccf9 feat(ui): 实现 Agent 模式切换和 Auto Edit 功能
- 添加 AgentModeSelector 组件,支持 Build/Plan 模式切换
- Build 模式下显示 Auto Edit 开关,自动授权文件写入/编辑
- 扩展 useChat hook 添加会话级别的 agentMode/autoApprove 状态
- 服务端支持解析和应用 Agent 模式配置
- 权限处理器实现 auto-approve 检查(仅 write/edit,不含 delete)
2025-12-15 19:42:51 +08:00
kurihada f09f8f2b03 fix: 修复多个问题
- fix(core): 修复 todoManager 未初始化导致待办工具无法使用
- fix(core): 为 LSP 初始化添加 10 秒超时,防止 write_file 卡住
- fix(ui): 停止对话时保留已流式输出的内容
2025-12-15 18:40:07 +08:00
kurihada 11d4abfc50 feat(core): 实现 Doom Loop 检测和软性 maxSteps 限制
- 新增 doom-loop.ts: 实现 Doom Loop 检测器,检测连续 3 次相同工具调用
- 修改 agent.ts: 集成检测器,添加 onDoomLoop 回调,动态 maxSteps
- 修改 registry.ts: 默认 maxSteps 从 10 改为 50
- 更新 index.ts: 导出新模块

参考 OpenCode/OpenHands 等开源项目的多层防御策略
2025-12-15 18:14:09 +08:00
kurihada 3fd8fd98b8 feat(ui): 优化流式输出工具调用渲染
- 添加 tool_start/tool_end WebSocket 事件支持
- 流式消息复用 ChatMessage 组件渲染工具调用卡片
- 修复 AI SDK v5 格式兼容问题(input/output 字段)
- 修复会话恢复时 tool-result 格式错误
- 放宽 ToolState schema 中 input 字段类型为 unknown
2025-12-15 17:35:39 +08:00
kurihada 865e0906b9 feat(ui): 实现 @ 文件提及自动补全功能
- Core: 添加 file-index 模块,使用 ripgrep 索引文件,fuzzysort 模糊搜索
- Server: 添加 /api/files/search 端点,支持文件模糊搜索
- Server: WebSocket 消息处理中将 @filepath 转换为 ./filepath 格式
- UI: 新增 FileMenu 组件,显示文件搜索结果列表
- UI: 新增 FileMentionTag 组件,高亮显示文件提及
- UI: 新增 useFileMention hook,管理文件提及状态
- UI: ChatInput 集成 @ 触发的文件自动补全
- UI: ChatMessage 用户消息中高亮显示 @filepath
2025-12-15 16:32:59 +08:00
kurihada 5b7b0ff1e4 feat(ui): 实现深色/浅色主题切换功能
- 添加 CSS 变量定义浅色和深色主题色板
- 扩展 Tailwind 配置支持语义化颜色 (surface-*, fg-*, line-*, code)
- 创建 useTheme hook 管理主题状态和持久化
- 创建 ThemeToggle 组件支持三种模式 (light/dark/system)
- 迁移所有组件从硬编码 gray-* 到语义化颜色
- 支持系统主题偏好检测 (prefers-color-scheme)
- 添加主题初始化脚本防止闪烁 (FOUC)
2025-12-15 15:47:32 +08:00
kurihada cd0dd814ab feat(ui): 实现消息 parts 有序渲染
- Server: API 返回 parts 数组保持原始顺序
- Server: 添加 MessagePart 类型定义 (text/tool/reasoning)
- UI: ChatMessage 按 parts 顺序交叉渲染文本和工具调用
- UI: 新增 ToolPartItem 组件渲染单个工具 part
- UI: useChat 创建消息时添加 parts 字段
2025-12-15 14:46:00 +08:00
kurihada 2150abde7c feat(agent): 实现流式输出取消功能
- 添加 AbortController 管理,支持取消正在进行的请求
- Agent.chat() 新增 abortSignal 参数,传递给 streamText/generateText
- 取消时保存用户消息和 AI 已输出的部分内容
- cancelProcessing 实际调用 abort() 中止流式请求
2025-12-15 14:21:24 +08:00
kurihada d54b9788fa fix(storage): 修复 multi-step 对话产生多个 assistant 消息问题
- 同一轮对话的后续 assistant 消息追加 Parts 到现有消息
- 修复 AI SDK tool-call 参数字段名(input 而非 args)
- 修复 setToolCompleted/setToolError 中 discriminated union 类型访问
2025-12-15 13:58:10 +08:00
kurihada 9f456c1029 refactor(storage): 重构消息存储为 2-message 格式
采用 OpenCode 风格的消息存储架构:
- 只有 user 和 assistant 两种角色,移除 tool/system
- ToolPart 使用状态机模式 (pending → running → completed/error)
- 新增 toModelMessages() 转换函数用于调用 AI SDK
- 删除 message-merger.ts,存储层直接返回正确格式

主要改动:
- parts.ts: ToolState 状态机(pending/running/completed/error)
- message.ts: 移除 system role,添加 parentId 关联
- converter.ts: 新增 toModelMessages() 格式转换
- manager.ts: 重构 syncMessages/partsToModelMessages
- sessions.ts: 简化路由,直接从 Core Storage 读取
2025-12-15 13:35:32 +08:00
kurihada eda2ccb171 feat(api): 实现消息合并 API,支持工具调用显示
- 新增 MergedMessage、ToolCallInfo 类型定义
- 创建 message-merger.ts 消息合并工具
- 更新 sessions 路由使用合并后的消息格式
- 前端新增 ToolCallsDisplay 组件展示工具调用
- 工具调用显示状态、时长,可展开查看参数和结果
2025-12-15 12:21:16 +08:00
kurihada e9e8bfa30a fix(storage): 修复 ToolPart args 字段缺失导致的验证错误 2025-12-15 11:30:43 +08:00
kurihada 527692ec03 refactor(storage): 采用 OpenCode 风格三层存储结构
重构消息存储系统,从"每条消息一个文件"改为分层存储:
- Session → Message → Parts 三层结构
- 12 种 Part 类型(TextPart, ToolPart, ReasoningPart 等)
- ToolPart 状态机(pending → running → completed/error)
- 通用 Storage API(read/write/list/remove)

新增文件:
- parts.ts: Part 类型定义(Zod schema)
- message.ts: MessageInfo 类型定义
- id.ts: ID 生成器
- storage/: 分层存储实现

删除旧文件:
- storage.ts, types.ts, migration.ts

存储路径:
~/.local/share/ai-assist/
├── session/{projectId}/{sessionId}.json
├── message/{sessionId}/{messageId}.json
├── part/{messageId}/{partId}.json
└── todo/{sessionId}.json
2025-12-15 11:16:10 +08:00
kurihada b8fcb65f73 fix(server): 修复消息持久化问题
- 在 adapter.ts 中为 Agent 绑定 Core SessionManager
- 添加 getOrCreateSessionManager 函数创建和恢复会话
- 修改 getOrCreateAgent 为异步函数
- 在 sessions.ts 中转换 AI SDK 消息格式为字符串
2025-12-15 10:45:58 +08:00
kurihada bbce1de60c fix(ui): 优化 WebSocket 连接和 PWA meta 标签
- WebSocket onopen 检查 isClosingRef,组件卸载后自动关闭连接
- 更新 PWA meta 标签为 mobile-web-app-capable
2025-12-15 10:29:49 +08:00
kurihada 9e55237dae fix(ui): 修复 React key 警告和 forwardRef 问题
- ChatMessage/SessionItem 使用 forwardRef 支持 AnimatePresence
- useChat 为 message_received/done 事件生成唯一消息 ID
- sessions API 为历史消息添加 ID 字段
- cli 添加 @types/inquirer 依赖
2025-12-15 10:24:45 +08:00
kurihada 842cf1a3e8 fix(ui): 优化会话初始化逻辑
- 首次启动时自动创建会话
- 用户删除所有会话后不再自动创建,显示空状态
- 使用 localStorage 记录是否曾有会话
2025-12-15 10:05:17 +08:00
kurihada 6342a46e59 refactor(storage): 统一消息存储到 Core 层
问题:Server 端只存储最终文本响应,工具调用的中间消息丢失。

解决方案:
- Agent.chat() 返回 ChatResult,包含完整消息链
- Server SessionManager 简化为只管理会话元数据
- 消息 API 改为从 Core Storage 读取
- 移除 Server 端的消息存储和 addMessage 方法

影响范围:
- core: Agent.chat() 返回类型变更
- server: SessionManager 接口变更,移除消息存储
- server: GET /sessions/:id/messages 从 Core 读取
- server: 移除 POST /sessions/:id/messages 端点
2025-12-15 10:04:22 +08:00
kurihada a657af9bb7 test(server): 修复 SessionManager 测试的 session 清理问题
- 添加 afterEach 钩子清理测试中创建的 session
- 引入 createTrackedSession 辅助函数追踪需要清理的 session
- 防止测试产生的 session 持久化到磁盘影响用户数据
2025-12-15 00:55:02 +08:00
kurihada 5b20420ccd test(server): 增强 auth/ws/sse 测试覆盖率
- auth/token.ts: 50% → 100%
  - 新增 authMiddleware 中间件完整测试
  - 覆盖本地 IP 检测、远程认证、跳过路径等场景
  - 新增 getAuthContext 测试

- ws.ts: 90% → 98%
  - 新增 Blob/非标准数据类型处理测试
  - 新增 addMessage 返回 null 场景测试
  - 新增 tool_response 和 permission_response 边界测试

- sse.ts: 新增事件格式化和统计测试

测试数量: 344 → 369 (+25)
总体覆盖率: 80.82% → 82.98%
2025-12-15 00:36:43 +08:00
kurihada 7bc4f006a0 test(server): 补充 checkpoints/mcp/sse/adapter 测试
- 新增 checkpoints.test.ts: 29 个测试覆盖 Checkpoint 管理 API
- 新增 mcp.test.ts: 15 个测试覆盖 MCP 服务器管理 API
- 扩展 sse.test.ts: 添加 handleSSE 路由测试
- 扩展 adapter.test.ts: 添加 cancelProcessing, getContextUsage,
  compressContext, processMessage 测试

覆盖率提升: 60% -> 80.82%
测试数量: 288 -> 344
2025-12-15 00:32:04 +08:00