nia-mcp-server 1.0.11__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.11"
+__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/api_client.py

@@ -75,7 +75,7 @@ class NIAApiClient:
                     "lifetime limit",
                     "no chat credits",
                     "free api requests",
-                    "25 free",
+                    "5 free",
                     "usage limit",
                 ]
             ):
@@ -99,7 +99,7 @@ class NIAApiClient:
                     for phrase in [
                         "lifetime limit",
                         "free api requests",
-                        "25 free",
+                        "5 free",
                         "usage limit",
                     ]
                 ):
@@ -623,4 +623,30 @@ class NIAApiClient:
         except httpx.HTTPStatusError as e:
             raise self._handle_api_error(e)
         except Exception as e:
-            raise APIError(f"Deep research failed: {str(e)}")
+            raise APIError(f"Deep research failed: {str(e)}")
+    
+    async def get_source_content(
+        self,
+        source_type: str,
+        source_identifier: str,
+        metadata: Dict[str, Any] = None
+    ) -> Dict[str, Any]:
+        """Get full content of a specific source file or document."""
+        try:
+            payload = {
+                "source_type": source_type,
+                "source_identifier": source_identifier,
+                "metadata": metadata or {}
+            }
+            
+            response = await self.client.post(
+                f"{self.base_url}/v2/sources/content",
+                json=payload
+            )
+            response.raise_for_status()
+            return response.json()
+            
+        except httpx.HTTPStatusError as e:
+            raise self._handle_api_error(e)
+        except Exception as e:
+            raise APIError(f"Failed to get source content: {str(e)}")

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/server.py

@@ -25,7 +25,7 @@ load_dotenv(env_path)
 
 # Configure logging
 logging.basicConfig(
-    level=logging.INFO,
+    level=logging.INFO,  # Changed from INFO to DEBUG for troubleshooting
     format='%(asctime)s [%(levelname)s] %(name)s: %(message)s',
 )
 logger = logging.getLogger(__name__)
@@ -122,7 +122,7 @@ async def index_repository(
     except APIError as e:
         logger.error(f"API Error indexing repository: {e} (status_code={e.status_code}, detail={e.detail})")
         if e.status_code == 403 or "free tier limit" in str(e).lower() or "free api requests" in str(e).lower():
-            if e.detail and "25 free API requests" in e.detail:
+            if e.detail and "5 free API requests" in e.detail:
                 return [TextContent(
                     type="text", 
                     text=f"❌ {e.detail}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited API access."
@@ -222,9 +222,11 @@ async def search_codebase(
                     response_parts.append(data["content"])
                 
                 if "sources" in data and data["sources"]:
+                    logger.debug(f"Received sources data: {type(data['sources'])}, count: {len(data['sources'])}")
                     sources_parts.extend(data["sources"])
                     
-            except json.JSONDecodeError:
+            except json.JSONDecodeError as e:
+                logger.warning(f"Failed to parse JSON chunk: {chunk}, error: {e}")
                 continue
         
         # Format the response
@@ -232,21 +234,53 @@ async def search_codebase(
         
         if sources_parts and include_sources:
             response_text += "\n\n## Sources\n\n"
-            for i, source in enumerate(sources_parts[:5], 1):  # Limit to 5 sources
+            for i, source in enumerate(sources_parts[:10], 1):  # Limit to 10 sources (matches backend)
                 response_text += f"### Source {i}\n"
-                if "repository" in source:
-                    response_text += f"**Repository:** {source['repository']}\n"
-                if "file" in source:
-                    response_text += f"**File:** `{source['file']}`\n"
-                if "preview" in source:
-                    response_text += f"```\n{source['preview']}\n```\n\n"
+                
+                # Handle both string sources (file paths) and dictionary sources
+                if isinstance(source, str):
+                    # Source is just a file path string
+                    response_text += f"**File:** `{source}`\n\n"
+                    continue
+                elif not isinstance(source, dict):
+                    logger.warning(f"Expected source to be dict or str, got {type(source)}: {source}")
+                    response_text += f"**Source:** {str(source)}\n\n"
+                    continue
+                
+                # Handle dictionary sources with metadata
+                metadata = source.get("metadata", {})
+                
+                # Repository name
+                repository = source.get("repository") or metadata.get("source_name") or metadata.get("repository")
+                if repository:
+                    response_text += f"**Repository:** {repository}\n"
+                
+                # File path
+                file_path = source.get("file") or source.get("file_path") or metadata.get("file_path")
+                if file_path:
+                    response_text += f"**File:** `{file_path}`\n"
+                
+                # Content/preview
+                content = source.get("preview") or source.get("content")
+                if content:
+                    # Truncate very long content
+                    if len(content) > 500:
+                        content = content[:500] + "..."
+                    response_text += f"```\n{content}\n```\n\n"
+                else:
+                    # If no content, at least show that this is a valid source
+                    response_text += f"*Referenced source*\n\n"
+            
+            # Add helpful text about read_source_content tool
+            response_text += "\n💡 **Need more details from a source?**\n\n"
+            response_text += "If you need more information from the source links provided above, use the `read_source_content` tool from the available tools provided by Nia to get full context about that particular source.\n"
         
         return [TextContent(type="text", text=response_text)]
         
     except APIError as e:
         logger.error(f"API Error searching codebase: {e} (status_code={e.status_code}, detail={e.detail})")
         if e.status_code == 403 or "free tier limit" in str(e).lower() or "free api requests" in str(e).lower():
-            if e.detail and "25 free API requests" in e.detail:
+            if e.detail and "5 free API requests" in e.detail:
                 return [TextContent(
                     type="text", 
                     text=f"❌ {e.detail}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited API access."
@@ -287,6 +321,9 @@ async def search_documentation(
         
     Returns:
         Search results with relevant documentation excerpts
+
+    Important:
+        - Always use Source ID. If you don't have it, use `list_documentation` tool to get it.
     """
     try:
         client = await ensure_api_client()
@@ -316,7 +353,7 @@ async def search_documentation(
             messages=messages,
             repositories=[],  # No repositories
             data_sources=sources,
-            search_mode="unified",  # Use unified for full answers with documentation
+            search_mode="unified",  # Use unified mode for intelligent LLM processing
             stream=True,
             include_sources=include_sources
         ):
@@ -327,9 +364,11 @@ async def search_documentation(
                     response_parts.append(data["content"])
                 
                 if "sources" in data and data["sources"]:
+                    logger.debug(f"Received doc sources data: {type(data['sources'])}, count: {len(data['sources'])}")
                     sources_parts.extend(data["sources"])
                     
-            except json.JSONDecodeError:
+            except json.JSONDecodeError as e:
+                logger.warning(f"Failed to parse JSON chunk in documentation search: {chunk}, error: {e}")
                 continue
         
         # Format the response
@@ -337,14 +376,42 @@ async def search_documentation(
         
         if sources_parts and include_sources:
             response_text += "\n\n## Sources\n\n"
-            for i, source in enumerate(sources_parts[:5], 1):  # Limit to 5 sources
+            for i, source in enumerate(sources_parts[:10], 1):  # Limit to 10 sources (matches backend)
                 response_text += f"### Source {i}\n"
-                if "url" in source:
-                    response_text += f"**URL:** {source['url']}\n"
-                elif "file" in source:
-                    response_text += f"**Page:** {source['file']}\n"
-                if "preview" in source:
-                    response_text += f"```\n{source['preview']}\n```\n\n"
+                
+                # Handle both string sources and dictionary sources
+                if isinstance(source, str):
+                    # Source is just a URL or file path string
+                    response_text += f"**Document:** {source}\n\n"
+                    continue
+                elif not isinstance(source, dict):
+                    logger.warning(f"Expected source to be dict or str, got {type(source)}: {source}")
+                    response_text += f"**Source:** {str(source)}\n\n"
+                    continue
+                
+                # Handle dictionary sources with metadata
+                metadata = source.get("metadata", {})
+                
+                # URL or file
+                url = source.get("url") or metadata.get("url") or metadata.get("source") or metadata.get("sourceURL")
+                file_path = source.get("file") or source.get("file_path") or metadata.get("file_path") or metadata.get("document_name")
+                
+                if url:
+                    response_text += f"**URL:** {url}\n"
+                elif file_path:
+                    response_text += f"**Document:** {file_path}\n"
+                
+                # Title if available
+                title = source.get("title") or metadata.get("title")
+                if title:
+                    response_text += f"**Title:** {title}\n"
+                
+                # Add spacing after each source
+                response_text += "\n"
+            
+            # Add helpful text about read_source_content tool
+            response_text += "\n💡 **Need more details from a source?**\n\n"
+            response_text += "If you need more information from the source links provided above, use the `read_source_content` tool from the available tools provided by Nia to get full context about that particular source.\n"
         
         return [TextContent(type="text", text=response_text)]
         
@@ -352,7 +419,7 @@ async def search_documentation(
         logger.error(f"API Error searching documentation: {e}")
         error_msg = f"❌ {str(e)}"
         if e.status_code == 403 and "lifetime limit" in str(e).lower():
-            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 25 API requests. Upgrade to Pro for unlimited access."
+            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 5 API requests. Upgrade to Pro for unlimited access."
         return [TextContent(type="text", text=error_msg)]
     except Exception as e:
         logger.error(f"Error searching documentation: {e}")
@@ -411,7 +478,7 @@ async def list_repositories() -> List[TextContent]:
         # Check for free tier limit errors
         if e.status_code == 403 or "free tier limit" in str(e).lower() or "free api requests" in str(e).lower():
             # Extract the specific limit message
-            if e.detail and "25 free API requests" in e.detail:
+            if e.detail and "5 free API requests" in e.detail:
                 return [TextContent(
                     type="text", 
                     text=f"❌ {e.detail}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited API access."
@@ -491,7 +558,7 @@ async def check_repository_status(repository: str) -> List[TextContent]:
         logger.error(f"API Error checking repository status: {e}")
         error_msg = f"❌ {str(e)}"
         if e.status_code == 403 and "lifetime limit" in str(e).lower():
-            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 25 API requests. Upgrade to Pro for unlimited access."
+            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 5 API requests. Upgrade to Pro for unlimited access."
         return [TextContent(type="text", text=error_msg)]
     except Exception as e:
         logger.error(f"Error checking repository status: {e}")
@@ -578,7 +645,7 @@ async def index_documentation(
         logger.error(f"API Error indexing documentation: {e}")
         error_msg = f"❌ {str(e)}"
         if e.status_code == 403 and "lifetime limit" in str(e).lower():
-            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 25 API requests. Upgrade to Pro for unlimited access."
+            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 5 API requests. Upgrade to Pro for unlimited access."
         return [TextContent(type="text", text=error_msg)]
     except Exception as e:
         logger.error(f"Error indexing documentation: {e}")
@@ -637,7 +704,7 @@ async def list_documentation() -> List[TextContent]:
         logger.error(f"API Error listing documentation: {e}")
         error_msg = f"❌ {str(e)}"
         if e.status_code == 403 and "lifetime limit" in str(e).lower():
-            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 25 API requests. Upgrade to Pro for unlimited access."
+            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 5 API requests. Upgrade to Pro for unlimited access."
         return [TextContent(type="text", text=error_msg)]
     except Exception as e:
         logger.error(f"Error listing documentation: {e}")
@@ -703,7 +770,7 @@ async def check_documentation_status(source_id: str) -> List[TextContent]:
         logger.error(f"API Error checking documentation status: {e}")
         error_msg = f"❌ {str(e)}"
         if e.status_code == 403 and "lifetime limit" in str(e).lower():
-            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 25 API requests. Upgrade to Pro for unlimited access."
+            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 5 API requests. Upgrade to Pro for unlimited access."
         return [TextContent(type="text", text=error_msg)]
     except Exception as e:
         logger.error(f"Error checking documentation status: {e}")
@@ -742,7 +809,7 @@ async def delete_documentation(source_id: str) -> List[TextContent]:
         logger.error(f"API Error deleting documentation: {e}")
         error_msg = f"❌ {str(e)}"
         if e.status_code == 403 and "lifetime limit" in str(e).lower():
-            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 25 API requests. Upgrade to Pro for unlimited access."
+            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 5 API requests. Upgrade to Pro for unlimited access."
         return [TextContent(type="text", text=error_msg)]
     except Exception as e:
         logger.error(f"Error deleting documentation: {e}")
@@ -781,7 +848,7 @@ async def delete_repository(repository: str) -> List[TextContent]:
         logger.error(f"API Error deleting repository: {e}")
         error_msg = f"❌ {str(e)}"
         if e.status_code == 403 and "lifetime limit" in str(e).lower():
-            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 25 API requests. Upgrade to Pro for unlimited access."
+            error_msg += "\n\n💡 Tip: You've reached the free tier limit of 5 API requests. Upgrade to Pro for unlimited access."
         return [TextContent(type="text", text=error_msg)]
     except Exception as e:
         logger.error(f"Error deleting repository: {e}")
@@ -1033,7 +1100,7 @@ async def nia_web_search(
     except APIError as e:
         logger.error(f"API Error in web search: {e}")
         if e.status_code == 403 or "free tier limit" in str(e).lower() or "free api requests" in str(e).lower():
-            if e.detail and "25 free API requests" in e.detail:
+            if e.detail and "5 free API requests" in e.detail:
                 return [TextContent(
                     type="text", 
                     text=f"❌ {e.detail}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited API access."
@@ -1212,7 +1279,7 @@ async def nia_deep_research_agent(
     except APIError as e:
         logger.error(f"API Error in deep research: {e}")
         if e.status_code == 403 or "free tier limit" in str(e).lower() or "free api requests" in str(e).lower():
-            if e.detail and "25 free API requests" in e.detail:
+            if e.detail and "5 free API requests" in e.detail:
                 return [TextContent(
                     type="text", 
                     text=f"❌ {e.detail}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited API access."
@@ -1388,6 +1455,134 @@ async def initialize_project(
                  "- The NIA MCP server is properly installed"
         )]
 
+@mcp.tool()
+async def read_source_content(
+    source_type: str,
+    source_identifier: str,
+    metadata: Optional[Dict[str, Any]] = None
+) -> List[TextContent]:
+    """
+    Read the full content of a specific source file or document.
+    
+    This tool allows AI to fetch complete content from sources identified during search,
+    enabling deeper analysis when the truncated search results are insufficient.
+    
+    Args:
+        source_type: Type of source - "repository" or "documentation"
+        source_identifier: 
+            - For repository: "owner/repo:path/to/file.py" (e.g., "facebook/react:src/React.js")
+            - For documentation: The source URL or document ID
+        metadata: Optional metadata from search results to help locate the source
+        
+    Returns:
+        Full content of the requested source with metadata
+        
+    Examples:
+        - read_source_content("repository", "langchain-ai/langchain:libs/core/langchain_core/runnables/base.py")
+        - read_source_content("documentation", "https://docs.python.org/3/library/asyncio.html")
+    """
+    try:
+        client = await ensure_api_client()
+        
+        logger.info(f"Reading source content - type: {source_type}, identifier: {source_identifier}")
+        
+        # Call the API to get source content
+        result = await client.get_source_content(
+            source_type=source_type,
+            source_identifier=source_identifier,
+            metadata=metadata or {}
+        )
+        
+        if not result or not result.get("success"):
+            error_msg = result.get("error", "Unknown error") if result else "Failed to fetch source content"
+            return [TextContent(
+                type="text",
+                text=f"❌ Error reading source: {error_msg}"
+            )]
+        
+        # Format the response
+        content = result.get("content", "")
+        source_metadata = result.get("metadata", {})
+        
+        # Build response with metadata header
+        response_lines = []
+        
+        if source_type == "repository":
+            repo_name = source_metadata.get("repository", "Unknown")
+            file_path = source_metadata.get("file_path", source_identifier.split(":", 1)[-1] if ":" in source_identifier else "Unknown")
+            branch = source_metadata.get("branch", "main")
+            
+            response_lines.extend([
+                f"# Source: {repo_name}",
+                f"**File:** `{file_path}`",
+                f"**Branch:** {branch}",
+                ""
+            ])
+            
+            if source_metadata.get("url"):
+                response_lines.append(f"**GitHub URL:** {source_metadata['url']}")
+                response_lines.append("")
+            
+            # Add file info if available
+            if source_metadata.get("size"):
+                response_lines.append(f"**Size:** {source_metadata['size']} bytes")
+            if source_metadata.get("language"):
+                response_lines.append(f"**Language:** {source_metadata['language']}")
+                
+            response_lines.extend(["", "## Content", ""])
+            
+            # Add code block with language hint
+            language = source_metadata.get("language", "").lower() or "text"
+            response_lines.append(f"```{language}")
+            response_lines.append(content)
+            response_lines.append("```")
+            
+        elif source_type == "documentation":
+            url = source_metadata.get("url", source_identifier)
+            title = source_metadata.get("title", "Documentation")
+            
+            response_lines.extend([
+                f"# Documentation: {title}",
+                f"**URL:** {url}",
+                ""
+            ])
+            
+            if source_metadata.get("last_updated"):
+                response_lines.append(f"**Last Updated:** {source_metadata['last_updated']}")
+                response_lines.append("")
+                
+            response_lines.extend(["## Content", "", content])
+        
+        else:
+            # Generic format for unknown source types
+            response_lines.extend([
+                f"# Source Content",
+                f"**Type:** {source_type}",
+                f"**Identifier:** {source_identifier}",
+                "",
+                "## Content",
+                "",
+                content
+            ])
+        
+        return [TextContent(type="text", text="\n".join(response_lines))]
+        
+    except APIError as e:
+        logger.error(f"API Error reading source content: {e}")
+        if e.status_code == 403 or "free tier limit" in str(e).lower():
+            return [TextContent(
+                type="text",
+                text=f"❌ {str(e)}\n\n💡 Tip: Upgrade to Pro at https://trynia.ai/billing for unlimited access."
+            )]
+        else:
+            return [TextContent(type="text", text=f"❌ {str(e)}")]
+    except Exception as e:
+        logger.error(f"Error reading source content: {e}")
+        return [TextContent(
+            type="text",
+            text=f"❌ Error reading source content: {str(e)}"
+        )]
+
 @mcp.tool()
 async def visualize_codebase(
     repository: str

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.11.dist-info → nia_mcp_server-1.0.13.dist-info}/RECORD

@@ -1,17 +1,19 @@
-nia_mcp_server/__init__.py,sha256=PIwoKKoMquY_o5GY6__kwKO3_G2yz-zYewTJRWJc9W0,85
-nia_mcp_server/__main__.py,sha256=XY11ESL4hctu-BBgtPATFZyd1o-O7wE7y-UOSoNs-hw,152
-nia_mcp_server/api_client.py,sha256=Ix-GB8tw2FCALqkwzjIQySeeGkY-PqnHX-n3UQBa5sk,24943
+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=cyPv_Za4HEUj00QCLeeadcqqmSWRzVUXBrttCqXHhqk,66799
+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.11.dist-info/METADATA,sha256=4okbTVyaTi-09qZC2su-R1iGeqqB4fGPjy6ZbpYayZQ,6767
-nia_mcp_server-1.0.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nia_mcp_server-1.0.11.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
-nia_mcp_server-1.0.11.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
-nia_mcp_server-1.0.11.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.11.dist-info/METADATA

@@ -1,246 +0,0 @@
-Metadata-Version: 2.4
-Name: nia-mcp-server
-Version: 1.0.11
-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.