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() {
|
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 {
|
for key in &output.surfaced_keys {
|
||||||
let args = serde_json::json!({"key": key});
|
let rendered = rt.block_on(async {
|
||||||
if let Ok(rendered) = consciousness::mcp_server::forward_to_daemon("memory_render", args) {
|
consciousness::agent::tools::memory::memory_render(None, key, None).await
|
||||||
|
});
|
||||||
|
if let Ok(rendered) = rendered {
|
||||||
if !rendered.trim().is_empty() {
|
if !rendered.trim().is_empty() {
|
||||||
use std::fmt::Write as _;
|
use std::fmt::Write as _;
|
||||||
writeln!(out, "--- {} (surfaced) ---", key).ok();
|
writeln!(out, "--- {} (surfaced) ---", key).ok();
|
||||||
|
|
|
||||||
|
|
@ -165,9 +165,21 @@ fn run_agent_and_parse(agent: &str, session_arg: &Option<String>) {
|
||||||
return;
|
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 {
|
for key in &keys {
|
||||||
let args = serde_json::json!({"key": key});
|
let rendered = rt.block_on(async {
|
||||||
match consciousness::mcp_server::forward_to_daemon("memory_render", args) {
|
consciousness::agent::tools::memory::memory_render(None, &key, None).await
|
||||||
|
});
|
||||||
|
match rendered {
|
||||||
Ok(content) if !content.trim().is_empty() => {
|
Ok(content) if !content.trim().is_empty() => {
|
||||||
println!("--- {} (surfaced) ---", key);
|
println!("--- {} (surfaced) ---", key);
|
||||||
print!("{}", content);
|
print!("{}", content);
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue