diff --git a/backend/director/agents/text_to_movie.py b/backend/director/agents/text_to_movie.py index 0b48b49..068c6fc 100644 --- a/backend/director/agents/text_to_movie.py +++ b/backend/director/agents/text_to_movie.py @@ -15,7 +15,7 @@ RoleTypes, VideoData, ) -from director.llm.openai import OpenAI, OpenaiConfig +from director.llm import get_default_llm from director.tools.kling import KlingAITool, PARAMS_CONFIG as KLING_PARAMS_CONFIG from director.tools.stabilityai import ( StabilityAITool, @@ -125,7 +125,7 @@ def __init__(self, session: Session, **kwargs): "Agent for generating movies from storylines using Gen AI models" ) self.parameters = TEXT_TO_MOVIE_AGENT_PARAMETERS - self.llm = OpenAI(OpenaiConfig(timeout=120)) + self.llm = get_default_llm() self.engine_configs = { "kling": EngineConfig(