klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve sqlalchemy session in threads

Browse Source
This commit is contained in:
junjun
2025-10-15 17:35:51 +08:00
committed by Junjun
parent 885086e224
commit 67f446e858
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/task/llm.py
+3 -3
View File
@@ -565,7 +565,7 @@ class LLMService:
self.record = save_sql_answer(session=_session, record_id=self.record.id,
answer=orjson.dumps({'content': full_sql_text}).decode())
def generate_with_sub_sql(self, _session: Session, sql, sub_mappings: list):
def generate_with_sub_sql(self, session: Session, sql, sub_mappings: list):
sub_query = json.dumps(sub_mappings, ensure_ascii=False)
self.chat_question.sql = sql
self.chat_question.sub_query = sub_query
@@ -573,7 +573,7 @@ class LLMService:
dynamic_sql_msg.append(SystemMessage(content=self.chat_question.dynamic_sys_question()))
dynamic_sql_msg.append(HumanMessage(content=self.chat_question.dynamic_user_question()))
self.current_logs[OperationEnum.GENERATE_DYNAMIC_SQL] = start_log(session=_session,
self.current_logs[OperationEnum.GENERATE_DYNAMIC_SQL] = start_log(session=session,
ai_modal_id=self.chat_question.ai_modal_id,
ai_modal_name=self.chat_question.ai_modal_name,
operate=OperationEnum.GENERATE_DYNAMIC_SQL,
@@ -595,7 +595,7 @@ class LLMService:
dynamic_sql_msg.append(AIMessage(full_dynamic_text))
self.current_logs[OperationEnum.GENERATE_DYNAMIC_SQL] = end_log(session=_session,
self.current_logs[OperationEnum.GENERATE_DYNAMIC_SQL] = end_log(session=session,
log=self.current_logs[
OperationEnum.GENERATE_DYNAMIC_SQL],
full_message=[