nia-mcp-server 1.0.7__py3-none-any.whl → 1.0.8__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.7"
+__version__ = "1.0.8"

nia_mcp_server/api_client.py

@@ -24,7 +24,7 @@ class NIAApiClient:
     def __init__(self, api_key: str, base_url: str = None):
         self.api_key = api_key
         # Remove trailing slash from base URL to prevent double slashes
-        self.base_url = (base_url or os.getenv("NIA_API_URL", "https://api.trynia.ai")).rstrip('/')
+        self.base_url = (base_url or os.getenv("NIA_API_URL", "https://apigcp.trynia.ai")).rstrip('/')
         self.client = httpx.AsyncClient(
             headers={
                 "Authorization": f"Bearer {api_key}",
@@ -400,6 +400,7 @@ class NIAApiClient:
         self, 
         url: str, 
         url_patterns: List[str] = None,
+        exclude_patterns: List[str] = None,
         max_age: int = None,
         only_main_content: bool = True
     ) -> Dict[str, Any]:
@@ -407,7 +408,8 @@ class NIAApiClient:
         try:
             payload = {
                 "url": url,
-                "url_patterns": url_patterns or []
+                "url_patterns": url_patterns or [],
+                "exclude_patterns": exclude_patterns or []
             }
             
             # Add optional parameters

nia_mcp_server/server.py

@@ -5,6 +5,7 @@ import os
 import logging
 import json
 import asyncio
+import webbrowser
 from typing import List, Optional, Dict, Any
 from datetime import datetime
 from urllib.parse import urlparse
@@ -503,6 +504,7 @@ async def check_repository_status(repository: str) -> List[TextContent]:
 async def index_documentation(
     url: str,
     url_patterns: Optional[List[str]] = None,
+    exclude_patterns: Optional[List[str]] = None,
     max_age: Optional[int] = None,
     only_main_content: Optional[bool] = True
 ) -> List[TextContent]:
@@ -512,6 +514,7 @@ async def index_documentation(
     Args:
         url: URL of the documentation site to index
         url_patterns: Optional list of URL patterns to include in crawling (e.g., ["/docs/*", "/guide/*"])
+        exclude_patterns: Optional list of URL patterns to exclude from crawling (e.g., ["/blog/*", "/changelog/*"])
         max_age: Maximum age of cached content in seconds (for fast scraping mode)
         only_main_content: Extract only main content (removes navigation, ads, etc.)
         
@@ -520,6 +523,8 @@ async def index_documentation(
 
     Important:
         - When started indexing, prompt users to either use check_documentation_status tool or go to app.trynia.ai to check the status.
+        - By default, crawls the entire domain (up to 10,000 pages)
+        - Use exclude_patterns to filter out unwanted sections like blogs, changelogs, etc.
     """
     try:
         client = await ensure_api_client()
@@ -529,6 +534,7 @@ async def index_documentation(
         result = await client.create_data_source(
             url=url, 
             url_patterns=url_patterns,
+            exclude_patterns=exclude_patterns,
             max_age=max_age,
             only_main_content=only_main_content
         )
@@ -1366,6 +1372,159 @@ async def initialize_project(
                  "- The NIA MCP server is properly installed"
         )]
 
+@mcp.tool()
+async def visualize_codebase(
+    repository: str
+) -> List[TextContent]:
+    """
+    Open the graph visualization for an indexed repository in a browser.
+    
+    This tool launches a browser with the interactive graph visualization
+    that shows the code structure, relationships, and dependencies of
+    the indexed codebase.
+    
+    Args:
+        repository: Repository in owner/repo format (e.g., "facebook/react")
+        
+    Returns:
+        Status message with the URL that was opened
+        
+    Examples:
+        - visualize_codebase("facebook/react")
+        - visualize_codebase("langchain-ai/langchain")
+    """
+    try:
+        client = await ensure_api_client()
+        
+        logger.info(f"Looking up repository: {repository}")
+        
+        # List all repositories to find the matching one
+        repositories = await client.list_repositories()
+        
+        # Find the repository by name
+        matching_repo = None
+        for repo in repositories:
+            if repo.get("repository") == repository:
+                matching_repo = repo
+                break
+        
+        if not matching_repo:
+            # Try case-insensitive match as fallback
+            repository_lower = repository.lower()
+            for repo in repositories:
+                if repo.get("repository", "").lower() == repository_lower:
+                    matching_repo = repo
+                    break
+        
+        if not matching_repo:
+            return [TextContent(
+                type="text",
+                text=f"❌ Repository '{repository}' not found.\n\n"
+                     f"Available repositories:\n" +
+                     "\n".join(f"- {r.get('repository')}" for r in repositories if r.get('repository')) +
+                     "\n\nUse `list_repositories` to see all indexed repositories."
+            )]
+        
+        # Check if the repository is fully indexed
+        status = matching_repo.get("status", "unknown")
+        # Use the actual project ID if available, fall back to repository_id
+        repository_id = matching_repo.get("id") or matching_repo.get("repository_id")
+        
+        if not repository_id:
+            return [TextContent(
+                type="text",
+                text=f"❌ No repository ID found for '{repository}'. This may be a data inconsistency."
+            )]
+        
+        if status != "completed":
+            warning_msg = f"⚠️ Note: Repository '{repository}' is currently {status}.\n"
+            if status == "indexing":
+                warning_msg += "The visualization may show incomplete data.\n\n"
+            elif status == "error":
+                error_msg = matching_repo.get("error", "Unknown error")
+                warning_msg += f"Error: {error_msg}\n\n"
+            else:
+                warning_msg += "The visualization may not be available.\n\n"
+        else:
+            warning_msg = ""
+        
+        # Determine the base URL based on the API URL
+        api_base_url = client.base_url
+        if "localhost" in api_base_url or "127.0.0.1" in api_base_url:
+            # Local development
+            app_base_url = "http://localhost:3000"
+        else:
+            # Production
+            app_base_url = "https://app.trynia.ai"
+        
+        # Construct the visualization URL
+        visualization_url = f"{app_base_url}/visualize/{repository_id}"
+        
+        # Try to open the browser
+        try:
+            webbrowser.open(visualization_url)
+            browser_opened = True
+            open_msg = "✅ Opening graph visualization in your default browser..."
+        except Exception as e:
+            logger.warning(f"Failed to open browser: {e}")
+            browser_opened = False
+            open_msg = "⚠️ Could not automatically open browser."
+        
+        # Format the response
+        response_lines = [
+            f"# Graph Visualization: {repository}",
+            "",
+            warning_msg if warning_msg else "",
+            open_msg,
+            "",
+            f"**URL:** {visualization_url}",
+            "",
+        ]
+        
+        if matching_repo.get("display_name"):
+            response_lines.append(f"**Display Name:** {matching_repo['display_name']}")
+        
+        response_lines.extend([
+            f"**Branch:** {matching_repo.get('branch', 'main')}",
+            f"**Status:** {status}",
+            "",
+            "## Features Available:",
+            "- 🔍 Interactive force-directed graph",
+            "- 🎨 Color-coded node types (functions, classes, files, etc.)",
+            "- 🔗 Relationship visualization (calls, imports, inherits, etc.)",
+            "- 💬 Click on any node to chat with that specific code element",
+            "- 🔎 Search and filter capabilities",
+            "- 📊 Graph statistics and insights"
+        ])
+        
+        if not browser_opened:
+            response_lines.extend([
+                "",
+                "**Manual Access:**",
+                f"Copy and paste this URL into your browser: {visualization_url}"
+            ])
+        
+        return [TextContent(
+            type="text",
+            text="\n".join(response_lines)
+        )]
+        
+    except APIError as e:
+        logger.error(f"API Error in visualize_codebase: {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 in visualize_codebase: {e}")
+        return [TextContent(
+            type="text",
+            text=f"❌ Error opening visualization: {str(e)}"
+        )]
+
 # Resources
 
 # Note: FastMCP doesn't have list_resources or read_resource decorators

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

@@ -1,18 +1,16 @@
 Metadata-Version: 2.4
 Name: nia-mcp-server
-Version: 1.0.7
-Summary: NIA Knowledge Agent - MCP server for intelligent codebase search
+Version: 1.0.8
+Summary: Nia Knowledge Agent
 Project-URL: Homepage, https://trynia.ai
 Project-URL: Documentation, https://docs.trynia.ai
-Project-URL: Repository, https://github.com/trynia/nia-mcp-server
-Project-URL: Issues, https://github.com/trynia/nia-mcp-server/issues
-Author-email: NIA Team <support@trynia.ai>
-License-Expression: MIT
+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 :: MIT License
+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
@@ -203,7 +201,7 @@ Add to your Cline settings:
 ## Environment Variables
 
 - `NIA_API_KEY` (required) - Your NIA API key
-- `NIA_API_URL` (optional) - API endpoint (defaults to https://api.trynia.ai)
+- `NIA_API_URL` (optional) - API endpoint (defaults to https://apigcp.trynia.ai)
 
 ## Pricing
 

{nia_mcp_server-1.0.7.dist-info → nia_mcp_server-1.0.8.dist-info}/RECORD

@@ -1,17 +1,16 @@
-nia_mcp_server/__init__.py,sha256=Q-MLB7q1vWZ-UKUj8MzVk2mtiydn9wTsFtQ7kicpOyo,84
+nia_mcp_server/__init__.py,sha256=7gOUBmOzahYrylDFhFFQLPegegf4dI-ZGt7UNfUVXho,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/api_client.py,sha256=zcTWTDkG9KYj5xunx8A8t1JDLctJJqNqB_ZiGZc9H1M,24362
 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/server.py,sha256=m8tvape27a72NyQEho8_tT9z0plRGBR0kcEB6gSrqsw,65816
 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.8.dist-info/METADATA,sha256=A5e26Zda8sTmFMYoQnfeIOM1LEfXxiWnclFva2YMMGo,6766
+nia_mcp_server-1.0.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nia_mcp_server-1.0.8.dist-info/licenses/LICENSE,sha256=XYY1C3d9XyjAGuntC4mVWGrEhWIm-044Ji9ouO7hfLA,1557
+nia_mcp_server-1.0.8.dist-info/RECORD,,

nia_mcp_server-1.0.8.dist-info/licenses/LICENSE

@@ -0,0 +1,31 @@
+GNU AFFERO GENERAL PUBLIC LICENSE
+Version 3, 19 November 2007
+
+Copyright (C) 2024 NIA (trynia.ai)
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+Additional permissions under GNU AGPL version 3 section 7:
+
+If you modify this Program, or any covered work, by linking or combining it 
+with other code, such other code is not for that reason alone subject to any 
+of the requirements of the GNU Affero General Public License.
+
+THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW.
+EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES
+PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED OR
+IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
+AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND
+PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE,
+YOU ASSUME THE COST OF ALL NECESSARY SERVICING, REPAIR OR CORRECTION.

nia_mcp_server-1.0.7.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-nia-mcp-server = nia_mcp_server.__main__:main

nia_mcp_server-1.0.7.dist-info/licenses/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2024 NIA (trynia.ai)
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.