"""
Simple agent for GitHub issue reproduction.
Minimal configuration with BuiltInPlanner matching westworld agent setup.
"""

from google.adk.agents import LlmAgent
from google.adk.planners import BuiltInPlanner
from google.genai.types import HttpRetryOptions, GenerateContentConfig, ThinkingConfig
from google.adk.models import Gemini


root_agent = LlmAgent(
    name="simple_agent",
    description="A simple helpful assistant for testing",
    instruction="You are a helpful assistant, respond to user's queries",
    model=Gemini(
        model="gemini-2.5-pro",
        retry_options=HttpRetryOptions(
            attempts=3,
            initial_delay=1.0,
            max_delay=60.0,
            exp_base=2.0,
            jitter=0.2,
        ),
    ),
    planner=BuiltInPlanner(
        thinking_config=ThinkingConfig(
            include_thoughts=True,
            thinking_budget=8192,
        ),
    ),
    generate_content_config=GenerateContentConfig(
        temperature=0.0,
    ),
    output_key="chat_response",
)