From 32d9b6181ce30424abbd16756e5142364b20820c Mon Sep 17 00:00:00 2001 From: takatost Date: Fri, 22 Sep 2023 12:43:06 +0800 Subject: [PATCH] fix: transaction not commit during long LLM calls (#1213) --- api/core/conversation_message_task.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/core/conversation_message_task.py b/api/core/conversation_message_task.py index 6eefff76d..49c9cba08 100644 --- a/api/core/conversation_message_task.py +++ b/api/core/conversation_message_task.py @@ -112,7 +112,7 @@ class ConversationMessageTask: ) db.session.add(self.conversation) - db.session.flush() + db.session.commit() self.message = Message( app_id=self.app_model_config.app_id, @@ -197,6 +197,7 @@ class ConversationMessageTask: def on_chain_end(self, message_chain: MessageChain, chain_result: ChainResult): message_chain.output = json.dumps(chain_result.completion) + db.session.commit() self._pub_handler.pub_chain(message_chain) @@ -262,6 +263,7 @@ class ConversationMessageTask: ) db.session.add(dataset_query) + db.session.commit() def on_dataset_query_finish(self, resource: List): if resource and len(resource) > 0: