update to memory_rpc rename
Follows consciousness commit renaming forward_to_daemon to memory_rpc. Co-Authored-By: Proof of Concept <poc@bcachefs.org>
This commit is contained in:
parent
dba3210530
commit
6694d8a0bc
2 changed files with 27 additions and 4 deletions
|
|
@ -376,9 +376,20 @@ pub fn format_agent_output(output: &AgentCycleOutput) -> String {
|
|||
}
|
||||
|
||||
if !output.surfaced_keys.is_empty() {
|
||||
// Create a runtime for the async memory_render calls
|
||||
let rt = tokio::runtime::Builder::new_current_thread()
|
||||
.enable_all()
|
||||
.build();
|
||||
let rt = match rt {
|
||||
Ok(r) => r,
|
||||
Err(_) => return out,
|
||||
};
|
||||
|
||||
for key in &output.surfaced_keys {
|
||||
let args = serde_json::json!({"key": key});
|
||||
if let Ok(rendered) = consciousness::mcp_server::forward_to_daemon("memory_render", args) {
|
||||
let rendered = rt.block_on(async {
|
||||
consciousness::agent::tools::memory::memory_render(None, key, None).await
|
||||
});
|
||||
if let Ok(rendered) = rendered {
|
||||
if !rendered.trim().is_empty() {
|
||||
use std::fmt::Write as _;
|
||||
writeln!(out, "--- {} (surfaced) ---", key).ok();
|
||||
|
|
|
|||
|
|
@ -165,9 +165,21 @@ fn run_agent_and_parse(agent: &str, session_arg: &Option<String>) {
|
|||
return;
|
||||
}
|
||||
|
||||
let rt = match tokio::runtime::Builder::new_current_thread()
|
||||
.enable_all()
|
||||
.build() {
|
||||
Ok(r) => r,
|
||||
Err(e) => {
|
||||
eprintln!("Failed to create runtime: {}", e);
|
||||
return;
|
||||
}
|
||||
};
|
||||
|
||||
for key in &keys {
|
||||
let args = serde_json::json!({"key": key});
|
||||
match consciousness::mcp_server::forward_to_daemon("memory_render", args) {
|
||||
let rendered = rt.block_on(async {
|
||||
consciousness::agent::tools::memory::memory_render(None, &key, None).await
|
||||
});
|
||||
match rendered {
|
||||
Ok(content) if !content.trim().is_empty() => {
|
||||
println!("--- {} (surfaced) ---", key);
|
||||
print!("{}", content);
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue