Skip to content

feat(integrations): add MongoDB toolset with vector and hybrid search - #7041

Open
theshanbhag wants to merge 2 commits into
google:mainfrom
theshanbhag:theshanbhag/feat/add-mongodb-support
Open

feat(integrations): add MongoDB toolset with vector and hybrid search#7041
theshanbhag wants to merge 2 commits into
google:mainfrom
theshanbhag:theshanbhag/feat/add-mongodb-support

Conversation

@theshanbhag

Copy link
Copy Markdown

Add an experimental MongoDbToolset under google.adk.integrations.mongodb that lets agents run Atlas Vector Search ($vectorSearch) and hybrid full-text + vector search ($rankFusion, reciprocal rank fusion) against a MongoDB database. Search results exclude raw embedding vectors by default and include a search_score; limits and defaults are controlled via MongoDbToolSettings.

Usage:
toolset = MongoDbToolset(
connection_string="mongodb+srv://...",
database_name="my_db",
)
agent = Agent(model="gemini-2.5-flash", tools=[toolset])

Requires the pymongo package: pip install google-adk[mongodb].

Please ensure you have read the contribution guide before creating a pull request.

Link to Issue or Description of Change

1. Link to an existing issue (if applicable):

  • Closes: #issue_number
  • Related: #issue_number

2. Or, if no issue exists, describe the change:

If applicable, please follow the issue templates to provide as much detail as
possible.

Problem:
A clear and concise description of what the problem is.

Solution:
A clear and concise description of what you want to happen and why you choose
this solution.

Testing Plan

Please describe the tests that you ran to verify your changes. This is required
for all PRs that are not small documentation or typo fixes.

Unit Tests:

  • I have added or updated unit tests for my change.
  • All unit tests pass locally.

Please include a summary of passed pytest results.

Manual End-to-End (E2E) Tests:

Please provide instructions on how to manually test your changes, including any
necessary setup or configuration. Please provide logs or screenshots to help
reviewers better understand the fix.

Checklist

  • I have read the CONTRIBUTING.md document.
  • I have performed a self-review of my own code.
  • I have commented my code, particularly in hard-to-understand areas.
  • I have added tests that prove my fix is effective or that my feature works.
  • New and existing unit tests pass locally with my changes.
  • I have manually tested my changes end-to-end.
  • Any dependent changes have been merged and published in downstream modules.

Additional context

Add any other context or screenshots about the feature request here.

Add an experimental MongoDbToolset under google.adk.integrations.mongodb
that lets agents run Atlas Vector Search ($vectorSearch) and hybrid
full-text + vector search ($rankFusion, reciprocal rank fusion) against
a MongoDB database. Search results exclude raw embedding vectors by
default and include a search_score; limits and defaults are controlled
via MongoDbToolSettings.

Usage:
  toolset = MongoDbToolset(
      connection_string="mongodb+srv://...",
      database_name="my_db",
  )
  agent = Agent(model="gemini-2.5-flash", tools=[toolset])

Requires the pymongo package: pip install google-adk[mongodb].
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants