@@ -208,6 +208,9 @@ def openrouter_run(version: str, messages: list, callback_data: dict, model_name input_tokens = data['usage']['prompt_tokens'] output_tokens = data['usage']['completion_tokens'] return (re.sub(r'\\+["n*]', '', answer), input_tokens, output_tokens) + if (data := resp.json()) and data.get('error', {}).get('message'): + if any(error in data['error']['message'] for error in ('PROHIBITED_CONTENT',)): + raise RequestBlocked logger.error(f'Error occured via model {model_name}. Data: {resp.content}') raise Exception(f'No answer from {model_name}, please retry later')