scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(lens): use current worker protocol in billing integration (#44760)

Browse source
This commit is contained in:
moe-berri 2026-10-05 18:39:38 -07:00 • committed by GitHub
parent 6764b9af12
commit d232b008bc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/integration/spend/test_lens_billing.py
View file

@ -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(