diff --git a/api/core/app/apps/advanced_chat/generate_task_pipeline.py b/api/core/app/apps/advanced_chat/generate_task_pipeline.py index b296a64fa3..2c35167bcd 100644 --- a/api/core/app/apps/advanced_chat/generate_task_pipeline.py +++ b/api/core/app/apps/advanced_chat/generate_task_pipeline.py @@ -8,7 +8,6 @@ from threading import Thread from typing import Any, Optional, Union from flask import Flask, current_app - from sqlalchemy import select from sqlalchemy.orm import Session @@ -51,7 +50,7 @@ from core.app.entities.queue_entities import ( QueueWorkflowStartedEvent, QueueWorkflowSucceededEvent, ) -from core.app.entities.queue_task_bridge import advance_chat_queue_task_map, ForwardQueueMessage +from core.app.entities.queue_task_bridge import ForwardQueueMessage, advance_chat_queue_task_map from core.app.entities.task_entities import ( ChatbotAppBlockingResponse, ChatbotAppStreamResponse, @@ -59,8 +58,9 @@ from core.app.entities.task_entities import ( MessageAudioEndStreamResponse, MessageAudioStreamResponse, MessageEndStreamResponse, + MessageStreamResponse, StreamResponse, - WorkflowTaskState, MessageStreamResponse, + WorkflowTaskState, ) from core.app.task_pipeline.based_generate_task_pipeline import BasedGenerateTaskPipeline from core.app.task_pipeline.message_cycle_manage import MessageCycleManage diff --git a/api/core/app/apps/workflow/generate_task_pipeline.py b/api/core/app/apps/workflow/generate_task_pipeline.py index e9744a2cc1..cdea32135f 100644 --- a/api/core/app/apps/workflow/generate_task_pipeline.py +++ b/api/core/app/apps/workflow/generate_task_pipeline.py @@ -4,8 +4,8 @@ import threading import time from collections.abc import Generator from typing import Optional, Union -from flask import Flask, current_app +from flask import Flask, current_app from sqlalchemy.orm import Session from constants.tts_auto_play_timeout import TTS_AUTO_PLAY_TIMEOUT, TTS_AUTO_PLAY_YIELD_CPU_TIME @@ -43,18 +43,19 @@ from core.app.entities.queue_entities import ( QueueWorkflowStartedEvent, QueueWorkflowSucceededEvent, ) -from core.app.entities.queue_task_bridge import workflow_queue_task_map, ForwardQueueMessage +from core.app.entities.queue_task_bridge import ForwardQueueMessage, workflow_queue_task_map from core.app.entities.task_entities import ( ErrorStreamResponse, MessageAudioEndStreamResponse, MessageAudioStreamResponse, + MessageStreamResponse, StreamResponse, TextChunkStreamResponse, WorkflowAppBlockingResponse, WorkflowAppStreamResponse, WorkflowFinishStreamResponse, WorkflowStartStreamResponse, - WorkflowTaskState, MessageStreamResponse, + WorkflowTaskState, ) from core.app.task_pipeline.based_generate_task_pipeline import BasedGenerateTaskPipeline from core.app.task_pipeline.workflow_cycle_manage import WorkflowCycleManage diff --git a/api/core/app/entities/queue_task_bridge.py b/api/core/app/entities/queue_task_bridge.py index 2e68669ddf..5496b55c26 100644 --- a/api/core/app/entities/queue_task_bridge.py +++ b/api/core/app/entities/queue_task_bridge.py @@ -1,4 +1,4 @@ -from core.app.entities.queue_entities import QueueEvent, AppQueueEvent +from core.app.entities.queue_entities import AppQueueEvent, QueueEvent from core.app.entities.task_entities import StreamEvent, StreamResponse workflow_queue_task_map = {