From 711c931c71933fd33fceea3fc8cb88e33db590c7 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 28 May 2025 21:24:35 -0700 Subject: [PATCH] test: fix test_key_generation_with_object_permission --- .../management_endpoints/test_key_management_endpoints.py | 4 ++-- .../proxy/management_endpoints/test_team_endpoints.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py index 27d2828201..755b833486 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py @@ -217,7 +217,7 @@ async def test_key_generation_with_object_permission(monkeypatch): # --- Import objects after monkeypatching --- from litellm.proxy._types import ( GenerateKeyRequest, - LiteLLM_ObjectPermissionTable, + LiteLLM_ObjectPermissionBase, LitellmUserRoles, ) from litellm.proxy.auth.user_api_key_auth import UserAPIKeyAuth @@ -227,7 +227,7 @@ async def test_key_generation_with_object_permission(monkeypatch): # --- Call generate_key_fn with object_permission --- request_data = GenerateKeyRequest( - object_permission=LiteLLM_ObjectPermissionTable(vector_stores=["my-vector"]) + object_permission=LiteLLM_ObjectPermissionBase(vector_stores=["my-vector"]) ) await generate_key_fn( diff --git a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py index 1925610305..6e641d2e1c 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py @@ -286,13 +286,13 @@ async def test_new_team_with_object_permission(mock_db_client, mock_admin_auth): # --- Import after mocks applied --- from fastapi import Request - from litellm.proxy._types import LiteLLM_ObjectPermissionTable, NewTeamRequest + from litellm.proxy._types import LiteLLM_ObjectPermissionBase, NewTeamRequest from litellm.proxy.management_endpoints.team_endpoints import new_team # Build request objects team_request = NewTeamRequest( team_alias="my-team", - object_permission=LiteLLM_ObjectPermissionTable(vector_stores=["my-vector"]), + object_permission=LiteLLM_ObjectPermissionBase(vector_stores=["my-vector"]), ) # Pass a dummy FastAPI Request object