Preserve provider refs across task lifecycle

Author: vamgan

src/runtime.ts

@@ -179,7 +179,7 @@ export class RuntimeService {
 
       await this.store.updateTask(task.id, {
         status: "running",
-        providerRefs: { ...task.providerRefs, ...started.providerRefs },
+        providerRefs: { ...(await this.latestProviderRefs(task.id)), ...started.providerRefs },
         updatedAt: nowIso()
       });
       for (const artifact of started.artifacts ?? []) {
@@ -245,6 +245,10 @@ export class RuntimeService {
     };
   }
 
+  private async latestProviderRefs(taskId: string): Promise<Record<string, unknown>> {
+    return (await this.store.getTask(taskId))?.providerRefs ?? {};
+  }
+
   private selectAdapter(request: DispatchRequest): BackendAdapter {
     const backendName = request.backend ?? this.defaultBackendFor(request);
     if (backendName) {

test/runtime.test.ts

@@ -142,8 +142,10 @@ describe("RuntimeService", () => {
           cleanupStatus: "pending",
           createdAt: timestamp,
           updatedAt: timestamp
-        }
-      })
+        },
+        providerRefs: { runtimeId: "provider_runtime_1" }
+      }),
+      startTask: async () => ({ providerRefs: { runtimeSessionId: "agentcore_session_1" }, result: { ok: true } })
     };
     const service = new RuntimeService({
       config: { accounts: { "dev-aws": { provider: "aws", credentialSource: "aws-sdk-default" } }, backends: {} },
@@ -159,6 +161,12 @@ describe("RuntimeService", () => {
       cleanupStatus: "completed",
       providerRefs: { runtimeId: "provider_runtime_1", cleanupId: "cleanup_1" }
     });
+    expect(store.tasks.get(handle.taskId)).toMatchObject({
+      providerRefs: {
+        runtimeId: "provider_runtime_1",
+        runtimeSessionId: "agentcore_session_1"
+      }
+    });
   });
 
   it("persists failed runtime cleanup without changing succeeded task result", async () => {