Skip to content

Commit b142c34

Browse files
committed
Use dynamic thinking budget in generateContentThinkingFunctionCalling
1 parent 959e2b8 commit b142c34

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -234,24 +234,23 @@ struct GenerateContentIntegrationTests {
234234
@Test(
235235
arguments: [
236236
(.vertexAI_v1beta_global, ModelNames.gemini2_5_Flash, ThinkingConfig(thinkingBudget: 0)),
237-
(.vertexAI_v1beta_global, ModelNames.gemini2_5_Flash, ThinkingConfig(thinkingBudget: 24576)),
237+
(.vertexAI_v1beta_global, ModelNames.gemini2_5_Flash, ThinkingConfig(thinkingBudget: -1)),
238238
(.vertexAI_v1beta_global, ModelNames.gemini2_5_Flash, ThinkingConfig(
239-
thinkingBudget: 24576, includeThoughts: true
239+
thinkingBudget: -1, includeThoughts: true
240240
)),
241-
(.vertexAI_v1beta_global, ModelNames.gemini2_5_Pro, ThinkingConfig(thinkingBudget: 128)),
242-
(.vertexAI_v1beta_global, ModelNames.gemini2_5_Pro, ThinkingConfig(thinkingBudget: 32768)),
241+
(.vertexAI_v1beta_global, ModelNames.gemini2_5_Pro, ThinkingConfig(thinkingBudget: -1)),
243242
(.vertexAI_v1beta_global, ModelNames.gemini2_5_Pro, ThinkingConfig(
244-
thinkingBudget: 32768, includeThoughts: true
243+
thinkingBudget: -1, includeThoughts: true
245244
)),
246245
(.googleAI_v1beta, ModelNames.gemini2_5_Flash, ThinkingConfig(thinkingBudget: 0)),
247-
(.googleAI_v1beta, ModelNames.gemini2_5_Flash, ThinkingConfig(thinkingBudget: 24576)),
246+
(.googleAI_v1beta, ModelNames.gemini2_5_Flash, ThinkingConfig(thinkingBudget: -1)),
248247
(.googleAI_v1beta, ModelNames.gemini2_5_Flash, ThinkingConfig(
249-
thinkingBudget: 24576, includeThoughts: true
248+
thinkingBudget: -1, includeThoughts: true
250249
)),
251-
(.googleAI_v1beta, ModelNames.gemini2_5_Pro, ThinkingConfig(thinkingBudget: 128)),
252-
(.googleAI_v1beta, ModelNames.gemini2_5_Pro, ThinkingConfig(thinkingBudget: 32768)),
250+
(.googleAI_v1beta, ModelNames.gemini2_5_Pro, ThinkingConfig(thinkingBudget: -1)),
251+
(.googleAI_v1beta, ModelNames.gemini2_5_Pro, ThinkingConfig(thinkingBudget: -1)),
253252
(.googleAI_v1beta, ModelNames.gemini2_5_Pro, ThinkingConfig(
254-
thinkingBudget: 32768, includeThoughts: true
253+
thinkingBudget: -1, includeThoughts: true
255254
)),
256255
] as [(InstanceConfig, String, ThinkingConfig)]
257256
)

0 commit comments

Comments
 (0)