vision-agent 1.0.3__py3-none-any.whl → 1.0.4__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.
vision_agent/sim/sim.py CHANGED
@@ -58,7 +58,10 @@ def stella_embeddings(prompts: List[str]) -> List[np.ndarray]:
58
58
  }
59
59
  url = f"{_LND_API_URL_v2}/embeddings"
60
60
  vision_agent_api_key = get_vision_agent_api_key()
61
- headers = {"Authorization": f"Basic {vision_agent_api_key}"}
61
+ headers = {
62
+ "Authorization": f"Basic {vision_agent_api_key}",
63
+ "X-Source": "vision_agent",
64
+ }
62
65
  session = _create_requests_session(
63
66
  url=url,
64
67
  num_retry=3,
@@ -56,7 +56,10 @@ def send_inference_request(
56
56
  url = os.environ["TOOL_ENDPOINT_URL"]
57
57
 
58
58
  vision_agent_api_key = get_vision_agent_api_key()
59
- headers = {"Authorization": f"Basic {vision_agent_api_key}"}
59
+ headers = {
60
+ "Authorization": f"Basic {vision_agent_api_key}",
61
+ "X-Source": "vision_agent",
62
+ }
60
63
  if "TOOL_ENDPOINT_AUTH" in os.environ:
61
64
  headers["Authorization"] = os.environ["TOOL_ENDPOINT_AUTH"]
62
65
  headers.pop("apikey")
@@ -90,7 +93,10 @@ def send_task_inference_request(
90
93
  ) -> Any:
91
94
  url = f"{_LND_API_URL_v2}/{task_name}"
92
95
  vision_agent_api_key = get_vision_agent_api_key()
93
- headers = {"Authorization": f"Basic {vision_agent_api_key}"}
96
+ headers = {
97
+ "Authorization": f"Basic {vision_agent_api_key}",
98
+ "X-Source": "vision_agent",
99
+ }
94
100
  session = _create_requests_session(
95
101
  url=url,
96
102
  num_retry=3,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vision-agent
3
- Version: 1.0.3
3
+ Version: 1.0.4
4
4
  Summary: Toolset for Vision Agent
5
5
  Author: Landing AI
6
6
  Author-email: dev@landing.ai
@@ -32,7 +32,7 @@ vision_agent/models/agent_types.py,sha256=vBZ9-ns5lHDdFMO7ulCGGeZ6OwRo3gK4O3vN08
32
32
  vision_agent/models/lmm_types.py,sha256=v04h-NjbczHOIN8UWa1vvO5-1BDuZ4JQhD2mge1cXmw,305
33
33
  vision_agent/models/tools_types.py,sha256=8hYf2OZhI58gvf65KGaeGkt4EQ56nwLFqIQDPHioOBc,2339
34
34
  vision_agent/sim/__init__.py,sha256=Aouz6HEPPTYcLxR5_0fTYCL1OvPKAH1RMWAF90QXAlA,135
35
- vision_agent/sim/sim.py,sha256=xkioTFNTPl7wAnNgIFfNzswKK_9dRQ6G8OL8-K4CqsI,9893
35
+ vision_agent/sim/sim.py,sha256=8y060Ck7qOFJDw9k9Vl2xQYbLkNaTd6lP1YzbcwkkXc,9944
36
36
  vision_agent/tools/__init__.py,sha256=H7FWx0OXGVIjrSOTpNH-YwE4LBuOfThZTG-SHFpo_Z8,2576
37
37
  vision_agent/tools/meta_tools.py,sha256=DNRXHX9nZ1GBeqeLiq87sBshoe0aiZeYasETbG-9neI,24053
38
38
  vision_agent/tools/planner_tools.py,sha256=mlpUODpsN9sg-OKdsBm-6maK3eA97FUgAWm1etfgx4E,19553
@@ -43,11 +43,11 @@ vision_agent/utils/agent.py,sha256=2ifTP5QElItnr4YHOJR6L5P1PUzV0GhChTTqVxuVyQg,1
43
43
  vision_agent/utils/exceptions.py,sha256=zis8smCbdEylBVZBTVfEUfAh7Rb7cWV3MSPambu6FsQ,1837
44
44
  vision_agent/utils/execute.py,sha256=QAql6KC2uEhX1o_44mMA77lCmMUs0itaaGMFSfJBki8,21520
45
45
  vision_agent/utils/image_utils.py,sha256=bJM2mEvB6E__M9pxi74yQYzAiZ7mu3KE2ptyVrp5vzQ,12533
46
- vision_agent/utils/tools.py,sha256=lGELSZ3xmrUa1cHR4ey_f9gyDvWsIA8Q2R7G9LNGKiE,8036
46
+ vision_agent/utils/tools.py,sha256=_XGcF-Zd527BWNkGopBZeJ4j5ehreAkNsKQSBFrlvjw,8138
47
47
  vision_agent/utils/tools_doc.py,sha256=yFue6KSXoa_Z1ngCdBEc4SdPZOWF1rVLeaHu02I8Wis,2523
48
48
  vision_agent/utils/video.py,sha256=rjsQ1sKKisaQ6AVjJz0zd_G4g-ovRweS_rs4JEhenoI,5340
49
49
  vision_agent/utils/video_tracking.py,sha256=DZLFpNCuzuPJQzbQoVNcp-m4dKxgiKdCNM5QTh_zURE,12245
50
- vision_agent-1.0.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
51
- vision_agent-1.0.3.dist-info/METADATA,sha256=DQaYakxwHNuuKQvA7MpaPZ8oCZT5hdhfcCWzB_9zXI4,5940
52
- vision_agent-1.0.3.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
53
- vision_agent-1.0.3.dist-info/RECORD,,
50
+ vision_agent-1.0.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
51
+ vision_agent-1.0.4.dist-info/METADATA,sha256=-m5UEp_BinJMo7IQqv63frcNGAQ9dUICs8bGnpZD_Ac,5940
52
+ vision_agent-1.0.4.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
53
+ vision_agent-1.0.4.dist-info/RECORD,,