OpenRouterセットアップガイド

# OpenAI SDK（OpenRouter互換）
pip install openai

# 環境変数読み込み用
pip install python-dotenv

# オプション: エージェントフレームワーク
pip install langchain langchain-openai

import os
import json
from openai import OpenAI
from dotenv import load_dotenv

load_dotenv()

client = OpenAI(
    base_url="https://openrouter.ai/api/v1",
    api_key=os.getenv("OPENROUTER_API_KEY"),
)

# ツール定義
tools = [
    {
        "type": "function",
        "function": {
            "name": "calculate",
            "description": "数学の計算を実行します",
            "parameters": {
                "type": "object",
                "properties": {
                    "expression": {
                        "type": "string",
                        "description": "計算する式（例: '2 + 2', '10 * 5'）"
                    }
                },
                "required": ["expression"]
            }
        }
    }
]

# ツール実行関数
def execute_tool(name: str, args: dict) -> str:
    if name == "calculate":
        try:
            result = eval(args["expression"])  # 注意: 本番環境ではsafe_evalを使用
            return str(result)
        except Exception as e:
            return f"エラー: {e}"
    return "未知のツール"

# エージェントループ
def run_agent(user_message: str):
    messages = [{"role": "user", "content": user_message}]

    while True:
        response = client.chat.completions.create(
            model="anthropic/claude-sonnet-4",
            messages=messages,
            tools=tools,
            tool_choice="auto"
        )

        assistant_message = response.choices[0].message

        # ツール呼び出しがない場合は終了
        if not assistant_message.tool_calls:
            return assistant_message.content

        # ツールを実行
        messages.append(assistant_message)
        for tool_call in assistant_message.tool_calls:
            result = execute_tool(
                tool_call.function.name,
                json.loads(tool_call.function.arguments)
            )
            messages.append({
                "role": "tool",
                "tool_call_id": tool_call.id,
                "content": result
            })

# テスト実行
result = run_agent("125 * 48 + 367 を計算してください")
print(result)
# 出力: 125 * 48 + 367 = 6,367 です。