outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: transaction not commit during long LLM calls (#1213)

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

4
api/core/conversation_message_task.py
View File

@ -112,7 +112,7 @@ class ConversationMessageTask:
) )
db.session.add(self.conversation) db.session.add(self.conversation)
db.session.flush() db.session.commit()
self.message = Message( self.message = Message(
app_id=self.app_model_config.app_id, 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): def on_chain_end(self, message_chain: MessageChain, chain_result: ChainResult):
message_chain.output = json.dumps(chain_result.completion) message_chain.output = json.dumps(chain_result.completion)
db.session.commit()
self._pub_handler.pub_chain(message_chain) self._pub_handler.pub_chain(message_chain)
@ -262,6 +263,7 @@ class ConversationMessageTask:
) )
db.session.add(dataset_query) db.session.add(dataset_query)
db.session.commit()
def on_dataset_query_finish(self, resource: List): def on_dataset_query_finish(self, resource: List):
if resource and len(resource) > 0: if resource and len(resource) > 0: