Skip to content

Conversation

doujiang24
Copy link
Collaborator

WIth this PR, we could do test with P2PHANDSHAKE, i.e.

MC_METADATA_SERVER=P2PHANDSHAKE TARGET_SERVER_NAME=$LOCALIP:12345 PROTOCOL=rdma python3 target.py

TARGET_SERVER_NAME=$TARGETIPPORT INITIATOR_SERVER_NAME=$LOCALIP:12347 MC_METADATA_SERVER=P2PHANDSHAKE PROTOCOL=rdma python3 initiator_test.py

Copy link

Summary of Changes

Hello @doujiang24, 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 TransferEngine's initiator test script to support the P2PHANDSHAKE metadata server. The primary purpose is to ensure that the test environment correctly configures the initiator's communication endpoint, specifically its RPC port, when operating in a peer-to-peer handshaking scenario. This change allows for more robust testing of the TransferEngine's capabilities under different metadata server configurations.

Highlights

  • P2PHANDSHAKE Compatibility: The initiator test script has been updated to correctly function when the P2PHANDSHAKE metadata server is utilized, enabling comprehensive testing for this configuration.
  • Dynamic Port Configuration: The initiator's server name is now dynamically configured to use the correct RPC port obtained from the adaptor when P2PHANDSHAKE is active, ensuring proper communication setup.
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

@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 updates the initiator_test script to support P2PHANDSHAKE mode by dynamically updating the initiator server name with the port assigned by the transfer engine. The change is correct in its intent. However, the implementation for parsing the hostname is not robust and will fail for IPv6 addresses. I've suggested a fix to handle both IPv4 and IPv6 addresses correctly.

Copy link
Collaborator

@alogfans alogfans left a comment

Choose a reason for hiding this comment

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

LGTM. You can consider the advise from AI assistant. Just for suggestion

Copy link
Collaborator

@ShangmingCai ShangmingCai left a comment

Choose a reason for hiding this comment

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

LGTM. But the CI fails when preparing the dependency, maybe push an empty commit to retrigger the CI.

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
@doujiang24
Copy link
Collaborator Author

Thanks @alogfans and @ShangmingCai , suggestion from from AI assistant accepted and CI is happy now, merging.

@doujiang24 doujiang24 merged commit 2d80ed1 into kvcache-ai:main Oct 5, 2025
13 checks passed
@doujiang24 doujiang24 deleted the initiator_test branch October 5, 2025 03:41
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