From 93c9ef81e80218b49fc68a15d01e687f4ba832c7 Mon Sep 17 00:00:00 2001 From: Karl Weinmeister Date: Tue, 19 Mar 2024 07:34:25 -0500 Subject: [PATCH] fix: skip failing test due to allowlist restrictions --- generative_ai/test_gemini_examples.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generative_ai/test_gemini_examples.py b/generative_ai/test_gemini_examples.py index c66624e2a280..f49f686ebcc8 100644 --- a/generative_ai/test_gemini_examples.py +++ b/generative_ai/test_gemini_examples.py @@ -26,6 +26,7 @@ import gemini_safety_config_example import gemini_single_turn_video_example +from pytest import skip PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT") LOCATION = "us-central1" @@ -132,9 +133,8 @@ def test_gemini_chat_example() -> None: assert any([_ in text for _ in ("hi", "hello", "greeting")]) +@skip("Unable to test Google Search grounding due to allowlist restrictions.") def test_gemini_grounding_example() -> None: - # Test Vertex AI Search Grounding - # Unable to test Google Search grounding due to allowlist restrictions. data_store_id = "test-search-engine_1689960780551" data_store_path = f"projects/{PROJECT_ID}/locations/{LOCATION}/collections/default_collection/dataStores/{data_store_id}" response = gemini_grounding_example.generate_text_with_grounding(