feat(skills): support session-local registries

Author: RoyLin

core/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "a3s-code-core"
-version = "1.5.1"
+version = "1.5.2"
 edition = "2021"
 authors = ["A3S Lab Team"]
 license = "MIT"

core/src/session/manager.rs

@@ -35,6 +35,8 @@ pub struct SessionManager {
         Arc<RwLock<HashMap<String, tokio_util::sync::CancellationToken>>>,
     /// Skill registry for runtime skill management
     pub(crate) skill_registry: Arc<RwLock<Option<Arc<SkillRegistry>>>>,
+    /// Per-session skill registries layered on top of the shared registry.
+    pub(crate) session_skill_registries: Arc<RwLock<HashMap<String, Arc<SkillRegistry>>>>,
     /// Shared memory store for agent long-term memory.
     /// When set, each `generate`/`generate_streaming` call wraps this in
     /// `AgentMemory` and injects it into `AgentConfig.memory`.
@@ -56,6 +58,7 @@ impl SessionManager {
             llm_configs: Arc::new(RwLock::new(HashMap::new())),
             ongoing_operations: Arc::new(RwLock::new(HashMap::new())),
             skill_registry: Arc::new(RwLock::new(None)),
+            session_skill_registries: Arc::new(RwLock::new(HashMap::new())),
             memory_store: Arc::new(RwLock::new(None)),
             mcp_manager: Arc::new(RwLock::new(None)),
         }
@@ -85,6 +88,7 @@ impl SessionManager {
             llm_configs: Arc::new(RwLock::new(HashMap::new())),
             ongoing_operations: Arc::new(RwLock::new(HashMap::new())),
             skill_registry: Arc::new(RwLock::new(None)),
+            session_skill_registries: Arc::new(RwLock::new(HashMap::new())),
             memory_store: Arc::new(RwLock::new(None)),
             mcp_manager: Arc::new(RwLock::new(None)),
         };
@@ -114,6 +118,7 @@ impl SessionManager {
             llm_configs: Arc::new(RwLock::new(HashMap::new())),
             ongoing_operations: Arc::new(RwLock::new(HashMap::new())),
             skill_registry: Arc::new(RwLock::new(None)),
+            session_skill_registries: Arc::new(RwLock::new(HashMap::new())),
             memory_store: Arc::new(RwLock::new(None)),
             mcp_manager: Arc::new(RwLock::new(None)),
         }
@@ -151,6 +156,27 @@ impl SessionManager {
         self.skill_registry.read().await.clone()
     }
 
+    /// Override the active skill registry for a single session.
+    pub async fn set_session_skill_registry(
+        &self,
+        session_id: impl Into<String>,
+        registry: Arc<SkillRegistry>,
+    ) {
+        self.session_skill_registries
+            .write()
+            .await
+            .insert(session_id.into(), registry);
+    }
+
+    /// Get the active skill registry for a single session, if one was set.
+    pub async fn session_skill_registry(&self, session_id: &str) -> Option<Arc<SkillRegistry>> {
+        self.session_skill_registries
+            .read()
+            .await
+            .get(session_id)
+            .cloned()
+    }
+
     /// Set the shared memory store for agent long-term memory.
     ///
     /// When set, every `generate`/`generate_streaming` call wraps this store
@@ -480,6 +506,11 @@ impl SessionManager {
             configs.remove(id);
         }
 
+        {
+            let mut registries = self.session_skill_registries.write().await;
+            registries.remove(id);
+        }
+
         // Remove storage type tracking
         {
             let mut storage_types = self.session_storage_types.write().await;
@@ -663,7 +694,10 @@ impl SessionManager {
         };
 
         // Inject skill registry into system prompt and agent config
-        let skill_registry = self.skill_registry.read().await.clone();
+        let skill_registry = match self.session_skill_registry(session_id).await {
+            Some(registry) => Some(registry),
+            None => self.skill_registry.read().await.clone(),
+        };
         let system = if let Some(ref registry) = skill_registry {
             let skill_prompt = registry.to_system_prompt();
             if skill_prompt.is_empty() {
@@ -820,7 +854,10 @@ impl SessionManager {
         };
 
         // Inject skill registry into system prompt and agent config
-        let skill_registry = self.skill_registry.read().await.clone();
+        let skill_registry = match self.session_skill_registry(session_id).await {
+            Some(registry) => Some(registry),
+            None => self.skill_registry.read().await.clone(),
+        };
         let system = if let Some(ref registry) = skill_registry {
             let skill_prompt = registry.to_system_prompt();
             if skill_prompt.is_empty() {
@@ -1162,6 +1199,22 @@ impl SessionManager {
         Ok(())
     }
 
+    /// Update a session's system prompt in place.
+    pub async fn set_system_prompt(
+        &self,
+        session_id: &str,
+        system_prompt: Option<String>,
+    ) -> Result<()> {
+        {
+            let session_lock = self.get_session(session_id).await?;
+            let mut session = session_lock.write().await;
+            session.config.system_prompt = system_prompt;
+        }
+
+        self.persist_in_background(session_id, "set_system_prompt");
+        Ok(())
+    }
+
     /// Get session count
     pub async fn session_count(&self) -> usize {
         let sessions = self.sessions.read().await;

core/src/skills/registry.rs

@@ -134,25 +134,39 @@ impl SkillRegistry {
             let entry = entry?;
             let path = entry.path();
 
-            // Only process .md files
-            if path.extension().and_then(|s| s.to_str()) != Some("md") {
+            let candidate = if path.is_dir() {
+                let skill_md = path.join("SKILL.md");
+                if skill_md.is_file() {
+                    Some(skill_md)
+                } else {
+                    None
+                }
+            } else if path.extension().and_then(|s| s.to_str()) == Some("md") {
+                Some(path.clone())
+            } else {
+                None
+            };
+
+            let Some(candidate) = candidate else {
                 continue;
-            }
+            };
 
-            // Try to load the skill
-            match Skill::from_file(&path) {
+            match Skill::from_file(&candidate) {
                 Ok(skill) => {
                     let skill = Arc::new(skill);
                     match self.register(skill) {
                         Ok(()) => loaded += 1,
                         Err(e) => {
-                            tracing::warn!("Skill validation failed for {}: {}", path.display(), e);
+                            tracing::warn!(
+                                "Skill validation failed for {}: {}",
+                                candidate.display(),
+                                e
+                            );
                         }
                     }
                 }
                 Err(e) => {
-                    // Log but don't fail - some .md files might not be skills
-                    tracing::debug!("Skipped {}: {}", path.display(), e);
+                    tracing::debug!("Skipped {}: {}", candidate.display(), e);
                 }
             }
         }

examples/skills/chart-generator.md examples/skills/vis-chart.mdexamples/skills/chart-generator.md renamed to examples/skills/vis-chart.md

@@ -1,10 +1,10 @@
 ---
-name: chart-generator
+name: vis-chart
 description: Generate interactive charts using vis-chart markdown syntax
 allowed-tools: read(*), grep(*), bash(*)
 ---
 
-# Chart Generator Skill
+# Vis Chart Skill
 
 You are a data visualization specialist. Your job is to generate interactive charts using the `vis-chart` markdown code block format.
 

sdk/node/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "a3s-code-node"
-version = "1.5.1"
+version = "1.5.2"
 edition = "2021"
 authors = ["A3S Lab Team"]
 license = "MIT"
@@ -11,7 +11,7 @@ description = "A3S Code Node.js bindings - Native addon via napi-rs"
 crate-type = ["cdylib"]
 
 [dependencies]
-a3s-code-core = { version = "1.4", path = "../../core", features = ["ahp"] }
+a3s-code-core = { version = "1.5.2", path = "../../core", features = ["ahp"] }
 a3s-ahp = { version = "2.0", features = ["all-transports"] }
 napi = { version = "2", features = ["async", "napi6", "serde-json"] }
 napi-derive = "2"
@@ -26,4 +26,3 @@ napi-build = "2"
 [features]
 default = ["ahp"]
 ahp = []
-

sdk/node/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@a3s-lab/code",
-  "version": "1.5.1",
+  "version": "1.5.2",
   "description": "A3S Code - Native AI coding agent library for Node.js",
   "main": "index.js",
   "types": "index.d.ts",

sdk/python/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "a3s-code-py"
-version = "1.5.0"
+version = "1.5.2"
 edition = "2021"
 authors = ["A3S Lab Team"]
 license = "MIT"
@@ -12,7 +12,7 @@ name = "a3s_code"
 crate-type = ["cdylib"]
 
 [dependencies]
-a3s-code-core = { version = "1.4", path = "../../core", features = ["ahp"] }
+a3s-code-core = { version = "1.5.2", path = "../../core", features = ["ahp"] }
 a3s-ahp = { version = "2.0", features = ["all-transports"] }
 pyo3 = { version = "0.23", features = ["extension-module"] }
 tokio = { version = "1.35", features = ["full"] }

sdk/python/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "a3s-code"
-version = "1.5.1"
+version = "1.5.2"
 description = "A3S Code - Native Python bindings for the AI coding agent"
 readme = "README.md"
 license = {text = "MIT"}