diff --git a/.github/workflows/ci_cd.yml b/.github/workflows/ci_cd.yml index 56c31d19..75c64236 100644 --- a/.github/workflows/ci_cd.yml +++ b/.github/workflows/ci_cd.yml @@ -12,6 +12,8 @@ jobs: python-version: [3.9, 3.11] os: [ ubuntu-22.04, windows-2022, macos-12 ] runs-on: ${{ matrix.os }} + env: + RUNTIME_TAG: ci_job steps: - uses: actions/checkout@v3 - uses: actions/setup-python@v4 diff --git a/vision_agent/tools/tool_utils.py b/vision_agent/tools/tool_utils.py index f0304653..b87689c4 100644 --- a/vision_agent/tools/tool_utils.py +++ b/vision_agent/tools/tool_utils.py @@ -14,14 +14,13 @@ 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", "") + if runtime_tag := os.environ.get("RUNTIME_TAG", ""): + payload["runtime_tag"] = 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, )