wangtinghu/agentUniverse
1
0
Fork 0
mirror of https://github.com/agentuniverse-ai/agentUniverse.git synced 2026-02-09 01:59:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #133 from alipay/fix_session_bug_20240729

Browse Source 
Fix: Multi session conflict in async service run
This commit is contained in:
AniviaTn
2024-07-29 20:18:52 +08:00
committed by GitHub
parent fc1ddb7dd5 3245a14d1d
commit a0881ba001
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
agentuniverse/agent_serve/web/dal/request_library.py
View File

@@ -76,7 +76,7 @@ class RequestLibrary:
if not self.session:
self.__init_request_table()
self.session = self.sqldb_wrapper.get_session()
return self.session
return self.session()
def query_request_by_request_id(self, request_id: str) -> RequestDO | None:

4
agentuniverse/database/sqldb_wrapper.py
View File

@@ -100,7 +100,7 @@ class SQLDBWrapper(ComponentBase):
Get a sqlalchemy session, used for operating with orm.
"""
if self.db_session:
return self.db_session()
return self.db_session
# Create database engine
self.db_session = sessionmaker(bind=self.sql_database._engine)
return self.db_session()
return self.db_session