Skip to content

Commit 8b8ac0a

Browse files
authored
test: use gemini-2.5-pro for ai accessor doctests (googleapis#17354)
Fixes #<519296335> 🦕
1 parent e128488 commit 8b8ac0a

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

  • packages/bigframes/bigframes/operations

packages/bigframes/bigframes/operations/ai.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def filter(
5050
>>> bpd.options.compute.ai_ops_confirmation_threshold = 25
5151
5252
>>> import bigframes.ml.llm as llm
53-
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.0-flash-001")
53+
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.5-pro")
5454
5555
>>> df = bpd.DataFrame({"country": ["USA", "Germany"], "city": ["Seattle", "Berlin"]})
5656
>>> df.ai.filter("{city} is the capital of {country}", model)
@@ -119,7 +119,7 @@ def map(
119119
>>> bpd.options.compute.ai_ops_confirmation_threshold = 25
120120
121121
>>> import bigframes.ml.llm as llm
122-
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.0-flash-001")
122+
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.5-pro")
123123
124124
>>> df = bpd.DataFrame({"ingredient_1": ["Burger Bun", "Soy Bean"], "ingredient_2": ["Beef Patty", "Bittern"]})
125125
>>> df.ai.map("What is the food made from {ingredient_1} and {ingredient_2}? One word only.", model=model, output_schema={"food": "string"})
@@ -137,7 +137,7 @@ def map(
137137
>>> bpd.options.compute.ai_ops_confirmation_threshold = 25
138138
139139
>>> import bigframes.ml.llm as llm
140-
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.0-flash-001")
140+
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.5-pro")
141141
142142
>>> df = bpd.DataFrame({"text": ["Elmo lives at 123 Sesame Street."]})
143143
>>> df.ai.map("{text}", model=model, output_schema={"person": "string", "address": "string"})
@@ -268,7 +268,7 @@ def classify(
268268
>>> bpd.options.compute.ai_ops_confirmation_threshold = 25
269269
270270
>>> import bigframes.ml.llm as llm
271-
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.0-flash-001")
271+
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.5-pro")
272272
273273
>>> df = bpd.DataFrame({
274274
... "feedback_text": [
@@ -357,7 +357,7 @@ def join(
357357
>>> bpd.options.compute.ai_ops_confirmation_threshold = 25
358358
359359
>>> import bigframes.ml.llm as llm
360-
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.0-flash-001")
360+
>>> model = llm.GeminiTextGenerator(model_name="gemini-2.5-pro")
361361
362362
>>> cities = bpd.DataFrame({'city': ['Seattle', 'Ottawa', 'Berlin', 'Shanghai', 'New Delhi']})
363363
>>> continents = bpd.DataFrame({'continent': ['North America', 'Africa', 'Asia']})

0 commit comments

Comments
 (0)