Skip to content

Commit 121b40b

Browse files
committed
Update models to use default
1 parent 1d783e0 commit 121b40b

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

discoveryengine/standalone_apis_sample.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def grounded_generation_inline_vais_sample(
143143
# Format: projects/{project_number}/locations/{location}
144144
location=client.common_location_path(project=project_number, location="global"),
145145
generation_spec=discoveryengine.GenerateGroundedContentRequest.GenerationSpec(
146-
model_id="gemini-1.5-flash",
146+
model_id="default",
147147
),
148148
# Conversation between user and model
149149
contents=[
@@ -215,7 +215,7 @@ def grounded_generation_google_search_sample(
215215
# Format: projects/{project_number}/locations/{location}
216216
location=client.common_location_path(project=project_number, location="global"),
217217
generation_spec=discoveryengine.GenerateGroundedContentRequest.GenerationSpec(
218-
model_id="gemini-1.5-flash",
218+
model_id="default",
219219
),
220220
# Conversation between user and model
221221
contents=[
@@ -274,7 +274,7 @@ def grounded_generation_streaming_sample(
274274
# Format: projects/{project_number}/locations/{location}
275275
location=client.common_location_path(project=project_number, location="global"),
276276
generation_spec=discoveryengine.GenerateGroundedContentRequest.GenerationSpec(
277-
model_id="gemini-1.5-flash",
277+
model_id="default",
278278
),
279279
# Conversation between user and model
280280
contents=[

discoveryengine/standalone_apis_sample_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@
1515

1616
import os
1717

18-
from discoveryengine import standalone_apis_sample
19-
2018
from google.cloud import resourcemanager_v3
2119

20+
from discoveryengine import standalone_apis_sample
21+
2222
project_id = os.environ["GOOGLE_CLOUD_PROJECT"]
2323

2424

0 commit comments

Comments
 (0)