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

fix: #8843 event: tts_message_end always return in api streaming resp… (#8846)

Browse Source
This commit is contained in:
chenxu9741
2024-09-29 16:13:20 +08:00
committed by GitHub
parent e4ed916baa
commit c531b4a911
4 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/app/apps/advanced_chat/generate_task_pipeline.py
View File

@@ -231,6 +231,7 @@ class AdvancedChatAppGenerateTaskPipeline(BasedGenerateTaskPipeline, WorkflowCyc
except Exception as e: except Exception as e:
logger.error(e) logger.error(e)
break break
if tts_publisher:
yield MessageAudioEndStreamResponse(audio="", task_id=task_id) yield MessageAudioEndStreamResponse(audio="", task_id=task_id)
def _process_stream_response( def _process_stream_response(

1
api/core/app/apps/workflow/generate_task_pipeline.py
View File

@@ -212,6 +212,7 @@ class WorkflowAppGenerateTaskPipeline(BasedGenerateTaskPipeline, WorkflowCycleMa
except Exception as e: except Exception as e:
logger.error(e) logger.error(e)
break break
if tts_publisher:
yield MessageAudioEndStreamResponse(audio="", task_id=task_id) yield MessageAudioEndStreamResponse(audio="", task_id=task_id)
def _process_stream_response( def _process_stream_response(

1
api/core/app/task_pipeline/easy_ui_based_generate_task_pipeline.py
View File

@@ -248,6 +248,7 @@ class EasyUIBasedGenerateTaskPipeline(BasedGenerateTaskPipeline, MessageCycleMan
else: else:
start_listener_time = time.time() start_listener_time = time.time()
yield MessageAudioStreamResponse(audio=audio.audio, task_id=task_id) yield MessageAudioStreamResponse(audio=audio.audio, task_id=task_id)
if publisher:
yield MessageAudioEndStreamResponse(audio="", task_id=task_id) yield MessageAudioEndStreamResponse(audio="", task_id=task_id)
def _process_stream_response( def _process_stream_response(

2
api/core/model_runtime/model_providers/zhipuai/zhipuai_sdk/core/_base_models.py
View File

@@ -48,7 +48,7 @@ from ._utils import (
) )
if TYPE_CHECKING: if TYPE_CHECKING:
from pydantic_core.core_schema import LiteralSchema, ModelField, ModelFieldsSchema from pydantic_core.core_schema import ModelField
__all__ = ["BaseModel", "GenericModel"] __all__ = ["BaseModel", "GenericModel"]
_BaseModelT = TypeVar("_BaseModelT", bound="BaseModel") _BaseModelT = TypeVar("_BaseModelT", bound="BaseModel")