nia-mcp-server 1.0.25__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.25"
+__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.25",
+                "User-Agent": "nia-mcp-server/1.0.27",
                 "Content-Type": "application/json"
             },
             timeout=720.0  # 12 minute timeout for deep research operations
@@ -432,7 +432,104 @@ 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_github_tree(
+        self,
+        owner_repo: str,
+        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 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:
+                # First, get the repository ID
+                status = await self.get_repository_status(owner_repo)
+                if not status:
+                    raise APIError(f"Repository {owner_repo} not found", 404)
+
+                # Extract the repository ID from status
+                repo_id = status.get("repository_id") or status.get("id")
+                if not repo_id:
+                    # Try to get it from list as fallback
+                    repos = await self.list_repositories()
+                    for repo in repos:
+                        if repo.get("repository") == owner_repo:
+                            repo_id = repo.get("repository_id") or repo.get("id")
+                            break
+
+                if not repo_id:
+                    raise APIError(f"No repository ID found for {owner_repo}", 404)
+
+                # 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}/github-tree",
+                    params=params
+                )
+                response.raise_for_status()
+                return response.json()
+            else:
+                # Assume it's already a repository 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/{owner_repo}/github-tree",
+                    params=params
+                )
+                response.raise_for_status()
+                return response.json()
+
+        except httpx.HTTPStatusError as e:
+            raise self._handle_api_error(e)
+        except APIError:
+            raise
+        except Exception as e:
+            logger.error(f"Failed to get GitHub tree: {e}")
+            raise APIError(f"Failed to get GitHub tree: {str(e)}")
+
     # Data Source methods
     
     async def create_data_source(
@@ -656,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,