From 53fff2dd43ebb8fe1f29840160decdcb3749b2a5 Mon Sep 17 00:00:00 2001 From: wuyiqunLu Date: Wed, 11 Sep 2024 11:43:51 +0800 Subject: [PATCH] change back test code --- vision_agent/agent/vision_agent.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index cc8b4f53..b60e8391 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -13,9 +13,8 @@ VA_CODE, ) from vision_agent.lmm import LMM, Message, OpenAILMM -from vision_agent.tools import META_TOOL_DOCSTRING, save_image +from vision_agent.tools import META_TOOL_DOCSTRING, save_image, load_image from vision_agent.tools.meta_tools import Artifacts, use_extra_vision_agent_args -from vision_agent.tools.tools import load_image from vision_agent.utils import CodeInterpreterFactory from vision_agent.utils.execute import CodeInterpreter, Execution @@ -184,7 +183,7 @@ def chat_with_code( self, chat: List[Message], artifacts: Optional[Artifacts] = None, - test_multi_plan: bool = False, + test_multi_plan: bool = True, customized_tool_names: Optional[List[str]] = None, ) -> Tuple[List[Message], Artifacts]: """Chat with VisionAgent, it will use code to execute actions to accomplish