Skip to content

chore(tools/mssql,tools/mysql,tools/neo4j,tools/oceanbase): decouple source config from tools#2067

Merged
Yuan325 merged 1 commit into
config-branchfrom
config-mssql-etc
Dec 13, 2025
Merged

chore(tools/mssql,tools/mysql,tools/neo4j,tools/oceanbase): decouple source config from tools#2067
Yuan325 merged 1 commit into
config-branchfrom
config-mssql-etc

Conversation

@Yuan325

@Yuan325 Yuan325 commented Nov 27, 2025

Copy link
Copy Markdown
Contributor

Within Invoke() and RequiresClientAuthorization(), get Source from ResourceManager and use it to call functions directly.

@Yuan325 Yuan325 requested a review from a team November 27, 2025 17:27
@gemini-code-assist

Copy link
Copy Markdown
Contributor

Summary of Changes

Hello @Yuan325, I'm Gemini Code Assist1! 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 introduces a significant architectural improvement by decoupling the database source configuration from the individual tools that utilize them. Instead of tools maintaining direct references to specific database connection pools, they now dynamically fetch the required source information from a central ResourceManager at runtime. This change enhances modularity, reduces tight coupling between tools and specific source implementations, and simplifies the overall structure of the tool definitions.

Highlights

  • Decoupled Source Configuration: Tools no longer directly import or store specific database source types (e.g., mssql, mysql, neo4j, oceanbase).
  • Dynamic Source Retrieval: Database connection details are now retrieved dynamically from a ResourceManager (or SourceProvider) during the Invoke and Initialize phases of a tool's lifecycle.
  • Simplified Tool Structs: The Tool structs across various database integrations have been streamlined by removing direct fields for database connection pools (e.g., *sql.DB, neo4j.DriverWithContext).
  • Updated Authorization Interface: The RequiresClientAuthorization method signature has been updated to return an error alongside the boolean, allowing for more robust error handling.
  • Removed Static Compatibility Checks: Static compatibleSources arrays and their associated checks during tool initialization have been removed, as compatibility is now verified dynamically when the source is retrieved.
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.

Footnotes

  1. 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 gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request refactors several tools to decouple them from the source configuration by moving the source retrieval logic from the Initialize method to the Invoke method. This is a good architectural improvement that makes the tools more stateless and flexible. The changes are consistent across MSSQL, MySQL, Neo4j, and OceanBase tools.

I found a critical bug in the mysqllistactivequeries tool where the refactoring was not done correctly, which would cause the tool to fail at runtime. I've provided comments with suggestions to fix this issue.

The rest of the changes look good and correctly implement the intended decoupling.

@averikitsch averikitsch assigned averikitsch and unassigned duwenxin99 Dec 1, 2025
Base automatically changed from config-server to config-branch December 9, 2025 01:36
@Yuan325 Yuan325 force-pushed the config-mssql-etc branch 2 times, most recently from 95d0b84 to 08cd7dc Compare December 11, 2025 18:20
@Yuan325

Yuan325 commented Dec 11, 2025

Copy link
Copy Markdown
Contributor Author

/gemini review

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request effectively decouples the source configuration from the tool initialization process by moving the source retrieval and validation logic into the Invoke method. This is a solid refactoring that enhances modularity and consistency across the various tools. The changes are applied consistently for the most part. However, I've identified one file where the refactoring appears incomplete, resulting in duplicated source validation logic. My review includes a specific comment to address this inconsistency. Additionally, the signature change for RequiresClientAuthorization to return an error is a good improvement for error handling.

@Yuan325 Yuan325 merged commit ad91605 into config-branch Dec 13, 2025
6 of 11 checks passed
@Yuan325 Yuan325 deleted the config-mssql-etc branch December 13, 2025 15:11
Yuan325 added a commit that referenced this pull request Dec 17, 2025
…source config from tools (#2067)

Within Invoke() and RequiresClientAuthorization(), get Source from
ResourceManager and use it to call functions directly.
Yuan325 added a commit that referenced this pull request Dec 18, 2025
…source config from tools (#2067)

Within Invoke() and RequiresClientAuthorization(), get Source from
ResourceManager and use it to call functions directly.
Yuan325 added a commit that referenced this pull request Dec 19, 2025
…source config from tools (#2067)

Within Invoke() and RequiresClientAuthorization(), get Source from
ResourceManager and use it to call functions directly.
Yuan325 added a commit that referenced this pull request Dec 19, 2025
…source config from tools (#2067)

Within Invoke() and RequiresClientAuthorization(), get Source from
ResourceManager and use it to call functions directly.
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.

3 participants