vision-agent 0.2.221__py3-none-any.whl → 0.2.222__py3-none-any.whl

vision_agent/utils/sim.py

@@ -12,7 +12,12 @@ import requests
 from openai import AzureOpenAI, OpenAI
 from scipy.spatial.distance import cosine  # type: ignore
 
-from vision_agent.tools.tools import TOOLS_DF, stella_embeddings
+from vision_agent.tools.tool_utils import (
+    _LND_API_KEY,
+    _create_requests_session,
+    _LND_API_URL_v2,
+)
+from vision_agent.tools.tools import TOOLS_DF
 
 
 @lru_cache(maxsize=1)
@@ -45,6 +50,24 @@ def load_cached_sim(
     return sim
 
 
+def stella_embeddings(prompts: List[str]) -> List[np.ndarray]:
+    # implement stella here so we don't display embeddings which can be costly
+    payload = {
+        "input": prompts,
+        "model": "stella1.5b",
+    }
+    url = f"{_LND_API_URL_v2}/embeddings"
+    headers = {"apikey": _LND_API_KEY}
+    session = _create_requests_session(
+        url=url,
+        num_retry=3,
+        headers=headers,
+    )
+    response = session.post(url, data=payload).json()
+    data = response["data"]
+    return [np.array(d["embedding"]) for d in data]
+
+
 class Sim:
     def __init__(
         self,

{vision_agent-0.2.221.dist-info → vision_agent-0.2.222.dist-info}/METADATA

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

{vision_agent-0.2.221.dist-info → vision_agent-0.2.222.dist-info}/RECORD

@@ -1,5 +1,5 @@
-vision_agent/.sim_tools/df.csv,sha256=aTpXEOmdIhomzCB4c_qprECxrxTdudet3FK_C3TTzKw,42927
-vision_agent/.sim_tools/embs.npy,sha256=22NYWI7NswR367TwyD3s8I8td7ai2ZumgRLWsE6ajtM,270464
+vision_agent/.sim_tools/df.csv,sha256=1cpUFKN48Iuq6HvaG5OhbHs2RghESicb3ouKVTLhm-s,43360
+vision_agent/.sim_tools/embs.npy,sha256=Nji50P_8aV0hhyxp-Kfh_YmXAFWTHSwwBGrNJWitHPU,270464
 vision_agent/__init__.py,sha256=EAb4-f9iyuEYkBrX4ag1syM8Syx8118_t0R6_C34M9w,57
 vision_agent/agent/README.md,sha256=Q4w7FWw38qaWosQYAZ7NqWx8Q5XzuWrlv7nLhjUd1-8,5527
 vision_agent/agent/__init__.py,sha256=M8CffavdIh8Zh-skznLHIaQkYGCGK7vk4dq1FaVkbs4,617
@@ -13,7 +13,7 @@ vision_agent/agent/vision_agent_coder_prompts_v2.py,sha256=9v5HwbNidSzYUEFl6ZMni
 vision_agent/agent/vision_agent_coder_v2.py,sha256=G3I8O89gzE2VczQGPWV149aYaOjbbfB1lmgGuwFWvo4,16118
 vision_agent/agent/vision_agent_planner.py,sha256=fFzjNkZBKkh8Y_oS06ATI4qz31xmIJvixb_tV1kX8KA,18590
 vision_agent/agent/vision_agent_planner_prompts.py,sha256=mn9NlZpRkW4XAvlNuMZwIs1ieHCFds5aYZJ55WXupZY,6733
-vision_agent/agent/vision_agent_planner_prompts_v2.py,sha256=TENjNF_mQHw6RHic24TrP5b7-Q5KWiJardtP9_hitdw,33420
+vision_agent/agent/vision_agent_planner_prompts_v2.py,sha256=lzfJFvBYW_-Ue4OevgljI8bAQxgKC4Rdv5SmP6UsAxE,34102
 vision_agent/agent/vision_agent_planner_v2.py,sha256=vvxfmGydBIKB8CtNSAJyPvdEXkG7nIO5-Hs2SjNc48Y,20465
 vision_agent/agent/vision_agent_prompts.py,sha256=NtGdCfzzilCRtscKALC9FK55d1h4CBpMnbhLzg0PYlc,13772
 vision_agent/agent/vision_agent_prompts_v2.py,sha256=-vCWat-ARlCOOOeIDIFhg-kcwRRwjTXYEwsvvqPeaCs,1972
@@ -26,21 +26,21 @@ vision_agent/fonts/default_font_ch_en.ttf,sha256=1YM0Z3XqLDjSNbF7ihQFSAIUdjF9m1r
 vision_agent/lmm/__init__.py,sha256=jyY1sJb_tYKg5-Wzs3p1lvwFkc-aUNZfMcLy3TOC4Zg,100
 vision_agent/lmm/lmm.py,sha256=x_nIyDNDZwq4-pfjnJTmcyyJZ2_B7TjkA5jZp88YVO8,17103
 vision_agent/lmm/types.py,sha256=ZEXR_ptBL0ZwDMTDYkgxUCmSZFmBYPQd2jreNzr_8UY,221
-vision_agent/tools/__init__.py,sha256=XrViKgxqMjY1Ep3QCBootXqftDBlrtH6M7qIVXOq_MY,2809
+vision_agent/tools/__init__.py,sha256=15O7eQVn0bitmzUO5OxKdA618PoiLt6Z02gmKsSNMFM,2765
 vision_agent/tools/meta_tools.py,sha256=TPeS7QWnc_PmmU_ndiDT03dXbQ5yDSP33E7U8cSj7Ls,28660
 vision_agent/tools/planner_tools.py,sha256=CvaJ2vGM8O_CYvsoSk1avxAMqpIu3tv4C2bY0p1X-X4,13519
 vision_agent/tools/prompts.py,sha256=V1z4YJLXZuUl_iZ5rY0M5hHc_2tmMEUKr0WocXKGt4E,1430
 vision_agent/tools/tool_utils.py,sha256=LAnrb_nY6PNVamqJahRN-J0cuOy4gsKvCtSuXJf0RsI,10075
-vision_agent/tools/tools.py,sha256=MctcjA3IB8ySO0d-IcTkdOmwpGiPRnamcFE4M0XWp8U,96408
+vision_agent/tools/tools.py,sha256=46XafZnutSNmis6qa9BFc_QVzo6056lY3BKb_1N2ot0,91119
 vision_agent/tools/tools_types.py,sha256=8hYf2OZhI58gvf65KGaeGkt4EQ56nwLFqIQDPHioOBc,2339
 vision_agent/utils/__init__.py,sha256=QKk4zVjMwGxQI0MQ-aZZA50N-qItxRY4EB9CwQkZ2HY,185
 vision_agent/utils/exceptions.py,sha256=booSPSuoULF7OXRr_YbC4dtKt6gM_HyiFQHBuaW86C4,2052
 vision_agent/utils/execute.py,sha256=vOEP5Ys7S2lc0_7pOJbgk7OaWi85hrCNu9_8Bo3zk6I,29356
 vision_agent/utils/image_utils.py,sha256=z_ONgcza125B10NkoGwPOzXnL470bpTWZbkB16NeeH0,12188
-vision_agent/utils/sim.py,sha256=u6a04nOc34FJcpMN0mb9eZImDIfNGYQscHWN53Hav6I,8482
+vision_agent/utils/sim.py,sha256=znsInUDrsyBi3OlgAlV3rDn5UQQRfJAWXTXm7D7eJA8,9125
 vision_agent/utils/type_defs.py,sha256=BE12s3JNQy36QvauXHjwyeffVh5enfcvd4vTzSwvEZI,1384
 vision_agent/utils/video.py,sha256=e1VwKhXzzlC5LcFMyrcQYrPnpnX4wxDpnQ-76sB4jgM,6001
-vision_agent-0.2.221.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vision_agent-0.2.221.dist-info/METADATA,sha256=kzNQBuLO-sH4LWcrhWb-2JzgynNoAziMRCRyOecb7XY,20039
-vision_agent-0.2.221.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-vision_agent-0.2.221.dist-info/RECORD,,
+vision_agent-0.2.222.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vision_agent-0.2.222.dist-info/METADATA,sha256=Rcgt4WGB0f0ke8f-leTCCvixPfvznUb-TuTxniecTLk,20039
+vision_agent-0.2.222.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+vision_agent-0.2.222.dist-info/RECORD,,