@musashishao/agent-kit 1.11.1 → 1.11.2

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",
         }
@@ -231,6 +231,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 +369,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 +385,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/package.json

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