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

test(langfuse_otel): drain wire batches per poll instead of accumulating

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-23 08:33:52 +00:00
parent 7328bb86af
commit bc6d6a516d
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/integration/observability/test_langfuse_otel_metadata.py
View file

@ -79,13 +79,10 @@ def _langfuse_rig(
def _observation_span(collector: Wire, response_id: str) -> dict[str, object]:
batches: Final[list[Request]] = [] # mutable-ok: drain() consumes the queue, so accumulating needs a mutable list
def spans() -> tuple[dict[str, object], ...]:
batches.extend(collector.drain())
return tuple(
attributes
for batch in batches
for batch in collector.drain()
for attributes in _span_attributes(batch.body)
if attributes.get("llm.response.id") == response_id
)