fix: generate not stop when pressing stop link (#1961)
This commit is contained in:
parent
a8cedea15a
commit
0c746f5c5a
8 changed files with 68 additions and 39 deletions
|
|
@ -6,6 +6,7 @@ from typing import Any, Optional, Dict
|
|||
from flask import current_app, Flask
|
||||
from pydantic import BaseModel
|
||||
|
||||
from core.application_queue_manager import PublishFrom
|
||||
from core.moderation.base import ModerationAction, ModerationOutputsResult
|
||||
from core.moderation.factory import ModerationFactory
|
||||
|
||||
|
|
@ -66,7 +67,7 @@ class OutputModerationHandler(BaseModel):
|
|||
final_output = result.text
|
||||
|
||||
if public_event:
|
||||
self.on_message_replace_func(final_output)
|
||||
self.on_message_replace_func(final_output, PublishFrom.TASK_PIPELINE)
|
||||
|
||||
return final_output
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue