From 8dd68e2034a847200ac5717e25c6cf260e79d980 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Mon, 29 Jul 2024 18:30:29 +0800 Subject: [PATCH] fix(api/core/moderation/output_moderation.py): Fix config call. (#6769) --- api/configs/app_config.py | 4 +++- api/core/moderation/output_moderation.py | 6 ++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/api/configs/app_config.py b/api/configs/app_config.py index d1099a9036..a5a4fc788d 100644 --- a/api/configs/app_config.py +++ b/api/configs/app_config.py @@ -64,4 +64,6 @@ class DifyConfig( return f'{self.HTTP_REQUEST_NODE_MAX_TEXT_SIZE / 1024 / 1024:.2f}MB' SSRF_PROXY_HTTP_URL: str | None = None - SSRF_PROXY_HTTPS_URL: str | None = None \ No newline at end of file + SSRF_PROXY_HTTPS_URL: str | None = None + + MODERATION_BUFFER_SIZE: int = Field(default=300, description='The buffer size for moderation.') diff --git a/api/core/moderation/output_moderation.py b/api/core/moderation/output_moderation.py index 4fd08ca021..9a4d8db4e2 100644 --- a/api/core/moderation/output_moderation.py +++ b/api/core/moderation/output_moderation.py @@ -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()