Skip to content

feat(integrations): add MongoDB-backed memory service #7512

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

Closed
theshanbhag wants to merge 14 commits into from
Closed

feat(integrations): add MongoDB-backed memory service #7512

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
77f705f
feat(integrations): support Atlas Automated Embedding in MongoDB search
theshanbhag Oct 12, 2026
0269f9f
feat(integrations): add MongoDB-backed session service
theshanbhag Oct 12, 2026
54dde6b
feat(integrations): add MongoDB-backed memory service
theshanbhag Oct 12, 2026
9bd2fcb
Merge remote-tracking branch 'refs/remotes/upstream-main' into mongod…
theshanbhag Oct 14, 2026
62902df
Merge branch 'mongodb-session-service' into mongodb-memory-service
theshanbhag Oct 14, 2026
4fbb99b
perf(integrations): batch user-state reads and guard index creation i…
theshanbhag Oct 14, 2026
5e9bd54
Merge branch 'mongodb-session-service' into mongodb-memory-service
theshanbhag Oct 14, 2026
495b288
perf(integrations): bulk-write memory ingestion and guard index creation
theshanbhag Oct 14, 2026
98a9829
chore: add mongomock pins to constraints-3.15.txt
theshanbhag Oct 14, 2026
97f76d8
chore: align constraint annotations with the stabilized generation flow
theshanbhag Oct 14, 2026
13fb967
Merge branch 'mongodb-session-service' into mongodb-memory-service
theshanbhag Oct 14, 2026
946282d
Merge remote-tracking branch 'refs/remotes/upstream-main' into mongod…
theshanbhag Oct 19, 2026
dac35e5
chore: untrack local sample files accidentally swept into the merge
theshanbhag Oct 19, 2026
453fa3a
Merge branch 'mongodb-session-service' into mongodb-memory-service
theshanbhag Oct 19, 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
12 changes: 11 additions & 1 deletion 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-10-14 --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-10-14 -o constraints-3.11.txt.new.tmp --constraint constraints-3.11.txt.stable.tmp
a2a-sdk==1.1.1
# via
# -c constraints-3.11.txt.stable.tmp
Expand Down Expand Up @@ -1127,6 +1127,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 @@ -1383,6 +1387,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# onnxruntime
# opentelemetry-instrumentation
# pyink
Expand Down Expand Up @@ -1766,6 +1771,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 @@ -1886,6 +1892,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
12 changes: 11 additions & 1 deletion 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-10-14 --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-10-14 -o constraints-3.12.txt.new.tmp --constraint constraints-3.12.txt.stable.tmp
a2a-sdk==1.1.1
# via
# -c constraints-3.12.txt.stable.tmp
Expand Down Expand Up @@ -987,6 +987,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 @@ -1206,6 +1210,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# opentelemetry-instrumentation
# pyink
# pyproject-api
Expand Down Expand Up @@ -1519,6 +1524,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 @@ -1624,6 +1630,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
12 changes: 11 additions & 1 deletion 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-10-14 --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-10-14 -o constraints-3.13.txt.new.tmp --constraint constraints-3.13.txt.stable.tmp
a2a-sdk==1.1.1
# via
# -c constraints-3.13.txt.stable.tmp
Expand Down Expand Up @@ -979,6 +979,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 @@ -1198,6 +1202,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# opentelemetry-instrumentation
# pyink
# pyproject-api
Expand Down Expand Up @@ -1511,6 +1516,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 @@ -1616,6 +1622,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
12 changes: 11 additions & 1 deletion 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-10-14 --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-10-14 -o constraints-3.14.txt.new.tmp --constraint constraints-3.14.txt.stable.tmp
a2a-sdk==1.1.1
# via
# -c constraints-3.14.txt.stable.tmp
Expand Down Expand Up @@ -979,6 +979,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 @@ -1198,6 +1202,7 @@ packaging==26.2
# langchain-core
# langsmith
# marshmallow
# mongomock
# opentelemetry-instrumentation
# pyink
# pyproject-api
Expand Down Expand Up @@ -1511,6 +1516,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 @@ -1616,6 +1622,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
12 changes: 11 additions & 1 deletion constraints-3.15.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.15 --no-emit-package google-adk --exclude-newer 2026-10-14 --index-url https://pypi.org/simple -o constraints-3.15.txt
# uv pip compile pyproject.toml --all-extras --python-version 3.15 --no-emit-package google-adk --exclude-newer 2026-10-14 -o constraints-3.15.txt.new.tmp --constraint constraints-3.15.txt.stable.tmp
a2a-sdk==1.2.2
# via
# -c constraints-3.15.txt.stable.tmp
Expand Down Expand Up @@ -1002,6 +1002,10 @@ mmh3==5.3.1
# via
# -c constraints-3.15.txt.stable.tmp
# google-cloud-spanner
mongomock==4.3.0
# via
# -c constraints-3.15.txt.stable.tmp
# google-adk (pyproject.toml)
multidict==6.9.1
# via
# -c constraints-3.15.txt.stable.tmp
Expand Down Expand Up @@ -1225,6 +1229,7 @@ packaging==26.3
# langchain-core
# langsmith
# marshmallow
# mongomock
# opentelemetry-instrumentation
# pyink
# pyproject-api
Expand Down Expand Up @@ -1556,6 +1561,7 @@ pytokens==0.4.1
pytz==2026.5
# via
# -c constraints-3.15.txt.stable.tmp
# mongomock
# oci
# pandas
pyyaml==6.0.3
Expand Down Expand Up @@ -1661,6 +1667,10 @@ scipy==1.18.1
# via
# -c constraints-3.15.txt.stable.tmp
# scikit-learn
sentinels==1.1.1
# via
# -c constraints-3.15.txt.stable.tmp
# mongomock
setuptools==84.0.0
# via
# -c constraints-3.15.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 @@ -369,6 +369,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
8 changes: 8 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 @@ -68,6 +68,8 @@ class FeatureName(str, Enum):
# 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_MEMORY_SERVICE = "MONGODB_MEMORY_SERVICE"
MONGODB_SESSION_SERVICE = "MONGODB_SESSION_SERVICE"
MONGODB_TOOLSET = "MONGODB_TOOLSET"
MONGODB_TOOL_SETTINGS = "MONGODB_TOOL_SETTINGS"
PROGRESSIVE_SSE_STREAMING = "PROGRESSIVE_SSE_STREAMING"
Expand Down Expand Up @@ -218,6 +220,12 @@ class FeatureConfig:
FeatureName._MCP_MODERN_PROTOCOL: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=False
),
FeatureName.MONGODB_MEMORY_SERVICE: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
FeatureName.MONGODB_SESSION_SERVICE: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
FeatureName.MONGODB_TOOLSET: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
Expand Down
Loading
Loading