feat(dataplex): Add support for lookup context tool. by AnujJhunjhunwala · Pull Request #2744 · googleapis/mcp-toolbox

AnujJhunjhunwala · 2026-03-13T16:46:29Z

Added support for lookup context tool in Dataplex.
Fixes #2743

gemini-code-assist · 2026-03-13T16:46:56Z

Summary of Changes

Hello, I'm Gemini Code Assist¹! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request significantly enhances Dataplex integration by introducing a new dataplex-lookup-context tool. This tool allows users to efficiently retrieve metadata for data assets from the Dataplex Catalog, improving data discovery and management capabilities. The changes include the core implementation, configuration, documentation, and robust testing to ensure reliability.

Highlights

New Dataplex Tool: Introduced the dataplex-lookup-context tool, designed to retrieve metadata for data assets within Dataplex Catalog.
Toolset Integration: Integrated the new lookup_context tool into the existing dataplex_tools toolset, making it available for use.
Documentation & Testing: Added comprehensive documentation for the dataplex-lookup-context tool and included integration tests to ensure its functionality and proper authorization handling.

Changelog

cmd/internal/imports.go
- Added an import for the new dataplexlookupcontext package.
cmd/internal/tools_file_test.go
- Updated the dataplex_tools test configuration to include the lookup_context tool.
docs/DATAPLEX_README.md
- Updated the Dataplex README to list the newly added lookup_context tool and its description.
docs/en/resources/tools/dataplex/dataplex-lookup-context.md
- Created a new documentation page detailing the dataplex-lookup-context tool, its purpose, parameters, IAM requirements, and examples.
internal/prebuiltconfigs/tools/dataplex.yaml
- Added a new lookup_context tool configuration.
- Included lookup_context in the dataplex_tools toolset.
internal/tools/dataplex/dataplexlookupcontext/dataplexlookupcontext.go
- Added the core implementation for the dataplex-lookup-context tool, including its configuration, initialization, and invocation logic.
internal/tools/dataplex/dataplexlookupcontext/dataplexlookupcontext_test.go
- Added unit tests for parsing dataplex-lookup-context tool configurations from YAML.
tests/dataplex/dataplex_integration_test.go
- Defined DataplexLookupContextToolType constant.
- Added a call to runDataplexLookupContextToolInvokeTest in TestDataplexToolEndpoints.
- Added configurations for my-dataplex-lookup-context-tool and my-auth-dataplex-lookup-context-tool.
- Implemented runDataplexLookupContextToolInvokeTest to verify the functionality and authorization of the new tool.

Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature	Command	Description
Code Review	`/gemini review`	Performs a code review for the current pull request in its current state.
Pull Request Summary	`/gemini summary`	Provides a summary of the current pull request in its current state.
Comment	@gemini-code-assist	Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help	`/gemini help`	Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for GitHub and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution. ↩

gemini-code-assist

Code Review

This pull request introduces a new lookup_context tool for Dataplex, including its implementation, documentation, and tests. The changes are well-structured and follow existing patterns. I've identified one high-severity issue in the tool's initialization logic where custom parameters from the configuration are ignored, and I've provided a suggestion to fix it.

estshy

Can we also correct recommended system prompt?

github-actions · 2026-03-13T17:13:54Z

🔎 Preview at https://googleapis.github.io/genai-toolbox/previews/PR-2744/

github-actions · 2026-03-19T20:17:56Z

🔎 Preview at https://googleapis.github.io/genai-toolbox/previews/PR-2744/

github-actions · 2026-03-23T09:13:36Z

🔎 Preview at https://googleapis.github.io/genai-toolbox/previews/PR-2744/

github-actions · 2026-03-24T13:43:46Z

🔎 Preview at https://googleapis.github.io/genai-toolbox/previews/PR-2744/

github-actions · 2026-03-24T15:26:40Z

🔎 Preview at https://googleapis.github.io/genai-toolbox/previews/PR-2744/

github-actions · 2026-03-24T15:42:46Z

🔎 Preview at https://googleapis.github.io/genai-toolbox/previews/PR-2744/

github-actions · 2026-03-24T15:59:00Z

🔎 Preview at https://googleapis.github.io/genai-toolbox/previews/PR-2744/

estshy · 2026-03-24T16:52:48Z

/gcbrun

github-actions · 2026-03-24T16:58:25Z

🔎 Preview at https://googleapis.github.io/genai-toolbox/previews/PR-2744/

Added support for lookup context tool in Dataplex. Fixes #2743 --------- Co-authored-by: Mateusz Nowak <matnow@google.com> Co-authored-by: Wenxin Du <117315983+duwenxin99@users.noreply.github.com> facb69d

…pis#2744) Added support for lookup context tool in Dataplex. Fixes googleapis#2743 --------- Co-authored-by: Mateusz Nowak <matnow@google.com> Co-authored-by: Wenxin Du <117315983+duwenxin99@users.noreply.github.com> facb69d

🤖 I have created a release *beep* *boop* --- ## [0.31.0](v0.30.0...v0.31.0) (2026-03-26) ### ⚠ BREAKING CHANGES * release upgraded docsite ([#2831](#2831)) * **http:** sanitize non-2xx error output ([#2654](#2654)) * add a new `enable-api` flag ([#2846](#2846)) * remove deprecations and update tools-file flag ([#2806](#2806)) ### Features * Add a new `enable-api` flag ([#2846](#2846)) ([7a070da](7a070da)) * **auth:** Add generic `authService` type for MCP ([#2619](#2619)) ([f6678f8](f6678f8)) * **auth:** Add Protected Resource Metadata endpoint ([#2698](#2698)) ([b53dcf2](b53dcf2)) * **auth:** Support manual PRM override ([#2717](#2717)) ([283e4e3](283e4e3)) * **dataplex:** Add support for lookup context tool. ([#2744](#2744)) ([facb69d](facb69d)) * Remove deprecations and update tools-file flag ([#2806](#2806)) ([ab64c95](ab64c95)) ### Bug Fixes * **ci:** Remove search index generation from preview deployment workflow ([#2859](#2859)) ([f8891b8](f8891b8)) * **docs:** Skip empty folders in pagination & reduce PR comment noise ([#2853](#2853)) ([9ebd93a](9ebd93a)) * **http:** Sanitize non-2xx error output ([#2654](#2654)) ([5bef954](5bef954)) * **skills:** Fix integer parameter parsing through agent skills ([#2847](#2847)) ([4564efe](4564efe)) ### Documentation * Release upgraded docsite ([#2831](#2831)) ([5b25ce0](5b25ce0)) --- This PR was generated with [Release Please](https://github.com/googleapis/release-please). See [documentation](https://github.com/googleapis/release-please#release-please). --------- Co-authored-by: release-please[bot] <55107282+release-please[bot]@users.noreply.github.com> Co-authored-by: Yuan Teoh <45984206+Yuan325@users.noreply.github.com>

🤖 I have created a release *beep* *boop* --- ## [0.31.0](v0.30.0...v0.31.0) (2026-03-26) ### ⚠ BREAKING CHANGES * release upgraded docsite ([#2831](#2831)) * **http:** sanitize non-2xx error output ([#2654](#2654)) * add a new `enable-api` flag ([#2846](#2846)) * remove deprecations and update tools-file flag ([#2806](#2806)) ### Features * Add a new `enable-api` flag ([#2846](#2846)) ([7a070da](7a070da)) * **auth:** Add generic `authService` type for MCP ([#2619](#2619)) ([f6678f8](f6678f8)) * **auth:** Add Protected Resource Metadata endpoint ([#2698](#2698)) ([b53dcf2](b53dcf2)) * **auth:** Support manual PRM override ([#2717](#2717)) ([283e4e3](283e4e3)) * **dataplex:** Add support for lookup context tool. ([#2744](#2744)) ([facb69d](facb69d)) * Remove deprecations and update tools-file flag ([#2806](#2806)) ([ab64c95](ab64c95)) ### Bug Fixes * **ci:** Remove search index generation from preview deployment workflow ([#2859](#2859)) ([f8891b8](f8891b8)) * **docs:** Skip empty folders in pagination & reduce PR comment noise ([#2853](#2853)) ([9ebd93a](9ebd93a)) * **http:** Sanitize non-2xx error output ([#2654](#2654)) ([5bef954](5bef954)) * **skills:** Fix integer parameter parsing through agent skills ([#2847](#2847)) ([4564efe](4564efe)) ### Documentation * Release upgraded docsite ([#2831](#2831)) ([5b25ce0](5b25ce0)) --- This PR was generated with [Release Please](https://github.com/googleapis/release-please). See [documentation](https://github.com/googleapis/release-please#release-please). --------- Co-authored-by: release-please[bot] <55107282+release-please[bot]@users.noreply.github.com> Co-authored-by: Yuan Teoh <45984206+Yuan325@users.noreply.github.com> c6b811c

🤖 I have created a release *beep* *boop* --- ## [0.31.0](googleapis/mcp-toolbox@v0.30.0...v0.31.0) (2026-03-26) ### ⚠ BREAKING CHANGES * release upgraded docsite ([googleapis#2831](googleapis#2831)) * **http:** sanitize non-2xx error output ([googleapis#2654](googleapis#2654)) * add a new `enable-api` flag ([googleapis#2846](googleapis#2846)) * remove deprecations and update tools-file flag ([googleapis#2806](googleapis#2806)) ### Features * Add a new `enable-api` flag ([googleapis#2846](googleapis#2846)) ([7a070da](googleapis@7a070da)) * **auth:** Add generic `authService` type for MCP ([googleapis#2619](googleapis#2619)) ([f6678f8](googleapis@f6678f8)) * **auth:** Add Protected Resource Metadata endpoint ([googleapis#2698](googleapis#2698)) ([b53dcf2](googleapis@b53dcf2)) * **auth:** Support manual PRM override ([googleapis#2717](googleapis#2717)) ([283e4e3](googleapis@283e4e3)) * **dataplex:** Add support for lookup context tool. ([googleapis#2744](googleapis#2744)) ([facb69d](googleapis@facb69d)) * Remove deprecations and update tools-file flag ([googleapis#2806](googleapis#2806)) ([ab64c95](googleapis@ab64c95)) ### Bug Fixes * **ci:** Remove search index generation from preview deployment workflow ([googleapis#2859](googleapis#2859)) ([f8891b8](googleapis@f8891b8)) * **docs:** Skip empty folders in pagination & reduce PR comment noise ([googleapis#2853](googleapis#2853)) ([9ebd93a](googleapis@9ebd93a)) * **http:** Sanitize non-2xx error output ([googleapis#2654](googleapis#2654)) ([5bef954](googleapis@5bef954)) * **skills:** Fix integer parameter parsing through agent skills ([googleapis#2847](googleapis#2847)) ([4564efe](googleapis@4564efe)) ### Documentation * Release upgraded docsite ([googleapis#2831](googleapis#2831)) ([5b25ce0](googleapis@5b25ce0)) --- This PR was generated with [Release Please](https://github.com/googleapis/release-please). See [documentation](https://github.com/googleapis/release-please#release-please). --------- Co-authored-by: release-please[bot] <55107282+release-please[bot]@users.noreply.github.com> Co-authored-by: Yuan Teoh <45984206+Yuan325@users.noreply.github.com> c6b811c

AnujJhunjhunwala and others added 4 commits March 11, 2026 16:16

Add lookup context tool

833b3e2

Merge branch 'googleapis:main' into tool/dataplex-lookup-context

c83cdf7

Add integration tests and update docs

094412a

Merge branch 'googleapis:main' into tool/dataplex-lookup-context

060e4c7

AnujJhunjhunwala requested review from a team as code owners March 13, 2026 16:46

blunderbuss-gcf Bot assigned duwenxin99 Mar 13, 2026

gemini-code-assist Bot reviewed Mar 13, 2026

View reviewed changes

Comment thread internal/tools/dataplex/dataplexlookupcontext/dataplexlookupcontext.go

estshy approved these changes Mar 13, 2026

View reviewed changes

Merge branch 'main' into tool/dataplex-lookup-context

378ef19

AnujJhunjhunwala and others added 4 commits March 16, 2026 11:51

Merge branch 'main' into tool/dataplex-lookup-context

ab328c2

Update docs

16ad3c6

Modify parameter descriptions

dada9f4

Modify year in header

3e71426

AnujJhunjhunwala changed the title ~~Add support for lookup context tool in Dataplex.~~ feat(dataplex): Add support for lookup context tool in Dataplex. Mar 16, 2026

AnujJhunjhunwala changed the title ~~feat(dataplex): Add support for lookup context tool in Dataplex.~~ feat(dataplex): Add support for lookup context tool. Mar 16, 2026

AnujJhunjhunwala and others added 2 commits March 17, 2026 13:39

Merge branch 'main' into tool/dataplex-lookup-context

35c9ad0

Merge branch 'main' into tool/dataplex-lookup-context

abe37cb

Merge branch 'main' into tool/dataplex-lookup-context

539c676

AnujJhunjhunwala added 5 commits March 24, 2026 08:48

Update client library

2ef4436

temp remove tool file test changes

06901ef

Merge branch 'main' into tool/dataplex-lookup-context

afd80d2

Update integration tests

15de6f6

Merge branch 'main' into tool/dataplex-lookup-context

edab691

duwenxin99 approved these changes Mar 24, 2026

View reviewed changes

Add tool to config tests

14e0cba

AnujJhunjhunwala added 2 commits March 24, 2026 15:37

Fix order of tools

c98a48e

Merge branch 'main' into tool/dataplex-lookup-context

9802c9b

AnujJhunjhunwala added 2 commits March 24, 2026 15:48

Update docs

d410173

Merge branch 'main' into tool/dataplex-lookup-context

735db1f

duwenxin99 enabled auto-merge (squash) March 24, 2026 15:54

Merge branch 'main' into tool/dataplex-lookup-context

1283731

duwenxin99 merged commit facb69d into googleapis:main Mar 24, 2026
21 checks passed

release-please Bot mentioned this pull request Mar 24, 2026

chore(main): release 0.31.0 #2814

Merged

estshy deleted the tool/dataplex-lookup-context branch March 25, 2026 17:53

BrewTestBot mentioned this pull request Mar 27, 2026

mcp-toolbox 0.31.0 Homebrew/homebrew-core#274467

Merged

Conversation

AnujJhunjhunwala commented Mar 13, 2026

Uh oh!

gemini-code-assist Bot commented Mar 13, 2026

Summary of Changes

Highlights

Footnotes

Uh oh!

gemini-code-assist Bot left a comment

Choose a reason for hiding this comment

Code Review

Uh oh!

Uh oh!

estshy left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

github-actions Bot commented Mar 13, 2026

Uh oh!

github-actions Bot commented Mar 19, 2026

Uh oh!

github-actions Bot commented Mar 23, 2026

Uh oh!

github-actions Bot commented Mar 24, 2026

Uh oh!

github-actions Bot commented Mar 24, 2026

Uh oh!

github-actions Bot commented Mar 24, 2026

Uh oh!

github-actions Bot commented Mar 24, 2026

Uh oh!

estshy commented Mar 24, 2026

Uh oh!

github-actions Bot commented Mar 24, 2026

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants