vision-agent 0.2.170__py3-none-any.whl → 0.2.172__py3-none-any.whl

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.
@@ -149,11 +149,11 @@ def execute_code_action(
149
149
  result = code_interpreter.exec_isolation(
150
150
  BoilerplateCode.add_boilerplate(code, remote_path=artifact_remote_path)
151
151
  )
152
- extract_and_save_files_to_artifacts(artifacts, code)
153
152
 
154
153
  obs = str(result.logs)
155
154
  if result.error:
156
155
  obs += f"\n{result.error}"
156
+ extract_and_save_files_to_artifacts(artifacts, code, obs)
157
157
  return result, obs
158
158
 
159
159
 
@@ -182,6 +182,7 @@ def execute_user_code_action(
182
182
  )
183
183
  if user_result.error:
184
184
  user_obs += f"\n{user_result.error}"
185
+ extract_and_save_files_to_artifacts(artifacts, user_code_action, user_obs)
185
186
  return user_result, user_obs
186
187
 
187
188
 
@@ -451,11 +451,6 @@ def generate_vision_code(
451
451
  custom_tool_names=custom_tool_names,
452
452
  )
453
453
 
454
- # capture and save any files that were saved in the code to the artifacts
455
- extract_and_save_files_to_artifacts(
456
- artifacts, response["code"] + "\n" + response["test"]
457
- )
458
-
459
454
  redisplay_results(response["test_result"])
460
455
  code = response["code"]
461
456
  artifacts[name] = code
@@ -536,10 +531,6 @@ def edit_vision_code(
536
531
  test_multi_plan=False,
537
532
  custom_tool_names=custom_tool_names,
538
533
  )
539
- # capture and save any files that were saved in the code to the artifacts
540
- extract_and_save_files_to_artifacts(
541
- artifacts, response["code"] + "\n" + response["test"]
542
- )
543
534
 
544
535
  redisplay_results(response["test_result"])
545
536
  code = response["code"]
@@ -766,7 +757,9 @@ def use_object_detection_fine_tuning(
766
757
  return diff
767
758
 
768
759
 
769
- def extract_and_save_files_to_artifacts(artifacts: Artifacts, code: str) -> None:
760
+ def extract_and_save_files_to_artifacts(
761
+ artifacts: Artifacts, code: str, obs: str
762
+ ) -> None:
770
763
  """Extracts and saves files used in the code to the artifacts object.
771
764
 
772
765
  Parameters:
@@ -776,12 +769,16 @@ def extract_and_save_files_to_artifacts(artifacts: Artifacts, code: str) -> None
776
769
  try:
777
770
  response = extract_json(
778
771
  AnthropicLMM()( # type: ignore
779
- f"""You are a helpful AI assistant. Your job is to look at a snippet of code and return the file paths that are being saved in the file. Below is the code snippet:
772
+ f"""You are a helpful AI assistant. Your job is to look at a snippet of code and the output of running that code and return the file paths that are being saved in the file. Below is the code snippet:
780
773
 
781
774
  ```python
782
775
  {code}
783
776
  ```
784
777
 
778
+ ```output
779
+ {obs}
780
+ ```
781
+
785
782
  Return the file paths in the following JSON format:
786
783
  {{"file_paths": ["/path/to/image1.jpg", "/other/path/to/data.json"]}}"""
787
784
  )
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vision-agent
3
- Version: 0.2.170
3
+ Version: 0.2.172
4
4
  Summary: Toolset for Vision Agent
5
5
  Author: Landing AI
6
6
  Author-email: dev@landing.ai
@@ -41,7 +41,11 @@ Project-URL: repository, https://github.com/landing-ai/vision-agent
41
41
  Description-Content-Type: text/markdown
42
42
 
43
43
  <div align="center">
44
- <img alt="vision_agent" height="200px" src="https://github.com/landing-ai/vision-agent/blob/main/assets/logo.svg?raw=true">
44
+ <picture>
45
+ <source media="(prefers-color-scheme: dark)" srcset="https://github.com/landing-ai/vision-agent/blob/main/assets/logo_light.svg?raw=true">
46
+ <source media="(prefers-color-scheme: light)" srcset="https://github.com/landing-ai/vision-agent/blob/main/assets/logo_dark.svg?raw=true">
47
+ <img alt="VisionAgent" height="200px" src="https://github.com/landing-ai/vision-agent/blob/main/assets/logo_light.svg?raw=true">
48
+ </picture>
45
49
 
46
50
  [![](https://dcbadge.vercel.app/api/server/wPdN8RCYew?compact=true&style=flat)](https://discord.gg/wPdN8RCYew)
47
51
  ![ci_status](https://github.com/landing-ai/vision-agent/actions/workflows/ci_cd.yml/badge.svg)
@@ -2,7 +2,7 @@ vision_agent/__init__.py,sha256=EAb4-f9iyuEYkBrX4ag1syM8Syx8118_t0R6_C34M9w,57
2
2
  vision_agent/agent/__init__.py,sha256=RRMPhH8mgm_pCtEKiVFSjJyDi4lCr4F7k05AhK01xlM,436
3
3
  vision_agent/agent/agent.py,sha256=2cjIOxEuSJrqbfPXYoV0qER5ihXsPFCoEFJa4jpqan0,597
4
4
  vision_agent/agent/agent_utils.py,sha256=eSgg8CwWylX_erLTqTg2pVhEEgVkMLRrQfYRyJzI3so,5443
5
- vision_agent/agent/vision_agent.py,sha256=OBwj1RuJDXAFmnN6RRiG8mxZC6Nqfa7LddymvbI958w,26135
5
+ vision_agent/agent/vision_agent.py,sha256=zzXLIOi5fNLiqCGt18kfOI4OQPq1qhEeKanfrzl7Qj0,26223
6
6
  vision_agent/agent/vision_agent_coder.py,sha256=aVkl0b9LKvy-auuHGYSag-ixYnue0iRQqD1PYLPBR-s,29312
7
7
  vision_agent/agent/vision_agent_coder_prompts.py,sha256=gPLVXQMNSzYnQYpNm0wlH_5FPkOTaFDV24bqzK3jQ40,12221
8
8
  vision_agent/agent/vision_agent_planner.py,sha256=mjmnXG9CvYf_ZA7ZJ3ri4H-2U_Km55gF1sZYRSOlxpY,19027
@@ -17,7 +17,7 @@ vision_agent/lmm/__init__.py,sha256=jyY1sJb_tYKg5-Wzs3p1lvwFkc-aUNZfMcLy3TOC4Zg,
17
17
  vision_agent/lmm/lmm.py,sha256=B5ClgwvbybVCWkf9opDMLjTtJZemUU4KUkQoRxGh43I,16787
18
18
  vision_agent/lmm/types.py,sha256=ZEXR_ptBL0ZwDMTDYkgxUCmSZFmBYPQd2jreNzr_8UY,221
19
19
  vision_agent/tools/__init__.py,sha256=u-vS5iORB4ccvxoAjbtpvhTALDhXGilcATIq1_eZhKo,2332
20
- vision_agent/tools/meta_tools.py,sha256=7XM3VP4EW4Dtg_Hvoov_laOAEaZLdSGOeA-iPb7CimU,28315
20
+ vision_agent/tools/meta_tools.py,sha256=zcqp3POpGcS4cfceMih0AD1c6MwS_PFLLA6EjTXhonM,28013
21
21
  vision_agent/tools/prompts.py,sha256=V1z4YJLXZuUl_iZ5rY0M5hHc_2tmMEUKr0WocXKGt4E,1430
22
22
  vision_agent/tools/tool_utils.py,sha256=VPGqGJ2ZYEJA6AW7K9X7hQv6vRlMtAQcybE4izdToCw,8196
23
23
  vision_agent/tools/tools.py,sha256=iKsBZxJ5--xWK-mqgZ1jbX_bfGS5HmAp-VRZ69m9yPg,77921
@@ -29,7 +29,7 @@ vision_agent/utils/image_utils.py,sha256=rm9GfXvD4JrjnqKrP_f2gfq4SzmqYC0IdC1kKwd
29
29
  vision_agent/utils/sim.py,sha256=ZuSS07TUXFGjipmiQoY8TKRmSes7XXCdtU9PI8PC1sw,5609
30
30
  vision_agent/utils/type_defs.py,sha256=BE12s3JNQy36QvauXHjwyeffVh5enfcvd4vTzSwvEZI,1384
31
31
  vision_agent/utils/video.py,sha256=fOPR48-SuwMbE5eB5rc2F7lVo6k1mVHn26eEJ0QCslc,5602
32
- vision_agent-0.2.170.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
33
- vision_agent-0.2.170.dist-info/METADATA,sha256=7jdWi4AYPZKvspmVd8KHJcScWq0MtB1Fpv8y0UQFjLw,18010
34
- vision_agent-0.2.170.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
35
- vision_agent-0.2.170.dist-info/RECORD,,
32
+ vision_agent-0.2.172.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
33
+ vision_agent-0.2.172.dist-info/METADATA,sha256=igL_7hfPKB2VSQgOobgmqVSLtmWOltFUsKE58MNYQDM,18342
34
+ vision_agent-0.2.172.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
35
+ vision_agent-0.2.172.dist-info/RECORD,,