| 84 | self.engine = getattr(self.config, 'animation_engine', 'remotion') |
| 85 | |
| 86 | async def create_messages(self, messages): |
| 87 | assert isinstance(messages, str) |
| 88 | system = self.system |
| 89 | |
| 90 | video_prompt = self.video_prompt if self.config.use_text2video else '' |
| 91 | video_prompt = video_prompt.format(animation_engine=self.engine) |
| 92 | system = system.format( |
| 93 | video_prompt=video_prompt, animation_engine=self.engine) |
| 94 | |
| 95 | return [ |
| 96 | Message(role='system', content=system), |
| 97 | Message(role='user', content=messages), |
| 98 | ] |
| 99 | |
| 100 | async def run(self, messages, **kwargs): |
| 101 | logger.info('Segmenting script to sentences.') |