source-context-mcp 0.1.dev1__tar.gz → 0.1.dev2__tar.gz

{source_context_mcp-0.1.dev1 → source_context_mcp-0.1.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: source-context-mcp
-Version: 0.1.dev1
+Version: 0.1.dev2
 Summary: Local Model Context Protocol (MCP) server for Source Context code intelligence.
 Author: Le Ngoc Tu, Dang Huu Hieu, Huynh Phat Tai
 Author-email: Le Ngoc Tu <tuleaibk@gmail.com>, Dang Huu Hieu <danghuuhieu038@gmail.com>, Huynh Phat Tai <huynhphtai@gmail.com>

{source_context_mcp-0.1.dev1 → source_context_mcp-0.1.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "source-context-mcp"
-version = "0.1.dev1"
+version = "0.1.dev2"
 description = "Local Model Context Protocol (MCP) server for Source Context code intelligence."
 readme = "README.md"
 authors = [

{source_context_mcp-0.1.dev1 → source_context_mcp-0.1.dev2}/src/source_context_mcp/core/api_client.py

@@ -4,6 +4,7 @@ from dataclasses import dataclass
 from typing import Any, Literal, overload
 
 import httpx
+from mcp.server.mcpserver.exceptions import ToolError
 from pydantic import TypeAdapter, ValidationError
 
 
@@ -15,6 +16,27 @@ class ApiResponse[T]:
     error_code: str | None
     message: str | None
 
+    def result(self) -> T:
+        """
+        Return the response data if the API request succeeded.
+
+        Raises:
+            ToolError:
+                If the API request failed. This exception is intended to propagate to the MCP framework, which
+                automatically converts it into an MCP tool error response (`isError=True`). It does not need to be
+                caught by tool implementations.
+        """
+        if self.success:
+            return self.data
+        if self.error_code is None and self.message is None:
+            msg = "UNKNOWN_ERROR"
+        else:
+            msg = (
+                f"{self.error_code if self.error_code is not None else ''} "
+                f"{self.message if self.message is not None else ''}"
+            )
+        raise ToolError(msg)
+
     @staticmethod
     def ok(data: T | None, message: str | None = None, status_code: int = 200) -> ApiResponse[T]:
         return ApiResponse(
@@ -112,6 +134,14 @@ class ApiClient:
                     error_code="CONNECTION_ERROR",
                 )
 
+            except httpx.HTTPStatusError as exc:
+                res = exc.response
+                try:
+                    res_json = res.json()
+                    return ApiResponse.error(exc.response.status_code, res_json.get("error_code", "UNKNOW_ERROR"))
+                except ValueError:
+                    return ApiResponse.error(0, str(exc))
+
             except httpx.HTTPError as exc:
                 return ApiResponse.error(0, str(exc))
 

{source_context_mcp-0.1.dev1 → source_context_mcp-0.1.dev2}/src/source_context_mcp/core/settings.py

@@ -23,6 +23,9 @@ class Settings(BaseSettings):
     DEFAULT_WORKSPACE_ID: int = 1
     """Default Workspace ID"""
 
+    PATH_WORKSPACE: dict[str, int] = {}
+    PATH_REPO: dict[str, int] = {}
+
     @classmethod
     def settings_customise_sources(
         cls,

{source_context_mcp-0.1.dev1 → source_context_mcp-0.1.dev2}/src/source_context_mcp/server.py

@@ -37,7 +37,9 @@ def create_server(
     life_span = make_life_span(
         base_url_override=base_url_override, pat_override=pat_override, workspace_id_override=workspace_id_override
     )
+    import sys
 
+    print("hello", file=sys.stderr)
     server = MCPServer(
         name="SourceContextMCP",
         lifespan=life_span,

source_context_mcp-0.1.dev2/src/source_context_mcp/tools/base.py

@@ -0,0 +1,75 @@
+from typing import Any
+
+from mcp.server import MCPServer
+from mcp.server.mcpserver import Context
+
+from source_context_mcp.core import ApiClientDep, AppContext
+
+
+def register_tools(mcp: MCPServer) -> None:
+    """
+    Register bases tools:
+        - list user's available workspaces
+        - list repos in workspace
+        - list branches in repo
+        - list projects in branch
+
+    Args:
+        mcp: MCPServer instance
+    """
+
+    @mcp.tool(description="List all available workspaces that current user can access")
+    async def list_workspaces(client: ApiClientDep) -> Any:
+        res = await client.get("/general/workspaces", response_model=list)
+        return res.result()
+
+    @mcp.tool(
+        description=(
+            "Get the default settings for specified directory.Include default workspace id adn default repository id."
+        )
+    )
+    async def get_path_settings(ctx: Context[AppContext], path: str) -> dict[str, Any]:
+        app_context: AppContext = ctx.request_context.lifespan_context
+        settings = app_context.settings
+        return {
+            "path": path,
+            "default_workspace_id": settings.PATH_WORKSPACE.get(path, None),
+            "default_repo_id": settings.PATH_REPO.get(path, None),
+        }
+
+    @mcp.tool(
+        description=(
+            "Get default workspace. "
+            "This is the default workspace use settings in case current path has no default config"
+        )
+    )
+    async def default_workspace(ctx: Context[AppContext]) -> Any:
+        app_context: AppContext = ctx.request_context.lifespan_context
+        return app_context.settings.DEFAULT_WORKSPACE_ID
+
+    @mcp.tool(description="List all available repositories in workspace")
+    async def list_repositories(
+        client: ApiClientDep,
+        workspace_id: int,
+    ) -> Any:
+        res = await client.post("/general/repositories", {"workspace_id": workspace_id})
+        return res.result()
+
+    @mcp.tool(description="List all available branches in specified repository.")
+    async def list_branches(client: ApiClientDep, workspace_id: int, repository_id: int) -> Any:
+        req_body = {
+            "workspace_id": workspace_id,
+            "repository_id": repository_id,
+        }
+        res = await client.post("/general/branches", req_body)
+        return res.result()
+
+    @mcp.tool(description="List all available projects in specific branch.")
+    async def list_projects(client: ApiClientDep, workspace_id: int, repository_id: int, branch_name: str) -> Any:
+        req_body = {
+            "workspace_id": workspace_id,
+            "repo_id": repository_id,
+            "branch_name": branch_name,
+        }
+        res = await client.post("/general/projects", req_body)
+        return res.result()

{source_context_mcp-0.1.dev1 → source_context_mcp-0.1.dev2}/src/source_context_mcp/tools/graph.py

@@ -15,24 +15,34 @@ def register_tools(mcp: MCPServer) -> None:
     @mcp.tool(description="List all files in specified project.")
     async def list_files_in_project(client: ApiClientDep, project_id: str) -> Any:
         res = await client.get(f"/graph/projects/{project_id}/files")
-        return res.data
+        return res.result()
 
     @mcp.tool(description="Inspect file structure(classes and methods")
     async def get_file_structure(client: ApiClientDep, file_id: str) -> Any:
-        res = await client.get(f"/graph/files/{file_id}/structures")
-        return res.data
+        res = await client.get(f"/graph/files/{file_id}/structure")
+        return res.result()
 
     @mcp.tool(description="Read file content.")
     async def get_file_content(client: ApiClientDep, file_id: str) -> Any:
         res = await client.get(f"/graph/files/{file_id}/content")
-        return res.data
+        return res.result()
+
+    @mcp.tool(description="Get node information: id, name, file_id, position in file")
+    async def get_node_info(client: ApiClientDep, node_id: str) -> Any:
+        res = await client.get(f"/graph/nodes/{node_id}")
+        return res.result()
+
+    @mcp.tool(description="Get node information.")
+    async def get_node_content(client: ApiClientDep, node_id: str) -> Any:
+        res = await client.get(f"/graph/nodes/{node_id}/content")
+        return res.result()
 
     @mcp.tool(description="Find all nodes that call/reference to the specified node.")
     async def find_node_usages(client: ApiClientDep, node_id: str) -> Any:
         res = await client.get(f"/graph/nodes/{node_id}/usages")
-        return res.data
+        return res.result()
 
     @mcp.tool(description="Find all nodes(or its children) that the specified node is calling or reference to.")
     async def find_node_callees(client: ApiClientDep, node_id: str) -> Any:
         res = await client.get(f"/graph/nodes/{node_id}/callees")
-        return res.data
+        return res.result()

{source_context_mcp-0.1.dev1 → source_context_mcp-0.1.dev2}/src/source_context_mcp/tools/vector.py

@@ -12,7 +12,7 @@ def register_tools(mcp: MCPServer) -> None:
         mcp: MCP server instance
     """
 
-    @mcp.tool()
+    @mcp.tool(description="Sematic search.")
     async def search_in_workspace(client: ApiClientDep, query: str, workspace_id: int, top_k: int = 5) -> Any:
         req_body = {
             "query": query,
@@ -21,9 +21,9 @@ def register_tools(mcp: MCPServer) -> None:
 
         res = await client.post(f"/vector/search/{workspace_id}", req_body)
 
-        return res.data
+        return res.result()
 
-    @mcp.tool()
+    @mcp.tool(description="Sematic search.")
     async def search_in_repo_and_branch(
         client: ApiClientDep, repository_id: int, branch_name: str, query: str, top_k: int = 5
     ) -> Any:
@@ -34,4 +34,4 @@ def register_tools(mcp: MCPServer) -> None:
 
         res = await client.post(f"/vector/search/{repository_id}/{branch_name}", req_body)
 
-        return res.data
+        return res.result()

source_context_mcp-0.1.dev1/src/source_context_mcp/tools/base.py

@@ -1,41 +0,0 @@
-from typing import Any
-
-from mcp.server import MCPServer
-
-from source_context_mcp.core import ApiClientDep
-
-
-def register_tools(mcp: MCPServer) -> None:
-    """
-    Register bases tools:
-        - list user's available workspaces
-        - list repos in workspace
-        - list branches in repo
-        - list projects in branch
-
-    Args:
-        mcp: MCPServer instance
-    """
-
-    @mcp.tool(description="List all available workspaces that current user can access")
-    async def list_workspaces(client: ApiClientDep) -> Any:
-        res = await client.get("/general/workspaces", response_model=list)
-        return res.data
-
-    @mcp.tool(description="List all available repositories in workspace")
-    async def list_repositories(
-        client: ApiClientDep,
-        workspace_id: int,
-    ) -> Any:
-        res = await client.post("/general/repositories", {"workspace_id": workspace_id})
-        return res.data
-
-    @mcp.tool(description="List all available branches in specified repository.")
-    async def list_branches(client: ApiClientDep, project_id: int) -> Any:
-        """This tool is not implemented yet"""
-        return ["This tool is not implemented yet"]
-
-    @mcp.tool(description="List all available projects in specific branch.")
-    def list_projects(client: ApiClientDep, branch_id: int) -> Any:
-        """This tool is not implemented yet"""
-        return ["This tool is not implemented yet"]