phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_alangfuse.py): fix race condition

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-20 10:55:47 -07:00
parent 713a6d4b32
commit 8766a58a16
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_alangfuse.py
View file

@ -362,8 +362,9 @@ async def test_aaalangfuse_logging_metadata(langfuse_client):
print(response) print(response)
metadata["existing_trace_id"] = trace_id metadata["existing_trace_id"] = trace_id
langfuse_client.flush() # make sure traces w/ generations are flushed in-order, prevents race condition of trace output being overwritten await asyncio.sleep(2)
await asyncio.sleep(10) langfuse_client.flush()
# await asyncio.sleep(10)
# Tests the metadata filtering and the override of the output to be the last generation # Tests the metadata filtering and the override of the output to be the last generation
for trace_id, generation_ids in trace_identifiers.items(): for trace_id, generation_ids in trace_identifiers.items():