nia-mcp-server 1.0.19__tar.gz → 1.0.20__tar.gz

{nia_mcp_server-1.0.19 → nia_mcp_server-1.0.20}/PKG-INFO

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

{nia_mcp_server-1.0.19 → nia_mcp_server-1.0.20}/pyproject.toml

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

{nia_mcp_server-1.0.19 → nia_mcp_server-1.0.20}/src/nia_mcp_server/__init__.py

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

{nia_mcp_server-1.0.19 → nia_mcp_server-1.0.20}/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.0",
+                "User-Agent": "nia-mcp-server/1.0.20",
                 "Content-Type": "application/json"
             },
             timeout=720.0  # 12 minute timeout for deep research operations
@@ -552,15 +552,23 @@ class NIAApiClient:
                 for repo in repositories:
                     repo_list.append({"repository": repo})
             
-            # Build data source list  
+            # Build data source list
             source_list = []
             if data_sources:
-                for source_id in data_sources:
-                    # Handle both list of IDs and list of dicts
-                    if isinstance(source_id, dict):
-                        source_list.append(source_id)
+                for source in data_sources:
+                    # Handle flexible identifier formats:
+                    # 1. String directly (display_name, URL, or source_id) - NEW
+                    # 2. Dict with "source_id" (backwards compatible)
+                    # 3. Dict with "identifier" (new format)
+                    if isinstance(source, str):
+                        # Pass string directly - backend will resolve it
+                        source_list.append(source)
+                    elif isinstance(source, dict):
+                        # Keep dict format as-is (backwards compatible)
+                        source_list.append(source)
                     else:
-                        source_list.append({"source_id": source_id})
+                        # Convert other types to string
+                        source_list.append(str(source))
             
             # Validate at least one source
             if not repo_list and not source_list:

{nia_mcp_server-1.0.19 → nia_mcp_server-1.0.20}/src/nia_mcp_server/server.py

@@ -186,34 +186,22 @@ async def search_codebase(
     try:
         client = await ensure_api_client()
         
-        # Get all indexed repositories if not specified
+        # Require explicit repository selection
         if not repositories:
-            all_repos = await client.list_repositories()
-            
-            # 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",
-                    text="❌ No indexed repositories found. Use `index_repository` to index a codebase first."
-                )]
+            return [TextContent(
+                type="text",
+                text="🔍 **Please specify which repositories to search:**\n\n"
+                     "1. Use `list_repositories` to see available repositories\n"
+                     "2. Then call `search_codebase(\"your query\", [\"owner/repo1\", \"owner/repo2\"])`\n\n"
+                     "**Example:**\n"
+                     "```\n"
+                     "search_codebase(\"How does auth work?\", [\"facebook/react\"])\n"
+                     "```\n\n"
+                     "**📌 Tip:** You can search specific folders using the exact format from `list_repositories`:\n"
+                     "```\n"
+                     "search_codebase(\"query\", [\"owner/repo/tree/branch/folder\"])\n"
+                     "```"
+            )]
         
         # Build messages for the query
         messages = [
@@ -343,31 +331,46 @@ async def search_documentation(
     include_sources: bool = True
 ) -> List[TextContent]:
     """
-    Search indexed documentation using natural language. 
-    
+    Search indexed documentation using natural language.
+
     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 source IDs to search. Use it based on user's query.
+        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:
-        - Always use Source ID. If you don't have it, use `list_documentation` tool to get it.
+        - 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()
         
-        # Get all indexed documentation sources if not specified
+        # Require explicit source selection
         if not sources:
-            all_sources = await client.list_data_sources()
-            sources = [source["id"] for source in all_sources if source.get("status") == "completed"]
-            if not sources:
-                return [TextContent(
-                    type="text",
-                    text="❌ No indexed documentation found. Use `index_documentation` to index documentation first."
-                )]
+            return [TextContent(
+                type="text",
+                text="📚 **Please specify which documentation sources to search:**\n\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"
+                     "- UUIDs: `\"550e8400-e29b-41d4-a716-446655440000\"`\n\n"
+                     "**Example:**\n"
+                     "```\n"
+                     "search_documentation(\"API reference\", [\"Vercel AI SDK - Core\"])\n"
+                     "```\n\n"
+                     "**📌 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
         messages = [
@@ -844,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: Documentation source ID
+            - 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:
@@ -852,7 +855,8 @@ async def rename_resource(
 
     Examples:
         - rename_resource("repository", "facebook/react", "React Framework")
-        - rename_resource("documentation", "doc-id-123", "Python Official Docs")
+        - rename_resource("documentation", "Vercel AI SDK - Core", "Python Official Docs")
+        - rename_resource("documentation", "https://docs.trynia.ai/", "NIA Documentation")
     """
     try:
         # Validate resource type
@@ -914,14 +918,15 @@ 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: Documentation source ID
+            - 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", "doc-id-123")
+        - delete_resource("documentation", "Vercel AI SDK - Core")
+        - delete_resource("documentation", "https://docs.trynia.ai/")
     """
     try:
         # Validate resource type
@@ -976,14 +981,18 @@ async def check_resource_status(
         resource_type: Type of resource - "repository" or "documentation"
         identifier:
             - For repository: Repository in owner/repo format (e.g., "facebook/react")
-            - For documentation: Documentation source ID
+            - For documentation: Source ID (UUID format only) - use list_resources to get the UUID
 
     Returns:
         Current status of the resource
 
     Examples:
         - check_resource_status("repository", "facebook/react")
-        - check_resource_status("documentation", "doc-id-123")
+        - check_resource_status("documentation", "550e8400-e29b-41d4-a716-446655440000")
+
+    Note:
+        - Documentation status checking requires UUID identifiers only
+        - Use list_resources("documentation") to find the UUID for a documentation source
     """
     try:
         # Validate resource type