nia-mcp-server 1.0.12__py3-none-any.whl → 1.0.13__py3-none-any.whl

nia_mcp_server/__init__.py

@@ -2,4 +2,4 @@
 NIA MCP Server - Proxy server for NIA Knowledge Agent
 """
 
-__version__ = "1.0.12"
+__version__ = "1.0.13"

nia_mcp_server/__main__.py

@@ -1,11 +1,21 @@
 """
 Entry point for NIA MCP Server
 """
+import sys
 from .server import run
+from .cli import main as cli_main
+
 
 def main():
     """Main entry point."""
-    run()
+    # Check if running as CLI command
+    if len(sys.argv) > 1 and sys.argv[1] in ["setup", "--help", "-h"]:
+        # Run CLI interface
+        cli_main()
+    else:
+        # Run MCP server
+        run()
+
 
 if __name__ == "__main__":
     main()

nia_mcp_server/cli.py

@@ -0,0 +1,69 @@
+"""
+CLI interface for NIA MCP Server
+"""
+import sys
+import argparse
+from typing import Optional
+from .setup import setup_mcp_config
+
+
+def validate_api_key(api_key: str) -> bool:
+    """Validate API key format."""
+    if not api_key:
+        return False
+    # Check if it starts with the expected prefix
+    if not api_key.startswith("nk_"):
+        return False
+    # Check minimum length (prefix + reasonable key length)
+    if len(api_key) < 10:
+        return False
+    return True
+
+
+def main():
+    """Main CLI entry point."""
+    parser = argparse.ArgumentParser(
+        description="NIA MCP Server - AI-powered code search",
+        prog="nia-mcp-server"
+    )
+    
+    subparsers = parser.add_subparsers(dest="command", help="Available commands")
+    
+    # Setup command
+    setup_parser = subparsers.add_parser(
+        "setup",
+        help="Set up NIA MCP Server for your IDE"
+    )
+    setup_parser.add_argument(
+        "api_key",
+        help="Your NIA API key (get it from https://app.trynia.ai/api-keys)"
+    )
+    setup_parser.add_argument(
+        "--ide",
+        choices=["cursor", "vscode", "continue"],
+        default="cursor",
+        help="IDE to configure (default: cursor)"
+    )
+    
+    # Parse arguments
+    args = parser.parse_args()
+    
+    # Handle commands
+    if args.command == "setup":
+        # Validate API key
+        if not validate_api_key(args.api_key):
+            print("❌ Invalid API key format. API key should start with 'nk_'")
+            print("   Get your API key from: https://app.trynia.ai/api-keys")
+            sys.exit(1)
+        
+        # Run setup
+        success = setup_mcp_config(args.api_key, args.ide)
+        sys.exit(0 if success else 1)
+    
+    # If no command specified, show help
+    parser.print_help()
+    sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

nia_mcp_server/setup.py

@@ -0,0 +1,177 @@
+"""
+Setup utilities for NIA MCP Server configuration
+"""
+import os
+import json
+import platform
+import shutil
+from pathlib import Path
+from typing import Dict, Optional, Any
+
+
+def find_mcp_config_path(ide: str = "cursor") -> Path:
+    """
+    Find the MCP configuration file path based on OS and IDE.
+    
+    Args:
+        ide: IDE to configure (cursor, vscode, continue)
+    
+    Returns:
+        Path to the MCP configuration file
+    """
+    system = platform.system()
+    home = Path.home()
+    
+    if ide == "cursor":
+        if system == "Darwin":  # macOS
+            return home / ".cursor" / "mcp.json"
+        elif system == "Windows":
+            appdata = os.environ.get("APPDATA", home / "AppData" / "Roaming")
+            return Path(appdata) / "Cursor" / "mcp.json"
+        else:  # Linux and others
+            return home / ".config" / "cursor" / "mcp.json"
+    
+    elif ide == "vscode":
+        # VS Code uses different config locations
+        if system == "Darwin":
+            return home / "Library" / "Application Support" / "Code" / "User" / "mcp.json"
+        elif system == "Windows":
+            appdata = os.environ.get("APPDATA", home / "AppData" / "Roaming")
+            return Path(appdata) / "Code" / "User" / "mcp.json"
+        else:
+            return home / ".config" / "Code" / "User" / "mcp.json"
+    
+    elif ide == "continue":
+        # Continue.dev uses .continue directory
+        return home / ".continue" / "config.json"
+    
+    else:
+        raise ValueError(f"Unsupported IDE: {ide}")
+
+
+def backup_config(config_path: Path) -> Optional[Path]:
+    """
+    Create a backup of existing configuration file.
+    
+    Args:
+        config_path: Path to the configuration file
+    
+    Returns:
+        Path to the backup file if created, None otherwise
+    """
+    if config_path.exists():
+        backup_path = config_path.with_suffix(".json.backup")
+        # If backup already exists, add timestamp
+        if backup_path.exists():
+            import time
+            timestamp = int(time.time())
+            backup_path = config_path.with_suffix(f".json.backup.{timestamp}")
+        
+        shutil.copy2(config_path, backup_path)
+        return backup_path
+    return None
+
+
+def create_nia_config(api_key: str) -> Dict[str, Any]:
+    """
+    Create NIA MCP server configuration.
+    
+    Args:
+        api_key: NIA API key
+    
+    Returns:
+        Dictionary with NIA server configuration
+    """
+    return {
+        "command": "pipx",
+        "args": ["run", "nia-mcp-server"],
+        "env": {
+            "NIA_API_KEY": api_key,
+            "NIA_API_URL": "https://apigcp.trynia.ai/"
+        }
+    }
+
+
+def update_mcp_config(config_path: Path, api_key: str) -> bool:
+    """
+    Update or create MCP configuration file with NIA server.
+    
+    Args:
+        config_path: Path to the MCP configuration file
+        api_key: NIA API key
+    
+    Returns:
+        True if successful, False otherwise
+    """
+    try:
+        # Ensure directory exists
+        config_path.parent.mkdir(parents=True, exist_ok=True)
+        
+        # Load existing config or create new one
+        if config_path.exists():
+            with open(config_path, 'r') as f:
+                config = json.load(f)
+        else:
+            config = {}
+        
+        # Ensure mcpServers section exists
+        if "mcpServers" not in config:
+            config["mcpServers"] = {}
+        
+        # Add or update NIA server configuration
+        config["mcpServers"]["nia"] = create_nia_config(api_key)
+        
+        # Write updated configuration
+        with open(config_path, 'w') as f:
+            json.dump(config, f, indent=2)
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ Error updating configuration: {e}")
+        return False
+
+
+def setup_mcp_config(api_key: str, ide: str = "cursor") -> bool:
+    """
+    Main setup function to configure NIA MCP Server.
+    
+    Args:
+        api_key: NIA API key
+        ide: IDE to configure
+    
+    Returns:
+        True if successful, False otherwise
+    """
+    print(f"\n🚀 Setting up NIA MCP Server for {ide.title()}...\n")
+    
+    # Find config path
+    config_path = find_mcp_config_path(ide)
+    print(f"📍 Configuration path: {config_path}")
+    
+    # Backup existing config
+    backup_path = backup_config(config_path)
+    if backup_path:
+        print(f"📦 Backed up existing config to: {backup_path}")
+    
+    # Update configuration
+    if update_mcp_config(config_path, api_key):
+        print(f"\n✅ NIA MCP Server setup complete!")
+        print(f"\n📝 Configuration written to: {config_path}")
+        print(f"🔑 API Key: {api_key[:10]}...")
+        
+        print("\n📌 Next steps:")
+        print(f"   1. Restart {ide.title()} to load the NIA MCP server")
+        print(f"   2. Test with: \"Claude, list my repositories\"")
+        print(f"   3. Get started: \"Claude, index https://github.com/owner/repo\"")
+        
+        print("\n💡 Learn more:")
+        print("   - Documentation: https://docs.trynia.ai")
+        print("   - Get help: https://discord.gg/BBSwUMrrfn")
+        
+        return True
+    else:
+        print(f"\n❌ Setup failed. Please check the error messages above.")
+        if backup_path:
+            print(f"   Your original config is safe at: {backup_path}")
+        return False

nia_mcp_server-1.0.13.dist-info/METADATA

@@ -0,0 +1,39 @@
+Metadata-Version: 2.4
+Name: nia-mcp-server
+Version: 1.0.13
+Summary: Nia Knowledge Agent
+Project-URL: Homepage, https://trynia.ai
+Project-URL: Documentation, https://docs.trynia.ai
+Author-email: Nia Team <founders@nozomio.com>
+License-Expression: AGPL-3.0
+License-File: LICENSE
+Keywords: ai,codebase,mcp,nia,search
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: GNU Affero General Public License v3
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: mcp>=0.1.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: python-dotenv>=1.0.0
+Description-Content-Type: text/markdown
+
+# NIA MCP Server
+
+The NIA MCP Server enables AI assistants like Claude to search and understand your indexed codebases through the Model Context Protocol (MCP).
+
+## Quick Start
+
+### Automatic Setup (Recommended) ✨
+
+Get your API key from [https://trynia.ai/api-keys](https://trynia.ai/api-keys) and run:
+
+```bash
+pipx run nia-mcp-server setup YOUR_API_KEY
+```

{nia_mcp_server-1.0.12.dist-info → nia_mcp_server-1.0.13.dist-info}/RECORD

@@ -1,17 +1,19 @@
-nia_mcp_server/__init__.py,sha256=u5VyiO3beS875PBflORPjfQPUa57WihSs7aisJLlXkU,85
-nia_mcp_server/__main__.py,sha256=XY11ESL4hctu-BBgtPATFZyd1o-O7wE7y-UOSoNs-hw,152
+nia_mcp_server/__init__.py,sha256=meeT4feLJfGvMAng90TkSl96ykJI_siO0HtnR-5RR5w,85
+nia_mcp_server/__main__.py,sha256=YQSpFtDeKp18r8mKr084cHnRFV4416_EKCu9FTM8_ik,394
 nia_mcp_server/api_client.py,sha256=6QFUg6jI9m0EAMrGm-HEAU5uYsUUxzhRau0iLCW9fpU,25815
+nia_mcp_server/cli.py,sha256=32VSPNIocXtDgVBDZNZsxvj3kytBn54_a1pIE84vOdY,1834
 nia_mcp_server/profiles.py,sha256=2DD8PFRr5Ij4IK4sPUz0mH8aKjkrEtkKLC1R0iki2bA,7221
 nia_mcp_server/project_init.py,sha256=T0-ziJhofL4L8APwnM43BLhxtlmOHaYH-V9PF2yXLw4,7138
 nia_mcp_server/rule_transformer.py,sha256=wCxoQ1Kl_rI9mUFnh9kG5iCXYU4QInrmFQOReZfAFVo,11000
 nia_mcp_server/server.py,sha256=CuXj-YWd4QHhhE5v_hEGHWGalO3whVlFUOMK359vqgs,75827
+nia_mcp_server/setup.py,sha256=nJXVY8NHGtWROtoH8DW-3uOgyuPs4F9dW0cNhcbCLrM,5355
 nia_mcp_server/assets/rules/claude_rules.md,sha256=HNL5GJMUbFxSpNbIAJUQWqAywjMl4lf530I1in69aNY,7380
 nia_mcp_server/assets/rules/cursor_rules.md,sha256=hd6lhzNrK1ULQUYIEVeOnyKnuLKq4hmwZPbMqGUI1Lk,1720
 nia_mcp_server/assets/rules/nia_rules.md,sha256=mvdYrkoiRgxeROhtnRXCV53TX5B9wqLiCJ6oYTqSPfY,6345
 nia_mcp_server/assets/rules/vscode_rules.md,sha256=fqn4aJO_bhftaCGkVoquruQHf3EaREQJQWHXq6a4FOk,6967
 nia_mcp_server/assets/rules/windsurf_rules.md,sha256=PzU2as5gaiVsV6PAzg8T_-GR7VCyRQGMjAHcSzYF_ms,3354
-nia_mcp_server-1.0.12.dist-info/METADATA,sha256=zpdgFJZ99Rm3AI_WUnBXg2aNCPem0HCDvHNuYf-JZDg,6767
-nia_mcp_server-1.0.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nia_mcp_server-1.0.12.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
-nia_mcp_server-1.0.12.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
-nia_mcp_server-1.0.12.dist-info/RECORD,,
+nia_mcp_server-1.0.13.dist-info/METADATA,sha256=uUDkIiPkBZLCuZk8uqUUo0aCIF-dw8iZKuXGqHqQwe4,1324
+nia_mcp_server-1.0.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nia_mcp_server-1.0.13.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
+nia_mcp_server-1.0.13.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
+nia_mcp_server-1.0.13.dist-info/RECORD,,

nia_mcp_server-1.0.12.dist-info/METADATA

@@ -1,246 +0,0 @@
-Metadata-Version: 2.4
-Name: nia-mcp-server
-Version: 1.0.12
-Summary: Nia Knowledge Agent
-Project-URL: Homepage, https://trynia.ai
-Project-URL: Documentation, https://docs.trynia.ai
-Author-email: Nia Team <founders@nozomio.com>
-License-Expression: AGPL-3.0
-License-File: LICENSE
-Keywords: ai,codebase,mcp,nia,search
-Classifier: Development Status :: 4 - Beta
-Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: GNU Affero General Public License v3
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Python: >=3.8
-Requires-Dist: httpx>=0.24.0
-Requires-Dist: mcp>=0.1.0
-Requires-Dist: pydantic>=2.0.0
-Requires-Dist: python-dotenv>=1.0.0
-Description-Content-Type: text/markdown
-
-# NIA MCP Server
-
-The NIA MCP Server enables AI assistants like Claude to search and understand your indexed codebases through the Model Context Protocol (MCP).
-
-## Quick Start
-
-### 1. Get your NIA API Key
-
-Sign up and get your API key at [https://trynia.ai/api-keys](https://trynia.ai/api-keys)
-
-### 2. Install via pip
-
-```bash
-pip install nia-mcp-server
-```
-
-### 3. Configure with Claude Desktop
-
-Add to your Claude Desktop configuration (`~/Library/Application Support/Claude/claude_desktop_config.json` on macOS):
-
-```json
-{
-  "mcpServers": {
-    "nia": {
-      "command": "nia-mcp-server",
-      "env": {
-        "NIA_API_KEY": "your-api-key-here"
-      }
-    }
-  }
-}
-```
-
-### 4. Restart Claude Desktop
-
-That's it! You can now ask Claude to index and search codebases.
-
-## Usage Examples
-
-### Index a repository
-```
-Claude, please index https://github.com/facebook/react
-```
-
-### Index documentation
-```
-Index the documentation at https://docs.python.org
-```
-
-### Search across everything
-```
-How does async/await work? Search both my code and documentation.
-```
-
-### Search only repositories
-```
-Find the authentication logic in my repositories
-```
-
-### Search only documentation
-```
-What are the best practices for error handling according to the docs?
-```
-
-### Search and index new content
-```
-Find the best RAG implementations out there
-```
-Claude will:
-1. Use the `nia_web_search` tool to find trending RAG repos
-2. Show you the results with summaries
-3. Prompt you to index the ones you want
-4. You say "Index the first two" and it indexes them!
-
-```
-What are the hottest new Rust web frameworks this week?
-```
-Claude searches trending repos and guides you through indexing them.
-
-Advanced search examples:
-```
-Find GitHub repos similar to langchain/langchain
-
-Search for AI papers published in the last 30 days
-
-What are the trending machine learning frameworks this month?
-```
-
-### Deep research questions
-```
-Compare the top 3 vector databases for RAG applications
-
-What are the pros and cons of different LLM orchestration frameworks?
-
-Research the latest developments in AI agent architectures
-```
-
-### List your resources
-```
-Show me all my indexed repositories and documentation
-```
-
-## Available Tools
-
-### Search & Research
-- **`nia_web_search`** - AI-powered search of repositories, documentation, and content
-  - Finds trending GitHub repos, relevant documentation, and more
-  - Returns structured results that guide you to index the best content
-  - Advanced options:
-    - `category`: Filter by type (github, company, research paper, news, etc.)
-    - `days_back`: Find content from the last N days (great for trending)
-    - `find_similar_to`: Search for content similar to a given URL
-  - Built into NIA's advanced search capabilities
-
-- **`nia_deep_research_agent`** - Multi-step AI research for complex questions
-  - Best for comparative analysis, comprehensive overviews
-  - Returns structured data with citations
-  - Examples: "Compare top RAG frameworks", "Analyze trends in AI safety"
-
-### Repository Management
-- **`index_repository`** - Index a GitHub repository
-- **`list_repositories`** - List all indexed repositories
-- **`check_repository_status`** - Check repository indexing progress
-- **`delete_repository`** - Remove an indexed repository
-
-### Documentation Management
-- **`index_documentation`** - Index documentation or any website
-- **`list_documentation`** - List all indexed documentation sources
-- **`check_documentation_status`** - Check documentation indexing progress
-- **`delete_documentation`** - Remove indexed documentation
-
-### Unified Search
-- **`search_codebase`** - Search across repositories and/or documentation
-  - `search_mode`: "repositories", "sources", or "unified" (default)
-  - Automatically searches all indexed content if not specified
-
-## Other MCP Clients
-
-### Continue.dev
-
-Add to your `~/.continue/config.json`:
-
-```json
-{
-  "models": [...],
-  "mcpServers": [
-    {
-      "name": "nia",
-      "command": "nia-mcp-server",
-      "env": {
-        "NIA_API_KEY": "your-api-key-here"
-      }
-    }
-  ]
-}
-```
-
-### VS Code Cline
-
-Add to your Cline settings:
-
-```json
-{
-  "mcpServers": {
-    "nia": {
-      "command": "nia-mcp-server",
-      "env": {
-        "NIA_API_KEY": "your-api-key-here"
-      }
-    }
-  }
-}
-```
-
-## Environment Variables
-
-- `NIA_API_KEY` (required) - Your NIA API key
-- `NIA_API_URL` (optional) - API endpoint (defaults to https://apigcp.trynia.ai)
-
-## Pricing
-
-NIA offers simple, transparent pricing:
-
-- **Free Tier**: Limited usage, public repos only
-- **Pro**: Unlimited API calls, private repos, advanced features
-
-See [https://trynia.ai/pricing](https://trynia.ai/pricing) for details.
-
-## Features
-
-- 🔍 **Unified Search** - Search across code AND documentation seamlessly
-- 📚 **Documentation Indexing** - Index any website or documentation
-- 🚀 **Fast Indexing** - Index repositories and websites quickly
-- 🔒 **Private Repos** - Support for private repositories (Pro)
-- 📊 **Smart Understanding** - AI-powered code and content comprehension
-- 🌐 **Works Everywhere** - Any MCP-compatible client
-
-## Troubleshooting
-
-### "No API key provided"
-Make sure `NIA_API_KEY` is set in your MCP client configuration.
-
-### "Invalid API key"
-Check your API key at [https://trynia.ai/api-keys](https://trynia.ai/api-keys)
-
-### "Rate limit exceeded"
-You've hit your monthly limit. Upgrade at [https://trynia.ai/billing](https://trynia.ai/billing)
-
-### Repository not indexing
-Large repositories can take a few minutes. Use `check_repository_status` to monitor progress.
-
-## Support
-
-- Documentation: [https://docs.trynia.ai](https://docs.trynia.ai)
-- Discord: [https://discord.gg/BBSwUMrrfn](https://discord.gg/BBSwUMrrfn)
-- Email: support@trynia.ai
-
-## License
-
-MIT License - see LICENSE file for details.