diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py index 6489652f..dc910300 100644 --- a/vision_agent/tools/meta_tools.py +++ b/vision_agent/tools/meta_tools.py @@ -116,7 +116,9 @@ def show(self, uploaded_file_path: Optional[Union[str, Path]] = None) -> str: ) output_str = "[Artifacts loaded]\n" for k in self.artifacts.keys(): - output_str += f"Artifact name: {k}, loaded to path: {str(loaded_path / k)}\n" + output_str += ( + f"Artifact name: {k}, loaded to path: {str(loaded_path / k)}\n" + ) output_str += "[End of artifacts]\n" print(output_str) return output_str diff --git a/vision_agent/tools/tool_utils.py b/vision_agent/tools/tool_utils.py index 924b96e6..b35d6fef 100644 --- a/vision_agent/tools/tool_utils.py +++ b/vision_agent/tools/tool_utils.py @@ -1,6 +1,6 @@ -import os import inspect import logging +import os from base64 import b64encode from typing import Any, Callable, Dict, List, MutableMapping, Optional, Tuple diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index 698b9ffd..bc73a9ae 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -1808,7 +1808,9 @@ def save_image(image: np.ndarray, file_path: str) -> None: """ from IPython.display import display - if not isinstance(image, np.ndarray) or (image.shape[0] == 0 and image.shape[1] == 0): + if not isinstance(image, np.ndarray) or ( + image.shape[0] == 0 and image.shape[1] == 0 + ): raise ValueError("The image is not a valid NumPy array with shape (H, W, C)") pil_image = Image.fromarray(image.astype(np.uint8)).convert("RGB") @@ -1844,7 +1846,9 @@ def save_video( if not isinstance(frame, np.ndarray) or ( frame.shape[0] == 0 and frame.shape[1] == 0 ): - raise ValueError("The frame is not a valid NumPy array with shape (H, W, C)") + raise ValueError( + "The frame is not a valid NumPy array with shape (H, W, C)" + ) if output_video_path is None: output_video_path = tempfile.NamedTemporaryFile( diff --git a/vision_agent/tools/tools_types.py b/vision_agent/tools/tools_types.py index 1cc765b6..b2812fc0 100644 --- a/vision_agent/tools/tools_types.py +++ b/vision_agent/tools/tools_types.py @@ -1,6 +1,6 @@ from enum import Enum -from uuid import UUID from typing import List, Optional, Tuple, Union +from uuid import UUID from pydantic import BaseModel, ConfigDict, Field, SerializationInfo, field_serializer