@@ -10,7 +10,7 @@ import requests from django.core.files import File from messages.models import Message -from ml_model.exceptions import RequestBlocked, GenerationException +from ml_model.exceptions import GenerationException, PromptLengthExceeded, RequestBlocked from ml_model.services.base import SimpleService from ml_model.tasks import replicate_run @@ -42,6 +42,8 @@ class Grok_Imagine_Video(SimpleService): return [msg] def make(self, input_message: Message, save: bool = True) -> list[Message]: + if len(input_message.content or '') > 2000: + raise PromptLengthExceeded(max_length=2000) duration = input_message.info.get('duration', 5) if (balance := PaymentPlanSelector(self.store.user).get_current_balance()) < ( cost := self.TOKENS_COST * duration