refactor: rename and update multi-module bug localization tool and request model

main.py

@@ -6,7 +6,7 @@ from loguru import logger
 from dataclasses import dataclass
 
 from utils.config import API_CONFIG
-from utils.model import BugLocalizationRequest
+from utils.model import MultiModuleRequest, SingleModuleRequest, SearchSpaceRoutingRequest
 
 @dataclass
 class DSContext:
@@ -27,36 +27,30 @@ async def ds_lifespan(server: FastMCP) -> AsyncIterator[DSContext]:
 mcp = FastMCP(name="defect-solver-server", lifespan=ds_lifespan)
 
 
+
 @mcp.tool(
-    name="full_bug_localization",
+    name="multi_module_bug_localization",
     description=(
         """
-        Given a BugLocalizationRequest, this tool analyzes the entire microservice architecture of the project, identifies the most likely microservices and code files that may contain the bug, and returns a list of candidate microservices and files for further investigation.
-        
+        Performs bug localization across the entire Java project by first identifying likely microservices 
+        (search spaces) and then locating the bug-related files within them.
+
+        Use this tool when you do not know which microservice is responsible for the bug.
+
         Args:
-            request (BugLocalizationRequest): The bug localization request data.
-            ctx (Context): The MCP context containing lifespan context.
-        
+            request (MultiModuleRequest): Issue key and descriptive details from the bug report.
+            ctx (Context): MCP context object.
+
         Returns:
-            dict: Predicted source code locations and the prompt used in the process, or error details.
+            dict: Includes selected microservices and localized file paths.
         """
     ),
 )
-async def full_bug_localization(request: BugLocalizationRequest, ctx: Context) -> dict:
-    """
-    Given a BugLocalizationRequest, this tool analyzes the entire microservice architecture of the project, identifies the most likely microservices and code files that may contain the bug, and returns a list of candidate microservices and files for further investigation.
-
-    Args:
-        request (BugLocalizationRequest): The bug localization request data.
-        ctx (Context): The MCP context containing lifespan context.
-
-    Returns:
-        dict: Predicted source code locations or error details.
-    """
+async def multi_module_bug_localization(request: MultiModuleRequest, ctx: Context) -> dict:
     ds_ctx: DSContext = ctx.request_context.lifespan_context
     client = ds_ctx.client
 
-    api_url = API_CONFIG["api_base_url"] + API_CONFIG["api_endpoint"]
+    api_url = API_CONFIG["api_base_url"] + API_CONFIG["api_multimodule_endpoint"]
     hf_token = API_CONFIG.get("hf_access_token", None)
 
     headers = {
@@ -65,7 +59,7 @@ async def full_bug_localization(request: BugLocalizationRequest, ctx: Context) -
     }
 
     try:
-        logger.info(f"Sending bug localization request to {api_url} with data: {request.model_dump()}")
+        logger.info(f"Sending multi-module bug localization request to {api_url} with data: {request.model_dump()}")
         response = await client.post(
             api_url,
             json={
@@ -81,13 +75,13 @@ async def full_bug_localization(request: BugLocalizationRequest, ctx: Context) -
         logger.info(f"Received response: {response.text}")
         return response.json()
     except httpx.HTTPStatusError as e:
-        logger.error(f"HTTP error {e.response.status_code} from bug localization endpoint: {e.response.text}")
+        logger.error(f"HTTP error {e.response.status_code} from multimodule bug localization endpoint: {e.response.text}")
         return {"error": f"HTTP error: {e.response.status_code}", "details": e.response.text}
     except httpx.RequestError as e:
-        logger.error(f"Request error while calling bug localization endpoint: {e}")
+        logger.error(f"Request error while calling multimodule bug localization endpoint: {e}")
         return {"error": "Request error", "details": str(e)}
     except Exception as e:
-        logger.exception("Unexpected error during bug localization request")
+        logger.exception("Unexpected error during multimodule bug localization request")
         return {"error": "Unexpected error", "details": str(e)}