Remove Gemma package due to incompatibility and pin google-adk to latest (#1520)

gemma requires "kauldron>=1.3.0" or else it fails when loading models.
Unfortunately this version of kauldron is not support by google-adk
causing issues with our "Five days of AI" notebooks. Somehow, our
package installer resolves it to a very old version (0.0.1)

let's remove Gemma for now.

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
@@ -49,6 +48,8 @@ ipympl
 ipywidgets==8.1.5
 isoweek
 jedi
+# jitler 0.11.1 breaks simulation image
+jiter==0.10.0
 # b/276358430: fix Jupyter lsp freezing up the jupyter server
 jupyter-lsp==1.5.1
 # b/333854354: pin jupyter-server to version 2.12.5; later versions break LSP (b/333854354)
@@ -60,8 +61,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_import_adk.py

@@ -0,0 +1,31 @@
+import json
+import unittest
+import threading
+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],
+        )

tests/test_jiter.py

@@ -0,0 +1,9 @@
+import unittest
+
+from distutils.version import StrictVersion
+
+import jiter
+
+class TestJiter(unittest.TestCase):
+    def test_version(self):
+        self.assertEqual(StrictVersion(jiter.__version__), StrictVersion("0.10.0"))