AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 15:09:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api/core/moderation/output_moderation.py): Fix config call. (#6769)

Browse Source
This commit is contained in:
-LAN- 2024-07-29 18:30:29 +08:00 committed by GitHub
parent 2cd662c43b
commit 8dd68e2034
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/configs/app_config.py
View File

@ -64,4 +64,6 @@ class DifyConfig(
return f'{self.HTTP_REQUEST_NODE_MAX_TEXT_SIZE / 1024 / 1024:.2f}MB' return f'{self.HTTP_REQUEST_NODE_MAX_TEXT_SIZE / 1024 / 1024:.2f}MB'
SSRF_PROXY_HTTP_URL: str | None = None SSRF_PROXY_HTTP_URL: str | None = None
SSRF_PROXY_HTTPS_URL: str | None = None SSRF_PROXY_HTTPS_URL: str | None = None
MODERATION_BUFFER_SIZE: int = Field(default=300, description='The buffer size for moderation.')

6
api/core/moderation/output_moderation.py
View File

@ -21,8 +21,6 @@ class ModerationRule(BaseModel):
class OutputModeration(BaseModel): class OutputModeration(BaseModel):
DEFAULT_BUFFER_SIZE: int = 300
tenant_id: str tenant_id: str
app_id: str app_id: str
@ -77,10 +75,10 @@ class OutputModeration(BaseModel):
return final_output return final_output
def start_thread(self) -> threading.Thread: 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={ thread = threading.Thread(target=self.worker, kwargs={
'flask_app': current_app._get_current_object(), '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() thread.start()