import requests
import os

MCP_URL = os.getenv("MCP_URL")


class MCPClient:

    def list_tools(self):
        r = requests.get(f"{MCP_URL}/tools")
        return r.json()

    def call_tool(self, tool, args):
        r = requests.post(
            f"{MCP_URL}/call",
            json={
                "tool": tool,
                "arguments": args
            }
        )
        return r.json()