跳转至

SQLAlchemy 会话

SQLAlchemySession 使用 SQLAlchemy 提供可用于生产环境的会话实现，支持使用 SQLAlchemy 兼容的任何数据库（PostgreSQL、MySQL、SQLite 等）存储会话。

安装

SQLAlchemy 会话需要 sqlalchemy 可选依赖 extra，以及与数据库 URL 匹配的异步数据库驱动程序。

对于以下 SQLite 代码示例（sqlite+aiosqlite://），请在安装该 extra 的同时安装 aiosqlite：

pip install 'openai-agents[sqlalchemy]' aiosqlite

该 extra 已包含 asyncpg，用于以 postgresql+asyncpg:// 开头的 PostgreSQL URL。对于以 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 参考