mirror of
https://github.com/BerriAI/litellm.git
synced 2025-12-06 11:33:26 +08:00
62b84d6aad
* add prompt security guardrails provider * cosmetic * small * add file sanitization and update context window * add pdf and OOXML files support * add system prompt support * add tests and documentation * remove print * fix PLR0915 Too many statements (96 > 50) * cosmetic * fix mypy error * Fix failed tests due to naming conflict of responses directory with same-named pip package * Fix mypy error: use 'aembedding' instead of 'embeddings' for async embedding call type * Fix: Install enterprise package into Poetry virtualenv for tests The GitHub Actions workflow was installing litellm-enterprise to system Python using 'python -m pip install -e .', but tests run in Poetry's virtualenv using 'poetry run pytest'. This caused ImportError for enterprise package types. Changed to 'poetry run pip install -e .' so the package is available in the same virtualenv where pytest executes. Fixes enterprise test collection errors in GitHub Actions CI. * Move Prompt Security guardrail tests to tests/test_litellm/ Per reviewer feedback, move test_prompt_security_guardrails.py from tests/guardrails_tests/ to tests/test_litellm/proxy/guardrails/ so it will be executed by GitHub Actions workflow test-litellm.yml. This ensures the Prompt Security integration tests run in CI. --------- Co-authored-by: Ori Tabac <oritabac@prompt.security> Co-authored-by: Vitaly Neyman <vitaly@prompt.security>
45 lines
1.3 KiB
YAML
45 lines
1.3 KiB
YAML
name: LiteLLM Mock Tests (folder - tests/test_litellm)
|
|
|
|
on:
|
|
pull_request:
|
|
branches: [ main ]
|
|
|
|
jobs:
|
|
test:
|
|
runs-on: ubuntu-latest
|
|
timeout-minutes: 25
|
|
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
|
|
- name: Thank You Message
|
|
run: |
|
|
echo "### 🙏 Thank you for contributing to LiteLLM!" >> $GITHUB_STEP_SUMMARY
|
|
echo "Your PR is being tested now. We appreciate your help in making LiteLLM better!" >> $GITHUB_STEP_SUMMARY
|
|
|
|
- name: Set up Python
|
|
uses: actions/setup-python@v4
|
|
with:
|
|
python-version: '3.12'
|
|
|
|
- name: Install Poetry
|
|
uses: snok/install-poetry@v1
|
|
|
|
- name: Install dependencies
|
|
run: |
|
|
poetry install --with dev,proxy-dev --extras "proxy semantic-router"
|
|
poetry run pip install "pytest-retry==1.6.3"
|
|
poetry run pip install pytest-xdist
|
|
poetry run pip install "google-genai==1.22.0"
|
|
poetry run pip install "google-cloud-aiplatform>=1.38"
|
|
poetry run pip install "fastapi-offline==1.7.3"
|
|
poetry run pip install "python-multipart==0.0.18"
|
|
- name: Setup litellm-enterprise as local package
|
|
run: |
|
|
cd enterprise
|
|
poetry run pip install -e .
|
|
cd ..
|
|
- name: Run tests
|
|
run: |
|
|
poetry run pytest tests/test_litellm --tb=short -vv --maxfail=10 -n 4 --durations=50
|