SQLAlchemy 세션
SQLAlchemySession은 SQLAlchemy를 사용해 프로덕션급 세션 구현을 제공합니다. 이를 통해 세션 저장소로 SQLAlchemy가 지원하는 모든 데이터베이스(PostgreSQL, MySQL, SQLite 등)를 사용할 수 있습니다.
설치
SQLAlchemy 세션에는sqlalchemy extra가 필요합니다:
빠른 시작
데이터베이스 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())기존 엔진 사용
기존 SQLAlchemy 엔진을 사용하는 애플리케이션의 경우:
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- 기본 세션 프로토콜