Procházet zdrojové kódy

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

tags/0.6.16
-LAN- před 1 rokem
rodič
revize
8dd68e2034
Žádný účet není propojen s e-mailovou adresou tvůrce revize
2 změnil soubory, kde provedl 5 přidání a 5 odebrání
  1. 3
    1
      api/configs/app_config.py
  2. 2
    4
      api/core/moderation/output_moderation.py

+ 3
- 1
api/configs/app_config.py Zobrazit soubor

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.')

+ 2
- 4
api/core/moderation/output_moderation.py Zobrazit soubor





class OutputModeration(BaseModel): class OutputModeration(BaseModel):
DEFAULT_BUFFER_SIZE: int = 300

tenant_id: str tenant_id: str
app_id: str app_id: str


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()

Načítá se…
Zrušit
Uložit