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