AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 00:05:57 +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

2
api/configs/app_config.py
View File

@ -65,3 +65,5 @@ class DifyConfig(
SSRF_PROXY_HTTP_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):
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()