コンテンツにスキップ

SQLAlchemy セッション

SQLAlchemySession は SQLAlchemy を使用して本番環境対応のセッション実装を提供し、SQLAlchemy がサポートする任意のデータベース（PostgreSQL、MySQL、SQLite など）をセッションストレージとして使用できるようにします。

インストール

SQLAlchemy セッションには、オプション依存関係の extra sqlalchemy と、データベース URL に対応する非同期データベースドライバーが必要です。

以下の SQLite の例（sqlite+aiosqlite://）では、この extra と併せて aiosqlite をインストールしてください。

pip install 'openai-agents[sqlalchemy]' aiosqlite

この extra には、postgresql+asyncpg:// で始まる PostgreSQL URL 用の asyncpg がすでに含まれています。mysql+aiomysql:// で始まる MySQL URL では、この extra と併せて aiomysql をインストールしてください。このドライバーの rsa extra は、MySQL の SHA-256 認証方式に必要な依存関係を提供します。

pip install 'openai-agents[sqlalchemy]' 'aiomysql[rsa]'

クイックスタート

データベース URL の使用

最も簡単に開始する方法は次のとおりです。

import asyncio
from agents import Agent, Runner
from agents.extensions.memory import SQLAlchemySession

async def main():
    agent = Agent("Assistant")

    # Create session using database URL
    session = SQLAlchemySession.from_url(
        "user-123",
        url="sqlite+aiosqlite:///:memory:",
        create_tables=True
    )

    result = await Runner.run(agent, "Hello", session=session)
    print(result.final_output)

if __name__ == "__main__":
    asyncio.run(main())

既存エンジンの使用

既存の SQLAlchemy エンジンを使用するアプリケーションの場合は、次のようにします。

import asyncio
from agents import Agent, Runner
from agents.extensions.memory import SQLAlchemySession
from sqlalchemy.ext.asyncio import create_async_engine

async def main():
    # Create your database engine
    engine = create_async_engine("postgresql+asyncpg://user:pass@localhost/db")

    agent = Agent("Assistant")
    session = SQLAlchemySession(
        "user-456",
        engine=engine,
        create_tables=True
    )

    result = await Runner.run(agent, "Hello", session=session)
    print(result.final_output)

    # Clean up
    await engine.dispose()

if __name__ == "__main__":
    asyncio.run(main())

非 ASCII テキストの保存

デフォルトでは、SQLAlchemySession はセッション項目を JSON にシリアライズするときに、非 ASCII 文字をエスケープします。これにより従来の保存形式を維持しながら、項目のロード時には元のテキストを完全に復元できます。

保存された JSON 内で多言語テキストを読みやすい状態に保つには、ensure_ascii=False を設定します。

session = SQLAlchemySession.from_url(
    "user-123",
    url="sqlite+aiosqlite:///conversations.db",
    create_tables=True,
    ensure_ascii=False,
)

既存のエンジンを使用する場合は、同じオプションを SQLAlchemySession(...) に直接渡すことができます。この設定で変更されるのはデータベースに保存される JSON 表現のみであり、セッションメソッドから返される値は変更されません。

API リファレンス