ktransformers/.github/workflows/release-sglang-kt.yml

name: Release sglang-kt to PyPI

on:
  push:
    branches:
      - main
    paths:
      - "third_party/sglang"
      - "version.py"
  workflow_dispatch:
    inputs:
      test_pypi:
        description: 'Publish to TestPyPI instead of PyPI (for testing)'
        required: false
        default: 'false'
        type: choice
        options:
          - 'true'
          - 'false'

permissions:
  contents: read

jobs:
  build-sglang-kt:
    name: Build sglang-kt wheel
    runs-on: [self-hosted, linux, x64]

    steps:
      - name: Checkout repository
        uses: actions/checkout@v4
        with:
          submodules: recursive

      - name: Set up Python
        uses: actions/setup-python@v4
        with:
          python-version: '3.12'

      - name: Install build tools
        run: |
          python -m pip install --upgrade pip
          pip install build wheel setuptools

      - name: Build sglang-kt wheel
        working-directory: third_party/sglang/python
        run: |
          # Read version from ktransformers version.py
          KT_VERSION=$(python3 -c "exec(open('${{ github.workspace }}/version.py').read()); print(__version__)")
          export SGLANG_KT_VERSION="$KT_VERSION"
          echo "Building sglang-kt v${KT_VERSION} wheel..."
          python -m build --wheel -v

      - name: Verify wheel
        working-directory: third_party/sglang/python
        run: |
          echo "Generated wheel:"
          ls -lh dist/
          # Verify the wheel has the correct package name
          ls dist/ | grep -q "sglang_kt" || (echo "ERROR: Wheel name does not contain sglang_kt" && exit 1)
          echo "Wheel name verified."

      - name: Upload artifact
        uses: actions/upload-artifact@v4
        with:
          name: sglang-kt-wheel
          path: third_party/sglang/python/dist/*.whl
          retention-days: 7

  publish-pypi:
    name: Publish sglang-kt to PyPI
    needs: [build-sglang-kt]
    runs-on: [self-hosted, linux, x64]
    if: github.repository == 'kvcache-ai/ktransformers' && github.ref == 'refs/heads/main'
    environment: prod
    permissions:
      id-token: write
      contents: read

    steps:
      - name: Download wheel artifact
        uses: actions/download-artifact@v4
        with:
          name: sglang-kt-wheel
          path: dist/

      - name: Display wheels
        run: |
          echo "Wheels to publish:"
          ls -lh dist/

      - name: Install twine
        run: |
          python -m pip install --upgrade pip
          pip install twine

      - name: Publish to TestPyPI (if requested)
        if: github.event.inputs.test_pypi == 'true'
        env:
          TWINE_USERNAME: __token__
          TWINE_PASSWORD: ${{ secrets.TEST_PYPI_API_TOKEN }}
        run: |
          python -m twine upload \
            --repository testpypi \
            --skip-existing \
            --verbose \
            dist/*.whl

      - name: Publish to PyPI
        if: github.event.inputs.test_pypi != 'true'
        env:
          TWINE_USERNAME: __token__
          TWINE_PASSWORD: ${{ secrets.PYPI_API_TOKEN }}
        run: |
          python -m twine upload \
            --skip-existing \
            --verbose \
            dist/*.whl

      - name: Create release summary
        run: |
          echo "## sglang-kt Published to PyPI" >> $GITHUB_STEP_SUMMARY
          echo "" >> $GITHUB_STEP_SUMMARY
          echo "### Installation" >> $GITHUB_STEP_SUMMARY
          echo '```bash' >> $GITHUB_STEP_SUMMARY
          echo "pip install sglang-kt" >> $GITHUB_STEP_SUMMARY
          echo '```' >> $GITHUB_STEP_SUMMARY
          echo "" >> $GITHUB_STEP_SUMMARY
          echo "This is the kvcache-ai fork of SGLang with kt-kernel support." >> $GITHUB_STEP_SUMMARY
          echo "PyPI link: https://pypi.org/project/sglang-kt/" >> $GITHUB_STEP_SUMMARY