fix: arg missing when call method on_message_replace_func in output… (#2106)
This commit is contained in:
parent
c7783dbd6c
commit
9c4bad8f1e
1 changed files with 1 additions and 1 deletions
|
|
@ -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
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue