feat: workflow/chatflow api support async

pull/18838/head
hobo.l 1 year ago
parent de0d9f96b8
commit 01943bca04

@ -8,7 +8,6 @@ from threading import Thread
from typing import Any, Optional, Union from typing import Any, Optional, Union
from flask import Flask, current_app from flask import Flask, current_app
from sqlalchemy import select from sqlalchemy import select
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
@ -51,7 +50,7 @@ from core.app.entities.queue_entities import (
QueueWorkflowStartedEvent, QueueWorkflowStartedEvent,
QueueWorkflowSucceededEvent, 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 ( from core.app.entities.task_entities import (
ChatbotAppBlockingResponse, ChatbotAppBlockingResponse,
ChatbotAppStreamResponse, ChatbotAppStreamResponse,
@ -59,8 +58,9 @@ from core.app.entities.task_entities import (
MessageAudioEndStreamResponse, MessageAudioEndStreamResponse,
MessageAudioStreamResponse, MessageAudioStreamResponse,
MessageEndStreamResponse, MessageEndStreamResponse,
MessageStreamResponse,
StreamResponse, StreamResponse,
WorkflowTaskState, MessageStreamResponse, WorkflowTaskState,
) )
from core.app.task_pipeline.based_generate_task_pipeline import BasedGenerateTaskPipeline from core.app.task_pipeline.based_generate_task_pipeline import BasedGenerateTaskPipeline
from core.app.task_pipeline.message_cycle_manage import MessageCycleManage from core.app.task_pipeline.message_cycle_manage import MessageCycleManage

@ -4,8 +4,8 @@ import threading
import time import time
from collections.abc import Generator from collections.abc import Generator
from typing import Optional, Union from typing import Optional, Union
from flask import Flask, current_app
from flask import Flask, current_app
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from constants.tts_auto_play_timeout import TTS_AUTO_PLAY_TIMEOUT, TTS_AUTO_PLAY_YIELD_CPU_TIME 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, QueueWorkflowStartedEvent,
QueueWorkflowSucceededEvent, 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 ( from core.app.entities.task_entities import (
ErrorStreamResponse, ErrorStreamResponse,
MessageAudioEndStreamResponse, MessageAudioEndStreamResponse,
MessageAudioStreamResponse, MessageAudioStreamResponse,
MessageStreamResponse,
StreamResponse, StreamResponse,
TextChunkStreamResponse, TextChunkStreamResponse,
WorkflowAppBlockingResponse, WorkflowAppBlockingResponse,
WorkflowAppStreamResponse, WorkflowAppStreamResponse,
WorkflowFinishStreamResponse, WorkflowFinishStreamResponse,
WorkflowStartStreamResponse, WorkflowStartStreamResponse,
WorkflowTaskState, MessageStreamResponse, WorkflowTaskState,
) )
from core.app.task_pipeline.based_generate_task_pipeline import BasedGenerateTaskPipeline from core.app.task_pipeline.based_generate_task_pipeline import BasedGenerateTaskPipeline
from core.app.task_pipeline.workflow_cycle_manage import WorkflowCycleManage from core.app.task_pipeline.workflow_cycle_manage import WorkflowCycleManage

@ -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 from core.app.entities.task_entities import StreamEvent, StreamResponse
workflow_queue_task_map = { workflow_queue_task_map = {

Loading…
Cancel
Save