mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-17 18:48:36 +00:00
73e9071311
* refactor(ui): extract auth state into AuthContext
Move auth state (token, userID, userRole, accessToken, premiumUser, userEmail,
disabledPersonalKeyCreation, showSSOBanner) out of src/app/page.tsx into a
new AuthProvider at src/contexts/AuthContext.tsx. Wrapped at the root layout
so login/onboarding/dashboard routes all have access via useAuth().
Day 1 foundation for the App Router migration: migrated (dashboard)/X/page.tsx
route entry points won't have a parent passing props, so shared auth state
must live in a context they can read from.
Sub-components are unchanged — they still receive accessToken/userID/userRole
as props from page.tsx (which now reads them from useAuth()). Only the
page.tsx → top-level-page-component handoff is de-drilled; deeper prop
drilling is left for the per-page migration to address.
Net change: -86 lines from page.tsx (state + two effects moved), +5 in
layout.tsx (provider wrap), new AuthContext.tsx (~140 lines), test update
to wrap CreateKeyPage in AuthProvider.
Fixes LIT-3366
Part of LIT-3128
* fix(ui): await getUiConfig before clearing authLoading
The AuthContext refactor flipped authLoading to false synchronously on mount
while letting getUiConfig() run fire-and-forget. On SERVER_ROOT_PATH deployments
this races the unauthenticated login-redirect effect: the redirect fires with
proxyBaseUrl still at its module-init value, sending users to /ui/login instead
of {SERVER_ROOT_PATH}/ui/login.
Restores the original sequencing inside AuthProvider's mount effect and adds a
Playwright spec wired into the existing SERVER_ROOT_PATH workflow matrix. The
spec delays the config endpoint via page.route() to make the race deterministic
across CI runners.
134 lines
4.1 KiB
YAML
134 lines
4.1 KiB
YAML
name: Test Proxy SERVER_ROOT_PATH Routing
|
|
permissions:
|
|
contents: read
|
|
|
|
on:
|
|
pull_request:
|
|
branches:
|
|
- main
|
|
- litellm_internal_staging
|
|
- litellm_oss_branch
|
|
- "litellm_**"
|
|
|
|
jobs:
|
|
test-server-root-path:
|
|
runs-on: ubuntu-latest
|
|
timeout-minutes: 30
|
|
|
|
strategy:
|
|
matrix:
|
|
root_path: ["/api/v1", "/llmproxy"]
|
|
|
|
steps:
|
|
- name: Checkout repository
|
|
uses: actions/checkout@08eba0b27e820071cde6df949e0beb9ba4906955 # v4.3.0
|
|
with:
|
|
persist-credentials: false
|
|
|
|
- name: Free up disk space
|
|
run: |
|
|
sudo rm -rf /usr/local/lib/android /usr/share/dotnet /opt/ghc /usr/local/share/boost
|
|
sudo apt-get clean
|
|
df -h /
|
|
|
|
- name: Set up Docker Buildx
|
|
uses: docker/setup-buildx-action@8d2750c68a42422c14e847fe6c8ac0403b4cbd6f # v3.12
|
|
|
|
- name: Build Docker image
|
|
uses: docker/build-push-action@0adf9959216b96bec444f325f1e493d4aa344497 #v6.14
|
|
with:
|
|
context: .
|
|
file: ./docker/Dockerfile.non_root
|
|
tags: litellm-test:${{ github.sha }}
|
|
load: true
|
|
cache-from: type=gha
|
|
cache-to: type=gha,mode=max
|
|
|
|
- name: Start LiteLLM container with SERVER_ROOT_PATH
|
|
run: |
|
|
docker run -d \
|
|
--name litellm-test \
|
|
-p 4000:4000 \
|
|
-e SERVER_ROOT_PATH="${{ matrix.root_path }}" \
|
|
-e LITELLM_MASTER_KEY="sk-1234" \
|
|
litellm-test:${{ github.sha }} \
|
|
--detailed_debug
|
|
|
|
- name: Wait for container to be healthy
|
|
run: |
|
|
echo "Waiting for LiteLLM to start..."
|
|
max_attempts=30
|
|
attempt=0
|
|
|
|
while [ $attempt -lt $max_attempts ]; do
|
|
if docker logs litellm-test 2>&1 | grep -q "Uvicorn running"; then
|
|
echo "LiteLLM started successfully"
|
|
break
|
|
fi
|
|
attempt=$((attempt + 1))
|
|
echo "Attempt $attempt/$max_attempts - waiting for server to start..."
|
|
sleep 2
|
|
done
|
|
|
|
if [ $attempt -eq $max_attempts ]; then
|
|
echo "Server failed to start within timeout"
|
|
docker logs litellm-test
|
|
exit 1
|
|
fi
|
|
|
|
sleep 5
|
|
|
|
- name: Show container logs
|
|
if: always()
|
|
run: docker logs litellm-test
|
|
|
|
- name: Test UI endpoint with root path
|
|
run: |
|
|
ROOT_PATH="${{ matrix.root_path }}"
|
|
echo "Testing UI at: http://localhost:4000${ROOT_PATH}/ui/"
|
|
|
|
for i in 1 2 3; do
|
|
content=$(curl -sL --max-time 5 -H "Authorization: Bearer sk-1234" "http://localhost:4000${ROOT_PATH}/ui/")
|
|
if echo "$content" | grep -q -E "(html|<!DOCTYPE|<head|<body)"; then
|
|
echo "UI page contains valid HTML content"
|
|
exit 0
|
|
fi
|
|
echo "Attempt $i/3 - no valid HTML, retrying in 5s..."
|
|
sleep 5
|
|
done
|
|
echo "UI page does not contain expected HTML content"
|
|
echo "Response: $content"
|
|
docker logs litellm-test
|
|
exit 1
|
|
|
|
- name: Setup Node for Playwright
|
|
uses: actions/setup-node@49933ea5288caeca8642d1e84afbd3f7d6820020 # v4.4.0
|
|
with:
|
|
node-version: "20"
|
|
|
|
- name: Install UI deps and Chromium
|
|
working-directory: ui/litellm-dashboard
|
|
run: |
|
|
npm ci
|
|
npx playwright install --with-deps chromium
|
|
|
|
- name: Run SERVER_ROOT_PATH redirect e2e
|
|
working-directory: ui/litellm-dashboard
|
|
env:
|
|
SERVER_ROOT_PATH: ${{ matrix.root_path }}
|
|
run: npx playwright test --config=e2e_tests/serverRootPath.config.ts
|
|
|
|
- name: Upload Playwright artifacts on failure
|
|
if: failure()
|
|
uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4.6.2
|
|
with:
|
|
name: playwright-trace-${{ strategy.job-index }}
|
|
path: ui/litellm-dashboard/test-results/
|
|
retention-days: 7
|
|
|
|
- name: Cleanup
|
|
if: always()
|
|
run: |
|
|
docker stop litellm-test || true
|
|
docker rm litellm-test || true
|