chore: add thinking_budget

DavidePaglieri · DavidePaglieri · commit e0afca7bf142 · 2025-11-25T11:35:26.000Z
diff --git a/balrog/client.py b/balrog/client.py
@@ -249,7 +249,7 @@ def _initialize_client(self):
 
             self.generation_config = genai.types.GenerateContentConfig(
                 **client_kwargs, 
-                thinking_config=types.ThinkingConfig(thinking_budget=)
+                thinking_config=types.ThinkingConfig(thinking_budget=thinking_budget)
             )
             self._initialized = True
 

Original file line number	Diff line number	Diff line change
`@@ -249,7 +249,7 @@ def _initialize_client(self):`
`249`	`249`
`250`	`250`	`self.generation_config = genai.types.GenerateContentConfig(`
`251`	`251`	`**client_kwargs,`
`252`		`- thinking_config=types.ThinkingConfig(thinking_budget=)`
	`252`	`+ thinking_config=types.ThinkingConfig(thinking_budget=thinking_budget)`
`253`	`253`	`)`
`254`	`254`	`self._initialized = True`
`255`	`255`