@musashishao/agent-kit 1.4.0 → 1.4.1

package/.agent/scripts/ak_cli.py

@@ -179,17 +179,25 @@ def cmd_init(args: argparse.Namespace) -> int:
     print(f"\n[{steps_completed + 1}/{total_steps}] Generating dependency graph...")
     graph_script = kit_path / "skills" / "graph-mapper" / "scripts" / "generate_graph.py"
     
-    src_dir = project_root / "src"
-    if not src_dir.exists():
-        src_dir = project_root / "app"
+    # Smart source detection
+    src_dir = None
+    for folder in ["src", "app", "lib", "scripts", "components", "docs"]:
+        if (project_root / folder).exists():
+            src_dir = project_root / folder
+            break
     
-    if src_dir.exists() and graph_script.exists():
+    if src_dir is None:
+        # Fallback to project root
+        src_dir = project_root
+    
+    if graph_script.exists():
         result = subprocess.run(
             [
                 "python3", str(graph_script),
                 "--src", str(src_dir),
                 "--output", str(agent_dir / "graph.json"),
                 "--format", "both",
+                "--lang", "universal",
             ],
             capture_output=True,
             text=True,
@@ -199,7 +207,7 @@ def cmd_init(args: argparse.Namespace) -> int:
         else:
             print("  ⚠️ No source code found or script failed")
     else:
-        print("  ⚠️ No src/ or app/ directory found, skipping")
+        print("  ⚠️ Graph script not found, skipping")
     steps_completed += 1
     
     # Step 5: Configure AI hosts
@@ -260,18 +268,25 @@ def cmd_init(args: argparse.Namespace) -> int:
 
 def find_source_dir(project_root: Path) -> Path:
     """Intelligently find the source directory."""
-    # Priority defaults
+    # Priority folders
     for folder in ["src", "app", "lib", "scripts", "components"]:
         if (project_root / folder).exists():
             return project_root / folder
     
-    # Fallback to root if there are source files in root
-    source_extensions = {".ts", ".tsx", ".js", ".jsx", ".py", ".go", ".c", ".cpp", ".cs"}
+    # Check if there are source files in root
+    source_extensions = {".ts", ".tsx", ".js", ".jsx", ".py", ".go", ".c", ".cpp", ".cs", ".md"}
+    has_source_files = False
     for item in project_root.iterdir():
         if item.is_file() and item.suffix in source_extensions:
-            return project_root
-            
-    return project_root / "src"  # Ultimate fallback
+            has_source_files = True
+            break
+    
+    # Check for docs folder
+    if (project_root / "docs").exists():
+        return project_root / "docs"
+    
+    # Fallback: use project root if it has any content, otherwise return root anyway
+    return project_root
 
 
 def cmd_sync(args: argparse.Namespace) -> int:
@@ -308,7 +323,7 @@ def cmd_sync(args: argparse.Namespace) -> int:
         if not graph_script.exists():
             print(f"  ❌ Graph script not found at: {graph_script}")
             success = False
-        elif not src_dir.exists() and src_dir != project_root:
+        elif not src_dir.exists():
             print(f"  ⚠️ Source directory {src_dir} not found")
         else:
             result = subprocess.run(
@@ -317,6 +332,7 @@ def cmd_sync(args: argparse.Namespace) -> int:
                     "--src", str(src_dir),
                     "--output", str(agent_dir / "graph.json"),
                     "--format", "both",
+                    "--lang", "universal",
                 ],
                 capture_output=True,
                 text=True,
@@ -335,12 +351,15 @@ def cmd_sync(args: argparse.Namespace) -> int:
         if not rag_script.exists():
             print(f"  ❌ RAG script not found at: {rag_script}")
             success = False
+        elif not src_dir.exists():
+            print(f"  ⚠️ Source directory {src_dir} not found")
         else:
             result = subprocess.run(
                 [
                     "python3", str(rag_script),
                     "--src", str(src_dir),
                     "--output", str(agent_dir / "rag" / "chunks.json"),
+                    "--lang", "universal",
                 ],
                 capture_output=True,
                 text=True,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@musashishao/agent-kit",
-  "version": "1.4.0",
+  "version": "1.4.1",
   "description": "AI Agent templates - Skills, Agents, Workflows, and AI-Ready Data Infrastructure Gateway",
   "main": "index.js",
   "bin": {