scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

test(spend): use collections.abc.Set for read-only set annotations

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-10-02 23:18:17 +00:00
parent ead0cededf
commit fbc49b3004
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/integration/spend/test_spend_log_request_tags_chaos.py
View file

@ -1,7 +1,7 @@
import json
import threading
import uuid
from collections.abc import AbstractSet, Mapping, Sequence
from collections.abc import Set, Mapping, Sequence
from itertools import chain
from concurrent.futures import ThreadPoolExecutor
from hashlib import sha256
@ -201,12 +201,12 @@ def test_sink_outage_does_not_lose_spend_log_tags(gateway: Gateway, tmp_path: Pa
(response.status_code, response.text[:200]) for response in first
]
def call_ids(responses: Sequence) -> AbstractSet[str]:
def call_ids(responses: Sequence) -> Set[str]:
return {response.headers["x-litellm-call-id"] for response in responses}
first_ids: Final = call_ids(first)
def events_for(ids: AbstractSet[str]) -> AbstractSet[str]:
def events_for(ids: Set[str]) -> Set[str]:
events: Final = chain.from_iterable(json.loads(batch.body) for batch in delivered)
return {event["litellm_call_id"] for event in events if event.get("litellm_call_id") in ids}