fix(BA-5650-F): align slice F call sites with renamed sokovan fields

- ``SessionWorkload`` now uses ``main_access_key`` / ``owner_id``; the
  ``PendingSessionData.to_session_workload`` bridge is updated accordingly.
- Scheduler db_source constructions of ``SessionDataForPull`` /
  ``SessionDataForStart`` use the new kwargs.
- ``cache_invalidation`` reverts to ``info.access_key``
  (``SessionTransitionInfo`` still carries the old field at slice F).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: jopemachine

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

@@ -3102,7 +3102,7 @@ async def _get_sessions_for_pull(
                 sessions_map[session_id] = SessionDataForPull(
                     session_id=session_id,
                     creation_id=row.creation_id,
-                    access_key=row.access_key,
+                    main_access_key=row.access_key,
                     kernels=[],
                 )
 
@@ -3294,13 +3294,13 @@ async def _get_sessions_for_start(
                 SessionDataForStart(
                     session_id=session_info["id"],
                     creation_id=session_info["creation_id"],
-                    access_key=session_info["access_key"],
+                    main_access_key=session_info["access_key"],
                     session_type=session_info["session_type"],
                     name=session_info["name"],
                     cluster_mode=session_info["cluster_mode"],
                     kernels=kernel_bindings,
                     environ=session_info.get("environ", {}),
-                    user_uuid=session_info["user_uuid"],
+                    owner_id=session_info["user_uuid"],
                     user_email=user_info.email,
                     user_name=user_info.username,
                 )
@@ -4074,7 +4074,7 @@ async def _fetch_sessions_for_pull_by_ids(
                 sessions_map[session_id] = SessionDataForPull(
                     session_id=session_id,
                     creation_id=row.creation_id,
-                    access_key=row.access_key,
+                    main_access_key=row.access_key,
                     kernels=[],
                 )
 
@@ -4293,13 +4293,13 @@ async def _fetch_sessions_for_start_by_ids(
                 SessionDataForStart(
                     session_id=session_info["id"],
                     creation_id=session_info["creation_id"],
-                    access_key=session_info["access_key"],
+                    main_access_key=session_info["access_key"],
                     session_type=session_info["session_type"],
                     name=session_info["name"],
                     cluster_mode=session_info["cluster_mode"],
                     kernels=kernel_bindings,
                     environ=session_info.get("environ", {}),
-                    user_uuid=session_info["user_uuid"],
+                    owner_id=session_info["user_uuid"],
                     user_email=user_info.email,
                     user_name=user_info.username,
                 )
@@ -4369,7 +4369,7 @@ async def search_sessions_with_kernels(
                 sessions_map[row.id] = SessionDataForPull(
                     session_id=row.id,
                     creation_id=row.creation_id,
-                    access_key=row.access_key,
+                    main_access_key=row.access_key,
                     kernels=[],
                 )
 
@@ -4625,13 +4625,13 @@ async def search_sessions_with_kernels_and_user(
                     SessionDataForStart(
                         session_id=session_info["id"],
                         creation_id=session_info["creation_id"],
-                        access_key=session_info["access_key"],
+                        main_access_key=session_info["access_key"],
                         session_type=session_info["session_type"],
                         name=session_info["name"],
                         cluster_mode=session_info["cluster_mode"],
                         kernels=session_info["kernels"],
                         environ=session_info.get("environ") or {},
-                        user_uuid=session_info["user_uuid"],
+                        owner_id=session_info["user_uuid"],
                         user_email=user_info.email,
                         user_name=user_info.username,
                     )

src/ai/backend/manager/repositories/scheduler/types/session.py

@@ -64,9 +64,9 @@ def to_session_workload(self) -> SessionWorkload:
         kernel_workloads = [k.to_kernel_workload() for k in self.kernels]
         return SessionWorkload(
             session_id=self.id,
-            access_key=self.main_access_key,
+            main_access_key=self.main_access_key,
             requested_slots=self.requested_slots,
-            user_uuid=self.owner_id,
+            owner_id=self.owner_id,
             group_id=self.group_id,
             domain_name=self.domain_name,
             scaling_group=self.scaling_group_name,

src/ai/backend/manager/sokovan/scheduler/post_processors/cache_invalidation.py

@@ -27,11 +27,11 @@ async def execute(self, context: PostProcessorContext) -> None:
         affected_keys: set[AccessKey] = set()
 
         for info in context.result.successes:
-            if info.main_access_key:
-                affected_keys.add(info.main_access_key)
+            if info.access_key:
+                affected_keys.add(info.access_key)
         for info in context.result.failures:
-            if info.main_access_key:
-                affected_keys.add(info.main_access_key)
+            if info.access_key:
+                affected_keys.add(info.access_key)
 
         if affected_keys:
             await self._repository.invalidate_kernel_related_cache(list(affected_keys))