nia-mcp-server 1.0.21__tar.gz → 1.0.22__tar.gz

{nia_mcp_server-1.0.21 → nia_mcp_server-1.0.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nia-mcp-server
-Version: 1.0.21
+Version: 1.0.22
 Summary: Nia Knowledge Agent
 Project-URL: Homepage, https://trynia.ai
 Project-URL: Documentation, https://docs.trynia.ai

{nia_mcp_server-1.0.21 → nia_mcp_server-1.0.22}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "nia-mcp-server"
-version = "1.0.21"
+version = "1.0.22"
 description = "Nia Knowledge Agent"
 readme = "README.md"
 requires-python = ">=3.8"

{nia_mcp_server-1.0.21 → nia_mcp_server-1.0.22}/src/nia_mcp_server/__init__.py

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

{nia_mcp_server-1.0.21 → nia_mcp_server-1.0.22}/src/nia_mcp_server/api_client.py

@@ -28,7 +28,7 @@ class NIAApiClient:
         self.client = httpx.AsyncClient(
             headers={
                 "Authorization": f"Bearer {api_key}",
-                "User-Agent": "nia-mcp-server/1.0.21",
+                "User-Agent": "nia-mcp-server/1.0.22",
                 "Content-Type": "application/json"
             },
             timeout=720.0  # 12 minute timeout for deep research operations
@@ -685,18 +685,44 @@ class NIAApiClient:
                 "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)}")
+
+    async def submit_bug_report(
+        self,
+        description: str,
+        bug_type: str = "bug",
+        additional_context: Optional[str] = None
+    ) -> Dict[str, Any]:
+        """Submit a bug report or feature request."""
+        try:
+            payload = {
+                "description": description,
+                "bug_type": bug_type,
+                "additional_context": additional_context
+            }
+
+            response = await self.client.post(
+                f"{self.base_url}/v2/bug-report",
+                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 submit bug report: {str(e)}")
     
     async def index_local_filesystem(
         self,

{nia_mcp_server-1.0.21 → nia_mcp_server-1.0.22}/src/nia_mcp_server/server.py

@@ -335,19 +335,18 @@ async def search_documentation(
 
     Args:
         query: Natural language search query. Don't just use keywords or unstrctured query, make a comprehensive question to get the best results possible.
-        sources: List of documentation identifiers to search. Preferred format is UUID, but also supports:
-            - Source UUIDs (e.g., "550e8400-e29b-41d4-a716-446655440000") - RECOMMENDED
+        sources: List of documentation identifiers to search. Can be:
             - Display names (e.g., "Vercel AI SDK - Core")
             - URLs (e.g., "https://sdk.vercel.ai/docs")
+            - Source IDs (UUID format for backwards compatibility)
         include_sources: Whether to include source references in results
 
     Returns:
         Search results with relevant documentation excerpts
 
     Important:
-        - UUIDs are the preferred identifier format for best performance
-        - Use `list_documentation` tool to see available sources and their UUIDs
-        - Display names and URLs are also supported for convenience
+        - You can now use friendly names instead of UUIDs! Try display names or URLs.
+        - If you don't know the identifiers, use `list_documentation` tool to see available options.
     """
     try:
         client = await ensure_api_client()
@@ -357,17 +356,20 @@ async def search_documentation(
             return [TextContent(
                 type="text",
                 text="📚 **Please specify which documentation sources to search:**\n\n"
-                     "1. Use `list_documentation` to see available sources and their UUIDs\n"
-                     "2. Then call `search_documentation(\"your query\", [\"uuid1\", \"uuid2\"])`\n\n"
-                     "**Supported identifier formats (UUIDs preferred):**\n"
-                     "- UUIDs: `\"550e8400-e29b-41d4-a716-446655440000\"` - RECOMMENDED\n"
+                     "1. Use `list_documentation` to see available sources\n"
+                     "2. Then call `search_documentation(\"your query\", [\"source1\", \"source2\"])`\n\n"
+                     "**You can use any of these identifier formats:**\n"
                      "- Display names: `\"Vercel AI SDK - Core\"`\n"
-                     "- URLs: `\"https://docs.trynia.ai/\"`\n\n"
-                     "**Example (preferred):**\n"
+                     "- URLs: `\"https://docs.trynia.ai/\"`\n"
+                     "- UUIDs: `\"550e8400-e29b-41d4-a716-446655440000\"`\n\n"
+                     "**Example:**\n"
                      "```\n"
-                     "search_documentation(\"API reference\", [\"550e8400-e29b-41d4-a716-446655440000\"])\n"
+                     "search_documentation(\"API reference\", [\"Vercel AI SDK - Core\"])\n"
                      "```\n\n"
-                     "**📌 Tip:** UUIDs provide best performance and reliability"
+                     "**📌 Tip:** Mix different identifier types in the same search:\n"
+                     "```\n"
+                     "search_documentation(\"query\", [\"Display Name\", \"https://docs.example.com/\"])\n"
+                     "```"
             )]
         
         # Build messages for the query
@@ -845,7 +847,7 @@ async def rename_resource(
         resource_type: Type of resource - "repository" or "documentation"
         identifier:
             - For repository: Repository in owner/repo format (e.g., "facebook/react")
-            - For documentation: UUID preferred, also supports display name or URL (e.g., "550e8400-e29b-41d4-a716-446655440000", "Vercel AI SDK - Core", or "https://docs.trynia.ai/")
+            - For documentation: Can be display name, URL, or UUID (e.g., "Vercel AI SDK - Core", "https://docs.trynia.ai/", or "doc-id-123")
         new_name: New display name for the resource (1-100 characters)
 
     Returns:
@@ -853,7 +855,7 @@ async def rename_resource(
 
     Examples:
         - rename_resource("repository", "facebook/react", "React Framework")
-        - rename_resource("documentation", "550e8400-e29b-41d4-a716-446655440000", "Python Official Docs")
+        - rename_resource("documentation", "Vercel AI SDK - Core", "Python Official Docs")
         - rename_resource("documentation", "https://docs.trynia.ai/", "NIA Documentation")
     """
     try:
@@ -916,14 +918,14 @@ async def delete_resource(
         resource_type: Type of resource - "repository" or "documentation"
         identifier:
             - For repository: Repository in owner/repo format (e.g., "facebook/react")
-            - For documentation: UUID preferred, also supports display name or URL (e.g., "550e8400-e29b-41d4-a716-446655440000", "Vercel AI SDK - Core", or "https://docs.trynia.ai/")
+            - For documentation: Can be display name, URL, or UUID (e.g., "Vercel AI SDK - Core", "https://docs.trynia.ai/", or "doc-id-123")
 
     Returns:
         Confirmation of deletion
 
     Examples:
         - delete_resource("repository", "facebook/react")
-        - delete_resource("documentation", "550e8400-e29b-41d4-a716-446655440000")
+        - delete_resource("documentation", "Vercel AI SDK - Core")
         - delete_resource("documentation", "https://docs.trynia.ai/")
     """
     try:
@@ -2979,6 +2981,112 @@ async def visualize_codebase(
             text=f"❌ Error opening visualization: {str(e)}"
         )]
 
+
+@mcp.tool()
+async def nia_bug_report(
+    description: str,
+    bug_type: str = "bug",
+    additional_context: Optional[str] = None
+) -> List[TextContent]:
+    """
+    Submit a bug report or feature request to the Nia development team.
+
+    This tool allows users to report bugs, request features, or provide feedback
+    directly to the development team. Reports are sent via email and Slack for
+    immediate attention.
+
+    Args:
+        description: Detailed description of the bug or feature request (10-5000 characters)
+        bug_type: Type of report - "bug", "feature-request", "improvement", or "other" (default: "bug")
+        additional_context: Optional additional context, steps to reproduce, or related information
+
+    Returns:
+        Confirmation of successful submission with reference ID
+
+    Examples:
+        - nia_bug_report("The search is not returning any results for my repository")
+        - nia_bug_report("Add support for searching within specific file types", "feature-request")
+        - nia_bug_report("Repository indexing fails with large repos", "bug", "Happens with repos over 1GB")
+    """
+    try:
+        client = await ensure_api_client()
+
+        # Validate input parameters
+        if not description or len(description.strip()) < 10:
+            return [
+                TextContent(
+                    type="text",
+                    text="❌ Error: Bug description must be at least 10 characters long."
+                )
+            ]
+
+        if len(description) > 5000:
+            return [
+                TextContent(
+                    type="text",
+                    text="❌ Error: Bug description must be 5000 characters or less."
+                )
+            ]
+
+        valid_types = ["bug", "feature-request", "improvement", "other"]
+        if bug_type not in valid_types:
+            return [
+                TextContent(
+                    type="text",
+                    text=f"❌ Error: bug_type must be one of: {', '.join(valid_types)}"
+                )
+            ]
+
+        if additional_context and len(additional_context) > 2000:
+            return [
+                TextContent(
+                    type="text",
+                    text="❌ Error: Additional context must be 2000 characters or less."
+                )
+            ]
+
+        logger.info(f"Submitting bug report: type={bug_type}, description_length={len(description)}")
+
+        # Submit bug report via API client
+        result = await client.submit_bug_report(
+            description=description.strip(),
+            bug_type=bug_type,
+            additional_context=additional_context.strip() if additional_context else None
+        )
+
+        if result.get("success"):
+            return [
+                TextContent(
+                    type="text",
+                    text=f"✅ Bug report submitted successfully!\n\n"
+                         f"Thank you for your feedback. Your report has been sent to the development team "
+                         f"and will be reviewed promptly.\n\n"
+                         f"Reference ID: {result.get('message', '').split(': ')[-1] if ': ' in result.get('message', '') else 'N/A'}\n"
+                         f"Type: {bug_type.title()}\n"
+                         f"Status: The team will be notified immediately via email and Slack.\n\n"
+                         f"You can also track issues and feature requests on our GitHub repository:\n"
+                         f"https://github.com/nozomio-labs/nia/issues"
+                )
+            ]
+        else:
+            return [
+                TextContent(
+                    type="text",
+                    text=f"❌ Failed to submit bug report: {result.get('message', 'Unknown error')}\n\n"
+                         f"Please try again or contact support directly at support@trynia.ai"
+                )
+            ]
+
+    except Exception as e:
+        logger.error(f"Error submitting bug report: {e}")
+        return [
+            TextContent(
+                type="text",
+                text=f"❌ Error submitting bug report: {str(e)}\n\n"
+                     f"Please try again or contact support directly at support@trynia.ai"
+            )
+        ]
+
 # Resources
 
 # Note: FastMCP doesn't have list_resources or read_resource decorators