17. 클라이언트가 서버를 발견·연결·사용하는 법

# client.py  —  공식 mcp SDK
import asyncio
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client

# 서버를 어떻게 띄울지 (07장의 "실행 명령 등록"과 같은 정보)
server_params = StdioServerParameters(
    command="python",          # 실행 파일
    args=["server.py"],        # 인자
    env=None,                  # 환경 변수 (비밀값 주입 지점)
)

async def main():
    async with stdio_client(server_params) as (read, write):
        async with ClientSession(read, write) as session:
            await session.initialize()                 # 핸드셰이크 (05장)

            # 발견
            tools = (await session.list_tools()).tools
            print("도구:", [t.name for t in tools])

            # 호출
            result = await session.call_tool("add", {"a": 3, "b": 4})
            print("결과:", result.content[0].text)      # "7"

            # 리소스 읽기 (11장)
            greeting = await session.read_resource("greeting://Alice")
            # 프롬프트 가져오기 (12장)
            prompt = await session.get_prompt("greet_user", {"name": "Alice"})

asyncio.run(main())

# http_client.py  —  공식 mcp SDK
import asyncio
from mcp import ClientSession
from mcp.client.streamable_http import streamablehttp_client

URL = "http://127.0.0.1:8000/mcp"

async def main():
    async with streamablehttp_client(URL) as (read, write, get_session_id):
        async with ClientSession(read, write) as session:
            await session.initialize()
            if (sid := get_session_id()) is not None:
                print("세션 ID:", sid)          # Mcp-Session-Id (08장)
            tools = (await session.list_tools()).tools
            result = await session.call_tool("add", {"a": 2, "b": 5})
            print(result.content)

asyncio.run(main())

# 인증 헤더 예 (원격)
headers = {"Authorization": f"Bearer {TOKEN}"}
async with streamablehttp_client(URL, headers=headers) as (read, write, _):
    ...

all_resources = []
cursor = None
while True:
    page = await session.list_resources(cursor=cursor)  # 버전에 따라 params 형태 다를 수 있음
    all_resources.extend(page.resources)
    if page.nextCursor:
        cursor = page.nextCursor
    else:
        break

from mcp import ClientSession, types

async def handle_sampling(message: types.CreateMessageRequestParams) -> types.CreateMessageResult:
    # 여기서 실제 LLM API를 호출하고 (보통 사용자 승인 후) 결과를 돌려준다
    return types.CreateMessageResult(
        role="assistant",
        content=types.TextContent(type="text", text="모델이 생성한 응답"),
        model="some-model",
        stopReason="endTurn",
    )

async with ClientSession(read, write, sampling_callback=handle_sampling) as session:
    await session.initialize()
    # 이제 세션이 서버의 sampling/createMessage 요청을 처리할 수 있다

sessions = {}
# 각 서버마다 (stdio_client/streamablehttp_client → ClientSession)를 열어 보관
sessions["files"] = file_session
sessions["github"] = gh_session
# 사용할 때 적절한 세션을 골라 호출
await sessions["github"].call_tool("search_issues", {"query": "bug"})