From b58e48d674cae8f6ee61a15c2dcc29ec7c6cc700 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 15 Oct 2024 20:35:57 -0700 Subject: [PATCH] fixed type errors --- vision_agent/agent/vision_agent.py | 10 ++++++++++ vision_agent/tools/meta_tools.py | 2 +- 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 761c90dd..1bb00621 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -335,6 +335,8 @@ def __init__( verbosity (int): The verbosity level of the agent. local_artifacts_path (Optional[Union[str, Path]]): The path to the local artifacts file. + remote_artifacts_path (Optional[Union[str, Path]]): The path to the remote + artifacts file. callback_message (Optional[Callable[[Dict[str, Any]], None]]): Callback function to send intermediate update messages. code_interpreter (Optional[Union[str, CodeInterpreter]]): For string values @@ -643,6 +645,7 @@ def __init__( agent: Optional[LMM] = None, verbosity: int = 0, local_artifacts_path: Optional[Union[str, Path]] = None, + remote_artifacts_path: Optional[Union[str, Path]] = None, callback_message: Optional[Callable[[Dict[str, Any]], None]] = None, code_interpreter: Optional[Union[str, CodeInterpreter]] = None, ) -> None: @@ -654,6 +657,8 @@ def __init__( verbosity (int): The verbosity level of the agent. local_artifacts_path (Optional[Union[str, Path]]): The path to the local artifacts file. + remote_artifacts_path (Optional[Union[str, Path]]): The path to the remote + artifacts file. callback_message (Optional[Callable[[Dict[str, Any]], None]]): Callback function to send intermediate update messages. code_interpreter (Optional[Union[str, CodeInterpreter]]): For string values @@ -667,6 +672,7 @@ def __init__( agent, verbosity, local_artifacts_path, + remote_artifacts_path, callback_message, code_interpreter, ) @@ -678,6 +684,7 @@ def __init__( agent: Optional[LMM] = None, verbosity: int = 0, local_artifacts_path: Optional[Union[str, Path]] = None, + remote_artifacts_path: Optional[Union[str, Path]] = None, callback_message: Optional[Callable[[Dict[str, Any]], None]] = None, code_interpreter: Optional[Union[str, CodeInterpreter]] = None, ) -> None: @@ -689,6 +696,8 @@ def __init__( verbosity (int): The verbosity level of the agent. local_artifacts_path (Optional[Union[str, Path]]): The path to the local artifacts file. + remote_artifacts_path (Optional[Union[str, Path]]): The path to the remote + artifacts file. callback_message (Optional[Callable[[Dict[str, Any]], None]]): Callback function to send intermediate update messages. code_interpreter (Optional[Union[str, CodeInterpreter]]): For string values @@ -702,6 +711,7 @@ def __init__( agent, verbosity, local_artifacts_path, + remote_artifacts_path, callback_message, code_interpreter, ) diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py index d13e3731..ffbfc204 100644 --- a/vision_agent/tools/meta_tools.py +++ b/vision_agent/tools/meta_tools.py @@ -835,7 +835,7 @@ def extract_and_save_files_to_artifacts( files = {} for res in result.results: for format in res.formats(): - if format in ["png", "jpeg", "mp4"]: # type: ignore + if format in ["png", "jpeg", "mp4"]: if format == "png": data = base64.b64decode(res.png) if res.png is not None else None elif format == "jpeg":