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: fix log persistence issues with concurrent question requests

Browse Source 
#940
This commit is contained in:
ulleo
2026-03-04 18:04:18 +08:00
committed by ulleo
parent ff60e1f8c9
commit 0ff44b4d94
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/task/llm.py
+11 -8
View File
@@ -70,8 +70,8 @@ class LLMService:
record: ChatRecord
config: LLMConfig
llm: BaseChatModel
sql_message: List[Union[BaseMessage, dict[str, Any]]] = []
chart_message: List[Union[BaseMessage, dict[str, Any]]] = []
sql_message: List[Union[BaseMessage, dict[str, Any]]]
chart_message: List[Union[BaseMessage, dict[str, Any]]]
# session: Session = db_session
current_user: CurrentUser
@@ -79,12 +79,10 @@ class LLMService:
out_ds_instance: Optional[AssistantOutDs] = None
change_title: bool = False
generate_sql_logs: List[ChatLog] = []
generate_chart_logs: List[ChatLog] = []
current_logs: dict[OperationEnum, ChatLog] = {}
chunk_list: List[str] = []
generate_sql_logs: List[ChatLog]
generate_chart_logs: List[ChatLog]
current_logs: dict[OperationEnum, ChatLog]
chunk_list: List[str]
future: Future
trans: I18nHelper = None
@@ -98,6 +96,11 @@ class LLMService:
def __init__(self, session: Session, current_user: CurrentUser, chat_question: ChatQuestion,
current_assistant: Optional[CurrentAssistant] = None, no_reasoning: bool = False,
embedding: bool = False, config: LLMConfig = None):
self.sql_message = []
self.chart_message = []
self.generate_sql_logs = []
self.generate_chart_logs = []
self.current_logs = {}
self.chunk_list = []
self.current_user = current_user
self.current_assistant = current_assistant