nia-mcp-server 1.0.23__py3-none-any.whl → 1.0.42__py3-none-any.whl

nia_mcp_server/cli.py

@@ -4,7 +4,7 @@ CLI interface for NIA MCP Server
 import sys
 import argparse
 from typing import Optional
-from .setup import setup_mcp_config
+from .setup import setup_mcp_config, SUPPORTED_IDES, REMOTE_SUPPORTED_IDES
 
 
 def validate_api_key(api_key: str) -> bool:
@@ -40,10 +40,21 @@ def main():
     )
     setup_parser.add_argument(
         "--ide",
-        choices=["cursor", "vscode", "continue"],
+        choices=SUPPORTED_IDES,
         default="cursor",
         help="IDE to configure (default: cursor)"
     )
+    setup_parser.add_argument(
+        "--remote",
+        action="store_true",
+        help="Use remote MCP server (no local installation, connects via HTTP)"
+    )
+    
+    # List command
+    list_parser = subparsers.add_parser(
+        "list-ides",
+        help="List all supported IDEs and their capabilities"
+    )
     
     # Parse arguments
     args = parser.parse_args()
@@ -57,13 +68,29 @@ def main():
             sys.exit(1)
         
         # Run setup
-        success = setup_mcp_config(args.api_key, args.ide)
+        success = setup_mcp_config(args.api_key, args.ide, args.remote)
         sys.exit(0 if success else 1)
     
+    elif args.command == "list-ides":
+        print("\n🖥️  Supported IDEs:\n")
+        print("Remote + Local (both modes supported):")
+        for ide in sorted(SUPPORTED_IDES):
+            if ide in REMOTE_SUPPORTED_IDES:
+                print(f"  ✅ {ide}")
+        
+        print("\nLocal only:")
+        for ide in sorted(SUPPORTED_IDES):
+            if ide not in REMOTE_SUPPORTED_IDES:
+                print(f"  📦 {ide}")
+        
+        print("\n💡 Use --remote for cloud-hosted MCP (no local deps)")
+        print("   Use default (local) for self-hosted MCP via pipx\n")
+        sys.exit(0)
+    
     # If no command specified, show help
     parser.print_help()
     sys.exit(1)
 
 
 if __name__ == "__main__":
-    main()
+    main()

nia_mcp_server/profiles.py

@@ -109,6 +109,22 @@ PROFILE_CONFIGS: Dict[str, Dict[str, Any]] = {
         }
     },
     
+    "antigravity": {
+        "name": "Google Antigravity",
+        "target_dir": ".gemini/antigravity",
+        "file_extension": ".md",
+        "file_map": {
+            "nia_rules.md": "nia_antigravity_guide.md"
+        },
+        "mcp_config": True,
+        "format": "markdown",
+        "features": ["ai_assistant", "code_generation", "multi_modal"],
+        "global_replacements": {
+            "# Nia": "# Nia for Google Antigravity",
+            "{{IDE}}": "Google Antigravity"
+        }
+    },
+    
     "zed": {
         "name": "Zed",
         "target_dir": ".zed",
@@ -187,9 +203,9 @@ TEMPLATE_CONFIGS = {
       "problemMatcher": []
     },
     {
-      "label": "Nia: Search Codebase",
+      "label": "Nia: Search",
       "type": "shell",
-      "command": "echo 'Run: search_codebase \\"${input:searchQuery}\\"'",
+      "command": "echo 'Run: search \\"${input:searchQuery}\\"'",
       "problemMatcher": []
     },
     {
@@ -223,8 +239,8 @@ TEMPLATE_CONFIGS = {
   },
   "Nia Search": {
     "prefix": "nia-search",
-    "body": ["search_codebase \\"${1:query}\\""],
-    "description": "Search indexed repositories"
+    "body": ["search \\"${1:query}\\""],
+    "description": "Search indexed repositories and docs"
   },
   "Nia Web Search": {
     "prefix": "nia-web",

nia_mcp_server/project_init.py

@@ -166,7 +166,7 @@ class NIAProjectInitializer:
         # General steps
         steps.extend([
             "Explore available commands with list_repositories",
-            "Search for code patterns with search_codebase",
+            "Search across code and docs with search",
             "Find new libraries with nia_web_search"
         ])
         

nia_mcp_server/rule_transformer.py

@@ -266,7 +266,7 @@ def enhance_mcp_code_blocks(content: str) -> str:
         code = match.group(2)
         
         # If it's a NIA command, add annotation
-        if any(cmd in code for cmd in ["index_repository", "search_codebase", "list_repositories"]):
+        if any(re.search(rf'\b{re.escape(cmd)}\b', code, re.IGNORECASE) for cmd in ["index_repository", "search", "list_repositories"]):
             return f'```{lang}\n# MCP Command - Run this in your AI assistant\n{code}\n```'
         return match.group(0)
     
@@ -327,7 +327,7 @@ Quick aliases for your terminal:
 ```bash
 # Add to your shell profile (.bashrc, .zshrc, etc.)
 alias nia-index='echo "index_repository"'
-alias nia-search='echo "search_codebase"'
+alias nia-search='echo "search"'
 alias nia-list='echo "list_repositories"'
 ```