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")

# ツールを定義（この Call はトレースされます）
@mcp.tool()
def add(a: int, b: int) -> int:
    """Add two numbers."""
    return a + b

# リソースを定義（この Call はトレースされます）
@mcp.resource("greeting://{name}")
def get_greeting(name: str) -> str:
    """Return a personalized greeting."""
    return f"Hello, {name}!"

# プロンプトを定義（この Call はトレースされます）
@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')"})