From d232b008bc8824b80ed8965174dc7039a6473ffe Mon Sep 17 00:00:00 2001 From: moe-berri Date: Mon, 5 Oct 2026 18:39:38 -0700 Subject: [PATCH] test(lens): use current worker protocol in billing integration (#44760) --- tests/integration/spend/test_lens_billing.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/integration/spend/test_lens_billing.py b/tests/integration/spend/test_lens_billing.py index 69a1899ba6d..e1e0e97c8fd 100644 --- a/tests/integration/spend/test_lens_billing.py +++ b/tests/integration/spend/test_lens_billing.py @@ -6,6 +6,7 @@ from typing import Final import pytest +from litellm.proxy.lens.release import PROTOCOL_VERSION from tests.integration._support.client import Gateway, eventually, object_value, string_value from tests.integration._support.database import read_rows, write_rows from tests.integration._support.process import owned_proxy @@ -71,7 +72,7 @@ def test_lens_bills_selected_key_and_rechecks_its_permissions( pool.map( lambda _: isolated.request( "POST", - "/lens/worker/claim?protocol_version=4&worker_release=" + RELEASE_TAG, + f"/lens/worker/claim?protocol_version={PROTOCOL_VERSION}&worker_release={RELEASE_TAG}", {}, key=worker_key, ), @@ -206,7 +207,7 @@ def test_worker_spend_logs_do_not_expose_investigation_content( scenario.cleanups.callback(delete_lens, lens_id) worker_token: Final = string_value(worker["token"]) claim: Final = isolated.post( - "/lens/worker/claim?protocol_version=4&worker_release=" + RELEASE_TAG, {}, key=worker_token + f"/lens/worker/claim?protocol_version={PROTOCOL_VERSION}&worker_release={RELEASE_TAG}", {}, key=worker_token ) job_id: Final = string_value(object_value(claim["job"])["id"]) result: Final = isolated.post(