From 2b018fade2177bb6bcf3f0edb113df5e8625a9eb Mon Sep 17 00:00:00 2001 From: takatost Date: Thu, 21 Sep 2023 11:22:10 +0800 Subject: [PATCH] fix: transaction hangs due to message commit block during long LLM calls (#1206) --- api/core/conversation_message_task.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/api/core/conversation_message_task.py b/api/core/conversation_message_task.py index 41db2d16c5..6eefff76d2 100644 --- a/api/core/conversation_message_task.py +++ b/api/core/conversation_message_task.py @@ -140,7 +140,7 @@ class ConversationMessageTask: ) db.session.add(self.message) - db.session.flush() + db.session.commit() def append_message_text(self, text: str): if text is not None: @@ -191,7 +191,7 @@ class ConversationMessageTask: ) db.session.add(message_chain) - db.session.flush() + db.session.commit() return message_chain @@ -217,7 +217,7 @@ class ConversationMessageTask: ) db.session.add(message_agent_thought) - db.session.flush() + db.session.commit() self._pub_handler.pub_agent_thought(message_agent_thought) @@ -249,7 +249,7 @@ class ConversationMessageTask: message_agent_thought.tokens = agent_loop.prompt_tokens + agent_loop.completion_tokens message_agent_thought.total_price = loop_total_price message_agent_thought.currency = agent_model_instance.get_currency() - db.session.flush() + db.session.commit() def on_dataset_query_end(self, dataset_query_obj: DatasetQueryObj): dataset_query = DatasetQuery( @@ -285,7 +285,7 @@ class ConversationMessageTask: created_by=self.user.id ) db.session.add(dataset_retriever_resource) - db.session.flush() + db.session.commit() self.retriever_resource = resource def message_end(self):