MongoDB Session Storage

[earthshakira]

Hey Folks 👋🏽!

Our Company started off with a mongoDB + python stack, I believe, a lot of startups or projects often follow the same trajectory, was wondering if I can contribute a MongoDBSessionService as an out of the box experience for the google-adk

To keep the installation lite, i was proposing to use a separate installations bit like

pip install google-adk[mongo]

and the session service can have a instantiation similar to the DatabaseSessionService like

MongoDBSessionService(mongouri,**options)

Though it's easy to extend the BaseSessionStorage this would be nice to have in there

Once I guage the teams feedback, I can use my existing code to open a PR
Thanks 🛸

[boyangsvl]

Thanks for your contribution! I'd suggest open a PR in the adk-python-community repo. @DeanChensj will work on a mechanism to register a session service under a url sheme. E.g. 'mangodb://...'

[DeanChensj]

Hi @earthshakira , with 391628f, you will be able to register a custom session service to use it in adk web, etc.

Example to register redis session service from community repo, you can do something like:

from google.adk.cli import cli_tools_click
from google.adk.cli.service_registry import get_service_registry
from google.adk_community.sessions import redis_session_service



def redis_session_service_factory(uri: str, **kwargs):
  """Factory for creating a RedisSessionService."""
  kwargs_copy = kwargs.copy()
  kwargs_copy.pop("agents_dir", None)
  return redis_session_service.RedisSessionService(**kwargs_copy)

"""Registers custom services with the ADK global registry."""
registry = get_service_registry()
registry.register_session_service("redis", redis_session_service_factory)


if __name__ == '__main__':
  cli_tools_click.main()

To run this script:

Install ADK-Community from git:
pip install git+https://github.com/google/adk-python-community.git@main
Run this script which registers the redisservice and runs the ADK CLI.
python test_run.py web contributing/samples/ --session_service_uri=redis://