diff --git a/mcp_server.py b/mcp_server.py index 650bdf3..3b7a3d9 100644 --- a/mcp_server.py +++ b/mcp_server.py @@ -1,6 +1,7 @@ import os from typing import Any from dateutil.parser import parse as parse_date +from urllib.parse import urljoin import httpx from mcp.server.fastmcp import FastMCP @@ -153,5 +154,17 @@ async def get_build_logs(job_name: str, build_id: str): } +@mcp.tool() +async def search_jobs(job: str): + # Construct the URL with parameters + endpoint = "/search" + url = urljoin(API_BASE_URL, endpoint) + + # Add parameters if provided + params = {} + if job: + params['job'] = job + return await make_request(url, data=params) + if __name__ == "__main__": mcp.run(transport=os.environ.get("MCP_TRANSPORT", "stdio"))