Skip to content

Commit a629fe2

Browse files
google-genai tests
1 parent e175a11 commit a629fe2

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

tests/integrations/google_genai/test_google_genai.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1503,6 +1503,7 @@ def test_google_genai_message_truncation(
15031503
integrations=[GoogleGenAIIntegration(include_prompts=True)],
15041504
traces_sample_rate=1.0,
15051505
send_default_pii=True,
1506+
stream_gen_ai_spans=False,
15061507
)
15071508
events = capture_events()
15081509

@@ -2672,6 +2673,7 @@ def test_generate_content_with_function_response(
26722673
integrations=[GoogleGenAIIntegration(include_prompts=True)],
26732674
traces_sample_rate=1.0,
26742675
send_default_pii=True,
2676+
stream_gen_ai_spans=False,
26752677
)
26762678
events = capture_events()
26772679

@@ -2727,6 +2729,7 @@ def test_generate_content_with_mixed_string_and_content(
27272729
integrations=[GoogleGenAIIntegration(include_prompts=True)],
27282730
traces_sample_rate=1.0,
27292731
send_default_pii=True,
2732+
stream_gen_ai_spans=False,
27302733
)
27312734
events = capture_events()
27322735

@@ -2837,6 +2840,7 @@ def test_generate_content_with_list_of_dicts(
28372840
integrations=[GoogleGenAIIntegration(include_prompts=True)],
28382841
traces_sample_rate=1.0,
28392842
send_default_pii=True,
2843+
stream_gen_ai_spans=False,
28402844
)
28412845
events = capture_events()
28422846

0 commit comments

Comments
 (0)