freedak
f7a720204a
- 移除 GovAI, nomifun-tauri, 算力盒子 的 submodule 引用 - 添加所有子项目的完整源代码 - 保留原始 .git 为 .git.bak 备份
64 lines
3.0 KiB
Rust
64 lines
3.0 KiB
Rust
//! E2E regression: the `knowledge_search` tool finds content in a base whose
|
|
//! directory carries a self-`.gitignore: *` — the exact scenario where ripgrep
|
|
//! (hidden-dir skip + gitignore) returned zero. Proves the hard bug is dead.
|
|
|
|
use std::sync::Arc;
|
|
|
|
use nomi_agent::knowledge_tools::KnowledgeSearchTool;
|
|
use nomi_tools::Tool;
|
|
use serde_json::json;
|
|
|
|
/// `nomifun_realtime` does not ship a public no-op broadcaster (its only one is
|
|
/// `pub(crate)` in nomifun-knowledge's testutil), so define a local one. The
|
|
/// `EventBroadcaster` trait is a single `broadcast` method; the emitter never
|
|
/// fires during a read-only search, but the service still requires one.
|
|
struct NoopBroadcaster;
|
|
|
|
impl nomifun_realtime::EventBroadcaster for NoopBroadcaster {
|
|
fn broadcast(&self, _event: nomifun_api_types::WebSocketMessage<serde_json::Value>) {}
|
|
}
|
|
|
|
async fn build_service() -> (Arc<nomifun_knowledge::KnowledgeService>, tempfile::TempDir) {
|
|
let db = nomifun_db::init_database_memory().await.expect("in-memory db");
|
|
let repo = Arc::new(nomifun_db::SqliteKnowledgeRepository::new(db.pool().clone()));
|
|
let tmp = tempfile::tempdir().unwrap();
|
|
let emitter = nomifun_knowledge::KnowledgeEventEmitter::new(Arc::new(NoopBroadcaster));
|
|
let svc = Arc::new(nomifun_knowledge::KnowledgeService::new(repo, tmp.path(), emitter));
|
|
(svc, tmp)
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn knowledge_search_tool_finds_topic_through_full_stack() {
|
|
let (svc, _tmp) = build_service().await;
|
|
let info = svc.create_base("运维手册", "", None, None).await.unwrap();
|
|
let root = svc.data_dir().join("knowledge").join(&info.id);
|
|
// The self-ignore the mount writes — must NOT blind the search.
|
|
std::fs::write(root.join(".gitignore"), "*\n").unwrap();
|
|
std::fs::write(root.join("rollback.md"), "# 回滚流程\n回滚分三步\n").unwrap();
|
|
|
|
let sink: Arc<dyn nomi_agent::knowledge_tools::KnowledgeRetrievalSink> =
|
|
Arc::new(nomifun_ai_agent::LiveKnowledgeRetrievalSink { service: svc });
|
|
let tool = KnowledgeSearchTool::new(sink, vec![info.id]);
|
|
|
|
let res = tool.execute(json!({"query": "回滚"})).await;
|
|
assert!(!res.is_error, "tool errored: {}", res.content);
|
|
assert!(res.content.contains("rollback.md"), "must surface the doc:\n{}", res.content);
|
|
assert!(res.content.contains("回滚流程"), "must include heading:\n{}", res.content);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn knowledge_search_tool_reports_no_match_cleanly() {
|
|
let (svc, _tmp) = build_service().await;
|
|
let info = svc.create_base("库", "", None, None).await.unwrap();
|
|
let root = svc.data_dir().join("knowledge").join(&info.id);
|
|
std::fs::write(root.join("a.md"), "# A\nunrelated content\n").unwrap();
|
|
|
|
let sink: Arc<dyn nomi_agent::knowledge_tools::KnowledgeRetrievalSink> =
|
|
Arc::new(nomifun_ai_agent::LiveKnowledgeRetrievalSink { service: svc });
|
|
let tool = KnowledgeSearchTool::new(sink, vec![info.id]);
|
|
|
|
let res = tool.execute(json!({"query": "完全不存在的主题词"})).await;
|
|
assert!(!res.is_error);
|
|
assert!(res.content.contains("No matches"), "{}", res.content);
|
|
}
|