Skip to content

feat(core): implement additionalContext for BeforeModel hooks and aggregate context from multiple hooks#23957

Closed
mbleigh wants to merge 5 commits intomainfrom
mb/beforemodel-context
Closed

feat(core): implement additionalContext for BeforeModel hooks and aggregate context from multiple hooks#23957
mbleigh wants to merge 5 commits intomainfrom
mb/beforemodel-context

Conversation

@mbleigh
Copy link
Copy Markdown
Contributor

@mbleigh mbleigh commented Mar 26, 2026

Summary

This PR implements the additionalContext field for BeforeModel hooks and ensures that additional context is properly aggregated when multiple hooks are executed. It also transitions the BeforeAgent hook's context handling to use the more robust BeforeModel infrastructure, removing legacy manual prompt concatenation.

Details

  • Hook Aggregation: Updated HookAggregator to collect additionalContext from all executed hooks (regardless of event type) and join them with newlines.
  • BeforeModel Application: Enhanced BeforeModelHookOutput.applyLLMRequestModifications to parse contents and append the aggregated context (wrapped in <hook_context> tags) to the last user message, or create a new one if necessary.
  • Refactoring: Removed legacy BeforeAgent context injection from HookRunner, relying instead on the standard aggregation and application flow.
  • Type Safety: Fixed several linting errors in packages/core/src/hooks/types.ts and its tests related to unsafe type assertions and any usage.
  • Documentation: Updated docs/hooks/reference.md to document the new additionalContext behavior for BeforeModel hooks.

Related Issues

Related to #24682 (Internal tracking for hook system enhancements)

How to Validate

  1. Run hook-related tests in core:
    npm test -w @google/gemini-cli-core -- src/hooks/hookAggregator.test.ts src/hooks/hookRunner.test.ts src/hooks/types.test.ts
  2. Verify that additionalContext from a BeforeModel hook appears in the model's transcript (wrapped in tags).
  3. Verify that multiple hooks providing context have their context merged with newlines.

Pre-Merge Checklist

  • Updated relevant documentation and README (if needed)
  • Added/updated tests (if needed)
  • Noted breaking changes (if any)
  • Validated on required platforms/methods:
    • MacOS
      • npm run
      • npx
      • Docker
      • Podman
      • Seatbelt
    • Windows
      • npm run
      • npx
      • Docker
    • Linux
      • npm run
      • npx
      • Docker

@mbleigh mbleigh requested review from a team as code owners March 26, 2026 23:05
@gemini-cli
Copy link
Copy Markdown
Contributor

gemini-cli bot commented Mar 26, 2026

Hi @mbleigh, thank you so much for your contribution to Gemini CLI! We really appreciate the time and effort you've put into this.

We're making some updates to our contribution process to improve how we track and review changes. Please take a moment to review our recent discussion post: Improving Our Contribution Process & Introducing New Guidelines.

Key Update: Starting January 26, 2026, the Gemini CLI project will require all pull requests to be associated with an existing issue. Any pull requests not linked to an issue by that date will be automatically closed.

Thank you for your understanding and for being a part of our community!

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 26, 2026

Size Change: +2.78 kB (+0.01%)

Total Size: 34 MB

Filename Size Change
./bundle/chunk-7CWTJVNG.js 0 B -3.15 MB (removed) 🏆
./bundle/chunk-COCTVCSZ.js 0 B -14.8 MB (removed) 🏆
./bundle/core-Q5PCQDKD.js 0 B -45.2 kB (removed) 🏆
./bundle/devtoolsService-JGVDKLVT.js 0 B -28.4 kB (removed) 🏆
./bundle/interactiveCli-SVC5NVFC.js 0 B -1.64 MB (removed) 🏆
./bundle/oauth2-provider-6ARXO6VQ.js 0 B -9.16 kB (removed) 🏆
./bundle/chunk-3R2QQP4M.js 3.15 MB +3.15 MB (new file) 🆕
./bundle/chunk-THR6KEJR.js 14.8 MB +14.8 MB (new file) 🆕
./bundle/core-CQRGOV7A.js 45.2 kB +45.2 kB (new file) 🆕
./bundle/devtoolsService-BRJESVYQ.js 28.4 kB +28.4 kB (new file) 🆕
./bundle/interactiveCli-BSKH2HAW.js 1.64 MB +1.64 MB (new file) 🆕
./bundle/oauth2-provider-UJLXT3NC.js 9.16 kB +9.16 kB (new file) 🆕
ℹ️ View Unchanged
Filename Size Change
./bundle/bundled/third_party/index.js 8 MB 0 B
./bundle/chunk-34MYV7JD.js 2.45 kB 0 B
./bundle/chunk-5AUYMPVF.js 858 B 0 B
./bundle/chunk-5PS3AYFU.js 1.18 kB 0 B
./bundle/chunk-664ZODQF.js 124 kB 0 B
./bundle/chunk-DAHVX5MI.js 206 kB 0 B
./bundle/chunk-GFUOVHXW.js 1.96 MB 0 B
./bundle/chunk-IUUIT4SU.js 56.5 kB 0 B
./bundle/chunk-RJTRUG2J.js 39.8 kB 0 B
./bundle/devtools-36NN55EP.js 696 kB 0 B
./bundle/dist-T73EYRDX.js 356 B 0 B
./bundle/events-XB7DADIJ.js 418 B 0 B
./bundle/gemini.js 552 kB -80 B (-0.01%)
./bundle/getMachineId-bsd-TXG52NKR.js 1.55 kB 0 B
./bundle/getMachineId-darwin-7OE4DDZ6.js 1.55 kB 0 B
./bundle/getMachineId-linux-SHIFKOOX.js 1.34 kB 0 B
./bundle/getMachineId-unsupported-5U5DOEYY.js 1.06 kB 0 B
./bundle/getMachineId-win-6KLLGOI4.js 1.72 kB 0 B
./bundle/memoryDiscovery-ACCRGPX3.js 980 B 0 B
./bundle/multipart-parser-KPBZEGQU.js 11.7 kB 0 B
./bundle/node_modules/@google/gemini-cli-devtools/dist/client/main.js 222 kB 0 B
./bundle/node_modules/@google/gemini-cli-devtools/dist/src/_client-assets.js 229 kB 0 B
./bundle/node_modules/@google/gemini-cli-devtools/dist/src/index.js 13.4 kB 0 B
./bundle/node_modules/@google/gemini-cli-devtools/dist/src/types.js 132 B 0 B
./bundle/sandbox-macos-permissive-open.sb 890 B 0 B
./bundle/sandbox-macos-permissive-proxied.sb 1.31 kB 0 B
./bundle/sandbox-macos-restrictive-open.sb 3.36 kB 0 B
./bundle/sandbox-macos-restrictive-proxied.sb 3.56 kB 0 B
./bundle/sandbox-macos-strict-open.sb 4.82 kB 0 B
./bundle/sandbox-macos-strict-proxied.sb 5.02 kB 0 B
./bundle/src-QVCVGIUX.js 47 kB 0 B
./bundle/tree-sitter-7U6MW5PS.js 274 kB 0 B
./bundle/tree-sitter-bash-34ZGLXVX.js 1.84 MB 0 B

compressed-size-action

@gemini-code-assist
Copy link
Copy Markdown
Contributor

Summary of Changes

Hello, 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 enhances the hook system by standardizing how additional context is provided to LLM requests. By moving context injection into the BeforeModel hook lifecycle, the system now ensures consistent aggregation across multiple hooks and improves maintainability by deprecating legacy manual prompt manipulation. These changes provide a more robust and type-safe way to influence model inputs before they are sent to the LLM.

Highlights

  • Additional Context Support: Implemented the additionalContext field for BeforeModel hooks, allowing developers to inject context into the LLM request wrapped in <hook_context> tags.
  • Hook Aggregation: Updated HookAggregator to automatically collect and merge additionalContext from multiple hooks using newline separators.
  • Refactoring and Cleanup: Transitioned BeforeAgent context handling to the BeforeModel infrastructure, removing legacy manual prompt concatenation and improving type safety.
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.

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.

Copy link
Copy Markdown
Contributor

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

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 introduces the additionalContext field for BeforeModel hooks, enabling hooks to append text to user messages wrapped in <hook_context> tags. The implementation includes logic to aggregate context from multiple hooks and updates the BeforeModelHookOutput to handle the injection into the LLM request. Feedback suggests refactoring the content normalization logic in packages/core/src/hooks/types.ts to improve type safety by removing unnecessary type assertions.

@mbleigh mbleigh force-pushed the mb/beforemodel-context branch from 526d1e5 to cda12a4 Compare March 26, 2026 23:37
@gemini-cli gemini-cli bot added the area/core Issues related to User Interface, OS Support, Core Functionality label Apr 4, 2026
@gemini-cli
Copy link
Copy Markdown
Contributor

gemini-cli bot commented Apr 10, 2026

Hi there! Thank you for your interest in contributing to Gemini CLI.

To ensure we maintain high code quality and focus on our prioritized roadmap, we have updated our contribution policy (see Discussion #17383).

We only guarantee review and consideration of pull requests for issues that are explicitly labeled as 'help wanted'. All other community pull requests are subject to closure after 14 days if they do not align with our current focus areas. For this reason, we strongly recommend that contributors only submit pull requests against issues explicitly labeled as 'help-wanted'.

This pull request is being closed as it has been open for 14 days without a 'help wanted' designation. We encourage you to find and contribute to existing 'help wanted' issues in our backlog! Thank you for your understanding and for being part of our community!

@gemini-cli gemini-cli bot closed this Apr 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area/core Issues related to User Interface, OS Support, Core Functionality

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant