Bladeren bron

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

takatost 1 jaar geleden
bovenliggende
commit
9c4bad8f1e
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  1. 1 1
      api/core/app_runner/moderation_handler.py

+ 1 - 1
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