Skip to content

Commit 62b84d6

Browse files
lior-pstabac-psvitaly-ps
authored
Prompt security litellm (#16365)
* 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 <[email protected]> Co-authored-by: Vitaly Neyman <[email protected]>
1 parent 559142a commit 62b84d6

File tree

12 files changed

+1726
-1
lines changed

12 files changed

+1726
-1
lines changed

.github/workflows/test-litellm.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ jobs:
3737
- name: Setup litellm-enterprise as local package
3838
run: |
3939
cd enterprise
40-
python -m pip install -e .
40+
poetry run pip install -e .
4141
cd ..
4242
- name: Run tests
4343
run: |

0 commit comments

Comments
 (0)