From 9c4bad8f1ef79d80770541118938ee5e40f3c163 Mon Sep 17 00:00:00 2001 From: takatost Date: Sat, 20 Jan 2024 17:53:38 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20arg=20missing=20when=20call=20method=20`?= =?UTF-8?q?on=5Fmessage=5Freplace=5Ffunc`=20in=20output=E2=80=A6=20(#2106)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/core/app_runner/moderation_handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/core/app_runner/moderation_handler.py b/api/core/app_runner/moderation_handler.py index ecb32ee43..24ea08561 100644 --- 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