vision-agent 0.2.7__tar.gz → 0.2.8__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (25) hide show
  1. {vision_agent-0.2.7 → vision_agent-0.2.8}/PKG-INFO +1 -1
  2. {vision_agent-0.2.7 → vision_agent-0.2.8}/pyproject.toml +1 -1
  3. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/vision_agent.py +5 -2
  4. {vision_agent-0.2.7 → vision_agent-0.2.8}/LICENSE +0 -0
  5. {vision_agent-0.2.7 → vision_agent-0.2.8}/README.md +0 -0
  6. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/__init__.py +0 -0
  7. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/__init__.py +0 -0
  8. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/agent.py +0 -0
  9. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/easytool.py +0 -0
  10. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/easytool_prompts.py +0 -0
  11. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/reflexion.py +0 -0
  12. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/reflexion_prompts.py +0 -0
  13. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/vision_agent_prompts.py +0 -0
  14. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/fonts/__init__.py +0 -0
  15. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/fonts/default_font_ch_en.ttf +0 -0
  16. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/image_utils.py +0 -0
  17. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/llm/__init__.py +0 -0
  18. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/llm/llm.py +0 -0
  19. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/lmm/__init__.py +0 -0
  20. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/lmm/lmm.py +0 -0
  21. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/tools/__init__.py +0 -0
  22. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/tools/prompts.py +0 -0
  23. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/tools/tools.py +0 -0
  24. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/tools/video.py +0 -0
  25. {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/type_defs.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vision-agent
3
- Version: 0.2.7
3
+ Version: 0.2.8
4
4
  Summary: Toolset for Vision Agent
5
5
  Author: Landing AI
6
6
  Author-email: dev@landing.ai
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
4
4
 
5
5
  [tool.poetry]
6
6
  name = "vision-agent"
7
- version = "0.2.7"
7
+ version = "0.2.8"
8
8
  description = "Toolset for Vision Agent"
9
9
  authors = ["Landing AI <dev@landing.ai>"]
10
10
  readme = "README.md"
@@ -346,7 +346,9 @@ def _handle_viz_tools(
346
346
  # 2. return a dictionary but not have the necessary keys
347
347
 
348
348
  if not isinstance(call_result, dict) or (
349
- "bboxes" not in call_result and "heat_map" not in call_result
349
+ "bboxes" not in call_result
350
+ and "mask" not in call_result
351
+ and "heat_map" not in call_result
350
352
  ):
351
353
  return image_to_data
352
354
 
@@ -366,7 +368,8 @@ def _handle_viz_tools(
366
368
  image_to_data[image]["scores"].extend(call_result.get("scores", []))
367
369
  image_to_data[image]["masks"].extend(call_result.get("masks", []))
368
370
  # only single heatmap is returned
369
- image_to_data[image]["heat_map"].append(call_result.get("heat_map", []))
371
+ if "heat_map" in call_result:
372
+ image_to_data[image]["heat_map"].append(call_result["heat_map"])
370
373
  if "mask_shape" in call_result:
371
374
  image_to_data[image]["mask_shape"] = call_result["mask_shape"]
372
375
 
File without changes
File without changes