fixed type errors

landing-ai · Oct 16, 2024 · b58e48d · b58e48d
1 parent 802c7e2
commit b58e48d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git 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
@@ -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":