nia-mcp-server 1.0.5__py3-none-any.whl → 1.0.7__py3-none-any.whl

nia_mcp_server/server.py

@@ -1,5 +1,5 @@
 """
-NIA MCP Proxy Server - Lightweight server that communicates with NIA API
+Nia MCP Proxy Server - Lightweight server that communicates with Nia API
 """
 import os
 import logging
@@ -12,6 +12,8 @@ from urllib.parse import urlparse
 from mcp.server.fastmcp import FastMCP
 from mcp.types import TextContent, Resource
 from .api_client import NIAApiClient, APIError
+from .project_init import initialize_nia_project
+from .profiles import get_supported_profiles
 from dotenv import load_dotenv
 import json
 
@@ -167,7 +169,26 @@ async def search_codebase(
         # Get all indexed repositories if not specified
         if not repositories:
             all_repos = await client.list_repositories()
-            repositories = [repo["repository"] for repo in all_repos if repo.get("status") == "completed"]
+            
+            # Ensure all_repos is a list and contains dictionaries
+            if not isinstance(all_repos, list):
+                logger.error(f"Unexpected type for all_repos: {type(all_repos)}")
+                return [TextContent(
+                    type="text",
+                    text="❌ Error retrieving repositories. The API returned an unexpected response."
+                )]
+            
+            repositories = []
+            for repo in all_repos:
+                if isinstance(repo, dict) and repo.get("status") == "completed":
+                    repo_name = repo.get("repository")
+                    if repo_name:
+                        repositories.append(repo_name)
+                    else:
+                        logger.warning(f"Repository missing 'repository' field: {repo}")
+                else:
+                    logger.warning(f"Unexpected repository format: {type(repo)}, value: {repo}")
+            
             if not repositories:
                 return [TextContent(
                     type="text",
@@ -364,7 +385,17 @@ async def list_repositories() -> List[TextContent]:
         
         for repo in repositories:
             status_icon = "✅" if repo.get("status") == "completed" else "⏳"
-            lines.append(f"\n## {status_icon} {repo['repository']}")
+            
+            # Show display name if available, otherwise show repository
+            display_name = repo.get("display_name")
+            repo_name = repo['repository']
+            
+            if display_name:
+                lines.append(f"\n## {status_icon} {display_name}")
+                lines.append(f"- **Repository:** {repo_name}")
+            else:
+                lines.append(f"\n## {status_icon} {repo_name}")
+            
             lines.append(f"- **Branch:** {repo.get('branch', 'main')}")
             lines.append(f"- **Status:** {repo.get('status', 'unknown')}")
             if repo.get("indexed_at"):
@@ -559,7 +590,17 @@ async def list_documentation() -> List[TextContent]:
         
         for source in sources:
             status_icon = "✅" if source.get("status") == "completed" else "⏳"
-            lines.append(f"\n## {status_icon} {source.get('url', 'Unknown URL')}")
+            
+            # Show display name if available, otherwise show URL
+            display_name = source.get("display_name")
+            url = source.get('url', 'Unknown URL')
+            
+            if display_name:
+                lines.append(f"\n## {status_icon} {display_name}")
+                lines.append(f"- **URL:** {url}")
+            else:
+                lines.append(f"\n## {status_icon} {url}")
+            
             lines.append(f"- **ID:** {source['id']}")
             lines.append(f"- **Status:** {source.get('status', 'unknown')}")
             lines.append(f"- **Type:** {source.get('source_type', 'web')}")
@@ -727,6 +768,100 @@ async def delete_repository(repository: str) -> List[TextContent]:
             text=f"❌ Error deleting repository: {str(e)}"
         )]
 
+@mcp.tool()
+async def rename_repository(repository: str, new_name: str) -> List[TextContent]:
+    """
+    Rename an indexed repository for better organization.
+    
+    Args:
+        repository: Repository in owner/repo format
+        new_name: New display name for the repository (1-100 characters)
+        
+    Returns:
+        Confirmation of rename operation
+    """
+    try:
+        # Validate name length
+        if not new_name or len(new_name) > 100:
+            return [TextContent(
+                type="text",
+                text="❌ Display name must be between 1 and 100 characters."
+            )]
+        
+        client = await ensure_api_client()
+        result = await client.rename_repository(repository, new_name)
+        
+        if result.get("success"):
+            return [TextContent(
+                type="text",
+                text=f"✅ Successfully renamed repository '{repository}' to '{new_name}'"
+            )]
+        else:
+            return [TextContent(
+                type="text",
+                text=f"❌ Failed to rename repository: {result.get('message', 'Unknown error')}"
+            )]
+            
+    except APIError as e:
+        logger.error(f"API Error renaming 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. Upgrade to Pro for unlimited access."
+        return [TextContent(type="text", text=error_msg)]
+    except Exception as e:
+        logger.error(f"Error renaming repository: {e}")
+        return [TextContent(
+            type="text",
+            text=f"❌ Error renaming repository: {str(e)}"
+        )]
+
+@mcp.tool()
+async def rename_documentation(source_id: str, new_name: str) -> List[TextContent]:
+    """
+    Rename a documentation source for better organization.
+    
+    Args:
+        source_id: Documentation source ID
+        new_name: New display name for the documentation (1-100 characters)
+        
+    Returns:
+        Confirmation of rename operation
+    """
+    try:
+        # Validate name length
+        if not new_name or len(new_name) > 100:
+            return [TextContent(
+                type="text",
+                text="❌ Display name must be between 1 and 100 characters."
+            )]
+        
+        client = await ensure_api_client()
+        result = await client.rename_data_source(source_id, new_name)
+        
+        if result.get("success"):
+            return [TextContent(
+                type="text",
+                text=f"✅ Successfully renamed documentation source to '{new_name}'"
+            )]
+        else:
+            return [TextContent(
+                type="text",
+                text=f"❌ Failed to rename documentation: {result.get('message', 'Unknown error')}"
+            )]
+            
+    except APIError as e:
+        logger.error(f"API Error renaming 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. Upgrade to Pro for unlimited access."
+        return [TextContent(type="text", text=error_msg)]
+    except Exception as e:
+        logger.error(f"Error renaming documentation: {e}")
+        return [TextContent(
+            type="text",
+            text=f"❌ Error renaming documentation: {str(e)}"
+        )]
+
 @mcp.tool()
 async def nia_web_search(
     query: str,
@@ -781,7 +916,7 @@ async def nia_web_search(
         other_content = result.get("other_content", [])
         
         # Format response to naturally guide next actions
-        response_text = f"## 🔍 NIA Web Search Results for: \"{query}\"\n\n"
+        response_text = f"## 🔍 Nia Web Search Results for: \"{query}\"\n\n"
         
         if days_back:
             response_text += f"*Showing results from the last {days_back} days*\n\n"
@@ -806,7 +941,7 @@ async def nia_web_search(
             
             # Be more aggressive based on query specificity
             if len(github_repos) == 1 or any(specific_word in query.lower() for specific_word in ["specific", "exact", "particular", "find me", "looking for"]):
-                response_text += "**🚀 RECOMMENDED ACTION - Index this repository with NIA:**\n"
+                response_text += "**🚀 RECOMMENDED ACTION - Index this repository with Nia:**\n"
                 response_text += f"```\nIndex {github_repos[0]['owner_repo']}\n```\n"
                 response_text += "✨ This will enable AI-powered code search, understanding, and analysis!\n\n"
             else:
@@ -1075,6 +1210,162 @@ async def nia_deep_research_agent(
                  "Try simplifying your question or using the regular nia_web_search tool."
         )]
 
+@mcp.tool()
+async def initialize_project(
+    project_root: str,
+    profiles: Optional[List[str]] = None
+) -> List[TextContent]:
+    """
+    Initialize a NIA-enabled project with IDE-specific rules and configurations.
+    
+    This tool sets up your project with NIA integration, creating configuration files
+    and rules tailored to your IDE or editor. It enables AI assistants to better
+    understand and work with NIA's knowledge search capabilities.
+    
+    Args:
+        project_root: Absolute path to the project root directory
+        profiles: List of IDE profiles to set up (default: ["cursor"]). 
+                 Options: cursor, vscode, claude, windsurf, cline, codex, zed, jetbrains, neovim, sublime
+        
+    Returns:
+        Status of the initialization with created files and next steps
+    
+    Examples:
+        - Basic: initialize_project("/path/to/project")
+        - Multiple IDEs: initialize_project("/path/to/project", profiles=["cursor", "vscode"])
+        - Specific IDE: initialize_project("/path/to/project", profiles=["windsurf"])
+    """
+    try:
+        # Validate project root
+        project_path = Path(project_root)
+        if not project_path.is_absolute():
+            return [TextContent(
+                type="text",
+                text=f"❌ Error: project_root must be an absolute path. Got: {project_root}"
+            )]
+        
+        # Default to cursor profile if none specified
+        if profiles is None:
+            profiles = ["cursor"]
+        
+        # Validate profiles
+        supported = get_supported_profiles()
+        invalid_profiles = [p for p in profiles if p not in supported]
+        if invalid_profiles:
+            return [TextContent(
+                type="text",
+                text=f"❌ Unknown profiles: {', '.join(invalid_profiles)}\n\n"
+                     f"Supported profiles: {', '.join(supported)}"
+            )]
+        
+        logger.info(f"Initializing NIA project at {project_root} with profiles: {profiles}")
+        
+        # Initialize the project
+        result = initialize_nia_project(
+            project_root=project_root,
+            profiles=profiles
+        )
+        
+        if not result.get("success"):
+            return [TextContent(
+                type="text",
+                text=f"❌ Failed to initialize project: {result.get('error', 'Unknown error')}"
+            )]
+        
+        # Format success response
+        response_lines = [
+            f"✅ Successfully initialized NIA project at: {project_root}",
+            "",
+            "## 📁 Created Files:",
+        ]
+        
+        for file in result.get("files_created", []):
+            response_lines.append(f"- {file}")
+        
+        if result.get("profiles_initialized"):
+            response_lines.extend([
+                "",
+                "## 🎨 Initialized Profiles:",
+            ])
+            for profile in result["profiles_initialized"]:
+                response_lines.append(f"- {profile}")
+        
+        if result.get("warnings"):
+            response_lines.extend([
+                "",
+                "## ⚠️ Warnings:",
+            ])
+            for warning in result["warnings"]:
+                response_lines.append(f"- {warning}")
+        
+        if result.get("next_steps"):
+            response_lines.extend([
+                "",
+                "## 🚀 Next Steps:",
+            ])
+            for i, step in enumerate(result["next_steps"], 1):
+                response_lines.append(f"{i}. {step}")
+        
+        # Add profile-specific instructions
+        response_lines.extend([
+            "",
+            "## 💡 Quick Start:",
+        ])
+        
+        if "cursor" in profiles:
+            response_lines.extend([
+                "**For Cursor:**",
+                "1. Restart Cursor to load the NIA MCP server",
+                "2. Run `list_repositories` to verify connection",
+                "3. Start indexing with `index_repository https://github.com/owner/repo`",
+                ""
+            ])
+        
+        if "vscode" in profiles:
+            response_lines.extend([
+                "**For VSCode:**",
+                "1. Reload the VSCode window (Cmd/Ctrl+R)",
+                "2. Open command palette (Cmd/Ctrl+Shift+P)",
+                "3. Run 'NIA: Index Repository' task",
+                ""
+            ])
+        
+        if "claude" in profiles:
+            response_lines.extend([
+                "**For Claude Desktop:**",
+                "1. The .claude directory has been created",
+                "2. Claude will now understand NIA commands",
+                "3. Try: 'Search for authentication patterns'",
+                ""
+            ])
+        
+        # Add general tips
+        response_lines.extend([
+            "## 📚 Tips:",
+            "- Use natural language for searches: 'How does X work?'",
+            "- Index repositories before searching them",
+            "- Use `nia_web_search` to discover new repositories",
+            "- Check `list_repositories` to see what's already indexed",
+            "",
+            "Ready to supercharge your development with AI-powered code search! 🚀"
+        ])
+        
+        return [TextContent(
+            type="text",
+            text="\n".join(response_lines)
+        )]
+        
+    except Exception as e:
+        logger.error(f"Error in initialize_project tool: {e}")
+        return [TextContent(
+            type="text",
+            text=f"❌ Error initializing project: {str(e)}\n\n"
+                 "Please check:\n"
+                 "- The project_root path is correct and accessible\n"
+                 "- You have write permissions to the directory\n"
+                 "- The NIA MCP server is properly installed"
+        )]
+
 # Resources
 
 # Note: FastMCP doesn't have list_resources or read_resource decorators

{nia_mcp_server-1.0.5.dist-info → nia_mcp_server-1.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nia-mcp-server
-Version: 1.0.5
+Version: 1.0.7
 Summary: NIA Knowledge Agent - MCP server for intelligent codebase search
 Project-URL: Homepage, https://trynia.ai
 Project-URL: Documentation, https://docs.trynia.ai

nia_mcp_server-1.0.7.dist-info/RECORD

@@ -0,0 +1,17 @@
+nia_mcp_server/__init__.py,sha256=Q-MLB7q1vWZ-UKUj8MzVk2mtiydn9wTsFtQ7kicpOyo,84
+nia_mcp_server/__main__.py,sha256=XY11ESL4hctu-BBgtPATFZyd1o-O7wE7y-UOSoNs-hw,152
+nia_mcp_server/api_client.py,sha256=dk9FkMljuekyjIw7Ij3athsoVNnu7E2b1l2xi2XtB1Y,24255
+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=B6aLTXkX2Mq5eBqDCV_WTumqtSm4bzDf2bHPVyVsgcA,59579
+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.7.dist-info/METADATA,sha256=mbOODzDtkXWQV9tXdYY2lVc5XrkarBsM1genV3dl3Jw,6910
+nia_mcp_server-1.0.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nia_mcp_server-1.0.7.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
+nia_mcp_server-1.0.7.dist-info/licenses/LICENSE,sha256=5jUPBVkZEicxSAZ91jOO7i8zXEPAHS6M0w8SSf6DftI,1071
+nia_mcp_server-1.0.7.dist-info/RECORD,,

nia_mcp_server-1.0.5.dist-info/RECORD

@@ -1,9 +0,0 @@
-nia_mcp_server/__init__.py,sha256=P3fqcVlJ9fp8ojx6xaX5lrdUNO8rkc8bGYEv8Qm3SAw,84
-nia_mcp_server/__main__.py,sha256=XY11ESL4hctu-BBgtPATFZyd1o-O7wE7y-UOSoNs-hw,152
-nia_mcp_server/api_client.py,sha256=E7x6W2zvFrfyR8yfk8X4z6WGF6wQk_EP4TfQhZAV_4w,20932
-nia_mcp_server/server.py,sha256=UJaP4Q7xlZt_xW4bDD17AgYhwAZ2sZaoTZ6nf_2mNOE,48694
-nia_mcp_server-1.0.5.dist-info/METADATA,sha256=lyWMV-006_5q1vb_aEnHWwUE8lAYcZMpbL9ypEi5kiI,6910
-nia_mcp_server-1.0.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nia_mcp_server-1.0.5.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
-nia_mcp_server-1.0.5.dist-info/licenses/LICENSE,sha256=5jUPBVkZEicxSAZ91jOO7i8zXEPAHS6M0w8SSf6DftI,1071
-nia_mcp_server-1.0.5.dist-info/RECORD,,