feat: AI model Gemini flash is going GA (GoogleCloudPlatform#11794)

msampathkumar · web-flow · commit 4e271eb5589c · 2024-05-24T14:11:40.000-04:00
Update model name to the latest version.
diff --git a/generative_ai/function_calling.py b/generative_ai/function_calling.py
@@ -166,7 +166,7 @@ def generate_function_call_advanced(project_id: str) -> GenerationResponse:
     )
 
     model = GenerativeModel(
-        "gemini-1.5-flash-preview-0514",
+        model_name="gemini-1.5-flash-001",
         tools=[retail_tool],
         tool_config=retail_tool_config,
     )
diff --git a/generative_ai/gemini_all_modalities.py b/generative_ai/gemini_all_modalities.py
@@ -24,7 +24,7 @@ def analyze_all_modalities(project_id: str) -> str:
 
     vertexai.init(project=project_id, location="us-central1")
 
-    model = GenerativeModel(model_name="gemini-1.5-flash-preview-0514")
+    model = GenerativeModel(model_name="gemini-1.5-flash-001")
 
     video_file_uri = (
         "gs://cloud-samples-data/generative-ai/video/behind_the_scenes_pixel.mp4"
diff --git a/generative_ai/gemini_audio.py b/generative_ai/gemini_audio.py
@@ -24,7 +24,7 @@ def summarize_audio(project_id: str) -> str:
 
     vertexai.init(project=project_id, location="us-central1")
 
-    model = GenerativeModel(model_name="gemini-1.5-flash-preview-0514")
+    model = GenerativeModel(model_name="gemini-1.5-flash-001")
 
     prompt = """
     Please provide a summary for the audio.
@@ -55,7 +55,7 @@ def transcript_audio(project_id: str) -> str:
 
     vertexai.init(project=project_id, location="us-central1")
 
-    model = GenerativeModel(model_name="gemini-1.5-flash-preview-0514")
+    model = GenerativeModel(model_name="gemini-1.5-flash-001")
 
     prompt = """
     Can you transcribe this interview, in the format of timecode, speaker, caption.
diff --git a/generative_ai/gemini_count_token_example.py b/generative_ai/gemini_count_token_example.py
@@ -58,7 +58,7 @@ def count_tokens_multimodal(project_id: str) -> GenerationResponse:
 
     vertexai.init(project=project_id, location="us-central1")
 
-    model = GenerativeModel(model_name="gemini-1.5-flash-preview-0514")
+    model = GenerativeModel(model_name="gemini-1.5-flash-001")
 
     contents = [
         Part.from_uri(
diff --git a/generative_ai/gemini_pdf_example.py b/generative_ai/gemini_pdf_example.py
@@ -24,7 +24,7 @@ def analyze_pdf(project_id: str) -> str:
 
     vertexai.init(project=project_id, location="us-central1")
 
-    model = GenerativeModel(model_name="gemini-1.5-flash-preview-0514")
+    model = GenerativeModel(model_name="gemini-1.5-flash-001")
 
     prompt = """
     You are a very professional document summarization specialist.
diff --git a/generative_ai/gemini_system_instruction.py b/generative_ai/gemini_system_instruction.py
@@ -25,7 +25,7 @@ def set_system_instruction(project_id: str) -> str:
     vertexai.init(project=project_id, location="us-central1")
 
     model = GenerativeModel(
-        model_name="gemini-1.5-flash-preview-0514",
+        model_name="gemini-1.5-flash-001",
         system_instruction=[
             "You are a helpful language translator.",
             "Your mission is to translate text in English to French.",
diff --git a/generative_ai/gemini_tuning_test.py b/generative_ai/gemini_tuning_test.py
@@ -20,7 +20,7 @@
 
 PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
 LOCATION = "us-central1"
-MODEL_ID = "gemini-1.5-flash-preview-0514"
+MODEL_ID = "gemini-1.5-flash-001"
 TUNING_JOB_ID = "4982013113894174720"
 
 
diff --git a/generative_ai/gemini_video_audio.py b/generative_ai/gemini_video_audio.py
@@ -24,7 +24,7 @@ def analyze_video_with_audio(project_id: str) -> str:
 
     vertexai.init(project=project_id, location="us-central1")
 
-    model = GenerativeModel(model_name="gemini-1.5-flash-preview-0514")
+    model = GenerativeModel(model_name="gemini-1.5-flash-001")
 
     prompt = """
     Provide a description of the video.
diff --git a/generative_ai/inference/inference_api_test.py b/generative_ai/inference/inference_api_test.py
@@ -21,7 +21,7 @@
 
 
 PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
-MODEL_ID = "gemini-1.5-flash-preview-0514"
+MODEL_ID = "gemini-1.5-flash-001"
 
 
 def test_non_stream_text_basic() -> None:

Original file line number	Diff line number	Diff line change
`@@ -166,7 +166,7 @@ def generate_function_call_advanced(project_id: str) -> GenerationResponse:`
`166`	`166`	`)`
`167`	`167`
`168`	`168`	`model = GenerativeModel(`
`169`		`- "gemini-1.5-flash-preview-0514",`
	`169`	`+ model_name="gemini-1.5-flash-001",`
`170`	`170`	`tools=[retail_tool],`
`171`	`171`	`tool_config=retail_tool_config,`
`172`	`172`	`)`