Skip to content

Commit 537dbfc

Browse files
authored
Merge pull request #167 from video-db/fix-editing-agent-default-llm
fix editing agent to use default llm
2 parents e4ce7d2 + 114996b commit 537dbfc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backend/director/agents/editing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
RoleTypes,
1111
)
1212
from director.tools.videodb_tool import VideoDBTool
13-
from director.llm.videodb_proxy import VideoDBProxy
13+
from director.llm import get_default_llm
1414

1515
from videodb.asset import VideoAsset, AudioAsset, ImageAsset, TextAsset
1616

@@ -151,7 +151,7 @@ def __init__(self, session: Session, **kwargs):
151151
self.editing_response = None
152152

153153
# TODO: benchmark different llm
154-
self.llm = VideoDBProxy()
154+
self.llm = get_default_llm()
155155

156156
# TODO: find a way to get the tool description from function/tool and not hardcode here
157157
self.tools = [

0 commit comments

Comments
 (0)