Set includeThoughts to True by default for Gemini API.

yifenglou · langfun authors · commit 7dc9e1264394 · 2025-06-03T20:23:45.000-07:00
PiperOrigin-RevId: 766939527
diff --git a/langfun/core/llms/gemini.py b/langfun/core/llms/gemini.py
@@ -695,7 +695,8 @@ def _generation_config(
       )
     if options.max_thinking_tokens is not None:
       config['thinkingConfig'] = {
-          'thinkingBudget': options.max_thinking_tokens
+          'includeThoughts': True,
+          'thinkingBudget': options.max_thinking_tokens,
       }
 
     return config
diff --git a/langfun/core/llms/gemini_test.py b/langfun/core/llms/gemini_test.py
@@ -173,7 +173,7 @@ def test_generation_config(self):
             responseLogprobs=False,
             logprobs=None,
             seed=None,
-            thinkingConfig={'thinkingBudget': 100},
+            thinkingConfig={'includeThoughts': True, 'thinkingBudget': 100},
         ),
     )
 

Original file line number	Diff line number	Diff line change
`@@ -695,7 +695,8 @@ def _generation_config(`
`695`	`695`	`)`
`696`	`696`	`if options.max_thinking_tokens is not None:`
`697`	`697`	`config['thinkingConfig'] = {`
`698`		`- 'thinkingBudget': options.max_thinking_tokens`
	`698`	`+ 'includeThoughts': True,`
	`699`	`+ 'thinkingBudget': options.max_thinking_tokens,`
`699`	`700`	`}`
`700`	`701`
`701`	`702`	`return config`
Original file line number	Diff line number	Diff line change
`@@ -173,7 +173,7 @@ def test_generation_config(self):`
`173`	`173`	`responseLogprobs=False,`
`174`	`174`	`logprobs=None,`
`175`	`175`	`seed=None,`
`176`		`- thinkingConfig={'thinkingBudget': 100},`
	`176`	`+ thinkingConfig={'includeThoughts': True, 'thinkingBudget': 100},`
`177`	`177`	`),`
`178`	`178`	`)`
`179`	`179`