Skip to content

refactor(acp): delegate prompt turn processing logic to GeminiClient#26222

Merged
ruomengz merged 1 commit intomainfrom
sripas_acp_refactor_phase2
Apr 30, 2026
Merged

refactor(acp): delegate prompt turn processing logic to GeminiClient#26222
ruomengz merged 1 commit intomainfrom
sripas_acp_refactor_phase2

Conversation

@sripasg
Copy link
Copy Markdown
Contributor

@sripasg sripasg commented Apr 29, 2026

Summary

Replaces low-level direct calls to chat modules in acpSession.ts with GeminiClient.sendMessageStream(). This establishes parity with core terminal features—including chat compression, token masking safeguards, context validation limits, and execution tracing.

Details

  • Prompt Pipeline Refactoring: Migrates Session.prompt() stream logic from calling GeminiChat.sendMessageStream() to GeminiClient.sendMessageStream().
  • Structured Events Mapping: Converts underlying core state notifications (GeminiEventType) to line-delimited protocol interactions format.
  • Enhanced Telemetry Tracking: Introduces variable client identifiers distinguishing incoming requests sources (tui, acp, cli-command, or a2a-server).
  • Automated Test Suites Extensions: Appends core integration case parameters tests into packages/cli/src/acp/acpSession.test.ts.

Before the change

Prior to this change, ACP interactive turn system bypassed the high-level client lifecycle manager (GeminiClient) and orchestrated low-level calls directly via internal packages/core components.

sequenceDiagram
    participant ACP Session
    participant Routing System
    participant GeminiChat

    ACP Session->>Routing System: router.route(RoutingContext)
    Routing System-->>ACP Session: Returns model target
    ACP Session->>GeminiChat: sendMessageStream(model, request parts)
    loop Stream Processing
        GeminiChat-->>ACP Session: yield Raw StreamEvent.CHUNK
        Note over ACP Session: - Manually parses usage data<br/>- Manually formats token info<br/>- Processes raw candidates content
    end
Loading

Pipeline Characteristics

  • Direct Interaction: Low-level consumption of raw response streams directly from GeminiChat.sendMessageStream().
  • Duplicated Responsibilities: Low-level token limits calculations, candidate conversions, and parsing were written directly inside the Session stream.
  • Bypassed Engine Features: ACP turn streams operated outside established core features:
    • Automated session history compression
    • Context optimization / Output text masking
    • Context constraints enforcement (overflow warnings)
    • Process steering (BeforeAgent & AfterAgent orchestration hooks)

After/With this change

With this commit, the interactive turn orchestrator in acpSession.ts was updated to use the high-level API provided by GeminiClient.sendMessageStream().

sequenceDiagram
    participant ACP Session
    participant GeminiClient
    participant Core Orchestrator

    ACP Session->>GeminiClient: sendMessageStream(currentParts)
    opt Client Interceptors
        GeminiClient->>Core Orchestrator: Triggers Hooks & History Management
    end
    loop Engine Events
        GeminiClient-->>ACP Session: yield GeminiEventType (Content, Thought, ToolCall, Finished)
        Note over ACP Session: Maps clean protocol payloads<br/>to frontends (e.g., VS Code IDE)
    end
Loading

Pipeline Characteristics

  • Abstracted Routing: Bounded turn orchestration is performed implicitly by the GeminiClient pipeline.
  • Decoupled Event Model: Stream consumers use high-level events defined in GeminiEventType rather than low-level JSON-derived candidate chunks.
  • Feature Parity: ACP flows leverage the identical capabilities driving primary terminal workflows:
    • Loop detection tracking and resolution hooks
    • Safety masking constraints and runtime compression optimizations
    • Trace tracking tagged with explicit contextual identification parameters (clientName set as acp, tui, cli-command, or acp-* IDE extensions)

Summary Comparison

Aspect Before After
Communication Target GeminiChat (Direct low-level interface) GeminiClient (Consolidated orchestrator)
Stream Event Model StreamEventType.CHUNK JSON objects GeminiEventType abstractions
Routing Strategy Manual call to ModelRouterService Automatic core tracking inside client pipeline
Feature Inclusions Basic prompts stream & execution Automated session compression, execution hooks, loop detection, output masking
Origin Tagging Unidentified client targets Explicitly configured sources (tui, acp, cli-command)

Related Issues

maintaince-gemini-cli#1604

How to Validate

Connect to ACP via Jetbrains and ensure all regular prompt workflows behave as expected.

Pre-Merge Checklist

  • [NA] Updated relevant documentation and README (if needed)
  • [✅] Added/updated tests (if needed)
  • [NA] 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

@sripasg sripasg requested a review from a team as a code owner April 29, 2026 20:17
@gemini-cli
Copy link
Copy Markdown
Contributor

gemini-cli Bot commented Apr 29, 2026

Hi @sripasg, 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!

@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 refactors the ACP session prompt processing logic to delegate directly to the GeminiClient. By moving away from low-level chat module calls, the implementation now benefits from centralized core features such as chat compression, context validation, and improved execution tracing. Additionally, the update introduces better telemetry tracking and robust error handling for stream interactions.

Highlights

  • Prompt Processing Refactor: Migrated session prompt logic to use GeminiClient.sendMessageStream(), ensuring consistency with core terminal features like token masking and context validation.
  • Event Handling: Updated event mapping to use GeminiEventType, improving the handling of structured protocol interactions and error states.
  • Telemetry Enhancements: Introduced granular client identifiers (tui, acp, cli-command, a2a-server) to improve request source tracking.
  • Test Suite Expansion: Extended integration tests in acpSession.test.ts to cover new event types and simulation scenarios.
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.

@sripasg sripasg self-assigned this Apr 29, 2026
@sripasg sripasg requested a review from ruomengz April 29, 2026 20:18
@github-actions
Copy link
Copy Markdown

github-actions Bot commented Apr 29, 2026

Size Change: +2.16 kB (+0.01%)

Total Size: 33.9 MB

Filename Size Change
./bundle/chunk-4J75CK26.js 0 B -14.7 MB (removed) 🏆
./bundle/chunk-5HWDHS7U.js 0 B -3.43 kB (removed) 🏆
./bundle/chunk-BN4GJ5PN.js 0 B -19.5 kB (removed) 🏆
./bundle/chunk-HZD6ICI2.js 0 B -2.72 MB (removed) 🏆
./bundle/chunk-LTJA22GD.js 0 B -12.6 kB (removed) 🏆
./bundle/chunk-RR7VWYUG.js 0 B -655 kB (removed) 🏆
./bundle/chunk-VQ3Z3CLF.js 0 B -3.8 kB (removed) 🏆
./bundle/chunk-ZW6ZBLYY.js 0 B -49.2 kB (removed) 🏆
./bundle/core-SZWLLU56.js 0 B -48.2 kB (removed) 🏆
./bundle/devtoolsService-SVA6HTB2.js 0 B -28 kB (removed) 🏆
./bundle/gemini-BYR4NDDO.js 0 B -577 kB (removed) 🏆
./bundle/interactiveCli-S3ZBBYPB.js 0 B -1.31 MB (removed) 🏆
./bundle/liteRtServerManager-IETE4ZMD.js 0 B -2.11 kB (removed) 🏆
./bundle/oauth2-provider-IG5O6TH4.js 0 B -9.16 kB (removed) 🏆
./bundle/chunk-3QIMIQKB.js 3.43 kB +3.43 kB (new file) 🆕
./bundle/chunk-6GGSHVQU.js 655 kB +655 kB (new file) 🆕
./bundle/chunk-AW4GRWBV.js 14.7 MB +14.7 MB (new file) 🆕
./bundle/chunk-E42IT5U4.js 12.6 kB +12.6 kB (new file) 🆕
./bundle/chunk-LHXFQ7F4.js 19.5 kB +19.5 kB (new file) 🆕
./bundle/chunk-MUCIFFJX.js 49.2 kB +49.2 kB (new file) 🆕
./bundle/chunk-UOPKNLH7.js 3.8 kB +3.8 kB (new file) 🆕
./bundle/chunk-ZSTNIISK.js 2.72 MB +2.72 MB (new file) 🆕
./bundle/core-XSFGKNN4.js 48.2 kB +48.2 kB (new file) 🆕
./bundle/devtoolsService-6ZNONWH5.js 28 kB +28 kB (new file) 🆕
./bundle/gemini-BYONNR3W.js 579 kB +579 kB (new file) 🆕
./bundle/interactiveCli-VJMC26XH.js 1.31 MB +1.31 MB (new file) 🆕
./bundle/liteRtServerManager-A4LOX5EI.js 2.11 kB +2.11 kB (new file) 🆕
./bundle/oauth2-provider-H2KE22KA.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-IUUIT4SU.js 56.5 kB 0 B
./bundle/chunk-RJTRUG2J.js 39.8 kB 0 B
./bundle/chunk-XRLFHCHC.js 1.97 MB 0 B
./bundle/cleanup-6ZLECNSB.js 0 B -932 B (removed) 🏆
./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/examples/hooks/scripts/on-start.js 188 B 0 B
./bundle/examples/mcp-server/example.js 1.43 kB 0 B
./bundle/gemini.js 5.1 kB 0 B
./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-FN3IAPBT.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/start-E4536TWT.js 0 B -652 B (removed) 🏆
./bundle/tree-sitter-7U6MW5PS.js 274 kB 0 B
./bundle/tree-sitter-bash-34ZGLXVX.js 1.84 MB 0 B
./bundle/cleanup-OOSJN7DQ.js 932 B +932 B (new file) 🆕
./bundle/start-YV7TKKDN.js 652 B +652 B (new file) 🆕

compressed-size-action

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 refactors the session handling logic in acpSession.ts and updates the corresponding tests to support new Gemini event types and improved stream processing. The changes introduce a loop-based turn management system and update the client interaction pattern. I have identified a critical issue regarding the missing GeminiChat instance in the sendMessageStream call, which will break conversation continuity, as well as an efficiency improvement regarding the Zod schema definition inside a loop.

Comment thread packages/cli/src/acp/acpSession.ts
Comment thread packages/cli/src/acp/acpSession.ts Outdated
@gemini-cli gemini-cli Bot added the status/need-issue Pull requests that need to have an associated issue. label Apr 29, 2026
@sripasg sripasg force-pushed the sripas_acp_refactor_phase2 branch from 32b4a85 to eb75669 Compare April 29, 2026 21:18
Replaces low-level direct calls to chat modules in `acpSession.ts` with `GeminiClient.sendMessageStream()`. This establishes parity with core terminal features—including chat compression, token masking safeguards, context validation limits, and execution tracing.

- **Prompt Pipeline Refactoring**: Migrates `Session.prompt()` stream logic from calling `GeminiChat.sendMessageStream()` to `GeminiClient.sendMessageStream()`.
- **Structured Events Mapping**: Converts underlying core state notifications (`GeminiEventType`) to line-delimited protocol interactions format.
- **Enhanced Telemetry Tracking**: Introduces variable client identifiers distinguishing incoming requests sources (`tui`, `acp`, `cli-command`, or `a2a-server`).
- **Automated Test Suites Extensions**: Appends core integration case parameters tests into `packages/cli/src/acp/acpSession.test.ts`.
@sripasg sripasg force-pushed the sripas_acp_refactor_phase2 branch from eb75669 to e68a560 Compare April 29, 2026 22:49
@ruomengz ruomengz added this pull request to the merge queue Apr 29, 2026
Merged via the queue into main with commit 0ccc5ce Apr 30, 2026
28 checks passed
@ruomengz ruomengz deleted the sripas_acp_refactor_phase2 branch April 30, 2026 00:11
kimjune01 pushed a commit to kimjune01/gemini-cli-claude that referenced this pull request May 6, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

status/need-issue Pull requests that need to have an associated issue.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants