feat(genai, generative_ai): Migration of Gemini samples to 2.X models

discoveryengine/answer_query_sample.py

@@ -11,9 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-#
 
-# NOTE: This snippet has been partially generated by `gemini-1.5-pro-001`
 
 # [START genappbuilder_answer_query]
 from google.api_core.client_options import ClientOptions
@@ -71,7 +69,7 @@ def answer_query_sample(
         ignore_non_answer_seeking_query=False,  # Optional: Ignore non-answer seeking query
         ignore_low_relevant_content=False,  # Optional: Return fallback answer when content is not relevant
         model_spec=discoveryengine.AnswerQueryRequest.AnswerGenerationSpec.ModelSpec(
-            model_version="gemini-1.5-flash-001/answer_gen/v2",  # Optional: Model to use for answer generation
+            model_version="gemini-2.0-flash-001/answer_gen/v2",  # Optional: Model to use for answer generation
         ),
         prompt_spec=discoveryengine.AnswerQueryRequest.AnswerGenerationSpec.PromptSpec(
             preamble="Give a detailed answer.",  # Optional: Natural language instructions for customizing the answer.

discoveryengine/session_sample.py

@@ -12,9 +12,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-#
 
-# NOTE: This snippet has been partially generated by `gemini-1.5-pro-001`
 
 # [START genappbuilder_create_session]
 from google.cloud import discoveryengine_v1 as discoveryengine

generative_ai/batch_predict/gemini_batch_predict_bigquery.py

@@ -37,7 +37,7 @@ def batch_predict_gemini_createjob(output_uri: str) -> str:
 
     # Submit a batch prediction job with Gemini model
     batch_prediction_job = BatchPredictionJob.submit(
-        source_model="gemini-1.5-flash-002",
+        source_model="gemini-2.0-flash-001",
         input_dataset=input_uri,
         output_uri_prefix=output_uri,
     )

generative_ai/batch_predict/gemini_batch_predict_gcs.py

@@ -38,7 +38,7 @@ def batch_predict_gemini_createjob(output_uri: str) -> str:
 
     # Submit a batch prediction job with Gemini model
     batch_prediction_job = BatchPredictionJob.submit(
-        source_model="gemini-1.5-flash-002",
+        source_model="gemini-2.0-flash-001",
         input_dataset=input_uri,
         output_uri_prefix=output_uri,
     )