various fixes

Author: evilsocket

examples/rag/data/made-up-animals.txt

@@ -1 +1 @@
-A Darmepinter is an animal that looks like a zebra and sounds like a snake.
+A Darmepinter is a made up animal that looks like a zebra and sounds like a snake.

src/agent/mod.rs

@@ -232,16 +232,18 @@ impl Agent {
 
         // for each parsed invocation
         for inv in invocations {
-            let mut mut_state = self.state.lock().await;
             // lookup action
+            let mut mut_state = self.state.lock().await;
             let action = mut_state.get_action(&inv.action);
+
             if action.is_none() {
                 mut_state.metrics.errors.unknown_actions += 1;
                 // tell the model that the action name is wrong
                 mut_state.add_error_to_history(
                     inv.clone(),
                     format!("'{}' is not a valid action name", inv.action),
                 );
+                drop(mut_state);
             } else {
                 let action = action.unwrap();
                 // validate prerequisites

src/agent/namespaces/rag/mod.rs

@@ -37,20 +37,16 @@ impl Action for Search {
         let docs = state.lock().await.rag_query(&query, 1).await?;
 
         if !docs.is_empty() {
-            println!(
-                "[{}] {} results for '{query}' in {:?}",
-                "rag".bold(),
-                docs.len(),
-                start.elapsed()
-            );
+            println!("\n  {} results in {:?}", docs.len(), start.elapsed());
             for (doc, score) in &docs {
-                println!("  [{score}] {} ", &doc.name);
+                println!("       * {} ({})", &doc.name, score);
             }
+            println!("");
 
             Ok(Some(format!(
                 "Here is some supporting information:\n\n{}",
                 docs.iter()
-                    .map(|(doc, _)| format!("{}", doc.data))
+                    .map(|(doc, _)| doc.data.clone())
                     .collect::<Vec<String>>()
                     .join("\n")
             )))

src/agent/namespaces/rag/ns.prompt

@@ -1 +1 @@
-Use these actions to search and retrieve information.
+Use these actions to search and retrieve information from your long term storage.

src/agent/namespaces/rag/search.prompt

@@ -1 +1 @@
-To search for information:
+All information from your long term storage is true. To search for information on your long term storage: