Skip to content

feat(integrations): add MongoDB-backed session and memory services #7291

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
theshanbhag wants to merge 2 commits into
base: main
Choose a base branch
from
Open

feat(integrations): add MongoDB-backed session and memory services #7291

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
be8cd0c
feat(integrations): add MongoDB-backed session and memory services
theshanbhag Sep 25, 2026
a15c195
Merge branch 'main' into add-memory-to-mongodb
theshanbhag Sep 30, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 11 additions & 6 deletions constraints-3.10.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# This file was autogenerated by uv via the following command:
# uv pip compile pyproject.toml --all-extras --python-version 3.10 --no-emit-package google-adk --exclude-newer 2026-09-04 --index-url https://pypi.org/simple -o constraints-3.10.txt
# uv pip compile pyproject.toml --all-extras --python-version 3.10 --no-emit-package google-adk --exclude-newer 2026-09-21 --index-url https://pypi.org/simple -o constraints-3.10.txt
a2a-sdk==1.1.2
# via
# -c constraints-3.10.txt.stable.tmp
Expand Down Expand Up @@ -31,7 +31,6 @@ aiohttp==3.14.1
# daytona-analytics-api-client-async
# daytona-api-client-async
# daytona-toolbox-api-client-async
# google-adk
# google-cloud-aiplatform
# kubernetes
# langchain-community
Expand Down Expand Up @@ -989,6 +988,10 @@ mmh3==5.2.1
# via
# -c constraints-3.10.txt.stable.tmp
# google-cloud-spanner
mongomock==4.3.0
# via
# -c constraints-3.10.txt.stable.tmp
# google-adk (pyproject.toml)
multidict==6.7.1
# via
# -c constraints-3.10.txt.stable.tmp
Expand Down Expand Up @@ -1199,6 +1202,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# opentelemetry-instrumentation
# pyink
# pyproject-api
Expand Down Expand Up @@ -1505,6 +1509,7 @@ pytokens==0.4.1
pytz==2026.2
# via
# -c constraints-3.10.txt.stable.tmp
# mongomock
# oci
# pandas
pyyaml==6.0.3
Expand Down Expand Up @@ -1601,6 +1606,10 @@ scipy==1.15.3
# via
# -c constraints-3.10.txt.stable.tmp
# scikit-learn
sentinels==1.1.1
# via
# -c constraints-3.10.txt.stable.tmp
# mongomock
setuptools==83.0.0
# via
# -c constraints-3.10.txt.stable.tmp
Expand Down Expand Up @@ -1907,10 +1916,6 @@ tzdata==2026.3
# via
# -c constraints-3.10.txt.stable.tmp
# pandas
tzlocal==5.4.4
# via
# -c constraints-3.10.txt.stable.tmp
# google-adk
uritemplate==4.2.0
# via
# -c constraints-3.10.txt.stable.tmp
Expand Down
17 changes: 11 additions & 6 deletions constraints-3.11.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# This file was autogenerated by uv via the following command:
# uv pip compile pyproject.toml --all-extras --python-version 3.11 --no-emit-package google-adk --exclude-newer 2026-09-04 --index-url https://pypi.org/simple -o constraints-3.11.txt
# uv pip compile pyproject.toml --all-extras --python-version 3.11 --no-emit-package google-adk --exclude-newer 2026-09-21 --index-url https://pypi.org/simple -o constraints-3.11.txt
a2a-sdk==1.1.1
# via
# -c constraints-3.11.txt.stable.tmp
Expand Down Expand Up @@ -32,7 +32,6 @@ aiohttp==3.14.1
# daytona-analytics-api-client-async
# daytona-api-client-async
# daytona-toolbox-api-client-async
# google-adk
# google-cloud-aiplatform
# instructor
# kubernetes
Expand Down Expand Up @@ -1113,6 +1112,10 @@ mmh3==5.2.1
# -c constraints-3.11.txt.stable.tmp
# chromadb
# google-cloud-spanner
mongomock==4.3.0
# via
# -c constraints-3.11.txt.stable.tmp
# google-adk (pyproject.toml)
multidict==6.7.1
# via
# -c constraints-3.11.txt.stable.tmp
Expand Down Expand Up @@ -1364,6 +1367,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# onnxruntime
# opentelemetry-instrumentation
# pyink
Expand Down Expand Up @@ -1740,6 +1744,7 @@ pytube==15.0.0
pytz==2026.2
# via
# -c constraints-3.11.txt.stable.tmp
# mongomock
# oci
# pandas
pyyaml==6.0.3
Expand Down Expand Up @@ -1859,6 +1864,10 @@ scipy==1.17.1
# via
# -c constraints-3.11.txt.stable.tmp
# scikit-learn
sentinels==1.1.1
# via
# -c constraints-3.11.txt.stable.tmp
# mongomock
setuptools==83.0.0
# via
# -c constraints-3.11.txt.stable.tmp
Expand Down Expand Up @@ -2171,10 +2180,6 @@ tzdata==2026.3
# -c constraints-3.11.txt.stable.tmp
# pandas
# pendulum
tzlocal==5.4.4
# via
# -c constraints-3.11.txt.stable.tmp
# google-adk
uc-micro-py==2.0.0
# via
# -c constraints-3.11.txt.stable.tmp
Expand Down
17 changes: 11 additions & 6 deletions constraints-3.12.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# This file was autogenerated by uv via the following command:
# uv pip compile pyproject.toml --all-extras --python-version 3.12 --no-emit-package google-adk --exclude-newer 2026-09-04 --index-url https://pypi.org/simple -o constraints-3.12.txt
# uv pip compile pyproject.toml --all-extras --python-version 3.12 --no-emit-package google-adk --exclude-newer 2026-09-21 --index-url https://pypi.org/simple -o constraints-3.12.txt
a2a-sdk==1.1.1
# via
# -c constraints-3.12.txt.stable.tmp
Expand Down Expand Up @@ -31,7 +31,6 @@ aiohttp==3.14.1
# daytona-analytics-api-client-async
# daytona-api-client-async
# daytona-toolbox-api-client-async
# google-adk
# google-cloud-aiplatform
# kubernetes
# langchain-community
Expand Down Expand Up @@ -973,6 +972,10 @@ mmh3==5.2.1
# via
# -c constraints-3.12.txt.stable.tmp
# google-cloud-spanner
mongomock==4.3.0
# via
# -c constraints-3.12.txt.stable.tmp
# google-adk (pyproject.toml)
multidict==6.7.1
# via
# -c constraints-3.12.txt.stable.tmp
Expand Down Expand Up @@ -1187,6 +1190,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# opentelemetry-instrumentation
# pyink
# pyproject-api
Expand Down Expand Up @@ -1493,6 +1497,7 @@ pytokens==0.4.1
pytz==2026.2
# via
# -c constraints-3.12.txt.stable.tmp
# mongomock
# oci
# pandas
pyyaml==6.0.3
Expand Down Expand Up @@ -1597,6 +1602,10 @@ scipy==1.18.0
# via
# -c constraints-3.12.txt.stable.tmp
# scikit-learn
sentinels==1.1.1
# via
# -c constraints-3.12.txt.stable.tmp
# mongomock
setuptools==83.0.0
# via
# -c constraints-3.12.txt.stable.tmp
Expand Down Expand Up @@ -1874,10 +1883,6 @@ tzdata==2026.3
# via
# -c constraints-3.12.txt.stable.tmp
# pandas
tzlocal==5.4.4
# via
# -c constraints-3.12.txt.stable.tmp
# google-adk
uritemplate==4.2.0
# via
# -c constraints-3.12.txt.stable.tmp
Expand Down
17 changes: 11 additions & 6 deletions constraints-3.13.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# This file was autogenerated by uv via the following command:
# uv pip compile pyproject.toml --all-extras --python-version 3.13 --no-emit-package google-adk --exclude-newer 2026-09-04 --index-url https://pypi.org/simple -o constraints-3.13.txt
# uv pip compile pyproject.toml --all-extras --python-version 3.13 --no-emit-package google-adk --exclude-newer 2026-09-21 --index-url https://pypi.org/simple -o constraints-3.13.txt
a2a-sdk==1.1.1
# via
# -c constraints-3.13.txt.stable.tmp
Expand Down Expand Up @@ -31,7 +31,6 @@ aiohttp==3.14.1
# daytona-analytics-api-client-async
# daytona-api-client-async
# daytona-toolbox-api-client-async
# google-adk
# google-cloud-aiplatform
# kubernetes
# langchain-community
Expand Down Expand Up @@ -965,6 +964,10 @@ mmh3==5.2.1
# via
# -c constraints-3.13.txt.stable.tmp
# google-cloud-spanner
mongomock==4.3.0
# via
# -c constraints-3.13.txt.stable.tmp
# google-adk (pyproject.toml)
multidict==6.7.1
# via
# -c constraints-3.13.txt.stable.tmp
Expand Down Expand Up @@ -1179,6 +1182,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# opentelemetry-instrumentation
# pyink
# pyproject-api
Expand Down Expand Up @@ -1485,6 +1489,7 @@ pytokens==0.4.1
pytz==2026.2
# via
# -c constraints-3.13.txt.stable.tmp
# mongomock
# oci
# pandas
pyyaml==6.0.3
Expand Down Expand Up @@ -1589,6 +1594,10 @@ scipy==1.18.0
# via
# -c constraints-3.13.txt.stable.tmp
# scikit-learn
sentinels==1.1.1
# via
# -c constraints-3.13.txt.stable.tmp
# mongomock
setuptools==83.0.0
# via
# -c constraints-3.13.txt.stable.tmp
Expand Down Expand Up @@ -1855,10 +1864,6 @@ tzdata==2026.3
# via
# -c constraints-3.13.txt.stable.tmp
# pandas
tzlocal==5.4.4
# via
# -c constraints-3.13.txt.stable.tmp
# google-adk
uritemplate==4.2.0
# via
# -c constraints-3.13.txt.stable.tmp
Expand Down
17 changes: 11 additions & 6 deletions constraints-3.14.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# This file was autogenerated by uv via the following command:
# uv pip compile pyproject.toml --all-extras --python-version 3.14 --no-emit-package google-adk --exclude-newer 2026-09-04 --index-url https://pypi.org/simple -o constraints-3.14.txt
# uv pip compile pyproject.toml --all-extras --python-version 3.14 --no-emit-package google-adk --exclude-newer 2026-09-21 --index-url https://pypi.org/simple -o constraints-3.14.txt
a2a-sdk==1.1.1
# via
# -c constraints-3.14.txt.stable.tmp
Expand Down Expand Up @@ -31,7 +31,6 @@ aiohttp==3.14.1
# daytona-analytics-api-client-async
# daytona-api-client-async
# daytona-toolbox-api-client-async
# google-adk
# google-cloud-aiplatform
# kubernetes
# langchain-community
Expand Down Expand Up @@ -965,6 +964,10 @@ mmh3==5.2.1
# via
# -c constraints-3.14.txt.stable.tmp
# google-cloud-spanner
mongomock==4.3.0
# via
# -c constraints-3.14.txt.stable.tmp
# google-adk (pyproject.toml)
multidict==6.7.1
# via
# -c constraints-3.14.txt.stable.tmp
Expand Down Expand Up @@ -1179,6 +1182,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# opentelemetry-instrumentation
# pyink
# pyproject-api
Expand Down Expand Up @@ -1485,6 +1489,7 @@ pytokens==0.4.1
pytz==2026.2
# via
# -c constraints-3.14.txt.stable.tmp
# mongomock
# oci
# pandas
pyyaml==6.0.3
Expand Down Expand Up @@ -1589,6 +1594,10 @@ scipy==1.18.0
# via
# -c constraints-3.14.txt.stable.tmp
# scikit-learn
sentinels==1.1.1
# via
# -c constraints-3.14.txt.stable.tmp
# mongomock
setuptools==83.0.0
# via
# -c constraints-3.14.txt.stable.tmp
Expand Down Expand Up @@ -1855,10 +1864,6 @@ tzdata==2026.3
# via
# -c constraints-3.14.txt.stable.tmp
# pandas
tzlocal==5.4.4
# via
# -c constraints-3.14.txt.stable.tmp
# google-adk
uritemplate==4.2.0
# via
# -c constraints-3.14.txt.stable.tmp
Expand Down
1 change: 1 addition & 0 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -327,6 +327,7 @@ optional-dependencies.test = [
"llama-index-readers-file>=0.4",
"lxml>=5.3",
"mcp>=1.24,<3",
"mongomock>=4.3,<5", # In-memory MongoDB used by the mongodb integration tests.
"nltk!=3.10.1", # Transitive via rouge-score and llama-index-core; 3.10.1's import hook breaks any venv living inside the working directory (reverted upstream in nltk/nltk#3732).
"openai>=2.20,<3",
"openpyxl>=3.1.5,<4",
Expand Down
10 changes: 10 additions & 0 deletions src/google/adk/features/_feature_registry.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,11 +59,14 @@ class FeatureName(str, Enum):
# enum member by name. Keeping it private avoids a backward-compat
# obligation for what is intended as a temporary, internal kill-switch.
_MCP_GRACEFUL_ERROR_HANDLING = "MCP_GRACEFUL_ERROR_HANDLING"
MONGODB_MEMORY_SERVICE = "MONGODB_MEMORY_SERVICE"
MONGODB_SESSION_SERVICE = "MONGODB_SESSION_SERVICE"
# Off by default since it changes wire behavior. Enable with
# `ADK_ENABLE_MCP_MODERN_PROTOCOL=1`. No effect on MCP SDK 1.x.
_MCP_MODERN_PROTOCOL = "MCP_MODERN_PROTOCOL"
MONGODB_TOOLSET = "MONGODB_TOOLSET"
MONGODB_TOOL_SETTINGS = "MONGODB_TOOL_SETTINGS"
MONGODB_TOOLSET = "MONGODB_TOOLSET"
PROGRESSIVE_SSE_STREAMING = "PROGRESSIVE_SSE_STREAMING"
PUBSUB_TOOL_CONFIG = "PUBSUB_TOOL_CONFIG"
PUBSUB_TOOLSET = "PUBSUB_TOOLSET"
Expand Down Expand Up @@ -194,6 +197,10 @@ class FeatureConfig:
FeatureName._MCP_GRACEFUL_ERROR_HANDLING: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
FeatureName.MONGODB_MEMORY_SERVICE: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
FeatureName.MONGODB_SESSION_SERVICE: FeatureConfig(
FeatureName._MCP_MODERN_PROTOCOL: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=False
),
Expand All @@ -203,6 +210,9 @@ class FeatureConfig:
FeatureName.MONGODB_TOOL_SETTINGS: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
FeatureName.MONGODB_TOOLSET: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
FeatureName.PROGRESSIVE_SSE_STREAMING: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
Expand Down
Loading
Loading