@@ -13,6 +13,7 @@ from ml_model.exceptions import ( NSFWDetectedException, RealPersonDetectedError, RequestBlocked, + PromptLengthExceeded, ) from poller.models import Proxy @@ -206,6 +207,9 @@ class BytedanceModelArkAdapter: timeout=600, ) as client: resp = client.post(cls.CONTENT_TYPE_TO_ENDPOINT[BytedanceContentType.CHAT], json=payload) + if resp.status_code == 413: + logger.error('Payload too large: %s', resp.text) + raise PromptLengthExceeded(max_length=202752) if resp.status_code >= 400: logger.error( 'Bytedance request failed status=%s route=%s body=%s', @@ -213,6 +217,7 @@ class BytedanceModelArkAdapter: cls.CONTENT_TYPE_TO_ENDPOINT[BytedanceContentType.CHAT], resp.text, ) + raise GenerationException try: data: BytedanceChatResponse = resp.json() except Exception as exc: