tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-01 16:21:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: update langfuse trace_id tests to use litellm_trace_id

Browse Source
This commit is contained in:
Yuta Saito
2026-01-22 06:19:43 +09:00
parent c0007bd418
commit 898cc3ff4f
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/litellm_utils_tests/test_utils.py
+2 -2
View File
@@ -973,11 +973,11 @@ def test_logging_trace_id(langfuse_trace_id, langfuse_existing_trace_id):
litellm_logging_obj._get_trace_id(service_name="langfuse")
== langfuse_trace_id
)
## if existing_trace_id exists
## if no trace_id or existing_trace_id is provided, use litellm_trace_id
else:
assert (
litellm_logging_obj._get_trace_id(service_name="langfuse")
== litellm_call_id
== litellm_logging_obj.litellm_trace_id
)
tests/logging_callback_tests/test_alerting.py
+2 -4
View File
@@ -866,11 +866,9 @@ async def test_langfuse_trace_id():
assert trace_url is not None
returned_trace_id = int(trace_url.split("/")[-1])
returned_trace_id = trace_url.split("/")[-1]
assert returned_trace_id == int(
litellm_logging_obj._get_trace_id(service_name="langfuse")
)
assert returned_trace_id == litellm_logging_obj._get_trace_id(service_name="langfuse")
@pytest.mark.asyncio