From 35dd2b1d4d545754e4bf03f1469360a99e3b0aee Mon Sep 17 00:00:00 2001
From: Asia <92344512+AsiaCao@users.noreply.github.com>
Date: Thu, 9 May 2024 16:50:22 -0700
Subject: [PATCH] Support identify internal calls via runtime tag (#79)

---
 vision_agent/tools/tool_utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/vision_agent/tools/tool_utils.py b/vision_agent/tools/tool_utils.py
index 480a7493..f0304653 100644
--- a/vision_agent/tools/tool_utils.py
+++ b/vision_agent/tools/tool_utils.py
@@ -1,4 +1,5 @@
 import logging
+import os
 from typing import Any, Dict
 
 import requests
@@ -13,11 +14,14 @@
 def _send_inference_request(
     payload: Dict[str, Any], endpoint_name: str
 ) -> Dict[str, Any]:
+    # runtime_tag is used to differentiate different internal callers
+    runtime_tag = os.environ.get("RUNTIME_TAG", "")
     res = requests.post(
         f"{_LND_API_URL}/model/{endpoint_name}",
         headers={
             "Content-Type": "application/json",
             "apikey": _LND_API_KEY,
+            "runtime-tag": runtime_tag,
         },
         json=payload,
     )