Skip to content

Commit cd02a14

Browse files
jopemachineclaude
andcommitted
fix(BA-5650-G): align slice G call sites after rebasing on F
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent d713fdd commit cd02a14

2 files changed

Lines changed: 7 additions & 4 deletions

File tree

src/ai/backend/manager/registry.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,7 @@ def __init__(
254254
event_producer,
255255
hook_plugin_ctx,
256256
self,
257+
user_repository,
257258
)
258259
self._client_pool = ClientPool(tcp_client_session_factory)
259260

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

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1150,8 +1150,8 @@ async def get_terminating_sessions_by_ids(
11501150
terminating_sessions.append(
11511151
TerminatingSessionData(
11521152
session_id=session_row.id,
1153-
main_access_key=AccessKey(session_row.access_key)
1154-
if session_row.access_key
1153+
main_access_key=AccessKey(owner_main_ak)
1154+
if owner_main_ak
11551155
else AccessKey(""),
11561156
creation_id=session_row.creation_id or "",
11571157
status=session_row.status,
@@ -1864,7 +1864,7 @@ async def allocate_sessions(
18641864
ScheduledSessionData(
18651865
session_id=allocation.session_id,
18661866
creation_id=creation_id,
1867-
main_access_key=access_key,
1867+
main_access_key=main_access_key,
18681868
reason="triggered-by-scheduler",
18691869
)
18701870
)
@@ -2924,7 +2924,9 @@ async def _get_sessions_by_statuses(
29242924
scheduled_session = ScheduledSessionData(
29252925
session_id=session.id,
29262926
creation_id=session.creation_id or "",
2927-
main_access_key=AccessKey(session.access_key) if session.access_key else AccessKey(""),
2927+
main_access_key=AccessKey(session.access_key)
2928+
if session.access_key
2929+
else AccessKey(""),
29282930
reason="triggered-by-scheduler",
29292931
)
29302932
scheduled_sessions.append(scheduled_session)

0 commit comments

Comments
 (0)