Odos MCP Server

""" Get Odos router contract address for specified blockchain network. """ import json from typing import Union from mcp import types from pydantic import BaseModel from ...helpers import make_odos_request, resolve_chain_id from ...mcp import mcp # pylint: disable=duplicate-code class GetRouterAddressArgs(BaseModel): """Arguments for getting router address""" version: str # v3 or v2 chain_id: Union[str, int] @mcp.tool( name="get_router_address", description="Get address of Odos router contract", ) async def get_router_address(args: GetRouterAddressArgs) -> list[types.TextContent]: """ Gets the Odos router contract address for a specific chain and version. """ try: if args.version not in ["v2", "v3"]: raise ValueError("Version must be 'v2' or 'v3'") chain_id = resolve_chain_id(args.chain_id) # Build URL with path parameters url = f"/info/router/{args.version}/{chain_id}" response_data = await make_odos_request(url, method="GET") output = { "message": f"Fetched router address for {args.version} on chain {chain_id}.", "version": args.version, "chainId": chain_id, "address": response_data.get("address"), } return [types.TextContent(type="text", text=json.dumps(output, indent=2))] except ConnectionError as e: return [ types.TextContent( type="text", text=f"❌ API Error in get_router_address: {str(e)}" ) ] except (ValueError, KeyError, TypeError) as e: return [ types.TextContent( type="text", text=f"❌ Unexpected Error in get_router_address: {str(e)}" ) ]