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

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.21",
                 "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/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,44 @@ 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. Preferred format is UUID, but also supports:
+            - Source UUIDs (e.g., "550e8400-e29b-41d4-a716-446655440000") - RECOMMENDED
+            - Display names (e.g., "Vercel AI SDK - Core")
+            - URLs (e.g., "https://sdk.vercel.ai/docs")
         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.
+        - 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
     """
     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 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"
+                     "- Display names: `\"Vercel AI SDK - Core\"`\n"
+                     "- URLs: `\"https://docs.trynia.ai/\"`\n\n"
+                     "**Example (preferred):**\n"
+                     "```\n"
+                     "search_documentation(\"API reference\", [\"550e8400-e29b-41d4-a716-446655440000\"])\n"
+                     "```\n\n"
+                     "**📌 Tip:** UUIDs provide best performance and reliability"
+            )]
         
         # Build messages for the query
         messages = [
@@ -844,7 +845,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: UUID preferred, also supports display name or URL (e.g., "550e8400-e29b-41d4-a716-446655440000", "Vercel AI SDK - Core", or "https://docs.trynia.ai/")
         new_name: New display name for the resource (1-100 characters)
 
     Returns:
@@ -852,7 +853,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", "550e8400-e29b-41d4-a716-446655440000", "Python Official Docs")
+        - rename_resource("documentation", "https://docs.trynia.ai/", "NIA Documentation")
     """
     try:
         # Validate resource type
@@ -914,14 +916,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: UUID preferred, also supports display name or URL (e.g., "550e8400-e29b-41d4-a716-446655440000", "Vercel AI SDK - Core", or "https://docs.trynia.ai/")
 
     Returns:
         Confirmation of deletion
 
     Examples:
         - delete_resource("repository", "facebook/react")
-        - delete_resource("documentation", "doc-id-123")
+        - delete_resource("documentation", "550e8400-e29b-41d4-a716-446655440000")
+        - delete_resource("documentation", "https://docs.trynia.ai/")
     """
     try:
         # Validate resource type
@@ -976,14 +979,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

{nia_mcp_server-1.0.19.dist-info → nia_mcp_server-1.0.21.dist-info}/METADATA

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

{nia_mcp_server-1.0.19.dist-info → nia_mcp_server-1.0.21.dist-info}/RECORD

@@ -1,19 +1,19 @@
-nia_mcp_server/__init__.py,sha256=pSWbf7f_Szm7PGv59EVlQ2OhXRKG7yy39pD01-DZ9pQ,85
+nia_mcp_server/__init__.py,sha256=U4ztjfv89tU3YlGCCf3CKfTS2ZEeFmjpjjSMHRujBVM,85
 nia_mcp_server/__main__.py,sha256=YQSpFtDeKp18r8mKr084cHnRFV4416_EKCu9FTM8_ik,394
-nia_mcp_server/api_client.py,sha256=tXxRMs1tHC4YaOs63xhfobo58tWnMo9VULSojQWQLQs,34549
+nia_mcp_server/api_client.py,sha256=e-U_RUGCGrC4DfjokIygWiprNQUAHt-stPxpwdsLG4I,35034
 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=kPUxhLlG4TYMeYbcNO6WRRYDinXFdIKPnt_MIbUJTu4,129525
+nia_mcp_server/server.py,sha256=BWSeAft-9El_nsf-69rDnt1HpUuz5_vB-9MrPHgsbr8,130666
 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=l6sx000uqoczoHYqOPp4hnNgyfpnhvO9NyT0fVx5nU0,8059
 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.19.dist-info/METADATA,sha256=efXnlGvpmf8kAjf9qgJRlPIzn0QCTIKMvJUN_GAFh-g,1324
-nia_mcp_server-1.0.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nia_mcp_server-1.0.19.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
-nia_mcp_server-1.0.19.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
-nia_mcp_server-1.0.19.dist-info/RECORD,,
+nia_mcp_server-1.0.21.dist-info/METADATA,sha256=v0nxEs-TwVywd4bN4HbdCHBLI9AgZkq_kmt84-r8J2o,1324
+nia_mcp_server-1.0.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nia_mcp_server-1.0.21.dist-info/entry_points.txt,sha256=V74FQEp48pfWxPCl7B9mihtqvIJNVjCSbRfCz4ww77I,64
+nia_mcp_server-1.0.21.dist-info/licenses/LICENSE,sha256=IrdVKi3bsiB2MTLM26MltBRpwyNi-8P6Cy0EnmAN76A,1557
+nia_mcp_server-1.0.21.dist-info/RECORD,,