fix(proxy): Enable partial matching for User ID filter in virtual keys page

[colesmcintosh]

Title

fix(proxy): Enable partial matching for User ID filter in virtual keys page

Relevant issues

Fixes LIT-268

Pre-Submission checklist

• I have Added testing in the tests/litellm/ directory, Adding at least 1 test is a hard requirement - see details
• I have added a screenshot of my new test passing locally
• My PR passes all unit tests on make test-unit
• My PR's scope is as isolated as possible, it only solves 1 specific problem

Type

🐛 Bug Fix

Changes

Problem

The virtual key page (/ui/?page=api-keys) required users to enter the complete, exact User ID to filter results. Partial searches returned no results, making it difficult to find keys when users only remembered part of a User ID.

Solution

Updated the _list_key_helper function in key_management_endpoints.py to use Prisma's contains operator with case-insensitive mode for user_id filtering instead of exact matching.

Implementation Details

• Changed user_id filter from exact match to partial substring match with case-insensitive mode
• Updated the Prisma query condition to use {"contains": user_id, "mode": "insensitive"}
• Added comprehensive test coverage in test_list_key_helper_user_id_partial_matching

Testing

Added tests to verify:

1. Partial substring matching works correctly
2. Case-insensitive matching works
3. Exact matches still work
4. Multiple results are returned for common substrings
5. No results are returned for non-existent patterns

Test passing locally:

tests/proxy_admin_ui_tests/test_key_management.py::test_list_key_helper_user_id_partial_matching PASSED

[vercel]

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name	Status	Preview	Comments	Updated (UTC)
litellm	✅ Ready (Inspect)	Visit Preview	💬 Add feedback	Jul 1, 2025 5:00pm