remove gemma for pin adk

Author: calderjo

kaggle_requirements.txt

@@ -33,9 +33,8 @@ fury
 fuzzywuzzy
 geojson
 gensim
-gemma
 # b/443054743,b/455550872
-google-adk[a2a,eval]
+google-adk[a2a,eval]>=1.21.0
 google-cloud-aiplatform
 google-cloud-videointelligence
 google-cloud-vision
@@ -60,8 +59,6 @@ jupyterlab-lsp
 # Colab base image is updated more frequently.
 kaggle>=1.7.4.2
 kaggle-environments
-# Lastest version to support Gemma
-kauldron>=1.3.0
 keras-cv
 keras-nlp
 keras-tuner

tests/test_google_adk.py

@@ -0,0 +1,31 @@
+import unittest
+
+# Marked as "necessary building blocks" for 5 Days of AI notebooks
+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
+
+class TestGoogleADK(unittest.TestCase):
+
+    # From "Day 1a - From Prompt to Action"
+    def test_define_agent(self):
+
+        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.5-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],
+        )