diff --git a/litellm/proxy/ui_crud_endpoints/proxy_setting_endpoints.py b/litellm/proxy/ui_crud_endpoints/proxy_setting_endpoints.py index fe83b0ae6ff..baf868d3e32 100644 --- a/litellm/proxy/ui_crud_endpoints/proxy_setting_endpoints.py +++ b/litellm/proxy/ui_crud_endpoints/proxy_setting_endpoints.py @@ -1187,7 +1187,7 @@ async def _validate_default_access_groups_exist(access_group_ids: Sequence[str]) ) existing_groups: Final = await find_many_in( - AccessGroupRepository(prisma_client).table, "access_group_id", dict.fromkeys(access_group_ids) + AccessGroupRepository(prisma_client, use_writer=True).table, "access_group_id", dict.fromkeys(access_group_ids) ) existing_ids: Final = {group.access_group_id for group in existing_groups} missing_ids: Final = tuple(dict.fromkeys(ag_id for ag_id in access_group_ids if ag_id not in existing_ids)) diff --git a/tests/unit/proxy/ui_crud_endpoints/test_proxy_setting_endpoints.py b/tests/unit/proxy/ui_crud_endpoints/test_proxy_setting_endpoints.py index 30c4538fea9..616fe6ac177 100644 --- a/tests/unit/proxy/ui_crud_endpoints/test_proxy_setting_endpoints.py +++ b/tests/unit/proxy/ui_crud_endpoints/test_proxy_setting_endpoints.py @@ -3494,7 +3494,7 @@ def mock_access_group_lookup(monkeypatch: pytest.MonkeyPatch) -> Callable[[froze find_many: Final = AsyncMock(side_effect=_find_many) fake_prisma: Final = MagicMock() - fake_prisma.db.litellm_accessgrouptable.find_many = find_many + fake_prisma.writer_db.litellm_accessgrouptable.find_many = find_many monkeypatch.setattr(proxy_server_module, "prisma_client", fake_prisma) monkeypatch.setattr("litellm.proxy.proxy_server.store_model_in_db", True)