|
@@ -116,7 +116,7 @@ class OutputModerationHandler(BaseModel):
|
|
|
|
|
|
# trigger replace event
|
|
# trigger replace event
|
|
if self.thread_running:
|
|
if self.thread_running:
|
|
- self.on_message_replace_func(final_output)
|
|
|
|
|
|
+ self.on_message_replace_func(final_output, PublishFrom.TASK_PIPELINE)
|
|
|
|
|
|
if result.action == ModerationAction.DIRECT_OUTPUT:
|
|
if result.action == ModerationAction.DIRECT_OUTPUT:
|
|
break
|
|
break
|