diff --git a/alloydb-ai-mcp/cymbal_logistic/app.py b/alloydb-ai-mcp/cymbal_logistic/app.py
index 6b50976..11b5ec7 100644
--- a/alloydb-ai-mcp/cymbal_logistic/app.py
+++ b/alloydb-ai-mcp/cymbal_logistic/app.py
@@ -111,7 +111,7 @@ def run_query_sync(request_text, cluster_name, location, instance_name, database
     When you answer questions about Cymbal Logistic activity
     use the toolset to run query in the AlloyDB cluster {cluster_name} instance {instance_name} in the location {location}
     in the project {project_id} in the database {database_name}
-    Use ai schema to use AI functions and models like gemini-3-flash-preview with the functions from the schema.
+    Use ai schema to use AI functions and the models like gemini-3.5-flash with the functions from the schema.
     """
     
     if summary:
diff --git a/alloydb-ai-mcp/cymbal_logistic/data_agent/agent.py b/alloydb-ai-mcp/cymbal_logistic/data_agent/agent.py
index 908b679..2005517 100644
--- a/alloydb-ai-mcp/cymbal_logistic/data_agent/agent.py
+++ b/alloydb-ai-mcp/cymbal_logistic/data_agent/agent.py
@@ -48,7 +48,7 @@
 mcp_toolset = McpToolset(connection_params=connection_params)
 
 
-MODEL_ID = "gemini-3-flash-preview"
+MODEL_ID = "gemini-3.5-flash"
 cluster_name="alloydb-aip-01"
 instance_name="alloydb-aip-01-pr"
 location="us-central1"
@@ -66,7 +66,7 @@
     When you answer questions about Cymbal Logistic activity
     use the toolset to run query in the AlloyDB cluster {cluster_name} instance {instance_name} in the location {location}
     in the project {project_id} in the database {database_name}
-    Use ai schema to use AI functions and models like gemini-3-flash-preview with the functions from the schema.
+    Use ai schema to use AI functions and models like gemini-3.5-flash with the functions from the schema.
     """,
     tools=[mcp_toolset],
 )
diff --git a/alloydb-ai-mcp/cymbal_logistic/pyproject.toml b/alloydb-ai-mcp/cymbal_logistic/pyproject.toml
index 315d4bd..63474f8 100644
--- a/alloydb-ai-mcp/cymbal_logistic/pyproject.toml
+++ b/alloydb-ai-mcp/cymbal_logistic/pyproject.toml
@@ -19,7 +19,7 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12.11"
 dependencies = [
-    "google-adk>=1.26.0",
+    "google-adk[mcp]>=1.26.0",
     "google-auth>=2.48.0",
     "matplotlib>=3.10.8",
     "mesop>=1.2.2",