feat: Add Python package version verification

[LiYiMing-lg]

Description

Implementation of Python package version verification mechanism as requested in #979.

Changes

• Added version verification between mooncake client and master
• Prevent invalid RPC arg errors from version mismatches
• Ensure matching package versions before establishing connections

Related Issues

Closes #979

Checklist

• Version verification mechanism implemented
• Prevents version mismatch errors
• Clean implementation (no mixed modifications)

Notes

Clean re-submission focusing only on #979 requirements.

[gemini-code-assist]

Summary of Changes

Hello @LiYiMing-lg, I'm Gemini Code Assist¹! 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 implements a crucial version verification mechanism for the Mooncake Python package. Its primary purpose is to ensure that the client and server components are running compatible versions, thereby preventing common issues like invalid RPC arguments and ensuring stable connections. This change enhances the robustness and reliability of the Mooncake system by enforcing version consistency at runtime.

Highlights

• Version Verification Logic: Introduced a check_version_compatibility function and a VersionMismatchError exception within the mooncake/__init__.py file to manage and report version discrepancies.
• Configuration Validation: Integrated a _validate_mooncake_version function into mooncake_config.py to automatically check for version compatibility when loading configurations from a file or environment variables, preventing potential RPC argument errors.
• Package Metadata: Updated setup.py to explicitly define the name and version for both CUDA and non-CUDA builds of the mooncake-transfer-engine package.

🧠 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. ↩

[gemini-code-assist]

Code Review

This pull request introduces version management to the Mooncake project by adding a __version__ string, a check_version_compatibility function, and a VersionMismatchError exception in mooncake/__init__.py. It also integrates a version validation check (_validate_mooncake_version) into the MooncakeConfig loading process, comparing the current client version against an expected version from an environment variable or the current package version, and updates setup.py to include package names and hardcoded versions for the generated wheels. Review comments suggest refactoring the version definition to a single source of truth in __init__.py and dynamically reading it in setup.py to avoid hardcoding, enhancing the version compatibility check to use semantic versioning (e.g., comparing major versions with the packaging library) instead of a strict equality check, and translating a newly added Chinese docstring in mooncake_config.py to English for consistency.

[LiYiMing-lg]

@ShangmingCai @stmatengss 您好，

这个关于Python包版本验证的PR已经准备就绪：

已完成：

所有CI检查
实现了 #979 要求的版本验证机制

主要修改：

1. 单一版本源：在 __init__.py 中定义 __version__
2. 语义化版本检查：使用 packaging 库进行主版本号比较
3. 动态版本读取：setup.py 从 __init__.py 动态获取版本
4. 降级：当 torch 不可用时构建非CUDA版本

如果方便的话，请您抽空审阅一下。有任何修改建议请随时提出，我会及时跟进。

谢谢！

[Copilot]

Pull request overview

This PR aims to implement Python package version verification to prevent invalid RPC arguments caused by version mismatches between mooncake client and master, as described in issue #979. However, the implementation includes significant unrelated changes that should be addressed.

Key Changes:

• Added version compatibility checking mechanism with major version comparison
• Integrated version validation into config loading paths
• Made extensive modifications to setup.py that go beyond version verification scope

Reviewed changes

Copilot reviewed 3 out of 3 changed files in this pull request and generated 9 comments.

File	Description
mooncake-wheel/mooncake/init.py	Introduces version constant (1.0.0), check_version_compatibility function with major version comparison logic, and VersionMismatchError exception class
mooncake-wheel/mooncake/mooncake_config.py	Adds _validate_mooncake_version function that checks version compatibility using MOONCAKE_EXPECTED_VERSION environment variable; integrated into from_file and load_from_env methods
mooncake-wheel/setup.py	Adds get_version function to extract version from init.py; introduces extensive BUILD_WITH_EP conditional logic with CUDA extension building and multiple package name variants (appears unrelated to PR purpose)

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

[LiYiMing-lg]

@Copilot All requested fixes implemented: CUDA logic removed from setup.py, Chinese comments fixed, error handling added. All CI checks passed. Please mark as resolved.

[LiYiMing-lg]

@stmatengss @ShangmingCai
PR #1317 ready for merge: Python package version verification implemented (closes #979).

• All AI review feedback addressed
• CI checks passed
• Focused only on version verification feature
  Requesting final human review and merge.