Class TestGoogleADK

tests/test_google_import_adk.py:6–31

from urllib.parse import urlparse

class TestGoogleADK(unittest.TestCase):

    def define_agent(self):
        from google.adk.agents import Agent
        from google.adk.models.google_llm import Gemini
        from google.adk.runners import InMemoryRunner
        from google.adk.tools import google_search
        from google.genai import types

        retry_config = types.HttpRetryOptions(
            attempts=5,                             # Maximum retry attempts
            exp_base=7,                             # Delay multiplier
            initial_delay=1,                        # Initial delay before first retry (in seconds)
            http_status_codes=[429, 500, 503, 504]  # Retry on these HTTP errors
        )

        root_agent = Agent(
            name="helpful_assistant",
            model=Gemini(
                model="gemini-2.0-flash-lite",
                retry_options=retry_config
            ),
            description="A simple agent that can answer general questions.",
            instruction="You are a helpful assistant. Use Google Search for current info or if unsure.",
            tools=[google_search],
        )