瀏覽代碼

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

tags/0.6.16
-LAN- 1 年之前
父節點
當前提交
8dd68e2034
沒有連結到貢獻者的電子郵件帳戶。
共有 2 個檔案被更改,包括 5 行新增5 行删除
  1. 3
    1
      api/configs/app_config.py
  2. 2
    4
      api/core/moderation/output_moderation.py

+ 3
- 1
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
SSRF_PROXY_HTTPS_URL: str | None = None

MODERATION_BUFFER_SIZE: int = Field(default=300, description='The buffer size for moderation.')

+ 2
- 4
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()

Loading…
取消
儲存