samples(discoveryengine): Add Model Prompt Spec and Model Spec to `se…

…arch_sample()` (#11274)
GoogleCloudPlatform · Mar 5, 2024 · af95f03 · af95f03
1 parent f832838
commit af95f03
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/discoveryengine/requirements-test.txt b/discoveryengine/requirements-test.txt
@@ -1,2 +1,2 @@
 pytest==7.2.1
-google-api-core==2.11.1
+google-api-core==2.17.1
diff --git a/discoveryengine/requirements.txt b/discoveryengine/requirements.txt
@@ -1,2 +1,2 @@
-google-cloud-discoveryengine==0.11.1
-google-api-core==2.11.1
+google-cloud-discoveryengine==0.11.8
+google-api-core==2.17.1
diff --git a/discoveryengine/search_sample.py b/discoveryengine/search_sample.py
@@ -68,6 +68,12 @@ def search_sample(
             include_citations=True,
             ignore_adversarial_query=True,
             ignore_non_summary_seeking_query=True,
+            model_prompt_spec=discoveryengine.SearchRequest.ContentSearchSpec.SummarySpec.ModelPromptSpec(
+                preamble="YOUR_CUSTOM_PROMPT"
+            ),
+            model_spec=discoveryengine.SearchRequest.ContentSearchSpec.SummarySpec.ModelSpec(
+                version="stable",  # Options: stable, preview
+            ),
         ),
     )