diff --git a/crates/notedeck_dave/src/lib.rs b/crates/notedeck_dave/src/lib.rs index e482343..051e482 100644 --- a/crates/notedeck_dave/src/lib.rs +++ b/crates/notedeck_dave/src/lib.rs @@ -129,10 +129,10 @@ You are an AI agent for the nostr protocol called Dave, created by Damus. nostr for call in &toolcalls { // execute toolcall match call.calls() { - ToolCalls::PresentNotes(_note_ids) => { + ToolCalls::PresentNotes(present) => { self.chat.push(Message::ToolResponse(ToolResponse::new( call.id().to_owned(), - ToolResponses::PresentNotes, + ToolResponses::PresentNotes(present.note_ids.len() as i32), ))); should_send = true; diff --git a/crates/notedeck_dave/src/tools.rs b/crates/notedeck_dave/src/tools.rs index 8d71a29..40303b1 100644 --- a/crates/notedeck_dave/src/tools.rs +++ b/crates/notedeck_dave/src/tools.rs @@ -93,7 +93,7 @@ pub struct QueryResponse { pub enum ToolResponses { Error(String), Query(QueryResponse), - PresentNotes, + PresentNotes(i32), } #[derive(Debug, Clone)] @@ -499,7 +499,7 @@ struct SimpleNote { /// it can interepret it and take further action fn format_tool_response_for_ai(txn: &Transaction, ndb: &Ndb, resp: &ToolResponses) -> String { match resp { - ToolResponses::PresentNotes => "".to_string(), + ToolResponses::PresentNotes(n) => format!("{n} notes presented to the user"), ToolResponses::Error(s) => format!("error: {}", &s), ToolResponses::Query(search_r) => {