From 4a50f5bdc798dffe3d450d78cd0796aee8db8e7c Mon Sep 17 00:00:00 2001
From: 0xrohitgarg <rohit@spext.co>
Date: Thu, 28 Nov 2024 12:01:19 +0530
Subject: [PATCH] use default llm in text to movie

---
 backend/director/agents/text_to_movie.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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(