[Issue]: Upload datasets trigger update session, and then the DB API connection will be closed.

[Aisuko]

Contact Details(optional)

No response

What is the issue you are having?

Upload datasets trigger update session, and then the DB API connection will be closed.

Relevant log output

2024-07-26 00:39:07.157 | INFO     | src.repository.vector_database:create_collection:53 - Vector Database --- Milvus: collection aisuko_squad01 created
2024-07-26 00:39:07,165 INFO sqlalchemy.engine.Engine SELECT session.id, session.uuid, session.account_id, session.name, session.session_type, session.dataset_name, session.created_at, session.updated_at 
FROM session 
WHERE session.uuid = $1::VARCHAR AND session.account_id = $2::INTEGER
2024-07-26 00:39:07,166 INFO sqlalchemy.engine.Engine [cached since 21.48s ago] ('4caee4f3-bd75-4651-a7ad-5bac62093e56', 3)
2024-07-26 00:39:07,176 INFO sqlalchemy.engine.Engine UPDATE session SET dataset_name=$1::VARCHAR, updated_at=now() WHERE session.uuid = $2::VARCHAR
2024-07-26 00:39:07,176 INFO sqlalchemy.engine.Engine [generated in 0.00029s] ('aisuko_squad01', '4caee4f3-bd75-4651-a7ad-5bac62093e56')
2024-07-26 00:39:07,177 INFO sqlalchemy.engine.Engine COMMIT
2024-07-26 00:39:07.178 | INFO     | src.repository.events:inspect_db_server_on_close:47 - Closing DB API Connection ---
 <AdaptedConnection <asyncpg.connection.Connection object at 0x795ae41afb50>>
2024-07-26 00:39:07.179 | INFO     | src.repository.events:inspect_db_server_on_close:48 - Closed Connection Record ---
 <sqlalchemy.pool.base._ConnectionRecord object at 0x795ac2f598d0>
2024-07-26 00:39:07.227 | INFO     | src.repository.events:inspect_db_server_on_connection:39 - New DB API Connection ---
 <AdaptedConnection <asyncpg.connection.Connection object at 0x795ac2fcb4c0>>
2024-07-26 00:39:07.228 | INFO     | src.repository.events:inspect_db_server_on_connection:40 - Connection Record ---
 <sqlalchemy.pool.base._ConnectionRecord object at 0x795ac2f59940>
2024-07-26 00:39:07,228 INFO sqlalchemy.engine.Engine BEGIN (implicit)
2024-07-26 00:39:07,229 INFO sqlalchemy.engine.Engine SELECT session.id, session.uuid, session.account_id, session.name, session.session_type, session.dataset_name, session.created_at, session.updated_at 
FROM session 
WHERE session.id = $1::INTEGER
2024-07-26 00:39:07,229 INFO sqlalchemy.engine.Engine [cached since 33.98s ago] (1,)
2024-07-26 00:39:07,237 INFO sqlalchemy.engine.Engine ROLLBACK
2024-07-26 00:39:07.237 | INFO     | src.repository.events:inspect_db_server_on_close:47 - Closing DB API Connection ---
 <AdaptedConnection <asyncpg.connection.Connection object at 0x795ac2fcb4c0>>
2024-07-26 00:39:07.238 | INFO     | src.repository.events:inspect_db_server_on_close:48 - Closed Connection Record ---
 <sqlalchemy.pool.base._ConnectionRecord object at 0x795ac2f59940>
INFO:     127.0.0.1:56864 - "POST /api/ds/load HTTP/1.1" 200 OK