fix(team): check default access groups on the writer so a fresh group is not rejected behind a read replica

This commit is contained in:
jesus 2026-11-06 03:19:00 +00:00
parent 75d9a5548e
commit 680db9baec
2 changed files with 2 additions and 2 deletions

View file

@ -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))

View file

@ -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)