tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-06 04:23:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: fix test_key_generation_with_object_permission

Browse Source
This commit is contained in:
Ishaan Jaff
2025-05-28 21:24:35 -07:00
parent c5a82b03f4
commit 711c931c71
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py
+2 -2
View File
@@ -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(
tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
+2 -2
View File
@@ -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