@@ -18,7 +18,7 @@ from deepl.translator import TextResult from requests import Response from backend import settings -from ml_model.exceptions import DeploymentDisabled, ModelTimeoutError, GenerationException +from ml_model.exceptions import DeploymentDisabled, ModelTimeoutError, GenerationException, RequestBlocked from ml_model.utils import count_openrouter_tokens from poller.models import Proxy @@ -121,6 +121,11 @@ def openrouter_run(version: str, messages: list, callback_data: dict, model_name if c.get('message') and c['message'].get('content') is not None ] if not raw_content: + if any( + c.get('native_finish_reason') == 'SAFETY_CHECK_TYPE_CSAM' + for c in (data.get('choices') or []) + ): + raise RequestBlocked raise GenerationException content = ','.join(raw_content) reasoning = ','.join(