fix: use kwargs to access generation data in workflow trace test

2026-03-12 14:36:35 +08:00 · 2026-03-12 14:36:35 +08:00 · 4202fd085e
parent 22da26e013
commit 4202fd085e
1 changed files with 1 additions and 1 deletions
--- a/api/tests/unit_tests/core/ops/test_langfuse_trace.py
+++ b/api/tests/unit_tests/core/ops/test_langfuse_trace.py
@ -124,7 +124,7 @@ class TestLangFuseDataTraceCompletionStartTime:
        ):
            trace.workflow_trace(trace_info)

-        generation = add_generation.call_args.args[0]
+        generation = add_generation.call_args.kwargs["langfuse_generation_data"]
        assert generation.completion_start_time == start_time + timedelta(seconds=1.2)

    def test_ignores_invalid_ttft_values(self):