|
|
|
|
@ -21,8 +21,6 @@ class ModerationRule(BaseModel):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class OutputModeration(BaseModel):
|
|
|
|
|
DEFAULT_BUFFER_SIZE: int = 300
|
|
|
|
|
|
|
|
|
|
tenant_id: str
|
|
|
|
|
app_id: str
|
|
|
|
|
|
|
|
|
|
@ -77,10 +75,10 @@ class OutputModeration(BaseModel):
|
|
|
|
|
return final_output
|
|
|
|
|
|
|
|
|
|
def start_thread(self) -> threading.Thread:
|
|
|
|
|
buffer_size = int(dify_config.config.MODERATION_BUFFER_SIZE)
|
|
|
|
|
buffer_size = dify_config.MODERATION_BUFFER_SIZE
|
|
|
|
|
thread = threading.Thread(target=self.worker, kwargs={
|
|
|
|
|
'flask_app': current_app._get_current_object(),
|
|
|
|
|
'buffer_size': buffer_size if buffer_size > 0 else self.DEFAULT_BUFFER_SIZE
|
|
|
|
|
'buffer_size': buffer_size if buffer_size > 0 else dify_config.MODERATION_BUFFER_SIZE
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
thread.start()
|
|
|
|
|
|