mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-04-28 20:00:06 +00:00
118 lines
4.1 KiB
Python
118 lines
4.1 KiB
Python
"""
|
|
Version command for kt-cli.
|
|
|
|
Displays version information for kt-cli and related packages.
|
|
"""
|
|
|
|
import platform
|
|
from typing import Optional
|
|
|
|
import typer
|
|
|
|
from kt_kernel.cli import __version__
|
|
from kt_kernel.cli.i18n import t
|
|
from kt_kernel.cli.utils.console import console, print_version_table
|
|
from kt_kernel.cli.utils.environment import detect_cuda_version, get_installed_package_version
|
|
|
|
|
|
def _get_sglang_info() -> str:
|
|
"""Get sglang version and installation source information."""
|
|
try:
|
|
import sglang
|
|
|
|
version = getattr(sglang, "__version__", None)
|
|
|
|
if not version:
|
|
version = get_installed_package_version("sglang")
|
|
|
|
if not version:
|
|
return t("version_not_installed")
|
|
|
|
# Try to detect installation source
|
|
from pathlib import Path
|
|
import subprocess
|
|
|
|
if hasattr(sglang, "__file__") and sglang.__file__:
|
|
location = Path(sglang.__file__).parent.parent
|
|
git_dir = location / ".git"
|
|
|
|
if git_dir.exists():
|
|
# Installed from git (editable install)
|
|
try:
|
|
# Get remote URL
|
|
result = subprocess.run(
|
|
["git", "remote", "get-url", "origin"],
|
|
cwd=location,
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=2,
|
|
)
|
|
if result.returncode == 0:
|
|
remote_url = result.stdout.strip()
|
|
# Simplify GitHub URLs
|
|
if "github.com" in remote_url:
|
|
repo_name = remote_url.split("/")[-1].replace(".git", "")
|
|
owner = remote_url.split("/")[-2]
|
|
return f"{version} [dim](GitHub: {owner}/{repo_name})[/dim]"
|
|
return f"{version} [dim](Git: {remote_url})[/dim]"
|
|
except (subprocess.TimeoutExpired, FileNotFoundError, OSError):
|
|
pass
|
|
|
|
# Default: installed from PyPI
|
|
return f"{version} [dim](PyPI)[/dim]"
|
|
|
|
except ImportError:
|
|
return t("version_not_installed")
|
|
|
|
|
|
def version(
|
|
verbose: bool = typer.Option(False, "--verbose", "-v", help="Show detailed version info"),
|
|
) -> None:
|
|
"""Show version information."""
|
|
console.print(f"\n[bold]{t('version_info')}[/bold] v{__version__}\n")
|
|
|
|
# Basic info
|
|
versions = {
|
|
t("version_python"): platform.python_version(),
|
|
t("version_platform"): f"{platform.system()} {platform.release()}",
|
|
}
|
|
|
|
# CUDA version
|
|
cuda_version = detect_cuda_version()
|
|
versions[t("version_cuda")] = cuda_version or t("version_cuda_not_found")
|
|
|
|
print_version_table(versions)
|
|
|
|
# Always show key packages with installation source
|
|
console.print("\n[bold]Packages:[/bold]\n")
|
|
|
|
sglang_info = _get_sglang_info()
|
|
key_packages = {
|
|
t("version_kt_kernel"): get_installed_package_version("kt-kernel") or t("version_not_installed"),
|
|
t("version_sglang"): sglang_info,
|
|
}
|
|
|
|
print_version_table(key_packages)
|
|
|
|
# Show SGLang installation hint if not installed
|
|
if sglang_info == t("version_not_installed"):
|
|
from kt_kernel.cli.utils.sglang_checker import print_sglang_install_instructions
|
|
|
|
console.print()
|
|
print_sglang_install_instructions()
|
|
|
|
if verbose:
|
|
console.print("\n[bold]Additional Packages:[/bold]\n")
|
|
|
|
package_versions = {
|
|
t("version_ktransformers"): get_installed_package_version("ktransformers") or t("version_not_installed"),
|
|
t("version_llamafactory"): get_installed_package_version("llamafactory") or t("version_not_installed"),
|
|
"typer": get_installed_package_version("typer") or t("version_not_installed"),
|
|
"rich": get_installed_package_version("rich") or t("version_not_installed"),
|
|
"torch": get_installed_package_version("torch") or t("version_not_installed"),
|
|
"transformers": get_installed_package_version("transformers") or t("version_not_installed"),
|
|
}
|
|
|
|
print_version_table(package_versions)
|
|
|
|
console.print()
|