Skip to content

Commit 2d7d64e

Browse files
committed
fix(BA-5709): keep delegate_endpoint_ownership signature in slice A
Slice A scope is only the new get_main_access_key_by_id helper and the SessionConditions rewrite. Accidentally dropping the target_main_access_key argument from delegate_endpoint_ownership belongs to a later slice that also updates EndpointRow and the user service; restoring the signature here so this PR builds standalone.
1 parent c5cf3cd commit 2d7d64e

2 files changed

Lines changed: 10 additions & 4 deletions

File tree

src/ai/backend/manager/repositories/user/db_source/db_source.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
from ai.backend.common.clients.valkey_client.valkey_stat.client import ValkeyStatClient
1919
from ai.backend.common.data.permission.types import RBACElementType
20-
from ai.backend.common.types import VFolderID
20+
from ai.backend.common.types import AccessKey, VFolderID
2121
from ai.backend.logging.utils import BraceStyleAdapter
2222
from ai.backend.manager.data.common.types import SearchResult
2323
from ai.backend.manager.data.keypair.types import (
@@ -672,10 +672,13 @@ async def delegate_endpoint_ownership(
672672
self,
673673
user_uuid: UUID,
674674
target_user_uuid: UUID,
675+
target_main_access_key: AccessKey,
675676
) -> None:
676677
"""Delegate endpoint ownership to another user."""
677678
async with self._db.begin_session() as session:
678-
await EndpointRow.delegate_endpoint_ownership(session, user_uuid, target_user_uuid)
679+
await EndpointRow.delegate_endpoint_ownership(
680+
session, user_uuid, target_user_uuid, target_main_access_key
681+
)
679682

680683
async def delete_endpoints(
681684
self,

src/ai/backend/manager/repositories/user/repository.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from ai.backend.common.resilience.policies.metrics import MetricArgs, MetricPolicy
1717
from ai.backend.common.resilience.policies.retry import BackoffStrategy, RetryArgs, RetryPolicy
1818
from ai.backend.common.resilience.resilience import Resilience
19-
from ai.backend.common.types import SlotName
19+
from ai.backend.common.types import AccessKey, SlotName
2020
from ai.backend.common.utils import nmget
2121
from ai.backend.logging.utils import BraceStyleAdapter
2222
from ai.backend.manager.data.common.types import SearchResult
@@ -191,9 +191,12 @@ async def delegate_endpoint_ownership(
191191
self,
192192
user_uuid: UUID,
193193
target_user_uuid: UUID,
194+
target_main_access_key: AccessKey,
194195
) -> None:
195196
"""Delegate endpoint ownership to another user."""
196-
await self._db_source.delegate_endpoint_ownership(user_uuid, target_user_uuid)
197+
await self._db_source.delegate_endpoint_ownership(
198+
user_uuid, target_user_uuid, target_main_access_key
199+
)
197200

198201
@user_repository_resilience.apply()
199202
async def delete_endpoints(

0 commit comments

Comments
 (0)