AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 17:39:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: transaction hangs due to message commit block during long LLM calls (#1206)

Browse Source
This commit is contained in:
takatost 2023-09-21 11:22:10 +08:00 committed by GitHub
parent e65f9cb17a
commit 2b018fade2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/conversation_message_task.py
View File

@ -140,7 +140,7 @@ class ConversationMessageTask:
) )
db.session.add(self.message) db.session.add(self.message)
db.session.flush() db.session.commit()
def append_message_text(self, text: str): def append_message_text(self, text: str):
if text is not None: if text is not None:
@ -191,7 +191,7 @@ class ConversationMessageTask:
) )
db.session.add(message_chain) db.session.add(message_chain)
db.session.flush() db.session.commit()
return message_chain return message_chain
@ -217,7 +217,7 @@ class ConversationMessageTask:
) )
db.session.add(message_agent_thought) db.session.add(message_agent_thought)
db.session.flush() db.session.commit()
self._pub_handler.pub_agent_thought(message_agent_thought) 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.tokens = agent_loop.prompt_tokens + agent_loop.completion_tokens
message_agent_thought.total_price = loop_total_price message_agent_thought.total_price = loop_total_price
message_agent_thought.currency = agent_model_instance.get_currency() 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): def on_dataset_query_end(self, dataset_query_obj: DatasetQueryObj):
dataset_query = DatasetQuery( dataset_query = DatasetQuery(
@ -285,7 +285,7 @@ class ConversationMessageTask:
created_by=self.user.id created_by=self.user.id
) )
db.session.add(dataset_retriever_resource) db.session.add(dataset_retriever_resource)
db.session.flush() db.session.commit()
self.retriever_resource = resource self.retriever_resource = resource
def message_end(self): def message_end(self):