litellm/tests/test_litellm/proxy/management_endpoints
Ishaan Jaff 60bcb26dc8
feat(agents): assign virtual keys to agents (#22045)
* 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
2026-02-24 18:28:16 -08:00
..
policy_endpoints feat(policy): test playground for AI policy suggestions (#21608) 2026-02-19 14:09:20 -08:00
scim fix: handle deprovisioning operations without path field (#21571) 2026-02-19 11:09:04 -08:00
search_endpoints masking sensitive values 2026-02-03 16:09:37 -08:00
usage_endpoints [Feat] UI - Allow using AI to understand Usage patterns (#22042) 2026-02-24 16:40:04 -08:00
test_access_group_endpoints.py cache adjustment for deleted teams and keys 2026-02-14 09:28:02 -08:00
test_access_group_management.py fix(proxy): add model_ids param to access group endpoints for precise deployment tagging (#21655) 2026-02-21 00:39:57 -08:00
test_budget_endpoints.py [Release - 02/10/2026] v1.81.10-nightly 2026-02-10 16:26:30 -08:00
test_cache_settings_endpoints.py [Feat] UI - Allow setting cache settings on UI (#16143) 2025-10-31 17:43:59 -07:00
test_callback_management_endpoints.py [Feat] AI Gateway Auth - Allow using JWTs for signing in with Proxy CLI (#16756) 2025-11-17 19:47:29 -08:00
test_common_daily_activity.py perf: use SQL GROUP BY for aggregated daily activity endpoints 2026-02-19 14:36:28 -08:00
test_common_utils.py allow team and org admins to call invitation/new 2026-02-11 11:23:27 -08:00
test_compliance_endpoints.py Add compliance checker endpoints + UI panel (#21432) 2026-02-17 18:22:26 -08:00
test_cost_tracking_settings.py [Feat] UI - Allow setting Provider Discounts on UI (#15550) 2025-10-14 20:08:18 -07:00
test_customer_budget.py test(customer): add unit tests for customer budget update functionality 2025-07-30 17:01:18 +00:00
test_customer_endpoints.py Adding tests 2025-12-04 12:36:15 -08:00
test_delete_callbacks_endpoint.py test_delete_callbacks_in_db 2025-11-08 14:05:44 -08:00
test_delete_verification_tokens_failed.py fix(key management): populate failed_tokens for admin partial-delete path 2026-02-19 19:28:03 -03:00
test_entraid_app_roles.py fix: app_roles missing from jwt payload (#16448) 2025-11-12 12:07:04 -08:00
test_internal_user_endpoints.py fix(tests): restore default_internal_user_params instead of delattr-ing it 2026-02-18 13:33:53 -03:00
test_key_management_endpoints.py feat(agents): assign virtual keys to agents (#22045) 2026-02-24 18:28:16 -08:00
test_mcp_management_endpoints.py MCP fixes 2026-02-11 18:07:24 -08:00
test_model_management_endpoints.py Fix: Make PATCH /model/{model_id}/update handle team_id consistently with POST /model/new (#15297) 2025-10-07 14:04:08 -07:00
test_organization_endpoints.py feat: expose user_email in organization members response 2026-02-20 21:41:53 -08:00
test_policy_endpoints.py feat: AI policy template suggestions (#21589) 2026-02-19 12:00:26 -08:00
test_router_settings_endpoints.py New endpoint for router fields + react query 2026-01-09 16:49:22 -08:00
test_tag_management_endpoints.py Fix tag management to preserve encrypted fields in litellm_params (#17484) 2025-12-18 02:28:39 +05:30
test_team_endpoints.py [Feature] Allow team members to view entire team usage 2026-02-18 21:38:40 -08:00
test_tool_management_endpoints.py feat(proxy): tool policies - auto-discover tools + policy enforcement guardrail (#22041) 2026-02-24 16:27:06 -08:00
test_ui_sso.py chore: resolve merge conflict with main 2026-02-19 01:46:13 -03:00