vision-agent 0.2.207__tar.gz → 0.2.208__tar.gz

{vision_agent-0.2.207 → vision_agent-0.2.208}/PKG-INFO

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

{vision_agent-0.2.207 → vision_agent-0.2.208}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.2.207"
+version = "0.2.208"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <dev@landing.ai>"]
 readme = "README.md"

{vision_agent-0.2.207 → vision_agent-0.2.208}/vision_agent/tools/tool_utils.py

@@ -41,11 +41,6 @@ def send_inference_request(
     metadata_payload: Optional[Dict[str, Any]] = None,
     is_form: bool = False,
 ) -> Any:
-    # TODO: runtime_tag and function_name should be metadata_payload and not included
-    # in the service payload
-    if runtime_tag := os.environ.get("RUNTIME_TAG", ""):
-        payload["runtime_tag"] = runtime_tag
-
     url = f"{_LND_API_URL_v2 if v2 else _LND_API_URL}/{endpoint_name}"
     if "TOOL_ENDPOINT_URL" in os.environ:
         url = os.environ["TOOL_ENDPOINT_URL"]
@@ -55,6 +50,9 @@ def send_inference_request(
         headers["Authorization"] = os.environ["TOOL_ENDPOINT_AUTH"]
         headers.pop("apikey")
 
+    if runtime_tag := os.environ.get("RUNTIME_TAG", ""):
+        headers["runtime_tag"] = runtime_tag
+
     session = _create_requests_session(
         url=url,
         num_retry=3,