chore: improve anda_assistant

Author: zensh

Cargo.toml

@@ -12,7 +12,7 @@ members = [
 ]
 
 [workspace.package]
-version = "0.9.1"
+version = "0.9.2"
 description = "Anda is an AI agent framework built with Rust, powered by ICP and TEEs."
 repository = "https://github.com/ldclabs/anda"
 homepage = "https://github.com/ldclabs/anda"

agents/anda_assistant/Cargo.toml

@@ -3,7 +3,7 @@ name = "anda_assistant"
 description = "Anda -- an AI Assistant powered by the Knowledge Interaction Protocol (KIP)."
 repository = "https://github.com/ldclabs/anda/tree/main/agents/anda_assistant"
 publish = true
-version = "0.4.2"
+version = "0.4.3"
 edition.workspace = true
 keywords.workspace = true
 categories.workspace = true

agents/anda_assistant/src/agent.rs

@@ -1,7 +1,8 @@
 use anda_cognitive_nexus::{CognitiveNexus, ConceptPK};
 use anda_core::{
-    Agent, AgentContext, AgentOutput, BoxError, CompletionRequest, Document, Documents, Message,
-    Principal, Resource, StateFeatures, Tool, ToolSet, Usage, evaluate_tokens, update_resources,
+    Agent, AgentContext, AgentOutput, BoxError, CacheExpiry, CacheFeatures, CompletionRequest,
+    Document, Documents, Message, Principal, Resource, StateFeatures, Tool, ToolSet, Usage,
+    evaluate_tokens, update_resources,
 };
 use anda_db::{database::AndaDB, index::BTree};
 use anda_engine::{
@@ -18,7 +19,7 @@ use anda_kip::{
     META_SELF_NAME, PERSON_SELF_KIP, PERSON_SYSTEM_KIP, PERSON_TYPE, SELF_INSTRUCTIONS,
     SYSTEM_INSTRUCTIONS, parse_kml,
 };
-use std::{collections::BTreeMap, sync::Arc};
+use std::{collections::BTreeMap, sync::Arc, time::Duration};
 
 #[derive(Clone)]
 pub struct Assistant {
@@ -188,29 +189,36 @@ impl Agent<AgentCtx> for Assistant {
             return Err("anonymous caller not allowed".into());
         }
 
-        let user_name = ctx
-            .meta()
-            .user
-            .clone()
-            .unwrap_or_else(|| caller.to_string());
+        let caller_key = format!("Running:{}", caller);
+        let created_at = unix_ms();
+        let ok = ctx
+            .cache_set_if_not_exists(
+                &caller_key,
+                (created_at, Some(CacheExpiry::TTL(Duration::from_secs(300)))),
+            )
+            .await;
+        if !ok {
+            return Err("Only one prompt can run at a time for you".into());
+        }
+
         let caller_info = self
             .memory
             .describe_caller(caller)
             .await
             .unwrap_or_else(|_| {
                 serde_json::json!({
-                    "id": caller.to_string(),
-                    "name": user_name
+                    "type": "Person",
+                    "name": caller.to_string(),
+                    "attributes": {},
                 })
             });
 
         let created_at = unix_ms();
         let primer = self.memory.describe_primer().await?;
         let instructions = format!(
             "{}\n---\n# Your Identity & Knowledge Domain Map\n{}\n",
-            SYSTEM_INSTRUCTIONS, primer
+            self.system_instructions, primer
         );
-
         let (mut conversations, mut cursor) = self
             .memory
             .list_conversations_by_user(caller, None, Some(7))
@@ -280,7 +288,7 @@ impl Agent<AgentCtx> for Assistant {
                 format!(
                     "Current Datetime: {}\n---\n{}",
                     rfc3339_datetime(created_at).unwrap_or_else(|| format!("unix_ms:{created_at}")),
-                    Documents::new("history".to_string(), history_docs)
+                    Documents::new("user_context".to_string(), history_docs)
                 )
                 .into(),
             ],
@@ -424,6 +432,7 @@ impl Agent<AgentCtx> for Assistant {
                 Ok::<(), BoxError>(())
             };
 
+            ctx.cache_delete(&caller_key).await;
             match rt().await {
                 Ok(_) => {}
                 Err(err) => {