refactor: consolidate hypotheses, agent_signals, and process_logs into action-based smart tools
This commit is contained in:
1 parent
79209da711
commit
3b146f91c2
17 files changed
+438
-359
No files matched your search
@@ -1223,7 +1223,7 @@ impl McpTool for SummarizeSubgraphHandler {
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::handlers::meta::{BroadcastAgentSignalHandler, QueryAgentSignalsHandler};
|
||||
use crate::handlers::meta::AgentSignalsHandler;
|
||||
use serde_json::json;
|
||||
|
||||
|
||||
@@ -1515,19 +1515,21 @@ mod tests {
|
||||
}), state.clone()).await.unwrap();
|
||||
assert_eq!(del_rel_res, "Relations deleted");
|
||||
|
||||
let bcast_handler = BroadcastAgentSignalHandler;
|
||||
let bcast_handler = AgentSignalsHandler;
|
||||
let bcast_res = bcast_handler.execute(json!({
|
||||
"action": "broadcast",
|
||||
"sender": "agent1",
|
||||
"signal_type": "task_completed",
|
||||
"payload": "fix_bug"
|
||||
}), state.clone()).await.unwrap();
|
||||
assert!(bcast_res.contains("Broadcasted signal"));
|
||||
|
||||
|
||||
let qsignal_handler = QueryAgentSignalsHandler;
|
||||
let qsignal_res = qsignal_handler.execute(json!({"sender": "agent1"}), state.clone()).await.unwrap();
|
||||
let qsignal_handler = AgentSignalsHandler;
|
||||
let qsignal_res = qsignal_handler.execute(json!({
|
||||
"action": "query",
|
||||
"sender": "agent1"
|
||||
}), state.clone()).await.unwrap();
|
||||
assert!(qsignal_res.contains("task_completed"));
|
||||
|
||||
let read_paged_handler = ReadGraphHandler;
|
||||
let paged_res = read_paged_handler
|
||||
.execute(json!({"limit": 1, "offset": 0}), state.clone())
|
||||
|
||||
Reference in new issue
Block a user