feat(anda_engine): add get_or_init_caller method to MemoryManagement

Author: zensh

agents/anda_assistant/src/agent.rs

@@ -203,17 +203,8 @@ impl Agent<AgentCtx> for Assistant {
 
         let caller_info = self
             .memory
-            .describe_caller(caller)
-            .await
-            .unwrap_or_else(|_| {
-                serde_json::json!({
-                    "type": "Person",
-                    "name": caller.to_string(),
-                    "attributes": {},
-                    "metadata": {},
-                })
-            });
-
+            .get_or_init_caller(caller, ctx.meta().user.clone())
+            .await?;
         let primer = self.memory.describe_primer().await?;
         let instructions = format!(
             "{}\n---\n# Your Identity & Knowledge Domain Map\n{}\n",

anda_engine/Cargo.toml

@@ -3,7 +3,7 @@ name = "anda_engine"
 description = "Agents engine for Anda -- an AI agent framework built with Rust, powered by ICP and TEEs."
 repository = "https://github.com/ldclabs/anda/tree/main/anda_engine"
 publish = true
-version = "0.9.10"
+version = "0.9.11"
 edition.workspace = true
 keywords.workspace = true
 categories.workspace = true

anda_engine/src/engine.rs

@@ -158,6 +158,12 @@ impl Engine {
             )
             .into());
         }
+        if let Some(user) = &meta.user {
+            let u = user.trim();
+            if u.is_empty() || u != user || u.len() > 32 {
+                return Err(format!("invalid user name {:?}", user).into());
+            }
+        }
 
         input.name = if input.name.is_empty() {
             self.default_agent.clone()
@@ -202,6 +208,12 @@ impl Engine {
             )
             .into());
         }
+        if let Some(user) = &meta.user {
+            let u = user.trim();
+            if u.is_empty() || u != user || u.len() > 32 {
+                return Err(format!("invalid user name {:?}", user).into());
+            }
+        }
 
         // manager can call any tool
         if !self.export_tools.contains(&input.name) && !self.management.is_manager(&caller) {

anda_engine/src/memory.rs

@@ -21,7 +21,7 @@ use ciborium::cbor;
 use ic_auth_types::ByteBufB64;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
-use serde_json::json;
+use serde_json::{Map, json};
 use std::{
     collections::BTreeMap,
     fmt,
@@ -395,6 +395,33 @@ impl MemoryManagement {
         Ok(user.to_concept_node())
     }
 
+    pub async fn get_or_init_caller(
+        &self,
+        id: &Principal,
+        name: Option<String>,
+    ) -> Result<Json, KipError> {
+        let mut attributes = Map::new();
+        let mut metadata = Map::new();
+        attributes.insert("id".to_string(), id.to_string().into());
+        attributes.insert("person_class".to_string(), "Human".into());
+        if let Some(name) = name {
+            attributes.insert("name".to_string(), name.into());
+        }
+        metadata.insert("author".to_string(), "$system".into());
+        metadata.insert("status".to_string(), "active".into());
+        let user = self
+            .nexus
+            .get_or_init_concept(
+                PERSON_TYPE.to_string(),
+                id.to_string(),
+                attributes,
+                metadata,
+            )
+            .await?;
+
+        Ok(user.to_concept_node())
+    }
+
     pub async fn add_resource(&self, resource: ResourceRef<'_>) -> Result<u64, DBError> {
         let id = self.resources.add_from(&resource).await?;
         self.resources.flush(unix_ms()).await?;