Add Tests for Grounded Generation

Author: holtskinner

discoveryengine/requirements-test.txt

@@ -1,2 +1,3 @@
 pytest==8.2.0
-google-api-core==2.17.1
+google-api-core==2.21.0
+google-cloud-resource-manager==1.12.5

discoveryengine/standalone_apis_sample_test.py

@@ -15,6 +15,8 @@
 
 import os
 
+from google.cloud import resourcemanager_v3
+
 from discoveryengine import standalone_apis_sample
 
 project_id = os.environ["GOOGLE_CLOUD_PROJECT"]
@@ -32,3 +34,29 @@ def test_rank():
     response = standalone_apis_sample.rank_sample(project_id)
     assert response
     assert response.records
+
+
+def test_grounded_generation_inline_vais_sample():
+
+    # Grounded Generation requires Project Number
+    client = resourcemanager_v3.ProjectsClient()
+    project = client.get_project(name=client.project_path(project_id))
+    project_number = client.parse_project_path(project.name)["project"]
+
+    response = standalone_apis_sample.grounded_generation_inline_vais_sample(
+        project_number, engine_id="test-search-engine_1689960780551"
+    )
+    assert response
+
+
+def test_grounded_generation_google_search_sample():
+
+    # Grounded Generation requires Project Number
+    client = resourcemanager_v3.ProjectsClient()
+    project = client.get_project(name=client.project_path(project_id))
+    project_number = client.parse_project_path(project.name)["project"]
+
+    response = standalone_apis_sample.grounded_generation_google_search_sample(
+        project_number
+    )
+    assert response