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.
- {vision_agent-0.2.7 → vision_agent-0.2.8}/PKG-INFO +1 -1
- {vision_agent-0.2.7 → vision_agent-0.2.8}/pyproject.toml +1 -1
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/vision_agent.py +5 -2
- {vision_agent-0.2.7 → vision_agent-0.2.8}/LICENSE +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/README.md +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/__init__.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/__init__.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/agent.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/easytool.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/easytool_prompts.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/reflexion.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/reflexion_prompts.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/agent/vision_agent_prompts.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/fonts/__init__.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/fonts/default_font_ch_en.ttf +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/image_utils.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/llm/__init__.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/llm/llm.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/lmm/__init__.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/lmm/lmm.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/tools/__init__.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/tools/prompts.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/tools/tools.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/tools/video.py +0 -0
- {vision_agent-0.2.7 → vision_agent-0.2.8}/vision_agent/type_defs.py +0 -0
@@ -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
|
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
|
-
|
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
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|