error handling

landing-ai · Mar 11, 2024 · e4f3b7e · e4f3b7e
1 parent 0c9d74b
commit e4f3b7e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -1,3 +1,4 @@
+import logging
 from abc import ABC, abstractmethod
 from pathlib import Path
 from typing import Any, Dict, List, Union, cast
@@ -7,6 +8,8 @@
 
 from vision_agent.image_utils import convert_to_b64
 
+_LOGGER = logging.getLogger(__name__)
+
 
 class ImageTool(ABC):
  @abstractmethod
@@ -56,8 +59,8 @@ def __call__(self, image: Union[str, Path, ImageType]) -> List[Dict]:
  json=data,
  )
  resp_json: Dict[str, Any] = res.json()
- # if resp_json["statusCode"] != 200:
- # _LOGGER.error(f"Request failed: {resp_json['data']}")
+ if resp_json["statusCode"] != 200:
+   _LOGGER.error(f"Request failed: {resp_json['data']}")
  return cast(str, resp_json["data"])