File tree Expand file tree Collapse file tree
posthog/test/ai/openai_agents Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -89,7 +89,10 @@ def test_initialization(self, mock_client):
8989
9090 def test_initialization_with_callable_distinct_id (self , mock_client , mock_trace ):
9191 """Test processor with callable distinct_id resolver."""
92- resolver = lambda trace : trace .metadata .get ("user_id" , "default" )
92+
93+ def resolver (trace ):
94+ return trace .metadata .get ("user_id" , "default" )
95+
9396 processor = PostHogTracingProcessor (
9497 client = mock_client ,
9598 distinct_id = resolver ,
@@ -640,7 +643,10 @@ def test_distinct_id_resolved_from_trace_for_spans(
640643 self , mock_client , mock_trace , mock_span
641644 ):
642645 """Test that spans use the distinct_id resolved at trace start."""
643- resolver = lambda trace : f"user-{ trace .name } "
646+
647+ def resolver (trace ):
648+ return f"user-{ trace .name } "
649+
644650 processor = PostHogTracingProcessor (
645651 client = mock_client ,
646652 distinct_id = resolver ,
You can’t perform that action at this time.
0 commit comments