From 393b5715b25057939810b5fb0ec7ee9ab973222e Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Thu, 26 Feb 2026 12:29:07 +0530 Subject: [PATCH] FIx : litellm/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py --- .../test_key_management_endpoints.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 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 05df3c2dcbb..fb71adc1085 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 @@ -6194,15 +6194,14 @@ async def test_generate_key_helper_fn_agent_id(): ) mock_prisma_client.insert_data = mock_insert - with patch.object(km, "prisma_client", mock_prisma_client): - with patch("litellm.proxy.proxy_server.prisma_client", mock_prisma_client): - await generate_key_helper_fn( - request_type="key", - agent_id="test-agent-456", - key_alias="test-agent-key", - models=[], - table_name="key", - ) + with patch("litellm.proxy.proxy_server.prisma_client", mock_prisma_client): + await generate_key_helper_fn( + request_type="key", + agent_id="test-agent-456", + key_alias="test-agent-key", + models=[], + table_name="key", + ) assert mock_insert.called, "insert_data was never called" # insert_data is called as insert_data(data=key_data, ...)