Skip to content

Conversation

@prasadskarmarkar
Copy link

@prasadskarmarkar prasadskarmarkar commented Jan 6, 2026

Fixes #677

Updated A2ASendMessageExecutor and A2ARemoteConfiguration to support dependency injection of session, artifact, and memory services. This allows persistent service implementations to replace hard-coded in-memory versions.

Changes:

  • Modified A2ASendMessageExecutor constructor to accept BaseSessionService, BaseArtifactService, and BaseMemoryService parameters
  • Updated A2ARemoteConfiguration to autowire service beans with fallback to in-memory defaults when custom beans are not provided
  • Added comprehensive test coverage for service injection scenarios

Tasks

  • Update A2ASendMessageExecutor to accept injected services
  • Modify A2ARemoteConfiguration to autowire service beans
  • Add unit tests for A2ASendMessageExecutor
  • Add integration tests for service injection
  • Add Spring configuration tests for custom services
  • Address code review feedback

@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello @prasadskarmarkar, 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 significantly enhances the A2ASendMessageExecutor by introducing dependency injection for its core services. This change allows for greater flexibility and extensibility, enabling developers to easily swap out default in-memory service implementations with custom, potentially persistent or distributed, alternatives. The A2ARemoteConfiguration has been updated to seamlessly integrate with this new design, automatically wiring available service beans while gracefully falling back to in-memory versions if no custom implementations are found. This refactoring improves the modularity and testability of the A2A runner component.

Highlights

  • Service Injection Enabled: The A2ASendMessageExecutor now supports dependency injection for BaseSessionService, BaseArtifactService, and BaseMemoryService, allowing replacement of hard-coded in-memory implementations.
  • Flexible Configuration: A2ARemoteConfiguration has been updated to autowire these services, providing a graceful fallback to in-memory defaults if custom beans are not explicitly provided in the Spring context.
  • Comprehensive Testing: New unit, integration, and Spring configuration tests have been added to thoroughly validate the correct behavior of service injection and fallback mechanisms.

🧠 New Feature in Public Preview: You can now enable Memory to help Gemini Code Assist learn from your team's feedback. This makes future code reviews more consistent and personalized to your project's style. Click here to enable Memory in your admin console.

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.

Copy link
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 successfully implements dependency injection for "A2ASendMessageExecutor", enhancing its flexibility by allowing external services to be provided. The changes in "A2ASendMessageExecutor.java" correctly update the constructor to accept "Base*Service" interfaces, and the extensive Javadoc clearly explains the new injection model. The "A2ARemoteConfiguration.java" effectively integrates these changes with Spring's autowiring, providing sensible fallbacks to in-memory implementations when custom beans are not supplied. Comprehensive test coverage, including unit, integration, and Spring configuration tests, ensures the robustness of the new functionality. The "pom.xml" updates correctly add necessary test dependencies.

@prasadskarmarkar prasadskarmarkar force-pushed the feat/a2a-configurable-services branch from f4356eb to f9abd0e Compare January 6, 2026 03:09
@prasadskarmarkar
Copy link
Author

/gemini review

Copy link
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 A2ASendMessageExecutor and A2ARemoteConfiguration to support dependency injection for services, which is a great improvement for flexibility and testability. The changes are well-implemented, using base interfaces and providing sensible defaults with @ConditionalOnMissingBean in the Spring configuration. The test coverage is comprehensive, including unit, integration, and Spring configuration tests. My feedback focuses on improving the maintainability and robustness of the new test code by reducing duplication and adding more specific assertions.

Updated A2ASendMessageExecutor and A2ARemoteConfiguration to support
dependency injection of session, artifact, and memory services. This
allows persistent service implementations to replace hard-coded
in-memory versions.

Changes:
- Modified A2ASendMessageExecutor constructor to accept BaseSessionService,
  BaseArtifactService, and BaseMemoryService parameters
- Updated A2ARemoteConfiguration to autowire service beans with fallback
  to in-memory defaults when custom beans are not provided
- Added comprehensive test coverage for service injection scenarios

Tasks:
[x] Update A2ASendMessageExecutor to accept injected services
[x] Modify A2ARemoteConfiguration to autowire service beans
[x] Add unit tests for A2ASendMessageExecutor
[x] Add integration tests for service injection
[x] Add Spring configuration tests for custom services
@prasadskarmarkar prasadskarmarkar force-pushed the feat/a2a-configurable-services branch from f9abd0e to c24c3b7 Compare January 6, 2026 05:21
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.

Cannot customize session/artifact/memory services for agents using A2A runner

1 participant