@minpeter/pss-runtime 0.1.0-next.0 → 0.1.0-next.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (161) hide show
  1. package/README.md +223 -191
  2. package/dist/agent-child-runs.js +16 -0
  3. package/dist/agent-child-runs.js.map +1 -0
  4. package/dist/agent-host-capabilities.js +9 -0
  5. package/dist/agent-host-capabilities.js.map +1 -0
  6. package/dist/agent-host-session-store.js +12 -0
  7. package/dist/agent-host-session-store.js.map +1 -0
  8. package/dist/agent-loop.js +59 -35
  9. package/dist/agent-loop.js.map +1 -1
  10. package/dist/agent-namespace.js +24 -0
  11. package/dist/agent-namespace.js.map +1 -0
  12. package/dist/agent-options.d.ts +35 -0
  13. package/dist/agent-options.js +16 -0
  14. package/dist/agent-options.js.map +1 -0
  15. package/dist/agent-resume.js +143 -0
  16. package/dist/agent-resume.js.map +1 -0
  17. package/dist/agent-session-entry.d.ts +13 -0
  18. package/dist/agent-validation.js +35 -0
  19. package/dist/agent-validation.js.map +1 -0
  20. package/dist/agent.d.ts +8 -33
  21. package/dist/agent.js +131 -55
  22. package/dist/agent.js.map +1 -1
  23. package/dist/child-session-cleanups.js +61 -0
  24. package/dist/child-session-cleanups.js.map +1 -0
  25. package/dist/execution/host.js +14 -0
  26. package/dist/execution/host.js.map +1 -0
  27. package/dist/execution/index.d.ts +4 -0
  28. package/dist/execution/index.js +3 -0
  29. package/dist/execution/memory-notifications.js +54 -0
  30. package/dist/execution/memory-notifications.js.map +1 -0
  31. package/dist/execution/memory-state.js +34 -0
  32. package/dist/execution/memory-state.js.map +1 -0
  33. package/dist/execution/memory-store.js +203 -0
  34. package/dist/execution/memory-store.js.map +1 -0
  35. package/dist/execution/memory.d.ts +7 -0
  36. package/dist/execution/memory.js +28 -0
  37. package/dist/execution/memory.js.map +1 -0
  38. package/dist/execution/run.js +55 -0
  39. package/dist/execution/run.js.map +1 -0
  40. package/dist/execution/types.d.ts +155 -0
  41. package/dist/index.d.ts +9 -10
  42. package/dist/index.js +1 -6
  43. package/dist/llm-tool-execution.d.ts +35 -0
  44. package/dist/llm-tool-execution.js +126 -0
  45. package/dist/llm-tool-execution.js.map +1 -0
  46. package/dist/llm.d.ts +11 -15
  47. package/dist/llm.js +5 -9
  48. package/dist/llm.js.map +1 -1
  49. package/dist/plugins.d.ts +20 -0
  50. package/dist/plugins.js +14 -0
  51. package/dist/plugins.js.map +1 -0
  52. package/dist/session/events.d.ts +26 -20
  53. package/dist/session/input-normalization.js +66 -0
  54. package/dist/session/input-normalization.js.map +1 -0
  55. package/dist/session/input.d.ts +0 -4
  56. package/dist/session/mapping.js +1 -2
  57. package/dist/session/mapping.js.map +1 -1
  58. package/dist/session/run.js +1 -0
  59. package/dist/session/run.js.map +1 -1
  60. package/dist/session/runtime-input.js +20 -58
  61. package/dist/session/runtime-input.js.map +1 -1
  62. package/dist/session/session-errors.js +18 -0
  63. package/dist/session/session-errors.js.map +1 -0
  64. package/dist/session/session-events.js +59 -0
  65. package/dist/session/session-events.js.map +1 -0
  66. package/dist/session/session-execution.js +88 -0
  67. package/dist/session/session-execution.js.map +1 -0
  68. package/dist/session/session-kill.js +23 -0
  69. package/dist/session/session-kill.js.map +1 -0
  70. package/dist/session/session-notification.js +58 -0
  71. package/dist/session/session-notification.js.map +1 -0
  72. package/dist/session/session-runtime-drain.js +22 -0
  73. package/dist/session/session-runtime-drain.js.map +1 -0
  74. package/dist/session/session-state.js +102 -0
  75. package/dist/session/session-state.js.map +1 -0
  76. package/dist/session/session-turn-error.js +35 -0
  77. package/dist/session/session-turn-error.js.map +1 -0
  78. package/dist/session/session-turn-processor.js +135 -0
  79. package/dist/session/session-turn-processor.js.map +1 -0
  80. package/dist/session/session.js +125 -335
  81. package/dist/session/session.js.map +1 -1
  82. package/dist/session/snapshot.js +5 -31
  83. package/dist/session/snapshot.js.map +1 -1
  84. package/dist/session/store/file.d.ts +1 -0
  85. package/dist/session/store/file.js +14 -0
  86. package/dist/session/store/file.js.map +1 -1
  87. package/dist/session/store/memory.d.ts +1 -0
  88. package/dist/session/store/memory.js +5 -0
  89. package/dist/session/store/memory.js.map +1 -1
  90. package/dist/session/store/types.d.ts +1 -0
  91. package/dist/subagent-background-child-run-state.js +51 -0
  92. package/dist/subagent-background-child-run-state.js.map +1 -0
  93. package/dist/subagent-background-child-run.js +103 -0
  94. package/dist/subagent-background-child-run.js.map +1 -0
  95. package/dist/subagent-background-in-process.js +98 -0
  96. package/dist/subagent-background-in-process.js.map +1 -0
  97. package/dist/subagent-background-notification-inbox.js +106 -0
  98. package/dist/subagent-background-notification-inbox.js.map +1 -0
  99. package/dist/subagent-background-notify.js +136 -0
  100. package/dist/subagent-background-notify.js.map +1 -0
  101. package/dist/subagent-background-resume-group.js +99 -0
  102. package/dist/subagent-background-resume-group.js.map +1 -0
  103. package/dist/subagent-background-runner.js +115 -0
  104. package/dist/subagent-background-runner.js.map +1 -0
  105. package/dist/subagent-background-schedule.js +43 -0
  106. package/dist/subagent-background-schedule.js.map +1 -0
  107. package/dist/subagent-child-run.js +68 -0
  108. package/dist/subagent-child-run.js.map +1 -0
  109. package/dist/subagent-job-cancel.js +84 -0
  110. package/dist/subagent-job-cancel.js.map +1 -0
  111. package/dist/subagent-job-observer.js +19 -0
  112. package/dist/subagent-job-observer.js.map +1 -0
  113. package/dist/subagent-job-output.js +87 -0
  114. package/dist/subagent-job-output.js.map +1 -0
  115. package/dist/subagent-job-state.js +66 -0
  116. package/dist/subagent-job-state.js.map +1 -0
  117. package/dist/subagent-jobs.js +96 -0
  118. package/dist/subagent-jobs.js.map +1 -0
  119. package/dist/subagent-prompt-schema.js +114 -0
  120. package/dist/subagent-prompt-schema.js.map +1 -0
  121. package/dist/subagent-run.js +111 -0
  122. package/dist/subagent-run.js.map +1 -0
  123. package/dist/subagents.js +125 -0
  124. package/dist/subagents.js.map +1 -0
  125. package/package.json +11 -6
  126. package/dist/plugins/compaction.d.ts +0 -15
  127. package/dist/plugins/compaction.js +0 -98
  128. package/dist/plugins/compaction.js.map +0 -1
  129. package/dist/plugins/index.d.ts +0 -5
  130. package/dist/plugins/index.js +0 -5
  131. package/dist/plugins/memory.d.ts +0 -11
  132. package/dist/plugins/memory.js +0 -146
  133. package/dist/plugins/memory.js.map +0 -1
  134. package/dist/plugins/runner.d.ts +0 -1
  135. package/dist/plugins/runner.js +0 -83
  136. package/dist/plugins/runner.js.map +0 -1
  137. package/dist/plugins/scope.js +0 -13
  138. package/dist/plugins/scope.js.map +0 -1
  139. package/dist/plugins/sessions.d.ts +0 -12
  140. package/dist/plugins/sessions.js +0 -34
  141. package/dist/plugins/sessions.js.map +0 -1
  142. package/dist/plugins/tool-hook-handlers.js +0 -77
  143. package/dist/plugins/tool-hook-handlers.js.map +0 -1
  144. package/dist/plugins/tool-hook-results.js +0 -64
  145. package/dist/plugins/tool-hook-results.js.map +0 -1
  146. package/dist/plugins/tool-hooks.js +0 -111
  147. package/dist/plugins/tool-hooks.js.map +0 -1
  148. package/dist/plugins/types.d.ts +0 -105
  149. package/dist/plugins/types.js +0 -20
  150. package/dist/plugins/types.js.map +0 -1
  151. package/dist/session/lifecycle.d.ts +0 -12
  152. package/dist/session/lifecycle.js +0 -126
  153. package/dist/session/lifecycle.js.map +0 -1
  154. package/dist/session/overlay-anchor.js +0 -151
  155. package/dist/session/overlay-anchor.js.map +0 -1
  156. package/dist/session/overlay.js +0 -141
  157. package/dist/session/overlay.js.map +0 -1
  158. package/dist/session/snapshot.d.ts +0 -1
  159. /package/dist/{agent-loop.d.ts → session/history.d.ts} +0 -0
  160. /package/dist/session/{runtime-input.d.ts → session-execution.d.ts} +0 -0
  161. /package/dist/{plugins/scope.d.ts → session/session-state.d.ts} +0 -0
@@ -0,0 +1,84 @@
1
+ import { cancelBackgroundChildRun } from "./subagent-background-child-run.js";
2
+ import { assertBackgroundTaskId, backgroundRunJobStatus, cancelJob, isActiveJob } from "./subagent-job-state.js";
3
+ import { jsonSchema, tool } from "ai";
4
+ //#region src/subagent-job-cancel.ts
5
+ function createBackgroundCancelTool(jobs, executionHost, scope) {
6
+ return tool({
7
+ description: "Cancel an active background subagent job.",
8
+ execute: async (input) => {
9
+ assertBackgroundTaskId(input.task_id, "background_cancel");
10
+ const job = jobs.get(input.task_id);
11
+ if (!job) {
12
+ const durableCancel = await durableBackgroundCancel(input.task_id, executionHost, scope);
13
+ if (durableCancel) return durableCancel;
14
+ throw new Error(`Unknown background subagent task ${input.task_id}.`);
15
+ }
16
+ const durableLocalCancel = await cancelDurableLocalJob(job, scope);
17
+ if (durableLocalCancel) return durableLocalCancel;
18
+ if (isActiveJob(job.status)) cancelJob(job);
19
+ return {
20
+ status: job.status,
21
+ task_id: job.id
22
+ };
23
+ },
24
+ inputSchema: jsonSchema({
25
+ additionalProperties: false,
26
+ properties: { task_id: { type: "string" } },
27
+ required: ["task_id"],
28
+ type: "object"
29
+ })
30
+ });
31
+ }
32
+ async function cancelDurableLocalJob(job, scope) {
33
+ if (!(job.executionHost && job.childRunId)) return null;
34
+ const record = await job.executionHost.store.runs.get(job.childRunId);
35
+ if (record?.kind !== "background-subagent") return null;
36
+ if (scope && !record.sessionKey.startsWith(scope.childSessionKeyPrefix)) return null;
37
+ const currentStatus = backgroundRunJobStatus(record.status);
38
+ if (!currentStatus) return null;
39
+ if (currentStatus === "pending" || currentStatus === "running") {
40
+ const nextStatus = backgroundRunJobStatus((await cancelBackgroundChildRun({
41
+ executionHost: job.executionHost,
42
+ runId: record.runId
43
+ }))?.status);
44
+ if (nextStatus === "cancelled") {
45
+ cancelJob(job);
46
+ return {
47
+ status: "cancelled",
48
+ task_id: job.id
49
+ };
50
+ }
51
+ return {
52
+ status: nextStatus ?? currentStatus,
53
+ task_id: job.id
54
+ };
55
+ }
56
+ return {
57
+ status: currentStatus,
58
+ task_id: job.id
59
+ };
60
+ }
61
+ async function durableBackgroundCancel(taskId, executionHost, scope) {
62
+ if (!executionHost) return null;
63
+ const host = executionHost;
64
+ const record = await host.store.runs.get(`background:${taskId}`);
65
+ if (record?.kind !== "background-subagent") return null;
66
+ if (scope && !record.sessionKey.startsWith(scope.childSessionKeyPrefix)) return null;
67
+ const currentStatus = backgroundRunJobStatus(record.status);
68
+ if (!currentStatus) return null;
69
+ if (currentStatus === "pending" || currentStatus === "running") return {
70
+ status: backgroundRunJobStatus((await cancelBackgroundChildRun({
71
+ executionHost: host,
72
+ runId: record.runId
73
+ }))?.status) ?? currentStatus,
74
+ task_id: taskId
75
+ };
76
+ return {
77
+ status: currentStatus,
78
+ task_id: taskId
79
+ };
80
+ }
81
+ //#endregion
82
+ export { createBackgroundCancelTool };
83
+
84
+ //# sourceMappingURL=subagent-job-cancel.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"subagent-job-cancel.js","names":[],"sources":["../src/subagent-job-cancel.ts"],"sourcesContent":["import { jsonSchema, tool } from \"ai\";\nimport type { ExecutionHost } from \"./execution/types\";\nimport { cancelBackgroundChildRun } from \"./subagent-background-child-run\";\nimport {\n assertBackgroundTaskId,\n backgroundRunJobStatus,\n cancelJob,\n isActiveJob,\n} from \"./subagent-job-state\";\nimport type { BackgroundCancelInput, SubagentJob } from \"./subagent-types\";\n\ninterface BackgroundToolScope {\n readonly childSessionKeyPrefix: string;\n}\n\nexport function createBackgroundCancelTool(\n jobs: Map<string, SubagentJob>,\n executionHost?: ExecutionHost,\n scope?: BackgroundToolScope\n) {\n return tool<BackgroundCancelInput, unknown, Record<string, unknown>>({\n description: \"Cancel an active background subagent job.\",\n execute: async (input: BackgroundCancelInput) => {\n assertBackgroundTaskId(input.task_id, \"background_cancel\");\n const job = jobs.get(input.task_id);\n if (!job) {\n const durableCancel = await durableBackgroundCancel(\n input.task_id,\n executionHost,\n scope\n );\n if (durableCancel) {\n return durableCancel;\n }\n\n throw new Error(`Unknown background subagent task ${input.task_id}.`);\n }\n\n const durableLocalCancel = await cancelDurableLocalJob(job, scope);\n if (durableLocalCancel) {\n return durableLocalCancel;\n }\n\n if (isActiveJob(job.status)) {\n cancelJob(job);\n }\n\n return {\n status: job.status,\n task_id: job.id,\n };\n },\n inputSchema: jsonSchema<BackgroundCancelInput>({\n additionalProperties: false,\n properties: {\n task_id: { type: \"string\" },\n },\n required: [\"task_id\"],\n type: \"object\",\n }),\n });\n}\n\nasync function cancelDurableLocalJob(\n job: SubagentJob,\n scope: BackgroundToolScope | undefined\n): Promise<Record<string, unknown> | null> {\n if (!(job.executionHost && job.childRunId)) {\n return null;\n }\n\n const record = await job.executionHost.store.runs.get(job.childRunId);\n if (record?.kind !== \"background-subagent\") {\n return null;\n }\n if (scope && !record.sessionKey.startsWith(scope.childSessionKeyPrefix)) {\n return null;\n }\n\n const currentStatus = backgroundRunJobStatus(record.status);\n if (!currentStatus) {\n return null;\n }\n if (currentStatus === \"pending\" || currentStatus === \"running\") {\n const cancelled = await cancelBackgroundChildRun({\n executionHost: job.executionHost,\n runId: record.runId,\n });\n const nextStatus = backgroundRunJobStatus(cancelled?.status);\n if (nextStatus === \"cancelled\") {\n cancelJob(job);\n return { status: \"cancelled\", task_id: job.id };\n }\n\n return { status: nextStatus ?? currentStatus, task_id: job.id };\n }\n\n return { status: currentStatus, task_id: job.id };\n}\n\nasync function durableBackgroundCancel(\n taskId: string,\n executionHost: ExecutionHost | undefined,\n scope: BackgroundToolScope | undefined\n): Promise<Record<string, unknown> | null> {\n if (!executionHost) {\n return null;\n }\n const host = executionHost;\n\n const record = await host.store.runs.get(`background:${taskId}`);\n if (record?.kind !== \"background-subagent\") {\n return null;\n }\n if (scope && !record.sessionKey.startsWith(scope.childSessionKeyPrefix)) {\n return null;\n }\n\n const currentStatus = backgroundRunJobStatus(record.status);\n if (!currentStatus) {\n return null;\n }\n\n if (currentStatus === \"pending\" || currentStatus === \"running\") {\n const cancelled = await cancelBackgroundChildRun({\n executionHost: host,\n runId: record.runId,\n });\n return {\n status: backgroundRunJobStatus(cancelled?.status) ?? currentStatus,\n task_id: taskId,\n };\n }\n\n return { status: currentStatus, task_id: taskId };\n}\n"],"mappings":";;;;AAeA,SAAgB,2BACd,MACA,eACA,OACA;CACA,OAAO,KAA8D;EACnE,aAAa;EACb,SAAS,OAAO,UAAiC;GAC/C,uBAAuB,MAAM,SAAS,mBAAmB;GACzD,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO;GAClC,IAAI,CAAC,KAAK;IACR,MAAM,gBAAgB,MAAM,wBAC1B,MAAM,SACN,eACA,KACF;IACA,IAAI,eACF,OAAO;IAGT,MAAM,IAAI,MAAM,oCAAoC,MAAM,QAAQ,EAAE;GACtE;GAEA,MAAM,qBAAqB,MAAM,sBAAsB,KAAK,KAAK;GACjE,IAAI,oBACF,OAAO;GAGT,IAAI,YAAY,IAAI,MAAM,GACxB,UAAU,GAAG;GAGf,OAAO;IACL,QAAQ,IAAI;IACZ,SAAS,IAAI;GACf;EACF;EACA,aAAa,WAAkC;GAC7C,sBAAsB;GACtB,YAAY,EACV,SAAS,EAAE,MAAM,SAAS,EAC5B;GACA,UAAU,CAAC,SAAS;GACpB,MAAM;EACR,CAAC;CACH,CAAC;AACH;AAEA,eAAe,sBACb,KACA,OACyC;CACzC,IAAI,EAAE,IAAI,iBAAiB,IAAI,aAC7B,OAAO;CAGT,MAAM,SAAS,MAAM,IAAI,cAAc,MAAM,KAAK,IAAI,IAAI,UAAU;CACpE,IAAI,QAAQ,SAAS,uBACnB,OAAO;CAET,IAAI,SAAS,CAAC,OAAO,WAAW,WAAW,MAAM,qBAAqB,GACpE,OAAO;CAGT,MAAM,gBAAgB,uBAAuB,OAAO,MAAM;CAC1D,IAAI,CAAC,eACH,OAAO;CAET,IAAI,kBAAkB,aAAa,kBAAkB,WAAW;EAK9D,MAAM,aAAa,wBAAuB,MAJlB,yBAAyB;GAC/C,eAAe,IAAI;GACnB,OAAO,OAAO;EAChB,CAAC,IACoD,MAAM;EAC3D,IAAI,eAAe,aAAa;GAC9B,UAAU,GAAG;GACb,OAAO;IAAE,QAAQ;IAAa,SAAS,IAAI;GAAG;EAChD;EAEA,OAAO;GAAE,QAAQ,cAAc;GAAe,SAAS,IAAI;EAAG;CAChE;CAEA,OAAO;EAAE,QAAQ;EAAe,SAAS,IAAI;CAAG;AAClD;AAEA,eAAe,wBACb,QACA,eACA,OACyC;CACzC,IAAI,CAAC,eACH,OAAO;CAET,MAAM,OAAO;CAEb,MAAM,SAAS,MAAM,KAAK,MAAM,KAAK,IAAI,cAAc,QAAQ;CAC/D,IAAI,QAAQ,SAAS,uBACnB,OAAO;CAET,IAAI,SAAS,CAAC,OAAO,WAAW,WAAW,MAAM,qBAAqB,GACpE,OAAO;CAGT,MAAM,gBAAgB,uBAAuB,OAAO,MAAM;CAC1D,IAAI,CAAC,eACH,OAAO;CAGT,IAAI,kBAAkB,aAAa,kBAAkB,WAKnD,OAAO;EACL,QAAQ,wBAAuB,MALT,yBAAyB;GAC/C,eAAe;GACf,OAAO,OAAO;EAChB,CAAC,IAE2C,MAAM,KAAK;EACrD,SAAS;CACX;CAGF,OAAO;EAAE,QAAQ;EAAe,SAAS;CAAO;AAClD"}
@@ -0,0 +1,19 @@
1
+ //#region src/subagent-job-observer.ts
2
+ function emitBackgroundJobUpdate(parentSession, job, event) {
3
+ if (!isParentVisibleJobUpdate(event)) return Promise.resolve();
4
+ return parentSession.emitObserverEvent({
5
+ eventType: event.type,
6
+ delegateToolCallId: job.delegateToolCallId,
7
+ status: job.status,
8
+ subagent: job.subagent,
9
+ task_id: job.id,
10
+ type: "subagent-job-update"
11
+ });
12
+ }
13
+ function isParentVisibleJobUpdate(event) {
14
+ return event.type === "assistant-text" || event.type === "tool-call" || event.type === "tool-result" || event.type === "turn-abort" || event.type === "turn-error";
15
+ }
16
+ //#endregion
17
+ export { emitBackgroundJobUpdate };
18
+
19
+ //# sourceMappingURL=subagent-job-observer.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"subagent-job-observer.js","names":[],"sources":["../src/subagent-job-observer.ts"],"sourcesContent":["import type { AgentEvent } from \"./session/events\";\nimport type { RuntimeInputSink, SubagentJob } from \"./subagent-types\";\n\nexport function emitBackgroundJobUpdate(\n parentSession: RuntimeInputSink,\n job: SubagentJob,\n event: AgentEvent\n): Promise<void> {\n if (!isParentVisibleJobUpdate(event)) {\n return Promise.resolve();\n }\n\n return parentSession.emitObserverEvent({\n eventType: event.type,\n delegateToolCallId: job.delegateToolCallId,\n status: job.status,\n subagent: job.subagent,\n task_id: job.id,\n type: \"subagent-job-update\" as const,\n });\n}\n\nfunction isParentVisibleJobUpdate(event: AgentEvent): boolean {\n return (\n event.type === \"assistant-text\" ||\n event.type === \"tool-call\" ||\n event.type === \"tool-result\" ||\n event.type === \"turn-abort\" ||\n event.type === \"turn-error\"\n );\n}\n"],"mappings":";AAGA,SAAgB,wBACd,eACA,KACA,OACe;CACf,IAAI,CAAC,yBAAyB,KAAK,GACjC,OAAO,QAAQ,QAAQ;CAGzB,OAAO,cAAc,kBAAkB;EACrC,WAAW,MAAM;EACjB,oBAAoB,IAAI;EACxB,QAAQ,IAAI;EACZ,UAAU,IAAI;EACd,SAAS,IAAI;EACb,MAAM;CACR,CAAC;AACH;AAEA,SAAS,yBAAyB,OAA4B;CAC5D,OACE,MAAM,SAAS,oBACf,MAAM,SAAS,eACf,MAAM,SAAS,iBACf,MAAM,SAAS,gBACf,MAAM,SAAS;AAEnB"}
@@ -0,0 +1,87 @@
1
+ import { assertBackgroundTaskId, cleanupJob, isActiveJob } from "./subagent-job-state.js";
2
+ import { jsonSchema, tool } from "ai";
3
+ //#region src/subagent-job-output.ts
4
+ function createBackgroundOutputTool(jobs, executionHost, scope) {
5
+ return tool({
6
+ description: "Retrieve compact output for a background subagent job.",
7
+ execute: async (input, { abortSignal }) => {
8
+ assertBackgroundTaskId(input.task_id, "background_output");
9
+ const job = jobs.get(input.task_id);
10
+ if (!job) {
11
+ const durableOutput = await durableBackgroundOutput(input.task_id, executionHost, scope);
12
+ if (durableOutput) return durableOutput;
13
+ throw new Error(`Unknown background subagent task ${input.task_id}.`);
14
+ }
15
+ if (input.block === true && isActiveJob(job.status)) await waitForJob(job, input.timeout, abortSignal);
16
+ const output = await durableBackgroundOutput(input.task_id, job.executionHost ?? executionHost, scope, job.subagent) ?? {
17
+ result: job.result,
18
+ status: job.status,
19
+ subagent: job.subagent,
20
+ task_id: job.id
21
+ };
22
+ if (!(isActiveJob(job.status) || !job.settled)) {
23
+ if (await cleanupJob(job).then(() => true, () => false)) jobs.delete(job.id);
24
+ }
25
+ return output;
26
+ },
27
+ inputSchema: jsonSchema({
28
+ additionalProperties: false,
29
+ properties: {
30
+ block: { type: "boolean" },
31
+ task_id: { type: "string" },
32
+ timeout: {
33
+ minimum: 0,
34
+ type: "number"
35
+ }
36
+ },
37
+ required: ["task_id"],
38
+ type: "object"
39
+ })
40
+ });
41
+ }
42
+ async function durableBackgroundOutput(taskId, executionHost, scope, fallbackSubagent = "subagent") {
43
+ const record = await executionHost?.store.runs.get(`background:${taskId}`);
44
+ if (record?.kind !== "background-subagent") return null;
45
+ if (scope && !record.sessionKey.startsWith(scope.childSessionKeyPrefix)) return null;
46
+ return {
47
+ result: record.output,
48
+ status: backgroundStatus(record.status),
49
+ subagent: subagentName(record.output, fallbackSubagent),
50
+ task_id: taskId
51
+ };
52
+ }
53
+ function backgroundStatus(status) {
54
+ if (status === "completed" || status === "cancelled" || status === "error") return status;
55
+ if (status === "running" || status === "leased") return "running";
56
+ return "pending";
57
+ }
58
+ function subagentName(output, fallback) {
59
+ if (typeof output === "object" && output !== null && "subagent" in output && typeof output.subagent === "string") return output.subagent;
60
+ return fallback;
61
+ }
62
+ async function waitForJob(job, timeout, abortSignal) {
63
+ if (abortSignal?.aborted) return;
64
+ const timeoutMs = Math.min(timeout ?? 6e4, 6e5);
65
+ let timeoutId;
66
+ let abortListener;
67
+ const abortPromise = abortSignal ? new Promise((resolve) => {
68
+ abortListener = resolve;
69
+ abortSignal.addEventListener("abort", abortListener, { once: true });
70
+ }) : void 0;
71
+ try {
72
+ await Promise.race([
73
+ job.promise,
74
+ ...abortPromise ? [abortPromise] : [],
75
+ new Promise((resolve) => {
76
+ timeoutId = setTimeout(resolve, timeoutMs);
77
+ })
78
+ ]);
79
+ } finally {
80
+ if (timeoutId) clearTimeout(timeoutId);
81
+ if (abortListener) abortSignal?.removeEventListener("abort", abortListener);
82
+ }
83
+ }
84
+ //#endregion
85
+ export { createBackgroundOutputTool };
86
+
87
+ //# sourceMappingURL=subagent-job-output.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"subagent-job-output.js","names":[],"sources":["../src/subagent-job-output.ts"],"sourcesContent":["import { jsonSchema, tool } from \"ai\";\nimport type { ExecutionHost, RunStatus } from \"./execution/types\";\nimport {\n assertBackgroundTaskId,\n cleanupJob,\n isActiveJob,\n} from \"./subagent-job-state\";\nimport type { BackgroundOutputInput, SubagentJob } from \"./subagent-types\";\n\ninterface BackgroundToolScope {\n readonly childSessionKeyPrefix: string;\n}\n\nexport function createBackgroundOutputTool(\n jobs: Map<string, SubagentJob>,\n executionHost?: ExecutionHost,\n scope?: BackgroundToolScope\n) {\n return tool<BackgroundOutputInput, unknown, Record<string, unknown>>({\n description: \"Retrieve compact output for a background subagent job.\",\n execute: async (input: BackgroundOutputInput, { abortSignal }) => {\n assertBackgroundTaskId(input.task_id, \"background_output\");\n const job = jobs.get(input.task_id);\n if (!job) {\n const durableOutput = await durableBackgroundOutput(\n input.task_id,\n executionHost,\n scope\n );\n if (durableOutput) {\n return durableOutput;\n }\n\n throw new Error(`Unknown background subagent task ${input.task_id}.`);\n }\n\n if (input.block === true && isActiveJob(job.status)) {\n await waitForJob(job, input.timeout, abortSignal);\n }\n\n const durableOutput = await durableBackgroundOutput(\n input.task_id,\n job.executionHost ?? executionHost,\n scope,\n job.subagent\n );\n const output = durableOutput ?? {\n result: job.result,\n status: job.status,\n subagent: job.subagent,\n task_id: job.id,\n };\n if (!(isActiveJob(job.status) || !job.settled)) {\n const cleaned = await cleanupJob(job).then(\n () => true,\n () => false\n );\n if (cleaned) {\n jobs.delete(job.id);\n }\n }\n\n return output;\n },\n inputSchema: jsonSchema<BackgroundOutputInput>({\n additionalProperties: false,\n properties: {\n block: { type: \"boolean\" },\n task_id: { type: \"string\" },\n timeout: { minimum: 0, type: \"number\" },\n },\n required: [\"task_id\"],\n type: \"object\",\n }),\n });\n}\n\nasync function durableBackgroundOutput(\n taskId: string,\n executionHost: ExecutionHost | undefined,\n scope: BackgroundToolScope | undefined,\n fallbackSubagent = \"subagent\"\n): Promise<Record<string, unknown> | null> {\n const record = await executionHost?.store.runs.get(`background:${taskId}`);\n if (record?.kind !== \"background-subagent\") {\n return null;\n }\n if (scope && !record.sessionKey.startsWith(scope.childSessionKeyPrefix)) {\n return null;\n }\n\n return {\n result: record.output,\n status: backgroundStatus(record.status),\n subagent: subagentName(record.output, fallbackSubagent),\n task_id: taskId,\n };\n}\n\nfunction backgroundStatus(status: RunStatus): string {\n if (status === \"completed\" || status === \"cancelled\" || status === \"error\") {\n return status;\n }\n\n if (status === \"running\" || status === \"leased\") {\n return \"running\";\n }\n\n return \"pending\";\n}\n\nfunction subagentName(output: unknown, fallback: string): string {\n if (\n typeof output === \"object\" &&\n output !== null &&\n \"subagent\" in output &&\n typeof output.subagent === \"string\"\n ) {\n return output.subagent;\n }\n\n return fallback;\n}\n\nasync function waitForJob(\n job: SubagentJob,\n timeout: number | undefined,\n abortSignal: AbortSignal | undefined\n) {\n if (abortSignal?.aborted) {\n return;\n }\n\n const timeoutMs = Math.min(timeout ?? 60_000, 600_000);\n let timeoutId: ReturnType<typeof setTimeout> | undefined;\n let abortListener: (() => void) | undefined;\n const abortPromise = abortSignal\n ? new Promise<void>((resolve) => {\n abortListener = resolve;\n abortSignal.addEventListener(\"abort\", abortListener, { once: true });\n })\n : undefined;\n try {\n await Promise.race([\n job.promise,\n ...(abortPromise ? [abortPromise] : []),\n new Promise<void>((resolve) => {\n timeoutId = setTimeout(resolve, timeoutMs);\n }),\n ]);\n } finally {\n if (timeoutId) {\n clearTimeout(timeoutId);\n }\n if (abortListener) {\n abortSignal?.removeEventListener(\"abort\", abortListener);\n }\n }\n}\n"],"mappings":";;;AAaA,SAAgB,2BACd,MACA,eACA,OACA;CACA,OAAO,KAA8D;EACnE,aAAa;EACb,SAAS,OAAO,OAA8B,EAAE,kBAAkB;GAChE,uBAAuB,MAAM,SAAS,mBAAmB;GACzD,MAAM,MAAM,KAAK,IAAI,MAAM,OAAO;GAClC,IAAI,CAAC,KAAK;IACR,MAAM,gBAAgB,MAAM,wBAC1B,MAAM,SACN,eACA,KACF;IACA,IAAI,eACF,OAAO;IAGT,MAAM,IAAI,MAAM,oCAAoC,MAAM,QAAQ,EAAE;GACtE;GAEA,IAAI,MAAM,UAAU,QAAQ,YAAY,IAAI,MAAM,GAChD,MAAM,WAAW,KAAK,MAAM,SAAS,WAAW;GASlD,MAAM,SAAS,MANa,wBAC1B,MAAM,SACN,IAAI,iBAAiB,eACrB,OACA,IAAI,QACN,KACgC;IAC9B,QAAQ,IAAI;IACZ,QAAQ,IAAI;IACZ,UAAU,IAAI;IACd,SAAS,IAAI;GACf;GACA,IAAI,EAAE,YAAY,IAAI,MAAM,KAAK,CAAC,IAAI;QAKhC,MAJkB,WAAW,GAAG,EAAE,WAC9B,YACA,KACR,GAEE,KAAK,OAAO,IAAI,EAAE;GAAA;GAItB,OAAO;EACT;EACA,aAAa,WAAkC;GAC7C,sBAAsB;GACtB,YAAY;IACV,OAAO,EAAE,MAAM,UAAU;IACzB,SAAS,EAAE,MAAM,SAAS;IAC1B,SAAS;KAAE,SAAS;KAAG,MAAM;IAAS;GACxC;GACA,UAAU,CAAC,SAAS;GACpB,MAAM;EACR,CAAC;CACH,CAAC;AACH;AAEA,eAAe,wBACb,QACA,eACA,OACA,mBAAmB,YACsB;CACzC,MAAM,SAAS,MAAM,eAAe,MAAM,KAAK,IAAI,cAAc,QAAQ;CACzE,IAAI,QAAQ,SAAS,uBACnB,OAAO;CAET,IAAI,SAAS,CAAC,OAAO,WAAW,WAAW,MAAM,qBAAqB,GACpE,OAAO;CAGT,OAAO;EACL,QAAQ,OAAO;EACf,QAAQ,iBAAiB,OAAO,MAAM;EACtC,UAAU,aAAa,OAAO,QAAQ,gBAAgB;EACtD,SAAS;CACX;AACF;AAEA,SAAS,iBAAiB,QAA2B;CACnD,IAAI,WAAW,eAAe,WAAW,eAAe,WAAW,SACjE,OAAO;CAGT,IAAI,WAAW,aAAa,WAAW,UACrC,OAAO;CAGT,OAAO;AACT;AAEA,SAAS,aAAa,QAAiB,UAA0B;CAC/D,IACE,OAAO,WAAW,YAClB,WAAW,QACX,cAAc,UACd,OAAO,OAAO,aAAa,UAE3B,OAAO,OAAO;CAGhB,OAAO;AACT;AAEA,eAAe,WACb,KACA,SACA,aACA;CACA,IAAI,aAAa,SACf;CAGF,MAAM,YAAY,KAAK,IAAI,WAAW,KAAQ,GAAO;CACrD,IAAI;CACJ,IAAI;CACJ,MAAM,eAAe,cACjB,IAAI,SAAe,YAAY;EAC7B,gBAAgB;EAChB,YAAY,iBAAiB,SAAS,eAAe,EAAE,MAAM,KAAK,CAAC;CACrE,CAAC,IACD,KAAA;CACJ,IAAI;EACF,MAAM,QAAQ,KAAK;GACjB,IAAI;GACJ,GAAI,eAAe,CAAC,YAAY,IAAI,CAAC;GACrC,IAAI,SAAe,YAAY;IAC7B,YAAY,WAAW,SAAS,SAAS;GAC3C,CAAC;EACH,CAAC;CACH,UAAU;EACR,IAAI,WACF,aAAa,SAAS;EAExB,IAAI,eACF,aAAa,oBAAoB,SAAS,aAAa;CAE3D;AACF"}
@@ -0,0 +1,66 @@
1
+ import { updateBackgroundRunStatus } from "./subagent-background-child-run.js";
2
+ //#region src/subagent-job-state.ts
3
+ function assertBackgroundTaskId(value, toolName) {
4
+ if (value.startsWith("bg_")) return;
5
+ throw new Error(`${toolName} expects a background task_id starting with bg_, not a session key: ${value}`);
6
+ }
7
+ function isActiveJob(status) {
8
+ return status === "pending" || status === "running";
9
+ }
10
+ function hasBackgroundJobCapacity({ jobs, maxActiveJobs, maxRetainedJobs }) {
11
+ if (jobs.size >= maxRetainedJobs) return false;
12
+ let activeJobs = 0;
13
+ for (const job of jobs.values()) if (isActiveJob(job.status) || !job.settled) activeJobs += 1;
14
+ return activeJobs < maxActiveJobs;
15
+ }
16
+ function cancelJob(job) {
17
+ job.status = "cancelled";
18
+ job.abort();
19
+ const statusUpdate = updateBackgroundRunStatus(job, "cancelled");
20
+ job.promise = Promise.allSettled([job.promise, statusUpdate]).then(() => void 0);
21
+ }
22
+ async function cleanupJob(job) {
23
+ await job.cleanup();
24
+ job.unregisterCleanup?.();
25
+ }
26
+ function backgroundLaunchOutput(job) {
27
+ return {
28
+ message: [
29
+ `Background subagent job ${job.id} started.`,
30
+ `Please wait for <system-reminder> before checking task ${job.id}.`,
31
+ `Do NOT call background_output({ task_id: "${job.id}" }) now; wait for <system-reminder> first.`
32
+ ].join(" "),
33
+ run_in_background: true,
34
+ status: job.status,
35
+ subagent: job.subagent,
36
+ task_id: job.id
37
+ };
38
+ }
39
+ function backgroundCancelledLaunchOutput({ id, subagent }) {
40
+ return {
41
+ message: `Background subagent job ${id} was cancelled before it started.`,
42
+ run_in_background: true,
43
+ status: "cancelled",
44
+ subagent,
45
+ task_id: id
46
+ };
47
+ }
48
+ function backgroundRunJobStatus(status) {
49
+ if (status === "cancelled" || status === "completed" || status === "error") return status;
50
+ if (status === "leased" || status === "running") return "running";
51
+ if (status === "needs-recovery" || status === "queued" || status === "suspended") return "pending";
52
+ return null;
53
+ }
54
+ function backgroundReplayOutput({ id, status, subagent }) {
55
+ return {
56
+ message: `Background subagent job ${id} was already ${status}. Reuse the existing task_id instead of launching it again.`,
57
+ run_in_background: true,
58
+ status,
59
+ subagent,
60
+ task_id: id
61
+ };
62
+ }
63
+ //#endregion
64
+ export { assertBackgroundTaskId, backgroundCancelledLaunchOutput, backgroundLaunchOutput, backgroundReplayOutput, backgroundRunJobStatus, cancelJob, cleanupJob, hasBackgroundJobCapacity, isActiveJob };
65
+
66
+ //# sourceMappingURL=subagent-job-state.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"subagent-job-state.js","names":[],"sources":["../src/subagent-job-state.ts"],"sourcesContent":["import type { RunStatus } from \"./execution/types\";\nimport { updateBackgroundRunStatus } from \"./subagent-background-child-run\";\nimport type { SubagentJob } from \"./subagent-types\";\n\nexport function assertBackgroundTaskId(value: string, toolName: string): void {\n if (value.startsWith(\"bg_\")) {\n return;\n }\n\n throw new Error(\n `${toolName} expects a background task_id starting with bg_, not a session key: ${value}`\n );\n}\n\nexport function isActiveJob(status: SubagentJob[\"status\"]): boolean {\n return status === \"pending\" || status === \"running\";\n}\n\nexport function hasBackgroundJobCapacity({\n jobs,\n maxActiveJobs,\n maxRetainedJobs,\n}: {\n readonly jobs: Map<string, SubagentJob>;\n readonly maxActiveJobs: number;\n readonly maxRetainedJobs: number;\n}): boolean {\n if (jobs.size >= maxRetainedJobs) {\n return false;\n }\n\n let activeJobs = 0;\n for (const job of jobs.values()) {\n if (isActiveJob(job.status) || !job.settled) {\n activeJobs += 1;\n }\n }\n\n return activeJobs < maxActiveJobs;\n}\n\nexport function cancelJob(job: SubagentJob): void {\n job.status = \"cancelled\";\n job.abort();\n const statusUpdate = updateBackgroundRunStatus(job, \"cancelled\");\n job.promise = Promise.allSettled([job.promise, statusUpdate]).then(\n () => undefined\n );\n}\n\nexport async function cleanupJob(job: SubagentJob): Promise<void> {\n await job.cleanup();\n job.unregisterCleanup?.();\n}\n\nexport function backgroundLaunchOutput(job: SubagentJob) {\n return {\n message: [\n `Background subagent job ${job.id} started.`,\n `Please wait for <system-reminder> before checking task ${job.id}.`,\n `Do NOT call background_output({ task_id: \"${job.id}\" }) now; wait for <system-reminder> first.`,\n ].join(\" \"),\n run_in_background: true,\n status: job.status,\n subagent: job.subagent,\n task_id: job.id,\n };\n}\n\nexport function backgroundCancelledLaunchOutput({\n id,\n subagent,\n}: {\n readonly id: string;\n readonly subagent?: string;\n}) {\n return {\n message: `Background subagent job ${id} was cancelled before it started.`,\n run_in_background: true,\n status: \"cancelled\",\n subagent,\n task_id: id,\n };\n}\n\nexport function backgroundRunJobStatus(\n status: RunStatus | undefined\n): Exclude<SubagentJob[\"status\"], \"aborted\"> | null {\n if (status === \"cancelled\" || status === \"completed\" || status === \"error\") {\n return status;\n }\n\n if (status === \"leased\" || status === \"running\") {\n return \"running\";\n }\n\n if (\n status === \"needs-recovery\" ||\n status === \"queued\" ||\n status === \"suspended\"\n ) {\n return \"pending\";\n }\n\n return null;\n}\n\nexport function backgroundReplayOutput({\n id,\n status,\n subagent,\n}: {\n readonly id: string;\n readonly status: Exclude<SubagentJob[\"status\"], \"aborted\">;\n readonly subagent: string;\n}) {\n return {\n message: `Background subagent job ${id} was already ${status}. Reuse the existing task_id instead of launching it again.`,\n run_in_background: true,\n status,\n subagent,\n task_id: id,\n };\n}\n"],"mappings":";;AAIA,SAAgB,uBAAuB,OAAe,UAAwB;CAC5E,IAAI,MAAM,WAAW,KAAK,GACxB;CAGF,MAAM,IAAI,MACR,GAAG,SAAS,sEAAsE,OACpF;AACF;AAEA,SAAgB,YAAY,QAAwC;CAClE,OAAO,WAAW,aAAa,WAAW;AAC5C;AAEA,SAAgB,yBAAyB,EACvC,MACA,eACA,mBAKU;CACV,IAAI,KAAK,QAAQ,iBACf,OAAO;CAGT,IAAI,aAAa;CACjB,KAAK,MAAM,OAAO,KAAK,OAAO,GAC5B,IAAI,YAAY,IAAI,MAAM,KAAK,CAAC,IAAI,SAClC,cAAc;CAIlB,OAAO,aAAa;AACtB;AAEA,SAAgB,UAAU,KAAwB;CAChD,IAAI,SAAS;CACb,IAAI,MAAM;CACV,MAAM,eAAe,0BAA0B,KAAK,WAAW;CAC/D,IAAI,UAAU,QAAQ,WAAW,CAAC,IAAI,SAAS,YAAY,CAAC,EAAE,WACtD,KAAA,CACR;AACF;AAEA,eAAsB,WAAW,KAAiC;CAChE,MAAM,IAAI,QAAQ;CAClB,IAAI,oBAAoB;AAC1B;AAEA,SAAgB,uBAAuB,KAAkB;CACvD,OAAO;EACL,SAAS;GACP,2BAA2B,IAAI,GAAG;GAClC,0DAA0D,IAAI,GAAG;GACjE,6CAA6C,IAAI,GAAG;EACtD,EAAE,KAAK,GAAG;EACV,mBAAmB;EACnB,QAAQ,IAAI;EACZ,UAAU,IAAI;EACd,SAAS,IAAI;CACf;AACF;AAEA,SAAgB,gCAAgC,EAC9C,IACA,YAIC;CACD,OAAO;EACL,SAAS,2BAA2B,GAAG;EACvC,mBAAmB;EACnB,QAAQ;EACR;EACA,SAAS;CACX;AACF;AAEA,SAAgB,uBACd,QACkD;CAClD,IAAI,WAAW,eAAe,WAAW,eAAe,WAAW,SACjE,OAAO;CAGT,IAAI,WAAW,YAAY,WAAW,WACpC,OAAO;CAGT,IACE,WAAW,oBACX,WAAW,YACX,WAAW,aAEX,OAAO;CAGT,OAAO;AACT;AAEA,SAAgB,uBAAuB,EACrC,IACA,QACA,YAKC;CACD,OAAO;EACL,SAAS,2BAA2B,GAAG,eAAe,OAAO;EAC7D,mBAAmB;EACnB;EACA;EACA,SAAS;CACX;AACF"}
@@ -0,0 +1,96 @@
1
+ import { createBackgroundTaskId, createDurableBackgroundTaskId, getBackgroundChildRun, getOrCreateBackgroundChildRun } from "./subagent-background-child-run.js";
2
+ import { backgroundCancelledLaunchOutput, backgroundLaunchOutput, backgroundReplayOutput, backgroundRunJobStatus, hasBackgroundJobCapacity } from "./subagent-job-state.js";
3
+ import { startInProcessBackgroundJob } from "./subagent-background-in-process.js";
4
+ import { scheduleDurableBackgroundJob } from "./subagent-background-schedule.js";
5
+ //#region src/subagent-jobs.ts
6
+ const maxBackgroundJobs = 64;
7
+ const maxRetainedBackgroundJobs = maxBackgroundJobs * 4;
8
+ async function startBackgroundJob({ abortSignal, description, executionHost, jobs, groupId, groups = /* @__PURE__ */ new Map(), delegateToolCallId, parentSession, parentRunId, parentSessionKey, ownerNamespace, prompt, registerCleanup, sessionKey, subagent }) {
9
+ const existingChildRun = await getBackgroundChildRun({
10
+ delegateToolCallId,
11
+ executionHost,
12
+ parentRunId,
13
+ parentSessionKey,
14
+ prompt,
15
+ sessionKey
16
+ });
17
+ const id = existingChildRun?.publicTaskId ?? (executionHost ? await createDurableBackgroundTaskId({
18
+ delegateToolCallId,
19
+ prompt,
20
+ sessionKey
21
+ }) : createBackgroundTaskId());
22
+ const existingJob = jobs.get(id);
23
+ if (existingJob) return backgroundLaunchOutput(existingJob);
24
+ const replayedStatus = backgroundRunJobStatus(existingChildRun?.status);
25
+ if (existingChildRun?.publicTaskId && replayedStatus) return backgroundReplayOutput({
26
+ id: existingChildRun.publicTaskId,
27
+ status: replayedStatus,
28
+ subagent: subagent.name ?? "subagent"
29
+ });
30
+ if (!hasBackgroundJobCapacity({
31
+ jobs,
32
+ maxActiveJobs: maxBackgroundJobs,
33
+ maxRetainedJobs: maxRetainedBackgroundJobs
34
+ })) return {
35
+ message: "Background subagent job was not started because the background job limit is full.",
36
+ run_in_background: true,
37
+ status: "cancelled",
38
+ subagent: subagent.name,
39
+ task_id: id
40
+ };
41
+ if (abortSignal.aborted) return backgroundCancelledLaunchOutput({
42
+ id,
43
+ subagent: subagent.name
44
+ });
45
+ const childRun = executionHost ? existingChildRun ?? await getOrCreateBackgroundChildRun({
46
+ delegateToolCallId,
47
+ description,
48
+ executionHost,
49
+ groupId,
50
+ ownerNamespace,
51
+ parentRunId,
52
+ parentSessionKey,
53
+ prompt,
54
+ publicTaskId: id,
55
+ sessionKey,
56
+ subagent: subagent.name ?? "subagent"
57
+ }) : void 0;
58
+ if (executionHost?.capabilities.backgroundSubagents === "durable" && childRun) return backgroundLaunchOutput(await scheduleDurableBackgroundJob({
59
+ childRun,
60
+ delegateToolCallId,
61
+ description,
62
+ executionHost,
63
+ groupId,
64
+ groups,
65
+ id,
66
+ jobs,
67
+ parentRunId,
68
+ parentSession,
69
+ parentSessionKey,
70
+ ownerNamespace,
71
+ subagent: subagent.name ?? "subagent"
72
+ }));
73
+ return await startInProcessBackgroundJob({
74
+ abortSignal,
75
+ childRun,
76
+ delegateToolCallId,
77
+ description,
78
+ executionHost,
79
+ groupId,
80
+ groups,
81
+ id,
82
+ jobs,
83
+ parentRunId,
84
+ parentSession,
85
+ parentSessionKey,
86
+ ownerNamespace,
87
+ prompt,
88
+ registerCleanup,
89
+ sessionKey,
90
+ subagent
91
+ });
92
+ }
93
+ //#endregion
94
+ export { startBackgroundJob };
95
+
96
+ //# sourceMappingURL=subagent-jobs.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"subagent-jobs.js","names":[],"sources":["../src/subagent-jobs.ts"],"sourcesContent":["import type { ExecutionHost } from \"./execution/types\";\nimport type { AgentInput } from \"./session/input\";\nimport {\n createBackgroundTaskId,\n createDurableBackgroundTaskId,\n getBackgroundChildRun,\n getOrCreateBackgroundChildRun,\n} from \"./subagent-background-child-run\";\nimport { startInProcessBackgroundJob } from \"./subagent-background-in-process\";\nimport { scheduleDurableBackgroundJob } from \"./subagent-background-schedule\";\nimport {\n backgroundCancelledLaunchOutput,\n backgroundLaunchOutput,\n backgroundReplayOutput,\n backgroundRunJobStatus,\n hasBackgroundJobCapacity,\n} from \"./subagent-job-state\";\nimport type {\n RuntimeInputSink,\n Subagent,\n SubagentJob,\n SubagentJobGroup,\n} from \"./subagent-types\";\n\nconst maxBackgroundJobs = 64;\nconst maxRetainedBackgroundJobs = maxBackgroundJobs * 4;\n\nexport async function startBackgroundJob({\n abortSignal,\n description,\n executionHost,\n jobs,\n groupId,\n groups = new Map<string, SubagentJobGroup>(),\n delegateToolCallId,\n parentSession,\n parentRunId,\n parentSessionKey,\n ownerNamespace,\n prompt,\n registerCleanup,\n sessionKey,\n subagent,\n}: {\n readonly abortSignal: AbortSignal;\n readonly description?: string;\n readonly executionHost?: ExecutionHost;\n readonly jobs: Map<string, SubagentJob>;\n readonly groupId?: string;\n readonly groups?: Map<string, SubagentJobGroup>;\n readonly delegateToolCallId?: string;\n readonly parentSession: RuntimeInputSink;\n readonly parentRunId?: string;\n readonly parentSessionKey?: string;\n readonly ownerNamespace?: string;\n readonly prompt: AgentInput;\n readonly registerCleanup: (cleanup: () => Promise<void>) => () => void;\n readonly sessionKey: string;\n readonly subagent: Subagent;\n}) {\n const existingChildRun = await getBackgroundChildRun({\n delegateToolCallId,\n executionHost,\n parentRunId,\n parentSessionKey,\n prompt,\n sessionKey,\n });\n const id =\n existingChildRun?.publicTaskId ??\n (executionHost\n ? await createDurableBackgroundTaskId({\n delegateToolCallId,\n prompt,\n sessionKey,\n })\n : createBackgroundTaskId());\n const existingJob = jobs.get(id);\n if (existingJob) {\n return backgroundLaunchOutput(existingJob);\n }\n const replayedStatus = backgroundRunJobStatus(existingChildRun?.status);\n if (existingChildRun?.publicTaskId && replayedStatus) {\n return backgroundReplayOutput({\n id: existingChildRun.publicTaskId,\n status: replayedStatus,\n subagent: subagent.name ?? \"subagent\",\n });\n }\n\n if (\n !hasBackgroundJobCapacity({\n jobs,\n maxActiveJobs: maxBackgroundJobs,\n maxRetainedJobs: maxRetainedBackgroundJobs,\n })\n ) {\n return {\n message:\n \"Background subagent job was not started because the background job limit is full.\",\n run_in_background: true,\n status: \"cancelled\",\n subagent: subagent.name,\n task_id: id,\n };\n }\n\n if (abortSignal.aborted) {\n return backgroundCancelledLaunchOutput({ id, subagent: subagent.name });\n }\n\n const childRun = executionHost\n ? (existingChildRun ??\n (await getOrCreateBackgroundChildRun({\n delegateToolCallId,\n description,\n executionHost,\n groupId,\n ownerNamespace,\n parentRunId,\n parentSessionKey,\n prompt,\n publicTaskId: id,\n sessionKey,\n subagent: subagent.name ?? \"subagent\",\n })))\n : undefined;\n if (\n executionHost?.capabilities.backgroundSubagents === \"durable\" &&\n childRun\n ) {\n const job = await scheduleDurableBackgroundJob({\n childRun,\n delegateToolCallId,\n description,\n executionHost,\n groupId,\n groups,\n id,\n jobs,\n parentRunId,\n parentSession,\n parentSessionKey,\n ownerNamespace,\n subagent: subagent.name ?? \"subagent\",\n });\n return backgroundLaunchOutput(job);\n }\n return await startInProcessBackgroundJob({\n abortSignal,\n childRun,\n delegateToolCallId,\n description,\n executionHost,\n groupId,\n groups,\n id,\n jobs,\n parentRunId,\n parentSession,\n parentSessionKey,\n ownerNamespace,\n prompt,\n registerCleanup,\n sessionKey,\n subagent,\n });\n}\n"],"mappings":";;;;;AAwBA,MAAM,oBAAoB;AAC1B,MAAM,4BAA4B,oBAAoB;AAEtD,eAAsB,mBAAmB,EACvC,aACA,aACA,eACA,MACA,SACA,yBAAS,IAAI,IAA8B,GAC3C,oBACA,eACA,aACA,kBACA,gBACA,QACA,iBACA,YACA,YAiBC;CACD,MAAM,mBAAmB,MAAM,sBAAsB;EACnD;EACA;EACA;EACA;EACA;EACA;CACF,CAAC;CACD,MAAM,KACJ,kBAAkB,iBACjB,gBACG,MAAM,8BAA8B;EAClC;EACA;EACA;CACF,CAAC,IACD,uBAAuB;CAC7B,MAAM,cAAc,KAAK,IAAI,EAAE;CAC/B,IAAI,aACF,OAAO,uBAAuB,WAAW;CAE3C,MAAM,iBAAiB,uBAAuB,kBAAkB,MAAM;CACtE,IAAI,kBAAkB,gBAAgB,gBACpC,OAAO,uBAAuB;EAC5B,IAAI,iBAAiB;EACrB,QAAQ;EACR,UAAU,SAAS,QAAQ;CAC7B,CAAC;CAGH,IACE,CAAC,yBAAyB;EACxB;EACA,eAAe;EACf,iBAAiB;CACnB,CAAC,GAED,OAAO;EACL,SACE;EACF,mBAAmB;EACnB,QAAQ;EACR,UAAU,SAAS;EACnB,SAAS;CACX;CAGF,IAAI,YAAY,SACd,OAAO,gCAAgC;EAAE;EAAI,UAAU,SAAS;CAAK,CAAC;CAGxE,MAAM,WAAW,gBACZ,oBACA,MAAM,8BAA8B;EACnC;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA,cAAc;EACd;EACA,UAAU,SAAS,QAAQ;CAC7B,CAAC,IACD,KAAA;CACJ,IACE,eAAe,aAAa,wBAAwB,aACpD,UAiBA,OAAO,uBAAuB,MAfZ,6BAA6B;EAC7C;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA,UAAU,SAAS,QAAQ;CAC7B,CAAC,CACgC;CAEnC,OAAO,MAAM,4BAA4B;EACvC;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;CACF,CAAC;AACH"}
@@ -0,0 +1,114 @@
1
+ //#region src/subagent-prompt-schema.ts
2
+ const contentArraySchema = {
3
+ items: { anyOf: [
4
+ {
5
+ additionalProperties: false,
6
+ properties: {
7
+ text: { type: "string" },
8
+ type: { const: "text" }
9
+ },
10
+ required: ["type", "text"],
11
+ type: "object"
12
+ },
13
+ {
14
+ additionalProperties: false,
15
+ properties: {
16
+ image: { type: "string" },
17
+ mediaType: { type: "string" },
18
+ type: { const: "image" }
19
+ },
20
+ required: ["type", "image"],
21
+ type: "object"
22
+ },
23
+ {
24
+ additionalProperties: false,
25
+ properties: {
26
+ data: { anyOf: [
27
+ { type: "string" },
28
+ {
29
+ additionalProperties: false,
30
+ properties: {
31
+ data: { type: "string" },
32
+ type: { const: "data" }
33
+ },
34
+ required: ["type", "data"],
35
+ type: "object"
36
+ },
37
+ {
38
+ additionalProperties: false,
39
+ properties: {
40
+ reference: {
41
+ additionalProperties: { type: "string" },
42
+ type: "object"
43
+ },
44
+ type: { const: "reference" }
45
+ },
46
+ required: ["type", "reference"],
47
+ type: "object"
48
+ },
49
+ {
50
+ additionalProperties: false,
51
+ properties: {
52
+ text: { type: "string" },
53
+ type: { const: "text" }
54
+ },
55
+ required: ["type", "text"],
56
+ type: "object"
57
+ },
58
+ {
59
+ additionalProperties: false,
60
+ properties: {
61
+ type: { const: "url" },
62
+ url: { type: "string" }
63
+ },
64
+ required: ["type", "url"],
65
+ type: "object"
66
+ }
67
+ ] },
68
+ filename: { type: "string" },
69
+ mediaType: { type: "string" },
70
+ type: { const: "file" }
71
+ },
72
+ required: [
73
+ "type",
74
+ "data",
75
+ "mediaType"
76
+ ],
77
+ type: "object"
78
+ }
79
+ ] },
80
+ type: "array"
81
+ };
82
+ const delegatePromptSchema = { anyOf: [
83
+ { type: "string" },
84
+ {
85
+ items: { type: "string" },
86
+ type: "array"
87
+ },
88
+ {
89
+ additionalProperties: false,
90
+ properties: {
91
+ text: { anyOf: [{ type: "string" }, {
92
+ items: { type: "string" },
93
+ type: "array"
94
+ }] },
95
+ type: { const: "user-text" }
96
+ },
97
+ required: ["type", "text"],
98
+ type: "object"
99
+ },
100
+ {
101
+ additionalProperties: false,
102
+ properties: {
103
+ content: contentArraySchema,
104
+ type: { const: "user-message" }
105
+ },
106
+ required: ["type", "content"],
107
+ type: "object"
108
+ },
109
+ contentArraySchema
110
+ ] };
111
+ //#endregion
112
+ export { delegatePromptSchema };
113
+
114
+ //# sourceMappingURL=subagent-prompt-schema.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"subagent-prompt-schema.js","names":[],"sources":["../src/subagent-prompt-schema.ts"],"sourcesContent":["const fileDataSchema = {\n anyOf: [\n { type: \"string\" },\n {\n additionalProperties: false,\n properties: {\n data: { type: \"string\" },\n type: { const: \"data\" },\n },\n required: [\"type\", \"data\"],\n type: \"object\",\n },\n {\n additionalProperties: false,\n properties: {\n reference: {\n additionalProperties: { type: \"string\" },\n type: \"object\",\n },\n type: { const: \"reference\" },\n },\n required: [\"type\", \"reference\"],\n type: \"object\",\n },\n {\n additionalProperties: false,\n properties: {\n text: { type: \"string\" },\n type: { const: \"text\" },\n },\n required: [\"type\", \"text\"],\n type: \"object\",\n },\n {\n additionalProperties: false,\n properties: {\n type: { const: \"url\" },\n url: { type: \"string\" },\n },\n required: [\"type\", \"url\"],\n type: \"object\",\n },\n ],\n};\n\nconst contentPartSchema = {\n anyOf: [\n {\n additionalProperties: false,\n properties: {\n text: { type: \"string\" },\n type: { const: \"text\" },\n },\n required: [\"type\", \"text\"],\n type: \"object\",\n },\n {\n additionalProperties: false,\n properties: {\n image: { type: \"string\" },\n mediaType: { type: \"string\" },\n type: { const: \"image\" },\n },\n required: [\"type\", \"image\"],\n type: \"object\",\n },\n {\n additionalProperties: false,\n properties: {\n data: fileDataSchema,\n filename: { type: \"string\" },\n mediaType: { type: \"string\" },\n type: { const: \"file\" },\n },\n required: [\"type\", \"data\", \"mediaType\"],\n type: \"object\",\n },\n ],\n};\n\nconst contentArraySchema = {\n items: contentPartSchema,\n type: \"array\",\n};\n\nexport const delegatePromptSchema = {\n anyOf: [\n { type: \"string\" },\n { items: { type: \"string\" }, type: \"array\" },\n {\n additionalProperties: false,\n properties: {\n text: {\n anyOf: [\n { type: \"string\" },\n { items: { type: \"string\" }, type: \"array\" },\n ],\n },\n type: { const: \"user-text\" },\n },\n required: [\"type\", \"text\"],\n type: \"object\",\n },\n {\n additionalProperties: false,\n properties: {\n content: contentArraySchema,\n type: { const: \"user-message\" },\n },\n required: [\"type\", \"content\"],\n type: \"object\",\n },\n contentArraySchema,\n ],\n};\n"],"mappings":";AAgFA,MAAM,qBAAqB;CACzB,OAAO,EAnCP,OAAO;EACL;GACE,sBAAsB;GACtB,YAAY;IACV,MAAM,EAAE,MAAM,SAAS;IACvB,MAAM,EAAE,OAAO,OAAO;GACxB;GACA,UAAU,CAAC,QAAQ,MAAM;GACzB,MAAM;EACR;EACA;GACE,sBAAsB;GACtB,YAAY;IACV,OAAO,EAAE,MAAM,SAAS;IACxB,WAAW,EAAE,MAAM,SAAS;IAC5B,MAAM,EAAE,OAAO,QAAQ;GACzB;GACA,UAAU,CAAC,QAAQ,OAAO;GAC1B,MAAM;EACR;EACA;GACE,sBAAsB;GACtB,YAAY;IACV,MAAM,EApEZ,OAAO;KACL,EAAE,MAAM,SAAS;KACjB;MACE,sBAAsB;MACtB,YAAY;OACV,MAAM,EAAE,MAAM,SAAS;OACvB,MAAM,EAAE,OAAO,OAAO;MACxB;MACA,UAAU,CAAC,QAAQ,MAAM;MACzB,MAAM;KACR;KACA;MACE,sBAAsB;MACtB,YAAY;OACV,WAAW;QACT,sBAAsB,EAAE,MAAM,SAAS;QACvC,MAAM;OACR;OACA,MAAM,EAAE,OAAO,YAAY;MAC7B;MACA,UAAU,CAAC,QAAQ,WAAW;MAC9B,MAAM;KACR;KACA;MACE,sBAAsB;MACtB,YAAY;OACV,MAAM,EAAE,MAAM,SAAS;OACvB,MAAM,EAAE,OAAO,OAAO;MACxB;MACA,UAAU,CAAC,QAAQ,MAAM;MACzB,MAAM;KACR;KACA;MACE,sBAAsB;MACtB,YAAY;OACV,MAAM,EAAE,OAAO,MAAM;OACrB,KAAK,EAAE,MAAM,SAAS;MACxB;MACA,UAAU,CAAC,QAAQ,KAAK;MACxB,MAAM;KACR;IACF,EA2ByB;IACnB,UAAU,EAAE,MAAM,SAAS;IAC3B,WAAW,EAAE,MAAM,SAAS;IAC5B,MAAM,EAAE,OAAO,OAAO;GACxB;GACA,UAAU;IAAC;IAAQ;IAAQ;GAAW;GACtC,MAAM;EACR;CACF,EAIuB;CACvB,MAAM;AACR;AAEA,MAAa,uBAAuB,EAClC,OAAO;CACL,EAAE,MAAM,SAAS;CACjB;EAAE,OAAO,EAAE,MAAM,SAAS;EAAG,MAAM;CAAQ;CAC3C;EACE,sBAAsB;EACtB,YAAY;GACV,MAAM,EACJ,OAAO,CACL,EAAE,MAAM,SAAS,GACjB;IAAE,OAAO,EAAE,MAAM,SAAS;IAAG,MAAM;GAAQ,CAC7C,EACF;GACA,MAAM,EAAE,OAAO,YAAY;EAC7B;EACA,UAAU,CAAC,QAAQ,MAAM;EACzB,MAAM;CACR;CACA;EACE,sBAAsB;EACtB,YAAY;GACV,SAAS;GACT,MAAM,EAAE,OAAO,eAAe;EAChC;EACA,UAAU,CAAC,QAAQ,SAAS;EAC5B,MAAM;CACR;CACA;AACF,EACF"}
@@ -0,0 +1,111 @@
1
+ //#region src/subagent-run.ts
2
+ const maxCompactTextLength = 2e4;
3
+ const maxStoredEvents = 200;
4
+ const childSessionKeySuffixPattern = /^[A-Za-z0-9_-]{1,80}$/;
5
+ async function runBlockingDelegation({ abortSignal, prompt, sessionKey, subagent }) {
6
+ const childSession = subagent.session(sessionKey);
7
+ if (abortSignal?.aborted) return {
8
+ eventCount: 0,
9
+ result: "aborted",
10
+ run_in_background: false,
11
+ subagent: subagent.name ?? "subagent",
12
+ text: ""
13
+ };
14
+ const abort = () => childSession.interrupt();
15
+ abortSignal?.addEventListener("abort", abort, { once: true });
16
+ try {
17
+ return await collectSubagentRun(await childSession.send(prompt), subagent.name ?? "subagent");
18
+ } finally {
19
+ abortSignal?.removeEventListener("abort", abort);
20
+ }
21
+ }
22
+ async function collectSubagentRun(run, subagent) {
23
+ return (await collectSubagentRunWithEvents(run, subagent)).result;
24
+ }
25
+ async function collectSubagentRunWithEvents(run, subagent, onEvent) {
26
+ let eventCount = 0;
27
+ let result = "completed";
28
+ const events = [];
29
+ const textParts = [];
30
+ let textLength = 0;
31
+ let textTruncated = false;
32
+ try {
33
+ for await (const event of run.events()) {
34
+ eventCount += 1;
35
+ if (events.length < maxStoredEvents) events.push(event);
36
+ await onEvent?.(event);
37
+ if (event.type === "assistant-text") {
38
+ const appended = appendCompactText(textParts, textLength, event.text);
39
+ textLength = appended.length;
40
+ textTruncated ||= appended.truncated;
41
+ } else if (event.type === "turn-abort") result = "aborted";
42
+ else if (event.type === "turn-error") return {
43
+ retainedEvents: events,
44
+ result: {
45
+ error: event.message,
46
+ eventCount,
47
+ result: "error",
48
+ run_in_background: false,
49
+ subagent,
50
+ text: compactText(textParts, textTruncated)
51
+ }
52
+ };
53
+ }
54
+ } catch (error) {
55
+ return {
56
+ retainedEvents: events,
57
+ result: {
58
+ error: errorMessage(error),
59
+ eventCount,
60
+ result: "error",
61
+ run_in_background: false,
62
+ subagent,
63
+ text: compactText(textParts, textTruncated)
64
+ }
65
+ };
66
+ }
67
+ return {
68
+ retainedEvents: events,
69
+ result: {
70
+ eventCount,
71
+ result,
72
+ run_in_background: false,
73
+ subagent,
74
+ text: compactText(textParts, textTruncated)
75
+ }
76
+ };
77
+ }
78
+ function defaultChildSessionKey(parentAgentNamespace, parentSessionKey, subagent) {
79
+ return `parent:${parentAgentNamespace}:${parentSessionKey}:subagent:${subagent}`;
80
+ }
81
+ function scopedChildSessionKey({ parentAgentNamespace, parentSessionKey, sessionKey, subagent }) {
82
+ const base = defaultChildSessionKey(parentAgentNamespace, parentSessionKey, subagent);
83
+ if (!sessionKey) return base;
84
+ if (!childSessionKeySuffixPattern.test(sessionKey)) throw new Error("delegate sessionKey must be a short alphanumeric child-session suffix");
85
+ return `${base}:${sessionKey}`;
86
+ }
87
+ function compactText(parts, truncated) {
88
+ const text = parts.join("");
89
+ return truncated ? `${text}…[truncated]` : text;
90
+ }
91
+ function appendCompactText(parts, currentLength, next) {
92
+ if (currentLength >= maxCompactTextLength) return {
93
+ length: currentLength,
94
+ truncated: next.length > 0
95
+ };
96
+ const remaining = maxCompactTextLength - currentLength;
97
+ const chunk = next.length > remaining ? next.slice(0, remaining) : next;
98
+ parts.push(chunk);
99
+ return {
100
+ length: currentLength + chunk.length,
101
+ truncated: next.length > remaining
102
+ };
103
+ }
104
+ function errorMessage(error) {
105
+ if (error instanceof Error) return error.message;
106
+ return String(error);
107
+ }
108
+ //#endregion
109
+ export { collectSubagentRunWithEvents, runBlockingDelegation, scopedChildSessionKey };
110
+
111
+ //# sourceMappingURL=subagent-run.js.map