Quellcode durchsuchen

using message_file event instead of message event (#22988)

tags/1.7.2
znn vor 2 Monaten
Ursprung
Commit
084dcd1a50
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden
1 geänderte Dateien mit 5 neuen und 0 gelöschten Zeilen
  1. 5
    0
      api/core/app/task_pipeline/message_cycle_manager.py

+ 5
- 0
api/core/app/task_pipeline/message_cycle_manager.py Datei anzeigen

@@ -23,6 +23,7 @@ from core.app.entities.task_entities import (
MessageFileStreamResponse,
MessageReplaceStreamResponse,
MessageStreamResponse,
StreamEvent,
WorkflowTaskState,
)
from core.llm_generator.llm_generator import LLMGenerator
@@ -180,11 +181,15 @@ class MessageCycleManager:
:param message_id: message id
:return:
"""
message_file = db.session.query(MessageFile).filter(MessageFile.id == message_id).first()
event_type = StreamEvent.MESSAGE_FILE if message_file else StreamEvent.MESSAGE

return MessageStreamResponse(
task_id=self._application_generate_entity.task_id,
id=message_id,
answer=answer,
from_variable_selector=from_variable_selector,
event=event_type,
)

def message_replace_to_stream_response(self, answer: str, reason: str = "") -> MessageReplaceStreamResponse:

Laden…
Abbrechen
Speichern