SQLAlchemy 会话
SQLAlchemySession 使用 SQLAlchemy 提供可用于生产环境的会话实现,使你可以使用 SQLAlchemy 支持的任意数据库(PostgreSQL、MySQL、SQLite 等)进行会话存储。
安装
SQLAlchemy 会话需要安装sqlalchemy 的额外依赖:
快速开始
使用数据库 URL
最简方式:
importasynciofromagentsimportAgent,Runnerfromagents.extensions.memoryimportSQLAlchemySessionasyncdefmain():agent=Agent("Assistant")# Create session using database URLsession=SQLAlchemySession.from_url("user-123",url="sqlite+aiosqlite:///:memory:",create_tables=True)result=awaitRunner.run(agent,"Hello",session=session)print(result.final_output)if__name__=="__main__":asyncio.run(main())使用已有 engine
适用于已有 SQLAlchemy engine 的应用:
importasynciofromagentsimportAgent,Runnerfromagents.extensions.memoryimportSQLAlchemySessionfromsqlalchemy.ext.asyncioimportcreate_async_engineasyncdefmain():# Create your database engineengine=create_async_engine("postgresql+asyncpg://user:pass@localhost/db")agent=Agent("Assistant")session=SQLAlchemySession("user-456",engine=engine,create_tables=True)result=awaitRunner.run(agent,"Hello",session=session)print(result.final_output)# Clean upawaitengine.dispose()if__name__=="__main__":asyncio.run(main())API 参考
SQLAlchemySession- 主类Session- 基础会话协议