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

9c4bad8f · takatost · GitHub · c7783dbd · 9c4bad8f
Unverified Commit 9c4bad8f authored Jan 20, 2024 by takatost Committed by GitHub Jan 20, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

moderation_handler.py api/core/app_runner/moderation_handler.py +1 -1

No files found.
--- 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