chore(generative-ai): Update Chat Completions Samples to Gemini 2.0 Flash GA (GoogleCloudPlatform#13153)

holtskinner · web-flow · commit bf826f55bd57 · 2025-02-13T17:29:21.000+01:00
Update Vertex SDK Chat Completions samples to use latest Gemini Model
diff --git 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-1.5-flash-002",
+        model="google/gemini-2.0-flash-001",
         messages=[{"role": "user", "content": "Why is the sky blue?"}],
     )
 
diff --git 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-1.5-flash-002",
+        model="google/gemini-2.0-flash-001",
         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
@@ -35,7 +35,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object:
     )
 
     response = client.chat.completions.create(
-        model="google/gemini-1.5-flash-002",
+        model="google/gemini-2.0-flash-001",
         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
@@ -35,7 +35,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object:
     )
 
     response = client.chat.completions.create(
-        model="google/gemini-1.5-flash-002",
+        model="google/gemini-2.0-flash-001",
         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
@@ -35,7 +35,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object:
     )
 
     response = client.chat.completions.create(
-        model="google/gemini-1.5-flash-002",
+        model="google/gemini-2.0-flash-001",
         messages=[{"role": "user", "content": "Why is the sky blue?"}],
         stream=True,
     )
diff --git a/generative_ai/chat_completions/requirements.txt b/generative_ai/chat_completions/requirements.txt
@@ -1,14 +1,2 @@
-pandas==1.3.5; python_version == '3.7'
-pandas==2.0.3; python_version == '3.8'
-pandas==2.1.4; python_version > '3.8'
-pillow==10.3.0; python_version < '3.8'
-pillow==10.3.0; python_version >= '3.8'
-google-cloud-aiplatform[all]==1.78.0
-sentencepiece==0.2.0
 google-auth==2.38.0
-anthropic[vertex]==0.28.0
-langchain-core==0.2.33
-langchain-google-vertexai==1.0.10
-numpy<2
 openai==1.60.0
-immutabledict==4.2.0

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object:`
`55`	`55`	`)`
`56`	`56`
`57`	`57`	`response = client.chat.completions.create(`
`58`		`- model="google/gemini-1.5-flash-002",`
	`58`	`+ model="google/gemini-2.0-flash-001",`
`59`	`59`	`messages=[{"role": "user", "content": "Why is the sky blue?"}],`
`60`	`60`	`)`
`61`	`61`
Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object:`
`36`	`36`	`)`
`37`	`37`
`38`	`38`	`response = client.chat.completions.create(`
`39`		`- model="google/gemini-1.5-flash-002",`
	`39`	`+ model="google/gemini-2.0-flash-001",`
`40`	`40`	`messages=[`
`41`	`41`	`{`
`42`	`42`	`"role": "user",`
Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ def generate_text(project_id: str, location: str = "us-central1") -> object:`
`35`	`35`	`)`
`36`	`36`
`37`	`37`	`response = client.chat.completions.create(`
`38`		`- model="google/gemini-1.5-flash-002",`
	`38`	`+ model="google/gemini-2.0-flash-001",`
`39`	`39`	`messages=[{"role": "user", "content": "Why is the sky blue?"}],`
`40`	`40`	`)`
`41`	`41`