mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-06 08:16:43 +00:00
128 lines
3.7 KiB
YAML
128 lines
3.7 KiB
YAML
name: LiteLLM Rust
|
|
|
|
on:
|
|
push:
|
|
paths:
|
|
- "litellm-rust/**"
|
|
- ".cargo/**"
|
|
- "pyproject.toml"
|
|
- "rust-toolchain.toml"
|
|
- ".github/scripts/smoke_test_native_wheel.py"
|
|
- ".github/scripts/verify_linux_native_wheel.py"
|
|
- "tests/test_litellm/rust_bridge/native_route_wheel_test.py"
|
|
- ".github/workflows/test-rust.yml"
|
|
pull_request:
|
|
branches:
|
|
- main
|
|
- litellm_internal_staging
|
|
- litellm_oss_staging
|
|
- "litellm_**"
|
|
paths:
|
|
- "litellm-rust/**"
|
|
- ".cargo/**"
|
|
- "pyproject.toml"
|
|
- "rust-toolchain.toml"
|
|
- ".github/scripts/smoke_test_native_wheel.py"
|
|
- ".github/scripts/verify_linux_native_wheel.py"
|
|
- "tests/test_litellm/rust_bridge/native_route_wheel_test.py"
|
|
- ".github/workflows/test-rust.yml"
|
|
|
|
permissions:
|
|
contents: read
|
|
|
|
concurrency:
|
|
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
|
|
cancel-in-progress: true
|
|
|
|
jobs:
|
|
rust-checks:
|
|
name: rustfmt, clippy, test
|
|
runs-on: ubuntu-latest
|
|
timeout-minutes: 10
|
|
defaults:
|
|
run:
|
|
working-directory: litellm-rust
|
|
env:
|
|
CARGO_TERM_COLOR: always
|
|
|
|
steps:
|
|
- name: Checkout repository
|
|
uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
|
|
with:
|
|
persist-credentials: false
|
|
|
|
- name: Set up Rust
|
|
run: rustup toolchain install
|
|
|
|
- name: Cache Cargo registry and target
|
|
uses: actions/cache@0057852bfaa89a56745cba8c7296529d2fc39830 # v4.3.0
|
|
with:
|
|
path: |
|
|
~/.cargo/registry
|
|
~/.cargo/git
|
|
litellm-rust/target
|
|
key: ${{ runner.os }}-cargo-${{ hashFiles('rust-toolchain.toml', 'litellm-rust/Cargo.lock') }}
|
|
restore-keys: |
|
|
${{ runner.os }}-cargo-
|
|
|
|
- name: Check Rust formatting
|
|
run: cargo fmt --check
|
|
|
|
- name: Run Clippy
|
|
run: cargo clippy --workspace --all-targets --locked -- -D warnings
|
|
|
|
- name: Run Clippy with Bedrock auth
|
|
run: cargo clippy -p litellm-core --all-targets --features bedrock-auth --locked -- -D warnings
|
|
|
|
- name: Run Rust tests
|
|
run: cargo test --workspace --locked
|
|
|
|
- name: Run core tests with Bedrock auth
|
|
run: cargo test -p litellm-core --features bedrock-auth --locked
|
|
|
|
release-wheel:
|
|
name: release wheel
|
|
runs-on: ubuntu-latest
|
|
timeout-minutes: 20
|
|
permissions:
|
|
contents: read
|
|
env:
|
|
CARGO_TERM_COLOR: always
|
|
|
|
steps:
|
|
- name: Checkout repository
|
|
uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
|
|
with:
|
|
persist-credentials: false
|
|
|
|
- name: Set up Python
|
|
uses: actions/setup-python@a26af69be951a213d495a4c3e4e4022e16d87065 # v5.6.0
|
|
with:
|
|
python-version: "3.12"
|
|
|
|
- name: Set up uv
|
|
uses: ./.github/actions/setup-uv-with-retries
|
|
with:
|
|
version: "0.10.9"
|
|
|
|
- name: Set up Rust
|
|
run: rustup toolchain install
|
|
|
|
- name: Build release wheel
|
|
run: uv build --wheel --out-dir dist
|
|
|
|
- name: Build panic contract wheel
|
|
run: >-
|
|
uv build --wheel --out-dir panic-dist
|
|
--config-setting "maturin.build-args=--features panic-test,extension-module"
|
|
|
|
- name: Smoke-test native panic unwinding
|
|
run: python .github/scripts/smoke_test_native_wheel.py panic-dist/*.whl
|
|
|
|
- name: Verify stripped native extension
|
|
env:
|
|
RELEASE_WHEEL_COMMIT_SHA: ${{ github.event.pull_request.head.sha || github.sha }}
|
|
run: python .github/scripts/verify_linux_native_wheel.py dist/*.whl
|
|
|
|
- name: Test native route wheel
|
|
run: python tests/test_litellm/rust_bridge/native_route_wheel_test.py dist/*.whl
|