vision-agent 0.2.79__py3-none-any.whl → 0.2.80__py3-none-any.whl

vision_agent/utils/execute.py

@@ -112,6 +112,11 @@ class Result:
         self.raw = copy.deepcopy(data)
 
         self.text = data.pop(MimeType.TEXT_PLAIN, None)
+        if self.text and (self.text.startswith("'") and self.text.endswith("'")):
+            # This is a workaround for the issue that str result is wrapped with single quotes by notebook.
+            # E.g. input text: "'flower'". what we want: "flower"
+            self.text = self.text[1:-1]
+
         self.html = data.pop(MimeType.TEXT_HTML, None)
         self.markdown = data.pop(MimeType.TEXT_MARKDOWN, None)
         self.svg = data.pop(MimeType.IMAGE_SVG, None)

vision_agent/utils/sim.py

@@ -1,4 +1,5 @@
 import os
+from functools import lru_cache
 from pathlib import Path
 from typing import Dict, List, Optional, Sequence, Union
 
@@ -33,11 +34,7 @@ class Sim:
             model: str: The model to use for embeddings.
         """
         self.df = df
-        if not api_key:
-            self.client = OpenAI()
-        else:
-            self.client = OpenAI(api_key=api_key)
-
+        self.client = OpenAI(api_key=api_key)
         self.model = model
         if "embs" not in df.columns and sim_key is None:
             raise ValueError("key is required if no column 'embs' is present.")
@@ -57,6 +54,7 @@ class Sim:
         df = df.drop("embs", axis=1)
         df.to_csv(sim_file / "df.csv", index=False)
 
+    @lru_cache(maxsize=256)
     def top_k(
         self, query: str, k: int = 5, thresh: Optional[float] = None
     ) -> Sequence[Dict]:

{vision_agent-0.2.79.dist-info → vision_agent-0.2.80.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.79
+Version: 0.2.80
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai

{vision_agent-0.2.79.dist-info → vision_agent-0.2.80.dist-info}/RECORD

@@ -13,12 +13,12 @@ vision_agent/tools/tool_utils.py,sha256=6z0jrvUnesJEFqDHZoAvbXPic8rzh0KfILL07tu0
 vision_agent/tools/tools.py,sha256=TkZqNYX-ocwdaCdXd6c6tysSa_HX2y6Nrgl4JKni4IQ,43661
 vision_agent/utils/__init__.py,sha256=CW84HnhqI6XQVuxf2KifkLnSuO7EOhmuL09-gAymAak,219
 vision_agent/utils/exceptions.py,sha256=JGiFPLAYnPpEvHfueQuJotv3rCkS-A7UAwEym56MmHM,1359
-vision_agent/utils/execute.py,sha256=DWFNslQVl4kpBrGJs8YGeUkOsZFykvdYNAo_DFvFIfQ,23029
+vision_agent/utils/execute.py,sha256=wt7YNVn4jEzU-9ljQg2RYTcFV8qNSrS30QN3tNDRct4,23326
 vision_agent/utils/image_utils.py,sha256=_cdiS5YrLzqkq_ZgFUO897m5M4_SCIThwUy4lOklfB8,7700
-vision_agent/utils/sim.py,sha256=ci6Eta73dDgLP1Ajtknbgmf1g8aAvBHqlVQvBuLMKXQ,4427
+vision_agent/utils/sim.py,sha256=1HTaiVaBiKeyXIy21IYGXlPw0TipOyw9FPOJDfyLI94,4409
 vision_agent/utils/type_defs.py,sha256=QeQRRIlklZMWzxROcCn5ELxP89nYdXGydy1rAiSpZZw,1384
 vision_agent/utils/video.py,sha256=rNmU9KEIkZB5-EztZNlUiKYN0mm_55A_2VGUM0QpqLA,8779
-vision_agent-0.2.79.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vision_agent-0.2.79.dist-info/METADATA,sha256=gWTSwXsZO3euICHmSJE99yiTXqnrvEf_93SK5Ry0qwU,9433
-vision_agent-0.2.79.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-vision_agent-0.2.79.dist-info/RECORD,,
+vision_agent-0.2.80.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vision_agent-0.2.80.dist-info/METADATA,sha256=xflnjK1RKkn82YM_yR21jEvLqlq0RpOm1JEgETYZXos,9433
+vision_agent-0.2.80.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+vision_agent-0.2.80.dist-info/RECORD,,