add imports for new meta tools

landing-ai · Sep 3, 2024 · 42cf172 · 42cf172
1 parent 7125e62
commit 42cf172
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)