fix type and format errors

landing-ai · Sep 17, 2024 · cabc88d · cabc88d
1 parent 5a8f6a9
commit cabc88d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -14,7 +14,13 @@
  VA_CODE,
 )
 from vision_agent.lmm import LMM, AnthropicLMM, Message, OpenAILMM
-from vision_agent.tools import META_TOOL_DOCSTRING, extract_frames, load_image, save_image
+from vision_agent.tools import (
+ META_TOOL_DOCSTRING,
+ extract_frames,
+ load_image,
+ save_image,
+ save_video,
+)
 from vision_agent.tools.meta_tools import (
  Artifacts,
  check_and_load_image,