feat: add search space routing tool for bug localization

main.py

@@ -143,6 +143,62 @@ async def single_module_bug_localization(request: SingleModuleRequest, ctx: Cont
         return {"error": "Unexpected error", "details": str(e)}
 
 
+@mcp.tool(
+    name="search_space_routing",
+    description=(
+        """
+        Identifies and returns the most likely microservices (search spaces) that could be the source of a reported bug.
+
+        Use this tool when the source of the bug is unknown and you want to narrow down the investigation to top candidate microservices.
+
+        Args:
+            request (SearchSpaceRoutingRequest): Bug description and optional issue key/summary.
+            ctx (Context): The MCP context with lifespan context.
+
+        Returns:
+            dict: A message and the list of selected search spaces (microservices).
+        """
+    ),
+)
+async def search_space_routing(request: SearchSpaceRoutingRequest, 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_searchspace_endpoint"]
+    hf_token = API_CONFIG.get("hf_access_token", None)
+
+    headers = {
+        "Authorization": f"Bearer {hf_token}",
+        "Content-Type": "application/json"
+    }
+
+    try:
+        logger.info(f"Sending search space routing request to {api_url} with data: {request.model_dump()}")
+        response = await client.post(
+            api_url,
+            json={
+                "key": request.issue_key,
+                "fields": {
+                    "summary": request.summary,
+                    "description": request.description,
+                }
+            },
+            headers=headers,
+        )
+        response.raise_for_status()
+        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 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 endpoint: {e}")
+        return {"error": "Request error", "details": str(e)}
+    except Exception as e:
+        logger.exception("Unexpected error during search space routing")
+        return {"error": "Unexpected error", "details": str(e)}
+
 
 
 if __name__ == "__main__":