Skip to content

Commit ff8dd79

Browse files
authored
fix(BA-5996): propagate session network_type/network_id to scheduler launcher (#11561)
1 parent 264e92e commit ff8dd79

2 files changed

Lines changed: 7 additions & 0 deletions

File tree

changes/11543.fix.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Propagate `SessionRow.network_type` and `SessionRow.network_id` through scheduler queries into `SessionDataForStart`, so the launcher correctly reuses pre-created networks for `PERSISTENT` sessions instead of calling `create_network`.

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2940,6 +2940,8 @@ async def _get_sessions_for_start(
29402940
SessionRow.environ,
29412941
SessionRow.cluster_mode,
29422942
SessionRow.user_uuid,
2943+
SessionRow.network_type,
2944+
SessionRow.network_id,
29432945
KernelRow.id.label("kernel_id"),
29442946
KernelRow.agent,
29452947
KernelRow.agent_addr,
@@ -2994,6 +2996,8 @@ async def _get_sessions_for_start(
29942996
"environ": row.environ,
29952997
"cluster_mode": row.cluster_mode,
29962998
"user_uuid": row.user_uuid,
2999+
"network_type": row.network_type,
3000+
"network_id": row.network_id,
29973001
}
29983002
if row.user_uuid:
29993003
user_uuids.add(row.user_uuid)
@@ -3088,6 +3092,8 @@ async def _get_sessions_for_start(
30883092
user_uuid=session_info["user_uuid"],
30893093
user_email=user_info.email,
30903094
user_name=user_info.username,
3095+
network_type=session_info["network_type"],
3096+
network_id=session_info["network_id"],
30913097
)
30923098
)
30933099

0 commit comments

Comments
 (0)