mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-29 16:22:08 +00:00
Ishaan JaffandGitHub
60bcb26dc8
* feat(agents): assign virtual keys to agents - Add agent_id field to LiteLLM_VerificationToken (schema.prisma + _types.py) - Pass agent_id through key generation endpoint so keys can be scoped to an agent - Refactor Add Agent wizard to 3-step flow (Configure → Assign Key → Ready) - Configure: all agent fields, custom/other type with just name+description - Assign Key: create new key or reassign existing key to agent - URL is now optional for easy discovery - Add "Agent" ownership option to Create Key modal on Virtual Keys page with agent selector dropdown - Extract CreatedKeyDisplay into shared component, reused in both flows - Add keyCreateForAgentCall networking helper - Add test for agent_id key generation * fix(agents): code quality fixes from self-review - Fix test_generate_key_helper_fn_agent_id: remove bare except clause, use explicit assert mock_insert.called, use .kwargs for clean arg access - Remove no-op conditional in handleNext (both branches were identical) - Validate selectedExistingKey before calling keyUpdateCall - Validate selectedAgentId before setting on formValues in create_key_button * fix(ui): replace deprecated Tremor Button with Ant Design Button in CreatedKeyDisplay
Testing for litellm/
This directory 1:1 maps the the litellm/ directory, and can only contain mocked tests.
The point of this is to:
- Increase test coverage of
litellm/ - Make it easy for contributors to add tests for the
litellm/package and easily run tests without needing LLM API keys.
File name conventions
litellm/proxy/test_caching_routes.pymaps tolitellm/proxy/caching_routes.pytest_<filename>.pymaps tolitellm/<filename>.py