nia-mcp-server 1.0.26__py3-none-any.whl → 1.0.27__py3-none-any.whl

nia_mcp_server/__init__.py

@@ -3,4 +3,4 @@ NIA MCP Server - Proxy server for NIA Knowledge Agent
 """
 
 
-__version__ = "1.0.26"
+__version__ = "1.0.27"

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.26",
+                "User-Agent": "nia-mcp-server/1.0.27",
                 "Content-Type": "application/json"
             },
             timeout=720.0  # 12 minute timeout for deep research operations
@@ -432,15 +432,31 @@ class NIAApiClient:
         except Exception as e:
             logger.error(f"Failed to rename repository: {e}")
             raise APIError(f"Failed to rename repository: {str(e)}")
-
-    async def get_repository_hierarchy(
+            
+    async def get_github_tree(
         self,
         owner_repo: str,
-        include_classes: bool = True,
-        include_methods: bool = False,
-        include_tree: bool = False
+        branch: Optional[str] = None,
+        include_paths: Optional[List[str]] = None,
+        exclude_paths: Optional[List[str]] = None,
+        file_extensions: Optional[List[str]] = None,
+        exclude_extensions: Optional[List[str]] = None,
+        show_full_paths: bool = False
     ) -> Dict[str, Any]:
-        """Get the file hierarchy for a repository."""
+        """Get file tree directly from GitHub API (no FalkorDB dependency).
+
+        Args:
+            owner_repo: Repository in owner/repo format or repository ID
+            branch: Optional branch name (defaults to repository's default branch)
+            include_paths: Only include files in these paths (e.g., ["src/", "lib/"])
+            exclude_paths: Exclude files in these paths (e.g., ["node_modules/", "dist/"])
+            file_extensions: Only include these file extensions (e.g., [".py", ".js"])
+            exclude_extensions: Exclude these file extensions (e.g., [".md", ".lock"])
+            show_full_paths: Show full file paths instead of hierarchical tree
+
+        Returns:
+            GitHub tree structure with files, directories, and stats
+        """
         try:
             # Check if this looks like owner/repo format (contains /)
             if '/' in owner_repo:
@@ -462,38 +478,46 @@ class NIAApiClient:
                 if not repo_id:
                     raise APIError(f"No repository ID found for {owner_repo}", 404)
 
-                # Get hierarchy using the ID
-                params = {
-                    "include_classes": include_classes,
-                    "include_methods": include_methods
-                }
-
-                headers = {}
-                if include_tree:
-                    headers["X-Include-Tree"] = "true"
+                # Get tree using the ID
+                params = {}
+                if branch:
+                    params["branch"] = branch
+                if include_paths:
+                    params["include_paths"] = ",".join(include_paths)
+                if exclude_paths:
+                    params["exclude_paths"] = ",".join(exclude_paths)
+                if file_extensions:
+                    params["file_extensions"] = ",".join(file_extensions)
+                if exclude_extensions:
+                    params["exclude_extensions"] = ",".join(exclude_extensions)
+                if show_full_paths:
+                    params["show_full_paths"] = "true"
 
                 response = await self.client.get(
-                    f"{self.base_url}/v2/repositories/{repo_id}/hierarchy",
-                    params=params,
-                    headers=headers
+                    f"{self.base_url}/v2/repositories/{repo_id}/github-tree",
+                    params=params
                 )
                 response.raise_for_status()
                 return response.json()
             else:
                 # Assume it's already a repository ID
-                params = {
-                    "include_classes": include_classes,
-                    "include_methods": include_methods
-                }
-
-                headers = {}
-                if include_tree:
-                    headers["X-Include-Tree"] = "true"
+                params = {}
+                if branch:
+                    params["branch"] = branch
+                if include_paths:
+                    params["include_paths"] = ",".join(include_paths)
+                if exclude_paths:
+                    params["exclude_paths"] = ",".join(exclude_paths)
+                if file_extensions:
+                    params["file_extensions"] = ",".join(file_extensions)
+                if exclude_extensions:
+                    params["exclude_extensions"] = ",".join(exclude_extensions)
+                if show_full_paths:
+                    params["show_full_paths"] = "true"
 
                 response = await self.client.get(
-                    f"{self.base_url}/v2/repositories/{owner_repo}/hierarchy",
-                    params=params,
-                    headers=headers
+                    f"{self.base_url}/v2/repositories/{owner_repo}/github-tree",
+                    params=params
                 )
                 response.raise_for_status()
                 return response.json()
@@ -503,8 +527,8 @@ class NIAApiClient:
         except APIError:
             raise
         except Exception as e:
-            logger.error(f"Failed to get repository hierarchy: {e}")
-            raise APIError(f"Failed to get repository hierarchy: {str(e)}")
+            logger.error(f"Failed to get GitHub tree: {e}")
+            raise APIError(f"Failed to get GitHub tree: {str(e)}")
 
     # Data Source methods
     
@@ -729,22 +753,86 @@ class NIAApiClient:
             payload = {
                 "query": query,
             }
-            
+
             if output_format:
                 payload["output_format"] = output_format
-            
+
             response = await self.client.post(
                 f"{self.base_url}/v2/deep-research",
                 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"Deep research failed: {str(e)}")
-    
+
+    async def regex_search(
+        self,
+        repositories: List[str],
+        query: str,
+        pattern: Optional[str] = None,
+        file_extensions: Optional[List[str]] = None,
+        languages: Optional[List[str]] = None,
+        max_results: int = 50,
+        include_context: bool = True,
+        context_lines: int = 3
+    ) -> Dict[str, Any]:
+        """
+        Perform regex pattern search over indexed repository source code.
+
+        Args:
+            repositories: List of repositories to search (owner/repo format)
+            query: Natural language query or regex pattern
+            pattern: Optional explicit regex pattern (overrides query extraction)
+            file_extensions: File extensions to filter (e.g., [".js", ".tsx"])
+            languages: Programming languages to filter
+            max_results: Maximum number of results to return
+            include_context: Include surrounding context lines
+            context_lines: Number of context lines before/after match
+
+        Returns:
+            Search results with matched patterns and locations
+        """
+        try:
+            # Build repository list
+            repo_list = []
+            for repo in repositories:
+                if isinstance(repo, dict):
+                    repo_list.append(repo)
+                else:
+                    repo_list.append({"repository": repo})
+
+            payload = {
+                "repositories": repo_list,
+                "query": query,
+                "max_results": max_results,
+                "include_context": include_context,
+                "context_lines": context_lines
+            }
+
+            # Add optional parameters
+            if pattern:
+                payload["pattern"] = pattern
+            if file_extensions:
+                payload["file_extensions"] = file_extensions
+            if languages:
+                payload["languages"] = languages
+
+            response = await self.client.post(
+                f"{self.base_url}/v2/regex-search",
+                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"Regex search failed: {str(e)}")
+
     async def get_source_content(
         self,
         source_type: str,