vision-agent 0.2.61__tar.gz → 0.2.63__tar.gz

{vision_agent-0.2.61 → vision_agent-0.2.63}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.61
+Version: 0.2.63
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai
@@ -242,7 +242,6 @@ You can then run Vision Agent using the Azure OpenAI models:
 
 ```python
 import vision_agent as va
-import vision_agent.tools as T
 agent = va.agent.AzureVisionAgent()
 ```
 

{vision_agent-0.2.61 → vision_agent-0.2.63}/README.md

@@ -206,6 +206,5 @@ You can then run Vision Agent using the Azure OpenAI models:
 
 ```python
 import vision_agent as va
-import vision_agent.tools as T
 agent = va.agent.AzureVisionAgent()
 ```

{vision_agent-0.2.61 → vision_agent-0.2.63}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.2.61"
+version = "0.2.63"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <dev@landing.ai>"]
 readme = "README.md"

{vision_agent-0.2.61 → vision_agent-0.2.63}/vision_agent/agent/vision_agent.py

@@ -97,6 +97,7 @@ def extract_json(json_str: str) -> Dict[str, Any]:
     try:
         json_dict = json.loads(json_str)
     except json.JSONDecodeError:
+        input_json_str = json_str
         if "```json" in json_str:
             json_str = json_str[json_str.find("```json") + len("```json") :]
             json_str = json_str[: json_str.find("```")]
@@ -104,7 +105,12 @@ def extract_json(json_str: str) -> Dict[str, Any]:
             json_str = json_str[json_str.find("```") + len("```") :]
             # get the last ``` not one from an intermediate string
             json_str = json_str[: json_str.find("}```")]
-        json_dict = json.loads(json_str)
+        try:
+            json_dict = json.loads(json_str)
+        except json.JSONDecodeError as e:
+            error_msg = f"Could not extract JSON from the given str: {json_str}.\nFunction input:\n{input_json_str}"
+            _LOGGER.exception(error_msg)
+            raise ValueError(error_msg) from e
     return json_dict  # type: ignore