Model Context Protocol (MCP) 및 Weave

# Weave 임포트 (트레이싱에 필요)
import weave
from mcp.server.fastmcp import FastMCP

# 프로젝트 이름으로 Weave 초기화
weave_client = weave.init("my-project")

# MCP 서버 설정
mcp = FastMCP("Demo")

# 도구 정의 (이 호출이 트레이스됨)
@mcp.tool()
def add(a: int, b: int) -> int:
    """Add two numbers."""
    return a + b

# 리소스 정의 (이 호출이 트레이스됨)
@mcp.resource("greeting://{name}")
def get_greeting(name: str) -> str:
    """Return a personalized greeting."""
    return f"Hello, {name}!"

# 프롬프트 정의 (이 호출이 트레이스됨)
@mcp.prompt()
def review_code(code: str) -> str:
    """Return a prompt for reviewing code."""
    return f"Please review this code:\n\n{code}"

# 서버 시작
mcp.run(transport="stdio")

# Weave 임포트 (트레이싱에 필요)
import weave
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client

# 프로젝트 이름으로 Weave 초기화
weave_client = weave.init("my-project")

# MCP 클라이언트 설정 및 실행
async with stdio_client(server_params) as (read, write):
    async with ClientSession(read, write) as session:
        # 세션 초기화
        await session.initialize()
        
        # 도구 호출 (트레이스 기록됨)
        result = await session.call_tool("add", arguments={"a": 1, "b": 2})
        
        # 리소스 조회 (트레이스 기록됨)
        resource = await session.read_resource("greeting://user")
        
        # 프롬프트 조회 (트레이스 기록됨)
        prompt = await session.get_prompt("review_code", arguments={"code": "print('Hello')"})