修复: 灵感 CRUD 工具补齐 + handler 透传 affected(B-260801-01)

This commit is contained in:
lxy
2026-08-01 14:43:27 +08:00
parent 6b68aa63e8
commit ba578dfe6c
5 changed files with 76 additions and 15 deletions
+8 -4
View File
@@ -1148,22 +1148,26 @@ mod tests {
// register_git_tools 3→6) // register_git_tools 3→6)
// BUG-2026-07-07: 修正 baseline 计数(原写死 41 与实际 48 不符,HEAD 上即失败)。 // BUG-2026-07-07: 修正 baseline 计数(原写死 41 与实际 48 不符,HEAD 上即失败)。
// 48 = 34 data + 13 file + 1 http。data 34 含 6 知识图谱 + 2 基础设施 + 1 module + 6 git。 // 48 = 34 data + 13 file + 1 http。data 34 含 6 知识图谱 + 2 基础设施 + 1 module + 6 git。
// update_idea/delete_idea(2026-08-01): data 层 34→36(补 AI 改/删灵感专用工具,
// idea_repo.update_field / soft_delete 早已具备,工具层此前缺失致 AI 误用 update_task)。
// 50 = 36 data + 13 file + 1 http。
assert_eq!( assert_eq!(
registry.len(), registry.len(),
48, 50,
"工具总数应为 48(34 data + 13 file + 1 http),实际 {}", registry.len() "工具总数应为 50(36 data + 13 file + 1 http),实际 {}", registry.len()
); );
// 工具名集合基线:防 rename / 漏注册 / 误删除。 // 工具名集合基线:防 rename / 漏注册 / 误删除。
// data 层 34 个(持 db):CRUD/状态机/工作流/知识图谱任务关联 + 项目事件流 + 基础设施配置 + git 工具 // data 层 36 个(持 db):CRUD/状态机/工作流/知识图谱任务关联 + 项目事件流 + 基础设施配置 + git 工具
// file 层 13 个(不持 db):命令/读/列/写/改/元/追加/删/移/搜/grep/环境探测/符号解析 // file 层 13 个(不持 db):命令/读/列/写/改/元/追加/删/移/搜/grep/环境探测/符号解析
// http 层 1 个(不持 db):http_request // http 层 1 个(不持 db):http_request
let mut expected: Vec<&str> = vec![ let mut expected: Vec<&str> = vec![
// ── data 层 (34) ── // ── data 层 (36) ──
"list_projects", "list_tasks", "list_ideas", "list_projects", "list_tasks", "list_ideas",
"update_project", "create_project", "bind_directory", "update_project", "create_project", "bind_directory",
"create_task", "update_task", "advance_task", "create_task", "update_task", "advance_task",
"run_workflow", "delete_task", "create_idea", "run_workflow", "delete_task", "create_idea",
"update_idea", "delete_idea",
"delete_project", "restore_project", "purge_project", "delete_project", "restore_project", "purge_project",
"list_trash", "get_project_count", "get_task_count", "list_trash", "get_project_count", "get_task_count",
// 知识图谱 Phase 1 任务关联工具(register_task_graph_tools 6 个) // 知识图谱 Phase 1 任务关联工具(register_task_graph_tools 6 个)
+53 -5
View File
@@ -1,7 +1,9 @@
//! 灵感类 AI 工具声明式注册(register_idea_tools 2 个:list + create 迁入)。 //! 灵感类 AI 工具声明式注册(list + create + update + delete 共 4 个)。
//! //!
//! 迁自 `tool_registry.rs::register_idea_tools`(原 2 个 list_ideas/create_idea), //! 迁自 `tool_registry.rs::register_idea_tools`(原 2 个 list_ideas/create_idea),
//! 改用 `declare_tool!` 宏。 //! 改用 `declare_tool!` 宏。新增 update_idea / delete_idea(AI 改/删灵感的专用工具,
//! 后端 idea_repo.update_field / soft_delete 早已具备,工具层此前缺失,致 AI 改/删灵感
//! 时被迫误用 update_task/delete_task)。
//! //!
//! 迁移策略(handler 逻辑零变更): //! 迁移策略(handler 逻辑零变更):
//! - handler body 逐字照搬原 `register_idea_tools` 内 async move 块(逻辑等价), //! - handler body 逐字照搬原 `register_idea_tools` 内 async move 块(逻辑等价),
@@ -11,7 +13,7 @@
//! - 复用类型/常量:new_id/IdeaStatus/IdeaRecord/now_millis(各 pub(crate) 来源,与原同源) + //! - 复用类型/常量:new_id/IdeaStatus/IdeaRecord/now_millis(各 pub(crate) 来源,与原同源) +
//! MAX_LIST_RESULTS(super::tool_registry 单真相源)。 //! MAX_LIST_RESULTS(super::tool_registry 单真相源)。
//! //!
//! 等价性验证:基线测试 `test_build_ai_tool_registry_baseline_tool_count` 断言 48 总量 + //! 等价性验证:基线测试 `test_build_ai_tool_registry_baseline_tool_count` 断言 50 总量 +
//! 工具名集合稳定(防 rename / 漏注册)。 //! 工具名集合稳定(防 rename / 漏注册)。
use std::sync::Arc; use std::sync::Arc;
@@ -25,11 +27,11 @@ use df_types::types::{new_id, IdeaStatus};
use crate::commands::now_millis; use crate::commands::now_millis;
use crate::commands::ai::tool_registry::MAX_LIST_RESULTS; use crate::commands::ai::tool_registry::MAX_LIST_RESULTS;
/// 注册 2 个灵感类工具(list/create)到 `$registry`。 /// 注册 4 个灵感类工具(list/create/update/delete)到 `$registry`。
/// ///
/// 与原手写 register(name, desc, schema, risk, handler) 语义 1:1: /// 与原手写 register(name, desc, schema, risk, handler) 语义 1:1:
/// - name/desc/schema 字符串与 JSON Schema 逐字照搬原定义 /// - name/desc/schema 字符串与 JSON Schema 逐字照搬原定义
/// - risk 与原一致(list=Low,create=Medium) /// - risk 与原一致(list=Low,create/update=Medium,delete=High)
/// - handler body 与原 async move 块逐字一致(逻辑零变更) /// - handler body 与原 async move 块逐字一致(逻辑零变更)
/// ///
/// 唯一差异:闭包包装改由 `declare_tool!` 宏生成,handler body 直接写业务逻辑。 /// 唯一差异:闭包包装改由 `declare_tool!` 宏生成,handler body 直接写业务逻辑。
@@ -84,4 +86,50 @@ pub fn register(registry: &mut AiToolRegistry, db: &Arc<Database>) {
Ok(serde_json::json!({ "id": id, "title": title, "status": "draft" })) Ok(serde_json::json!({ "id": id, "title": title, "status": "draft" }))
} }
); );
// update_idea / delete_idea:AI 改/删灵感的专用工具(对标 update_task / delete_task)。
// 后端 idea_repo.update_field / soft_delete 早已具备,工具层此前缺失,致 AI 改/删灵感
// 时被迫误用 update_task(任务表)——既改错表又触发 status 收口误报。
// 与 update_task 同源:白名单复用 df-storage CRUD(is_allowed_column("ideas", field)),
// 双重防御(schema 限制 + 白名单校验)。返回 updated 透传 affected bool(不硬编码 true),
// 0 行(id 不存在 / 已删)→ updated=false + title 空,让 AI 知道 id 不对而非假成功。
declare_tool!(
registry,
db: Arc<Database>,
"update_idea",
"更新灵感的指定字段(title/description/priority/tags/source/score 等),需要提供灵感 ID、字段名和新值。",
RiskLevel::Medium,
schema: object_schema(vec![("id", "string", true), ("field", "string", true), ("value", "string", true)]),
args => {
let id = args["id"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 id"))?;
let field = args["field"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 field"))?;
let value = args["value"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 value"))?;
// 复用 df-storage CRUD 白名单(按表隔离),与 update_field 校验同源
if !df_storage::crud::is_allowed_column("ideas", field) {
anyhow::bail!("不允许更新字段 '{}'", field);
}
let repo = df_storage::crud::IdeaRepo::new(&db);
let affected = repo.update_field(id, field, value).await?;
// 对齐 update_task:返回 title 供前端卡片友好展示(防裸 UUID 辨识度低)。
// 0 行(id 不存在 / 已软删)→ affected=false 且 get_by_id 返 None→title 空,
// 透传 affected 让 AI 知道 id 不对而非假成功。
let title = repo.get_by_id(id).await?.map(|i| i.title).unwrap_or_default();
Ok(serde_json::json!({ "id": id, "title": title, "field": field, "updated": affected }))
}
);
declare_tool!(
registry,
db: Arc<Database>,
"delete_idea",
"删除灵感(软删除,移入回收站,可恢复)",
RiskLevel::Medium,
schema: object_schema(vec![("id", "string", true)]),
args => {
let id = args["id"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 id"))?;
let repo = df_storage::crud::IdeaRepo::new(&db);
let affected = repo.soft_delete(id).await?;
Ok(serde_json::json!({ "id": id, "deleted": affected }))
}
);
} }
+7 -2
View File
@@ -52,8 +52,13 @@ pub fn register(registry: &mut AiToolRegistry, db: &Arc<Database>) {
anyhow::bail!("不允许更新字段 '{}'", field); anyhow::bail!("不允许更新字段 '{}'", field);
} }
let repo = df_storage::crud::ProjectRepo::new(&db); let repo = df_storage::crud::ProjectRepo::new(&db);
repo.update_field(id, field, value).await?; // 透传 affected:0 行(id 不存在 / 已软删)→ updated=false + name 空,
Ok(serde_json::json!({ "id": id, "field": field, "updated": true })) // 让 AI 知道 id 不对而非假成功(对齐 update_idea / update_task)。
let updated = repo.update_field(id, field, value).await?;
// 返回 name 供前端卡片友好展示(对齐 update_task 返 title);update_field 返 bool 不含
// record,单独 get_by_id 取 name;0 行时 get_by_id 返 None→name 空,配合 updated:false。
let name = repo.get_by_id(id).await?.map(|p| p.name).unwrap_or_default();
Ok(serde_json::json!({ "id": id, "name": name, "field": field, "updated": updated }))
} }
); );
+4 -2
View File
@@ -177,11 +177,13 @@ pub fn register(registry: &mut AiToolRegistry, db: &Arc<Database>) {
anyhow::bail!("不允许更新字段 '{}'", field); anyhow::bail!("不允许更新字段 '{}'", field);
} }
let repo = df_storage::crud::TaskRepo::new(&db); let repo = df_storage::crud::TaskRepo::new(&db);
repo.update_field(id, field, value).await?; // 透传 affected:0 行(id 不存在 / 已软删)→ updated=false + title 空,
// 让 AI 知道 id 不对而非假成功(对齐 update_idea)。
let updated = repo.update_field(id, field, value).await?;
// UX-260619-04: 返回 title 供前端卡片友好展示(对齐 advance_task 返 TaskRecord, // UX-260619-04: 返回 title 供前端卡片友好展示(对齐 advance_task 返 TaskRecord,
// 防裸 UUID id 辨识度低)。update_field 返 bool 不含 record,单独 get_by_id 取 title。 // 防裸 UUID id 辨识度低)。update_field 返 bool 不含 record,单独 get_by_id 取 title。
let title = repo.get_by_id(id).await?.map(|t| t.title).unwrap_or_default(); let title = repo.get_by_id(id).await?.map(|t| t.title).unwrap_or_default();
Ok(serde_json::json!({ "id": id, "title": title, "field": field, "updated": true })) Ok(serde_json::json!({ "id": id, "title": title, "field": field, "updated": updated }))
} }
); );
@@ -228,9 +228,11 @@ pub fn register(registry: &mut AiToolRegistry, db: &Arc<Database>) {
.map_err(|e| anyhow::anyhow!("content_json 不是合法 JSON: {e}"))?; .map_err(|e| anyhow::anyhow!("content_json 不是合法 JSON: {e}"))?;
} }
let repo = df_storage::crud::TaskRepo::new(&db); let repo = df_storage::crud::TaskRepo::new(&db);
repo.update_field(id, "content_json", content_json).await?; // 透传 affected:0 行(id 不存在 / 已软删)→ updated=false + title 空,
// 让 AI 知道 id 不对而非假成功(对齐 update_idea / update_task)。
let updated = repo.update_field(id, "content_json", content_json).await?;
let title = repo.get_by_id(id).await?.map(|t| t.title).unwrap_or_default(); let title = repo.get_by_id(id).await?.map(|t| t.title).unwrap_or_default();
Ok(serde_json::json!({ "id": id, "title": title, "field": "content_json", "updated": true })) Ok(serde_json::json!({ "id": id, "title": title, "field": "content_json", "updated": updated }))
} }
); );