04. MCP 프로토콜 심화

class MockMCPServer:
    """최소한의 MCP 서버 구현"""
 
    def __init__(self, name: str):
        self.name = name
        self.tools = {}
 
    def register_tool(self, name: str, description: str, handler: Callable):
        self.tools[name] = {
            "description": description,
            "handler": handler
        }
 
    def list_tools(self) -> List[Dict]:
        """Discovery 엔드포인트 - 이 서버가 무엇을 할 수 있나요?"""
        return [
            {"name": name, "description": tool["description"]}
            for name, tool in self.tools.items()
        ]
 
    def call_tool(self, name: str, arguments: Dict) -> Any:
        """이름으로 도구 실행"""
        if name not in self.tools:
            raise ValueError(f"Unknown tool: {name}")
        return self.tools[name]["handler"](**arguments)

class MCPClient:
    """MCP 서버에 연결하는 클라이언트"""
 
    def __init__(self, servers: List[MockMCPServer]):
        self.servers = servers
 
    def discover_all_tools(self) -> List[Dict]:
        """연결된 모든 서버에서 도구 탐색"""
        all_tools = []
        for server in self.servers:
            tools = server.list_tools()
            for tool in tools:
                tool["server"] = server.name
            all_tools.extend(tools)
        return all_tools
 
    def execute(self, server_name: str, tool_name: str, args: Dict):
        """특정 서버에서 도구 실행"""
        server = next(s for s in self.servers if s.name == server_name)
        return server.call_tool(tool_name, args)