Skip to content

Commit 23aeac6

Browse files
jopemachineclaude
andcommitted
fix(BA-5653): resolve cascaded rebase conflicts on top of slice I
- Drop conflict markers in scheduler db_source and user/service.py introduced by rebase against the updated slice I (preserve the user-table-joined main_access_key resolution). - Drop duplicate ``resolve_main_access_keys`` definition. - Use ``spec.access_key`` for SessionCreationSpec in scheduler db_source. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent a64ac5c commit 23aeac6

2 files changed

Lines changed: 3 additions & 37 deletions

File tree

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

Lines changed: 3 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -1488,7 +1488,7 @@ async def fetch_session_creation_data(
14881488
db_sess,
14891489
spec.user_scope.domain_name,
14901490
str(spec.user_scope.group_id),
1491-
spec.main_access_key,
1491+
spec.access_key,
14921492
)
14931493
image_infos = await self._resolve_image_info(db_sess, image_refs)
14941494

@@ -1509,7 +1509,7 @@ async def fetch_session_creation_data(
15091509
dotfile_data = await self._fetch_dotfiles(
15101510
db_sess,
15111511
spec.user_scope,
1512-
spec.main_access_key,
1512+
spec.access_key,
15131513
vfolder_mounts,
15141514
)
15151515

@@ -1553,7 +1553,7 @@ async def fetch_session_creation_context(
15531553
db_sess,
15541554
spec.user_scope.domain_name,
15551555
str(spec.user_scope.group_id),
1556-
spec.main_access_key,
1556+
spec.access_key,
15571557
)
15581558
image_infos = await self._resolve_image_info(db_sess, image_refs)
15591559

@@ -2927,11 +2927,6 @@ async def _get_sessions_by_statuses(
29272927
session_id=session.id,
29282928
creation_id=session.creation_id or "",
29292929
main_access_key=AccessKey(owner_main_ak) if owner_main_ak else AccessKey(""),
2930-
else AccessKey(""),
2931-
>>>>>>> e35caec08 (refactor(BA-5650-I): drop stray non-BA-5650 changes from slice)
2932-
=======
2933-
main_access_key=AccessKey(owner_main_ak) if owner_main_ak else AccessKey(""),
2934-
>>>>>>> 9fbd63dda (fix(BA-5650-I): resolve cascaded rebase conflicts and align tests)
29352930
reason="triggered-by-scheduler",
29362931
)
29372932
scheduled_sessions.append(scheduled_session)
@@ -2979,11 +2974,6 @@ async def _get_scheduled_sessions(self, db_sess: SASession) -> list[ScheduledSes
29792974
session_id=session.id,
29802975
creation_id=session.creation_id or "",
29812976
main_access_key=AccessKey(owner_main_ak) if owner_main_ak else AccessKey(""),
2982-
else AccessKey(""),
2983-
>>>>>>> e35caec08 (refactor(BA-5650-I): drop stray non-BA-5650 changes from slice)
2984-
=======
2985-
>>>>>>> c2dad7307 (refactor(BA-5650-I): test and remaining ORM updates)
2986-
>>>>>>> f0233454a (refactor(BA-5650-I): test and remaining ORM updates)
29872977
reason="triggered-by-scheduler",
29882978
)
29892979
)
@@ -4827,21 +4817,3 @@ async def _get_db_now_in_session(self, db_sess: SASession) -> datetime:
48274817
"""
48284818
result = await db_sess.execute(sa.select(sa.func.now()))
48294819
return result.scalar_one()
4830-
4831-
async def resolve_main_access_keys(
4832-
self, session_ids: Sequence[SessionId]
4833-
) -> dict[SessionId, AccessKey]:
4834-
"""Resolve the main access key for each session's owner.
4835-
4836-
Joins sessions → users to look up the owner's main_access_key.
4837-
"""
4838-
if not session_ids:
4839-
return {}
4840-
async with self._db.begin_readonly_session() as db_sess:
4841-
stmt = (
4842-
sa.select(SessionRow.id, UserRow.main_access_key)
4843-
.join(UserRow, SessionRow.user_uuid == UserRow.uuid)
4844-
.where(SessionRow.id.in_([sid for sid in session_ids]))
4845-
)
4846-
rows = (await db_sess.execute(stmt)).all()
4847-
return {SessionId(row[0]): AccessKey(row[1]) for row in rows if row[1] is not None}

src/ai/backend/manager/services/user/service.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -242,10 +242,7 @@ async def purge_user(self, action: PurgeUserAction) -> PurgeUserActionResult:
242242
await self._user_repository.delegate_endpoint_ownership(
243243
user_uuid=user_uuid,
244244
target_user_uuid=action.user_info_ctx.uuid,
245-
<<<<<<< HEAD
246245
target_main_access_key=AccessKey(target_main_ak),
247-
=======
248-
>>>>>>> d59487b17 (refactor(BA-5650-I): test and remaining ORM updates)
249246
)
250247
await self._user_repository.delete_endpoints(
251248
user_uuid=user_uuid,
@@ -326,10 +323,7 @@ async def _purge_single_user(
326323
await self._user_repository.delegate_endpoint_ownership(
327324
user_uuid=user_uuid,
328325
target_user_uuid=user_info_ctx.uuid,
329-
<<<<<<< HEAD
330326
target_main_access_key=AccessKey(target_main_ak),
331-
=======
332-
>>>>>>> d59487b17 (refactor(BA-5650-I): test and remaining ORM updates)
333327
)
334328
await self._user_repository.delete_endpoints(
335329
user_uuid=user_uuid,

0 commit comments

Comments
 (0)