vision-agent 0.2.160__py3-none-any.whl → 0.2.161__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.
@@ -197,6 +197,7 @@ class VisionAgent(Agent):
197
197
  local_artifacts_path: Optional[Union[str, Path]] = None,
198
198
  code_sandbox_runtime: Optional[str] = None,
199
199
  callback_message: Optional[Callable[[Dict[str, Any]], None]] = None,
200
+ code_interpreter: Optional[CodeInterpreter] = None,
200
201
  ) -> None:
201
202
  """Initialize the VisionAgent.
202
203
 
@@ -207,12 +208,14 @@ class VisionAgent(Agent):
207
208
  local_artifacts_path (Optional[Union[str, Path]]): The path to the local
208
209
  artifacts file.
209
210
  code_sandbox_runtime (Optional[str]): The code sandbox runtime to use.
211
+ code_interpreter (Optional[CodeInterpreter]): if not None, use this CodeInterpreter
210
212
  """
211
213
 
212
214
  self.agent = AnthropicLMM(temperature=0.0) if agent is None else agent
213
215
  self.max_iterations = 12
214
216
  self.verbosity = verbosity
215
217
  self.code_sandbox_runtime = code_sandbox_runtime
218
+ self.code_interpreter = code_interpreter
216
219
  self.callback_message = callback_message
217
220
  if self.verbosity >= 1:
218
221
  _LOGGER.setLevel(logging.INFO)
@@ -284,9 +287,14 @@ class VisionAgent(Agent):
284
287
  # this is setting remote artifacts path
285
288
  artifacts = Artifacts(WORKSPACE / "artifacts.pkl")
286
289
 
287
- with CodeInterpreterFactory.new_instance(
288
- code_sandbox_runtime=self.code_sandbox_runtime,
289
- ) as code_interpreter:
290
+ code_interpreter = (
291
+ self.code_interpreter
292
+ if self.code_interpreter is not None
293
+ else CodeInterpreterFactory.new_instance(
294
+ code_sandbox_runtime=self.code_sandbox_runtime,
295
+ )
296
+ )
297
+ with code_interpreter:
290
298
  orig_chat = copy.deepcopy(chat)
291
299
  int_chat = copy.deepcopy(chat)
292
300
  last_user_message = chat[-1]
@@ -28,6 +28,7 @@ CURRENT_FILE = None
28
28
  CURRENT_LINE = 0
29
29
  DEFAULT_WINDOW_SIZE = 100
30
30
  ZMQ_PORT = os.environ.get("ZMQ_PORT", None)
31
+ VERBOSITY = os.environ.get("VERBOSITY", 0)
31
32
 
32
33
 
33
34
  def report_progress_callback(port: int, inp: Dict[str, Any]) -> None:
@@ -375,7 +376,7 @@ def generate_vision_code(
375
376
  )
376
377
  )
377
378
  else:
378
- agent = va.agent.VisionAgentCoder()
379
+ agent = va.agent.VisionAgentCoder(verbosity=int(VERBOSITY))
379
380
 
380
381
  fixed_chat: List[Message] = [{"role": "user", "content": chat, "media": media}]
381
382
  response = agent.chat_with_workflow(
@@ -438,7 +439,7 @@ def edit_vision_code(
438
439
  return dogs
439
440
  """
440
441
 
441
- agent = va.agent.VisionAgentCoder()
442
+ agent = va.agent.VisionAgentCoder(verbosity=int(VERBOSITY))
442
443
  if name not in artifacts:
443
444
  print(f"[Artifact {name} does not exist]")
444
445
  return f"[Artifact {name} does not exist]"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vision-agent
3
- Version: 0.2.160
3
+ Version: 0.2.161
4
4
  Summary: Toolset for Vision Agent
5
5
  Author: Landing AI
6
6
  Author-email: dev@landing.ai
@@ -2,7 +2,7 @@ vision_agent/__init__.py,sha256=EAb4-f9iyuEYkBrX4ag1syM8Syx8118_t0R6_C34M9w,57
2
2
  vision_agent/agent/__init__.py,sha256=NF2LABqHixLvbsOIO-fe-VKZ7awvShLtcT0oQT4eWtI,235
3
3
  vision_agent/agent/agent.py,sha256=2cjIOxEuSJrqbfPXYoV0qER5ihXsPFCoEFJa4jpqan0,597
4
4
  vision_agent/agent/agent_utils.py,sha256=eIpLz2NunEqEsBBrECJaD34-2uY0bsFNnW-XKfqqohs,2518
5
- vision_agent/agent/vision_agent.py,sha256=etqyLMZHJJz_A6tkonoYGlYvFvEW0uUHs5D1gsYwkSs,20412
5
+ vision_agent/agent/vision_agent.py,sha256=x0oSuQk-rcERUUdZp29FB77Ua-eyQD3fLi_UfmsBTV0,20761
6
6
  vision_agent/agent/vision_agent_coder.py,sha256=2ZoGikn2nakGDfs20XRshZjQUyvbw6l47UhExJAYkqI,38515
7
7
  vision_agent/agent/vision_agent_coder_prompts.py,sha256=BmbTMhth4v1qLexuoSeyo47QQ0kPQvL1pLbCJHMsWDw,18910
8
8
  vision_agent/agent/vision_agent_prompts.py,sha256=LZ9Bnx7ZFkqbNOMqwfdiWZU4niND9Z1ArcFHNSn_jzA,11187
@@ -15,7 +15,7 @@ vision_agent/lmm/__init__.py,sha256=jyY1sJb_tYKg5-Wzs3p1lvwFkc-aUNZfMcLy3TOC4Zg,
15
15
  vision_agent/lmm/lmm.py,sha256=B5ClgwvbybVCWkf9opDMLjTtJZemUU4KUkQoRxGh43I,16787
16
16
  vision_agent/lmm/types.py,sha256=ZEXR_ptBL0ZwDMTDYkgxUCmSZFmBYPQd2jreNzr_8UY,221
17
17
  vision_agent/tools/__init__.py,sha256=PLVbfTMjKxQlHIRWnq9b785W9a52AXQS_tOa0tkQ0ZY,2420
18
- vision_agent/tools/meta_tools.py,sha256=TG4vXN7W2M3rOJhzVsCrDAhbctd3RJcuOmbAeXsk2Sw,25798
18
+ vision_agent/tools/meta_tools.py,sha256=BF5-fVshLhhpck5lJErKxnfPu9YxudBhR7ar_qA9Mjo,25889
19
19
  vision_agent/tools/prompts.py,sha256=V1z4YJLXZuUl_iZ5rY0M5hHc_2tmMEUKr0WocXKGt4E,1430
20
20
  vision_agent/tools/tool_utils.py,sha256=VPGqGJ2ZYEJA6AW7K9X7hQv6vRlMtAQcybE4izdToCw,8196
21
21
  vision_agent/tools/tools.py,sha256=vS1yCk3Fza9eYOTHPFwwroo_ULdw2ztMQMb81x1U5f8,78524
@@ -27,7 +27,7 @@ vision_agent/utils/image_utils.py,sha256=rm9GfXvD4JrjnqKrP_f2gfq4SzmqYC0IdC1kKwd
27
27
  vision_agent/utils/sim.py,sha256=ZuSS07TUXFGjipmiQoY8TKRmSes7XXCdtU9PI8PC1sw,5609
28
28
  vision_agent/utils/type_defs.py,sha256=BE12s3JNQy36QvauXHjwyeffVh5enfcvd4vTzSwvEZI,1384
29
29
  vision_agent/utils/video.py,sha256=xbMEoRk13l4fHeQlbvMQhLCn8RNndYmsDhUf01TUeR8,4781
30
- vision_agent-0.2.160.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
31
- vision_agent-0.2.160.dist-info/METADATA,sha256=yfWiwd_dUVN2AQOkCSDM9A41NPqACMoj8NaVgOHOmNQ,17753
32
- vision_agent-0.2.160.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
33
- vision_agent-0.2.160.dist-info/RECORD,,
30
+ vision_agent-0.2.161.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
31
+ vision_agent-0.2.161.dist-info/METADATA,sha256=81mQ74IJal478wgLMjkneAVg0kE89VVjX9Wa_hL0lMo,17753
32
+ vision_agent-0.2.161.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
33
+ vision_agent-0.2.161.dist-info/RECORD,,