use default llm in text to movie (#86)

video-db · Nov 28, 2024 · 0908ce3 · 0908ce3
1 parent d70913b
commit 0908ce3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(