fix: Change Gen App Builder test to use a different engine. (GoogleCloudPlatform#10446)

holtskinner · web-flow · commit 4d124c072521 · 2023-07-25T10:24:31.000-05:00
- Some engines created in Preview no longer work after the GA Launch.
- Updated discovery engine client to use latest version
- Reduced page size default
diff --git a/discoveryengine/requirements.txt b/discoveryengine/requirements.txt
@@ -1,2 +1,2 @@
-google-cloud-discoveryengine==0.8.1
+google-cloud-discoveryengine==0.9.1
 google-api-core==2.11.1
diff --git a/discoveryengine/search_sample.py b/discoveryengine/search_sample.py
@@ -46,8 +46,7 @@ def search_sample(
     )
 
     request = discoveryengine.SearchRequest(
-        serving_config=serving_config,
-        query=search_query,
+        serving_config=serving_config, query=search_query, page_size=10
     )
     response = client.search(request)
     for result in response.results:
diff --git a/discoveryengine/search_sample_test.py b/discoveryengine/search_sample_test.py
@@ -19,7 +19,7 @@
 
 project_id = os.environ["GOOGLE_CLOUD_PROJECT"]
 location = "global"
-search_engine_id = "test-search-engine"
+search_engine_id = "test-search-engine_1689960780551"
 serving_config_id = "default_config"
 search_query = "Google"
 

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1,2 @@`
`1`		`-google-cloud-discoveryengine==0.8.1`
	`1`	`+google-cloud-discoveryengine==0.9.1`
`2`	`2`	`google-api-core==2.11.1`
Original file line number	Diff line number	Diff line change
`@@ -46,8 +46,7 @@ def search_sample(`
`46`	`46`	`)`
`47`	`47`
`48`	`48`	`request = discoveryengine.SearchRequest(`
`49`		`- serving_config=serving_config,`
`50`		`- query=search_query,`
	`49`	`+ serving_config=serving_config, query=search_query, page_size=10`
`51`	`50`	`)`
`52`	`51`	`response = client.search(request)`
`53`	`52`	`for result in response.results:`