diff --git a/generative_ai/chat_completions/chat_completions_credentials_refresher.py b/generative_ai/chat_completions/chat_completions_credentials_refresher.py index a60d2391a1c..dd4185d6d1a 100644 --- a/generative_ai/chat_completions/chat_completions_credentials_refresher.py +++ b/generative_ai/chat_completions/chat_completions_credentials_refresher.py @@ -55,7 +55,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object: ) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=[{"role": "user", "content": "Why is the sky blue?"}], ) diff --git a/generative_ai/chat_completions/chat_completions_function_calling_basic.py b/generative_ai/chat_completions/chat_completions_function_calling_basic.py index d64c9aa1494..b9a75534242 100644 --- a/generative_ai/chat_completions/chat_completions_function_calling_basic.py +++ b/generative_ai/chat_completions/chat_completions_function_calling_basic.py @@ -68,7 +68,7 @@ def generate_text() -> object: messages.append({"role": "user", "content": "What is the weather in Boston?"}) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=messages, tools=tools, ) diff --git a/generative_ai/chat_completions/chat_completions_function_calling_config.py b/generative_ai/chat_completions/chat_completions_function_calling_config.py index 80b00ac993d..01f3d255d97 100644 --- a/generative_ai/chat_completions/chat_completions_function_calling_config.py +++ b/generative_ai/chat_completions/chat_completions_function_calling_config.py @@ -68,7 +68,7 @@ def generate_text() -> object: messages.append({"role": "user", "content": "What is the weather in Boston, MA?"}) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=messages, tools=tools, tool_choice="auto", diff --git a/generative_ai/chat_completions/chat_completions_non_streaming_image.py b/generative_ai/chat_completions/chat_completions_non_streaming_image.py index 2bfe8cf96fa..f946b235dae 100644 --- a/generative_ai/chat_completions/chat_completions_non_streaming_image.py +++ b/generative_ai/chat_completions/chat_completions_non_streaming_image.py @@ -36,7 +36,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object: ) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=[ { "role": "user", diff --git a/generative_ai/chat_completions/chat_completions_non_streaming_text.py b/generative_ai/chat_completions/chat_completions_non_streaming_text.py index 20de139d62c..8a91a59d732 100644 --- a/generative_ai/chat_completions/chat_completions_non_streaming_text.py +++ b/generative_ai/chat_completions/chat_completions_non_streaming_text.py @@ -35,7 +35,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object: ) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=[{"role": "user", "content": "Why is the sky blue?"}], ) diff --git a/generative_ai/chat_completions/chat_completions_streaming_image.py b/generative_ai/chat_completions/chat_completions_streaming_image.py index 05630ef15fe..8d03f081f04 100644 --- a/generative_ai/chat_completions/chat_completions_streaming_image.py +++ b/generative_ai/chat_completions/chat_completions_streaming_image.py @@ -35,7 +35,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object: ) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=[ { "role": "user", diff --git a/generative_ai/chat_completions/chat_completions_streaming_text.py b/generative_ai/chat_completions/chat_completions_streaming_text.py index 42e98809ad9..34a7e21c160 100644 --- a/generative_ai/chat_completions/chat_completions_streaming_text.py +++ b/generative_ai/chat_completions/chat_completions_streaming_text.py @@ -35,7 +35,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object: ) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=[{"role": "user", "content": "Why is the sky blue?"}], stream=True, ) diff --git a/generative_ai/evaluation/pairwise_summarization_quality.py b/generative_ai/evaluation/pairwise_summarization_quality.py index 88c89871904..87b33de23dc 100644 --- a/generative_ai/evaluation/pairwise_summarization_quality.py +++ b/generative_ai/evaluation/pairwise_summarization_quality.py @@ -52,11 +52,11 @@ def evaluate_output() -> EvalResult: eval_dataset = pd.DataFrame({"prompt": [prompt]}) # Baseline model for pairwise comparison - baseline_model = GenerativeModel("gemini-2.0-flash-lite-001") + baseline_model = GenerativeModel("gemini-1.5-flash") # Candidate model for pairwise comparison candidate_model = GenerativeModel( - "gemini-2.0-flash-001", generation_config={"temperature": 0.4} + "gemini-1.5-pro", generation_config={"temperature": 0.4} ) prompt_template = MetricPromptTemplateExamples.get_prompt_template( diff --git a/generative_ai/function_calling/chat_function_calling_basic.py b/generative_ai/function_calling/chat_function_calling_basic.py index 9731a41582f..ac2b7192018 100644 --- a/generative_ai/function_calling/chat_function_calling_basic.py +++ b/generative_ai/function_calling/chat_function_calling_basic.py @@ -71,7 +71,7 @@ def generate_text() -> object: messages.append({"role": "user", "content": "What is the weather in Boston?"}) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=messages, tools=tools, ) diff --git a/generative_ai/function_calling/chat_function_calling_config.py b/generative_ai/function_calling/chat_function_calling_config.py index 720d72db70e..6e1f68375e1 100644 --- a/generative_ai/function_calling/chat_function_calling_config.py +++ b/generative_ai/function_calling/chat_function_calling_config.py @@ -71,7 +71,7 @@ def generate_text() -> object: messages.append({"role": "user", "content": "What is the weather in Boston, MA?"}) response = client.chat.completions.create( - model="google/gemini-2.0-flash-001", + model="google/gemini-1.5-flash", messages=messages, tools=tools, tool_choice="auto", diff --git a/generative_ai/labels/labels_example.py b/generative_ai/labels/labels_example.py index 23168e7d461..6caf7ef0872 100644 --- a/generative_ai/labels/labels_example.py +++ b/generative_ai/labels/labels_example.py @@ -28,7 +28,7 @@ def generate_content() -> GenerationResponse: # PROJECT_ID = "your-project-id" vertexai.init(project=PROJECT_ID, location="us-central1") - model = GenerativeModel("gemini-2.0-flash-001") + model = GenerativeModel("gemini-1.5-flash") prompt = "What is Generative AI?" response = model.generate_content( diff --git a/generative_ai/model_tuning/supervised_advanced_example.py b/generative_ai/model_tuning/supervised_advanced_example.py index 9e0a7ef11cd..926a6ae055d 100644 --- a/generative_ai/model_tuning/supervised_advanced_example.py +++ b/generative_ai/model_tuning/supervised_advanced_example.py @@ -40,7 +40,7 @@ def gemini_tuning_advanced() -> sft.SupervisedTuningJob: # vertexai.init(encryption_spec_key_name="your-kms-key") sft_tuning_job = sft.train( - source_model="gemini-2.0-flash-001", + source_model="gemini-1.5-flash", # 1.5 and 2.0 models use the same JSONL format train_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_train_data.jsonl", # The following parameters are optional diff --git a/generative_ai/model_tuning/supervised_example.py b/generative_ai/model_tuning/supervised_example.py index f537a51bdbb..1d777f4d5ba 100644 --- a/generative_ai/model_tuning/supervised_example.py +++ b/generative_ai/model_tuning/supervised_example.py @@ -32,7 +32,7 @@ def gemini_tuning_basic() -> sft.SupervisedTuningJob: vertexai.init(project=PROJECT_ID, location="us-central1") sft_tuning_job = sft.train( - source_model="gemini-2.0-flash-001", + source_model="gemini-1.5-flash", # 1.5 and 2.0 models use the same JSONL format train_dataset="gs://cloud-samples-data/ai-platform/generative_ai/gemini-1_5/text/sft_train_data.jsonl", ) diff --git a/generative_ai/prompts/prompt_create.py b/generative_ai/prompts/prompt_create.py index a18fbd986f8..0353d0c27eb 100644 --- a/generative_ai/prompts/prompt_create.py +++ b/generative_ai/prompts/prompt_create.py @@ -39,7 +39,7 @@ def prompt_create() -> Prompt: {"movie1": "The Lion King", "movie2": "Frozen"}, {"movie1": "Inception", "movie2": "Interstellar"}, ], - model_name="gemini-2.0-flash-001", + model_name="gemini-1.5-flash", system_instruction="You are a movie critic. Answer in a short sentence.", # generation_config=GenerationConfig, # Optional, # safety_settings=SafetySetting, # Optional, diff --git a/generative_ai/prompts/prompt_delete.py b/generative_ai/prompts/prompt_delete.py index 80c2f6940f1..0f48a4bfe28 100644 --- a/generative_ai/prompts/prompt_delete.py +++ b/generative_ai/prompts/prompt_delete.py @@ -35,7 +35,7 @@ def delete_prompt() -> None: {"movie1": "The Lion King", "movie2": "Frozen"}, {"movie1": "Inception", "movie2": "Interstellar"}, ], - model_name="gemini-2.0-flash-001", + model_name="gemini-1.5-flash", system_instruction="You are a movie critic. Answer in a short sentence.", ) diff --git a/generative_ai/prompts/prompt_get.py b/generative_ai/prompts/prompt_get.py index 59cf9c0bbc7..4185291904f 100644 --- a/generative_ai/prompts/prompt_get.py +++ b/generative_ai/prompts/prompt_get.py @@ -33,7 +33,7 @@ def get_prompt() -> Prompt: prompt = Prompt( prompt_name="meteorologist", prompt_data="How should I dress for weather in August?", - model_name="gemini-2.0-flash-001", + model_name="gemini-1.5-flash", system_instruction="You are a meteorologist. Answer in a short sentence.", ) diff --git a/generative_ai/prompts/prompt_list_version.py b/generative_ai/prompts/prompt_list_version.py index 1fc200673fc..21c79baefde 100644 --- a/generative_ai/prompts/prompt_list_version.py +++ b/generative_ai/prompts/prompt_list_version.py @@ -32,7 +32,7 @@ def list_prompt_version() -> list: prompt = Prompt( prompt_name="zoologist", prompt_data="Which animal is the fastest on earth?", - model_name="gemini-2.0-flash-001", + model_name="gemini-1.5-flash", system_instruction="You are a zoologist. Answer in a short sentence.", ) # Save Prompt to online resource. diff --git a/generative_ai/prompts/prompt_restore_version.py b/generative_ai/prompts/prompt_restore_version.py index f2496dfccb8..6a538b68fe9 100644 --- a/generative_ai/prompts/prompt_restore_version.py +++ b/generative_ai/prompts/prompt_restore_version.py @@ -32,7 +32,7 @@ # prompt = Prompt( # prompt_name="zoologist", # prompt_data="Which animal is the fastest on earth?", -# model_name="gemini-2.0-flash-001", +# model_name="gemini-1.5-flash", # system_instruction="You are a zoologist. Answer in a short sentence.", # ) # # Save Prompt to online resource. diff --git a/generative_ai/prompts/prompt_template.py b/generative_ai/prompts/prompt_template.py index 7517c7bb666..475fd4a2d08 100644 --- a/generative_ai/prompts/prompt_template.py +++ b/generative_ai/prompts/prompt_template.py @@ -38,7 +38,7 @@ def prompt_template_example() -> list[GenerationResponse]: # define prompt template prompt = Prompt( prompt_data="Do {animal} {activity}?", - model_name="gemini-2.0-flash-001", + model_name="gemini-1.5-flash", variables=variables, system_instruction="You are a helpful zoologist" # generation_config=generation_config, # Optional diff --git a/generative_ai/provisioned_throughput/provisioned_throughput_with_txt.py b/generative_ai/provisioned_throughput/provisioned_throughput_with_txt.py index 8da294ab6aa..c9038e09b95 100644 --- a/generative_ai/provisioned_throughput/provisioned_throughput_with_txt.py +++ b/generative_ai/provisioned_throughput/provisioned_throughput_with_txt.py @@ -33,7 +33,7 @@ def generate_content() -> str: request_metadata=[("x-vertex-ai-llm-request-type", "shared")], ) - model = GenerativeModel("gemini-2.0-flash-001") + model = GenerativeModel("gemini-1.5-flash") response = model.generate_content( "What's a good name for a flower shop that specializes in selling bouquets of dried flowers?" diff --git a/generative_ai/rag/generate_content_example.py b/generative_ai/rag/generate_content_example.py index 6bff7cb1910..eedea7812f1 100644 --- a/generative_ai/rag/generate_content_example.py +++ b/generative_ai/rag/generate_content_example.py @@ -53,7 +53,7 @@ def generate_content_with_rag( genai_client = genai.Client(enterprise=True, project=PROJECT_ID, location="us-central1") response = genai_client.models.generate_content( - model="gemini-2.5-pro", + model="gemini-2.5-flash", contents="Why is the sky blue?", config=genai_types.GenerateContentConfig( tools=[rag_retrieval_tool] diff --git a/generative_ai/text_generation/text_example01.py b/generative_ai/text_generation/text_example01.py index 744ec4ee1ed..3a516f6f96a 100644 --- a/generative_ai/text_generation/text_example01.py +++ b/generative_ai/text_generation/text_example01.py @@ -25,7 +25,7 @@ def generate_from_text_input() -> str: # PROJECT_ID = "your-project-id" vertexai.init(project=PROJECT_ID, location="us-central1") - model = GenerativeModel("gemini-2.0-flash-001") + model = GenerativeModel("gemini-1.5-flash") response = model.generate_content( "What's a good name for a flower shop that specializes in selling bouquets of dried flowers?"