@musashishao/agent-kit 1.11.1 → 1.11.3

package/.agent/scripts/setup_notebooklm_mcp.py

@@ -43,7 +43,7 @@ def get_platform_info() -> Dict[str, Path]:
             "vscode_config": home / ".vscode/mcp.json",
             "agent_kit_config": cwd / ".agent/mcp/config/mcp-config.json",
             "antigravity_config": home / ".gemini/antigravity/mcp_config.json",
-            "codex_config": home / ".codex/mcp_config.json",
+            "codex_config": home / ".codex/config.toml",  # Codex uses TOML format
             "python_exe": "python",
             "pip_exe": "pip",
         }
@@ -56,7 +56,7 @@ def get_platform_info() -> Dict[str, Path]:
             "vscode_config": home / ".vscode/mcp.json",
             "agent_kit_config": cwd / ".agent/mcp/config/mcp-config.json",
             "antigravity_config": Path(os.environ.get("LOCALAPPDATA", home)) / "Google/Antigravity/mcp_config.json",
-            "codex_config": Path(os.environ.get("LOCALAPPDATA", home)) / "Codex/mcp_config.json",
+            "codex_config": Path(os.environ.get("LOCALAPPDATA", home)) / "Codex/config.toml",  # Codex uses TOML
             "python_exe": "python.exe",
             "pip_exe": "pip.exe",
         }
@@ -159,10 +159,28 @@ def install_package(python_path: str) -> bool:
 
 def get_wrapper_path() -> Optional[str]:
     """Get absolute path to wrapper script."""
-    wrapper_path = Path.cwd() / WRAPPER_SCRIPT_REL
+    # Search in multiple locations
+    search_paths = [
+        # 1. Project's installed .agent folder
+        Path.cwd() / WRAPPER_SCRIPT_REL,
+        # 2. Package source (when running from node_modules or repo)
+        Path(__file__).parent.parent / "mcp" / "servers" / "notebooklm" / "run_notebooklm_mcp.py",
+        # 3. Alongside this script's .agent folder  
+        Path(__file__).parent.parent.parent / WRAPPER_SCRIPT_REL,
+    ]
+    
+    wrapper_path = None
+    for path in search_paths:
+        if path.exists():
+            wrapper_path = path
+            break
     
-    if not wrapper_path.exists():
-        print(f"❌ Không tìm thấy wrapper script tại: {wrapper_path}")
+    if not wrapper_path:
+        print(f"❌ Không tìm thấy wrapper script.")
+        print(f"   Đã tìm tại:")
+        for p in search_paths:
+            print(f"   - {p}")
+        print(f"\n   Hãy chạy 'npx @musashishao/agent-kit init' trước.")
         return None
     
     # Make executable on Unix
@@ -172,7 +190,7 @@ def get_wrapper_path() -> Optional[str]:
         except Exception:
             pass
     
-    return str(wrapper_path)
+    return str(wrapper_path.resolve())
 
 
 def select_profile() -> str:
@@ -231,6 +249,65 @@ def update_json_config(file_path: Path, server_name: str, config_data: Dict) ->
         return False
 
 
+def update_toml_config(file_path: Path, server_name: str, python_path: str, wrapper_path: str, profile: str) -> bool:
+    """Update a TOML config file with server config (for Codex CLI)."""
+    try:
+        file_path.parent.mkdir(parents=True, exist_ok=True)
+        
+        # Read existing content
+        lines = []
+        if file_path.exists():
+            with open(file_path, 'r', encoding='utf-8') as f:
+                lines = f.readlines()
+        
+        # Check if server already exists
+        server_header = f'[mcp_servers.{server_name}]'
+        server_exists = any(server_header in line for line in lines)
+        
+        if server_exists:
+            # Remove existing server config
+            new_lines = []
+            skip_section = False
+            for line in lines:
+                if server_header in line:
+                    skip_section = True
+                    continue
+                elif skip_section and line.strip().startswith('['):
+                    skip_section = False
+                
+                if not skip_section:
+                    new_lines.append(line)
+            lines = new_lines
+        
+        # Build TOML config for server
+        # Escape backslashes for Windows paths
+        py_path_escaped = python_path.replace('\\', '\\\\')
+        wrapper_escaped = wrapper_path.replace('\\', '\\\\')
+        
+        toml_config = f'''
+[mcp_servers.{server_name}]
+command = "{py_path_escaped}"
+args = ["-u", "-W", "ignore", "{wrapper_escaped}"]
+env = {{ "PYTHONUNBUFFERED" = "1", "PYTHONWARNINGS" = "ignore", "NOTEBOOKLM_PROFILE" = "{profile}" }}
+startup_timeout_sec = 60
+'''
+        
+        # Append new config
+        content = ''.join(lines)
+        if not content.endswith('\n'):
+            content += '\n'
+        content += toml_config
+        
+        with open(file_path, 'w', encoding='utf-8') as f:
+            f.write(content)
+        
+        return True
+        
+    except Exception as e:
+        print(f"  ⚠️ Không thể cập nhật {file_path.name}: {e}")
+        return False
+
+
 def run_auth_check(python_path: str) -> bool:
     """Check if NotebookLM auth is configured."""
     try:
@@ -310,15 +387,15 @@ def main():
     # 8. Update client configs
     print("\n🔗 Đang cấu hình các app AI...")
     
-    configs_to_update = [
+    # JSON-based configs
+    json_configs = [
         (platform_info["claude_config"], "notebooklm-mcp", "Claude"),
         (platform_info["cursor_config"], "notebooklm-mcp", "Cursor"),
         (platform_info["vscode_config"], "notebooklm-mcp", "VS Code"),
         (platform_info.get("antigravity_config"), "notebooklm", "Antigravity"),
-        (platform_info.get("codex_config"), "notebooklm", "Codex"),
     ]
     
-    for config_path, server_name, app_name in configs_to_update:
+    for config_path, server_name, app_name in json_configs:
         if not config_path:
             continue
         if update_json_config(config_path, server_name, server_config):
@@ -326,6 +403,14 @@ def main():
         else:
             print(f"  ⚠️ {app_name} (bỏ qua)")
     
+    # TOML-based configs (Codex CLI)
+    codex_config = platform_info.get("codex_config")
+    if codex_config:
+        if update_toml_config(codex_config, "notebooklm", python_path, wrapper_path, profile):
+            print(f"  ✅ Codex CLI")
+        else:
+            print(f"  ⚠️ Codex CLI (bỏ qua)")
+    
     # 9. Update Agent Kit config
     ak_config = platform_info["agent_kit_config"]
     if ak_config.exists():

package/.agent/workflows/ai-agent.md

@@ -6,6 +6,8 @@ description_vi: Xây dựng workflow cho AI Agent. Từ thiết kế hệ thốn
 
 # 🤖 /ai-agent - AI Agent Builder Workflow
 
+$ARGUMENTS
+
 This workflow guides you through designing and implementing sophisticated AI agents and multi-agent systems.
 
 ## 📋 Steps

package/.agent/workflows/engineering-brief.md

@@ -6,6 +6,8 @@ description_vi: "Tạo Engineering Brief từ tài liệu thô qua NotebookLM. C
 
 # Engineering Brief Workflow
 
+$ARGUMENTS
+
 Distill PRD documents, specs, and meeting notes into a concise Engineering Brief to inject into the Agent context.
 
 ---

package/.agent/workflows/gamekit-init.md

@@ -6,6 +6,8 @@ description_en: "Initialize a new game project with Spec Ecosystem structure."
 
 # /gamekit.init - Game Project Initialization
 
+$ARGUMENTS
+
 Initialize a new game project with the complete Spec Ecosystem documentation structure.
 
 ---

package/.agent/workflows/gamekit-launch.md

@@ -6,6 +6,8 @@ description_en: "Mobile game launch workflow. From pre-launch to App Store submi
 
 # /gamekit-launch - Mobile Game Launch Workflow
 
+$ARGUMENTS
+
 Complete workflow to launch your mobile game to iOS App Store and Google Play.
 
 ---

package/.agent/workflows/gamekit-plan.md

@@ -6,6 +6,8 @@ description_en: "Generate implementation plan from approved game specification."
 
 # /gamekit.plan - Game Implementation Plan
 
+$ARGUMENTS
+
 Generate a detailed implementation plan from an approved specification. Includes task breakdown, agent assignments, and file structure.
 
 ---

package/.agent/workflows/gamekit-qa.md

@@ -6,6 +6,8 @@ description_en: "Playtest and QA loop for iterative game refinement."
 
 # /gamekit-qa - Evaluative Loop (QA & Refinement)
 
+$ARGUMENTS
+
 Playtest, collect feedback, and iteratively refine game feel and balance.
 
 > 💡 **Migrated from:** `/game-hybrid` Tier 3

package/.agent/workflows/gamekit-spec.md

@@ -6,6 +6,8 @@ description_en: "Create game specification with Explained Q&A for all stakeholde
 
 # /gamekit.spec - Game Specification Creation
 
+$ARGUMENTS
+
 Create a comprehensive game specification using the Explained Q&A system. Designed for ALL stakeholders (Developer, PM, Designer).
 
 ---

package/.agent/workflows/gamekit-tasks.md

@@ -6,6 +6,8 @@ description_en: "Generate executable task list from approved game plan."
 
 # /gamekit.tasks - Game Task List Generation
 
+$ARGUMENTS
+
 Generate an executable task list with checkboxes, session tracking, and progress overview from an approved implementation plan.
 
 ---

package/.agent/workflows/generative-ui-init.md

@@ -5,6 +5,8 @@ description_vi: "Khởi tạo dự án mới với kiến trúc Generative UI. C
 
 # /generative-ui-init Workflow
 
+$ARGUMENTS
+
 > **Purpose:** Scaffold a project optimized for Generative UI patterns (agent-driven, A2UI-ready).
 
 ---

package/.agent/workflows/marketing.md

@@ -6,6 +6,8 @@ description_vi: Workflow lập kế hoạch marketing tăng trưởng và chiế
 
 # 🚀 /marketing - Campaign Planning Workflow
 
+$ARGUMENTS
+
 This workflow guides you through the process of planning and executing a high-impact marketing campaign for your product.
 
 ## 📋 Steps

package/.agent/workflows/pentest.md

@@ -6,6 +6,8 @@ description_vi: Workflow kiểm thử xâm nhập và đánh giá bảo mật. P
 
 # 🕵️ /pentest - Security Audit Workflow
 
+$ARGUMENTS
+
 This workflow executes a systematic security assessment of your application or infrastructure.
 
 ## 📋 Steps

package/.agent/workflows/research.md

@@ -6,6 +6,8 @@ description_vi: "Workflow nghiên cứu sâu với NotebookLM. Quét web, phân
 
 # Deep Research Workflow
 
+$ARGUMENTS
+
 Use NotebookLM MCP to deeply research a topic, import sources, and synthesize insights.
 
 ---

package/.agent/workflows/saas.md

@@ -6,6 +6,8 @@ description_vi: Workflow ra mắt sản phẩm SaaS. Từ trang landing page đ
 
 # 🚀 /saas - SaaS Launch Workflow
 
+$ARGUMENTS
+
 This workflow helps you turn a project into a recurring revenue business by integrating essential SaaS components.
 
 ## 📋 Steps

package/.agent/workflows/ui-verify.md

@@ -4,6 +4,8 @@ description: "Run Vibe Alignment Audit on a URL or local file. Uses Playwright t
 
 # /ui-verify - UI/UX Quality Verification
 
+$ARGUMENTS
+
 Use this workflow to verify the visual quality of a UI before shipping.
 
 ---

package/.agent/workflows/vibe-audit.md

@@ -5,6 +5,8 @@ description_vi: "Đánh giá thẩm mỹ và cảm xúc của một trang hoặc
 
 # /vibe-audit Workflow
 
+$ARGUMENTS
+
 > **Purpose:** Audit any UI for "vibe" alignment and generate a quality report.
 
 ---

package/.agent/workflows/vibe-game.md

@@ -4,6 +4,8 @@ description: "Unified entry point for AI VibeCoding Game Ecosystem. Concept -> S
 
 # 🎸 /vibe-game Workflow
 
+$ARGUMENTS
+
 This workflow activates the **Sage AI Game Conductor** to transform a creative idea into a professional game architecture.
 
 ## 🌊 Steps

package/bin/agent-kit

@@ -18,6 +18,7 @@ const VERSION = packageJson.version;
 
 // Commands handled by cli.js (Node.js)
 const CLI_JS_COMMANDS = [
+  'setup',  // Full setup: init + NotebookLM + Codex sync
   'init',
   'update',
   'status',
@@ -46,7 +47,6 @@ const PYTHON_CLI_COMMANDS = [
   'setup-notion-mcp',
   'memory-sync',
   'notebooklm-status',
-  'setup',
   'install'
 ];
 

package/bin/cli.js

@@ -35,6 +35,7 @@ const translations = {
     usage: 'Usage:',
     usageText: `  npx ${PACKAGE_NAME} <command> [options]`,
     commands: 'Commands:',
+    setup: '⭐ FULL SETUP: Install everything in one command',
     init: 'Install .agent folder into current project',
     ai: 'AI Infrastructure management (init, sync, status)',
     setupCodex: 'Sync workflows with Codex CLI slash commands',
@@ -164,6 +165,7 @@ ${colors.bright}${t.usage}${colors.reset}
 ${t.usageText}
 
 ${colors.bright}${t.commands}${colors.reset}
+  ${colors.bright}${colors.yellow}setup${colors.reset}         ${t.setup || '⭐ Full setup: init + NotebookLM + Codex'}
   ${colors.cyan}init${colors.reset}          ${t.init}
   ${colors.cyan}ai${colors.reset}            ${t.ai}
   ${colors.cyan}setup-codex${colors.reset}   ${t.setupCodex}
@@ -314,6 +316,92 @@ ${colors.bright}📚 Documentation:${colors.reset}
   }
 }
 
+/**
+ * Setup command - Install everything in one go
+ * Runs: init + setup-notebooklm + setup-codex
+ */
+function setupCommand(options) {
+  log.title('🚀 Agent Kit Full Setup');
+  console.log(`
+╔══════════════════════════════════════════════════════════════╗
+║           AGENT KIT - FULL INSTALLATION                      ║
+║     Cài đặt hoàn chỉnh chỉ với 1 lệnh duy nhất               ║
+╚══════════════════════════════════════════════════════════════╝
+  `);
+  
+  const steps = [
+    { name: 'Agent Kit Core', desc: 'Cài đặt .agent folder' },
+    { name: 'NotebookLM MCP', desc: 'Cài MCP server cho research' },
+    { name: 'Codex CLI', desc: 'Sync slash commands' }
+  ];
+  
+  console.log(`${colors.bright}📋 Các bước sẽ thực hiện:${colors.reset}`);
+  steps.forEach((s, i) => {
+    console.log(`   ${i + 1}. ${s.name} - ${s.desc}`);
+  });
+  console.log('');
+  
+  // Step 1: Init (install .agent folder)
+  console.log(`\n${colors.bright}[1/3] Installing Agent Kit Core...${colors.reset}`);
+  try {
+    // Check if already exists
+    const agentDir = path.join(process.cwd(), '.agent');
+    if (fs.existsSync(agentDir) && !options.force) {
+      log.info('.agent folder already exists, skipping...');
+    } else {
+      initCommand({ ...options, quiet: true, force: options.force });
+      log.success('Agent Kit Core installed!');
+    }
+  } catch (e) {
+    log.error(`Init failed: ${e.message}`);
+    return;
+  }
+  
+  // Step 2: Setup NotebookLM MCP
+  console.log(`\n${colors.bright}[2/3] Setting up NotebookLM MCP...${colors.reset}`);
+  try {
+    const scriptPath = path.join(process.cwd(), '.agent/scripts/setup_notebooklm_mcp.py');
+    if (fs.existsSync(scriptPath)) {
+      execSync(`python3 "${scriptPath}"`, { stdio: 'inherit' });
+      log.success('NotebookLM MCP configured!');
+    } else {
+      log.warn('NotebookLM setup script not found, skipping...');
+    }
+  } catch (e) {
+    log.warn(`NotebookLM setup skipped: ${e.message}`);
+  }
+  
+  // Step 3: Setup Codex CLI
+  console.log(`\n${colors.bright}[3/3] Syncing with Codex CLI...${colors.reset}`);
+  try {
+    setupCodexCommand({ ...options, quiet: true });
+    log.success('Codex CLI synced!');
+  } catch (e) {
+    log.warn(`Codex sync skipped: ${e.message}`);
+  }
+  
+  // Final summary
+  console.log(`
+╔══════════════════════════════════════════════════════════════╗
+║                    🎉 SETUP HOÀN TẤT!                        ║
+╚══════════════════════════════════════════════════════════════╝
+
+${colors.bright}📌 Bước tiếp theo:${colors.reset}
+   1. Xác thực NotebookLM (nếu muốn dùng):
+      ${colors.cyan}notebooklm-mcp-auth${colors.reset}
+
+   2. Restart AI tools của bạn (Codex, Cursor, Claude...)
+
+   3. Bắt đầu sử dụng:
+      ${colors.cyan}/brainstorm${colors.reset} - Lên ý tưởng
+      ${colors.cyan}/create${colors.reset}     - Tạo app mới
+      ${colors.cyan}/debug${colors.reset}      - Fix lỗi
+
+${colors.bright}📚 Docs:${colors.reset} .agent/ARCHITECTURE.md
+${colors.bright}🆘 Help:${colors.reset} npx @musashishao/agent-kit doctor
+  `);
+}
+
 function setupCodexCommand(options) {
   const lang = getSystemLanguage();
   const t = translations[lang] || translations.en;
@@ -1581,6 +1669,9 @@ if (templateIndex !== -1 && args[templateIndex + 1]) {
 
 // Execute command
 switch (command) {
+  case 'setup':
+    setupCommand(options);
+    break;
   case 'init':
     initCommand(options);
     break;

package/package.json

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