From 42cf17237fcd15b50422735a8cea9ce79120cd8b Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 3 Sep 2024 15:13:12 -0700 Subject: [PATCH] add imports for new meta tools --- vision_agent/agent/vision_agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 77237954..4733bb24 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -30,7 +30,7 @@ class BoilerplateCode: pre_code = [ "from typing import *", "from vision_agent.utils.execute import CodeInterpreter", - "from vision_agent.tools.meta_tools import Artifacts, open_code_artifact, create_code_artifact, edit_code_artifact, get_tool_descriptions, generate_vision_code, edit_vision_code, write_media_artifact", + "from vision_agent.tools.meta_tools import Artifacts, open_code_artifact, create_code_artifact, edit_code_artifact, get_tool_descriptions, generate_vision_code, edit_vision_code, write_media_artifact, florence2_fine_tuning, use_florence2_fine_tuning", "artifacts = Artifacts('{remote_path}')", "artifacts.load('{remote_path}')", ] @@ -197,7 +197,7 @@ def chat_with_code( artifacts = Artifacts(WORKSPACE / "artifacts.pkl") with CodeInterpreterFactory.new_instance( - code_sandbox_runtime=self.code_sandbox_runtime + code_sandbox_runtime=self.code_sandbox_runtime, ) as code_interpreter: orig_chat = copy.deepcopy(chat) int_chat = copy.deepcopy(chat)