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, )