How to assign thinking budget to gemini models?

General
,
1 
from agno.agent import Agent
from agno.models.google import Gemini
from agno.tools.duckduckgo import DuckDuckGoTools

agent = Agent(
    model=Gemini(id="gemini-2.5-flash"),
    tools=[DuckDuckGoTools()],
    markdown=True,
    debug_mode=True,
)
agent.print_response("give me an update on Israel and Iran war", stream=True)
2

Hey @Prajwal
thanks for reaching out and supporting Agno. I’ve shared this with the team, we’re working through all requests one by one and will get back to you soon.
If it’s urgent, please let us know. We appreciate your patience!

3

Hi @Prajwal ! I just released a PR that should make it easier to do this : PR

4 
from agno.agent import Agent
from agno.models.google import Gemini
from agno.tools import python
from agno.tools.yfinance import YFinanceTools
from google.ai.generativelanguage_v1beta import ThinkingConfig
from agno.tools.python import PythonTools

agent = Agent(
    model=Gemini(id="gemini-2.5-flash"),
    tools=[PythonTools()],
    instructions="Use tables to display data. Don't include any other text.",
    markdown=True,
    debug_mode=True,
    monitoring=True,
    thinking_budget=10000,
    include_thoughts=True,
    stream_intermediate_steps=True
)
agent.print_response("write a snake game in python use pyqt for ui", stream=True)

[{
	"resource": "/c:/Users/prajw/Downloads/AI-OS/python-backend/assistant.py",
	"owner": "pyright",
	"code": {
		"value": "reportArgumentType",
		"target": {
			"$mid": 1,
			"path": "/v1.29.4/configuration/config-files/",
			"scheme": "https",
			"authority": "docs.basedpyright.com",
			"fragment": "reportArgumentType"
		}
	},
	"severity": 8,
	"message": "Argument of type \"Literal['exact']\" cannot be assigned to parameter \"count\" of type \"CountMethod | None\" in function \"select\"\n  Type \"Literal['exact']\" is not assignable to type \"CountMethod | None\"\n    \"Literal['exact']\" is not assignable to \"CountMethod\"\n    \"Literal['exact']\" is not assignable to \"None\"",
	"source": "basedpyright",
	"startLineNumber": 82,
	"startColumn": 37,
	"endLineNumber": 82,
	"endColumn": 44,
	"modelVersionId": 67
},{
	"resource": "/C:/Users/prajw/Downloads/AI-OS/python-backend/test.py",
	"owner": "pyright",
	"code": {
		"value": "reportCallIssue",
		"target": {
			"$mid": 1,
			"path": "/v1.29.4/configuration/config-files/",
			"scheme": "https",
			"authority": "docs.basedpyright.com",
			"fragment": "reportCallIssue"
		}
	},
	"severity": 8,
	"message": "No parameter named \"include_thoughts\"",
	"source": "basedpyright",
	"startLineNumber": 16,
	"startColumn": 5,
	"endLineNumber": 16,
	"endColumn": 21,
	"modelVersionId": 104
},{
	"resource": "/C:/Users/prajw/Downloads/AI-OS/python-backend/test.py",
	"owner": "pyright",
	"code": {
		"value": "reportCallIssue",
		"target": {
			"$mid": 1,
			"path": "/v1.29.4/configuration/config-files/",
			"scheme": "https",
			"authority": "docs.basedpyright.com",
			"fragment": "reportCallIssue"
		}
	},
	"severity": 8,
	"message": "No parameter named \"thinking_budget\"",
	"source": "basedpyright",
	"startLineNumber": 15,
	"startColumn": 5,
	"endLineNumber": 15,
	"endColumn": 20,
	"modelVersionId": 104
}]

i copied the entire “gemini.py” which have your modifed code

5

Hey @Prajwal the import looks incorrect. it should be

from google.genai.types import ThinkingConfig

Could you refer again to the PR and take reference from that

6

Hey @Prajwal The PR is merged. Once released in Agno you should be able to directly use Thinking Budget as shown in cookbook/models/google/gemini/agent_with_thinking_budget.py