mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-14 09:03:16 +00:00
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:
parent
75d9a5548e
commit
680db9baec
2 changed files with 2 additions and 2 deletions
|
|
@ -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))
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue