jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: regenerate in Chat, agent and Chatflow app (#7661)

Browse source
This commit is contained in:
Hash Brown 2024-09-22 03:15:11 +08:00 • committed by GitHub
commit 8c51d06222
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
51 changed files with 606 additions and 181 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/core/agent/base_agent_runner.py
View file

@ -32,6 +32,7 @@ from core.model_runtime.entities.message_entities import (
from core.model_runtime.entities.model_entities import ModelFeature
from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
from core.model_runtime.utils.encoders import jsonable_encoder
from core.prompt.utils.extract_thread_messages import extract_thread_messages
from core.tools.entities.tool_entities import (
ToolParameter,
ToolRuntimeVariablePool,
@ -441,10 +442,12 @@ class BaseAgentRunner(AppRunner):
.filter(
Message.conversation_id == self.message.conversation_id,
)
.order_by(Message.created_at.asc())
.order_by(Message.created_at.desc())
.all()
)
messages = list(reversed(extract_thread_messages(messages)))
for message in messages:
if message.id == self.message.id:
continue