fix: arg missing when call method `on_message_replace_func` in output… (#2106)

2 years ago · 9c4bad8f1e
parent c7783dbd6c
commit 9c4bad8f1e
1 changed files with 1 additions and 1 deletions
--- a/api/core/app_runner/moderation_handler.py
+++ b/api/core/app_runner/moderation_handler.py
@ -116,7 +116,7 @@ class OutputModerationHandler(BaseModel):
                # trigger replace event
                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:
                    break