fix: improve sqlalchemy session in threads

#294

Author: ulleo

backend/apps/chat/api/chat.py

@@ -118,7 +118,7 @@ def _return_empty():
 
         request_question = ChatQuestion(chat_id=record.chat_id, question=record.question if record.question else '')
 
-        llm_service = await LLMService.create(current_user, request_question, current_assistant, True)
+        llm_service = await LLMService.create(session, current_user, request_question, current_assistant, True)
         llm_service.set_record(record)
         llm_service.run_recommend_questions_task_async()
     except Exception as e:
@@ -147,8 +147,9 @@ async def stream_sql(session: SessionDep, current_user: CurrentUser, request_que
     """
 
     try:
-        llm_service = await LLMService.create(current_user, request_question, current_assistant, embedding=True)
-        llm_service.init_record()
+        llm_service = await LLMService.create(session, current_user, request_question, current_assistant,
+                                              embedding=True)
+        llm_service.init_record(session=session)
         llm_service.run_task_async()
     except Exception as e:
         traceback.print_exc()
@@ -189,8 +190,8 @@ async def analysis_or_predict(session: SessionDep, current_user: CurrentUser, ch
 
         request_question = ChatQuestion(chat_id=record.chat_id, question=record.question)
 
-        llm_service = await LLMService.create(current_user, request_question, current_assistant)
-        llm_service.run_analysis_or_predict_task_async(action_type, record)
+        llm_service = await LLMService.create(session, current_user, request_question, current_assistant)
+        llm_service.run_analysis_or_predict_task_async(session, action_type, record)
     except Exception as e:
         traceback.print_exc()