@assemblyline-agents/runtime 0.1.0

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 (331) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +19 -0
  3. package/dist/agent-harness.d.ts +108 -0
  4. package/dist/agent-harness.d.ts.map +1 -0
  5. package/dist/agent-harness.js +2 -0
  6. package/dist/agent-harness.js.map +1 -0
  7. package/dist/attachments-legacy.d.ts +17 -0
  8. package/dist/attachments-legacy.d.ts.map +1 -0
  9. package/dist/attachments-legacy.js +273 -0
  10. package/dist/attachments-legacy.js.map +1 -0
  11. package/dist/attachments.d.ts +49 -0
  12. package/dist/attachments.d.ts.map +1 -0
  13. package/dist/attachments.js +832 -0
  14. package/dist/attachments.js.map +1 -0
  15. package/dist/builtin-tool-executors.d.ts +96 -0
  16. package/dist/builtin-tool-executors.d.ts.map +1 -0
  17. package/dist/builtin-tool-executors.js +501 -0
  18. package/dist/builtin-tool-executors.js.map +1 -0
  19. package/dist/builtin-tools.d.ts +30 -0
  20. package/dist/builtin-tools.d.ts.map +1 -0
  21. package/dist/builtin-tools.js +690 -0
  22. package/dist/builtin-tools.js.map +1 -0
  23. package/dist/capability-catalog.d.ts +27 -0
  24. package/dist/capability-catalog.d.ts.map +1 -0
  25. package/dist/capability-catalog.js +193 -0
  26. package/dist/capability-catalog.js.map +1 -0
  27. package/dist/channel-http-util.d.ts +34 -0
  28. package/dist/channel-http-util.d.ts.map +1 -0
  29. package/dist/channel-http-util.js +93 -0
  30. package/dist/channel-http-util.js.map +1 -0
  31. package/dist/channel-ingress.d.ts +56 -0
  32. package/dist/channel-ingress.d.ts.map +1 -0
  33. package/dist/channel-ingress.js +273 -0
  34. package/dist/channel-ingress.js.map +1 -0
  35. package/dist/channel.d.ts +146 -0
  36. package/dist/channel.d.ts.map +1 -0
  37. package/dist/channel.js +2 -0
  38. package/dist/channel.js.map +1 -0
  39. package/dist/checkpoints.d.ts +34 -0
  40. package/dist/checkpoints.d.ts.map +1 -0
  41. package/dist/checkpoints.js +86 -0
  42. package/dist/checkpoints.js.map +1 -0
  43. package/dist/code-mode.d.ts +20 -0
  44. package/dist/code-mode.d.ts.map +1 -0
  45. package/dist/code-mode.js +658 -0
  46. package/dist/code-mode.js.map +1 -0
  47. package/dist/computer-relay-protocol.d.ts +36 -0
  48. package/dist/computer-relay-protocol.d.ts.map +1 -0
  49. package/dist/computer-relay-protocol.js +127 -0
  50. package/dist/computer-relay-protocol.js.map +1 -0
  51. package/dist/concurrency.d.ts +68 -0
  52. package/dist/concurrency.d.ts.map +1 -0
  53. package/dist/concurrency.js +193 -0
  54. package/dist/concurrency.js.map +1 -0
  55. package/dist/connection-auth-util.d.ts +56 -0
  56. package/dist/connection-auth-util.d.ts.map +1 -0
  57. package/dist/connection-auth-util.js +154 -0
  58. package/dist/connection-auth-util.js.map +1 -0
  59. package/dist/connection-clients.d.ts +50 -0
  60. package/dist/connection-clients.d.ts.map +1 -0
  61. package/dist/connection-clients.js +402 -0
  62. package/dist/connection-clients.js.map +1 -0
  63. package/dist/connection-stores.d.ts +108 -0
  64. package/dist/connection-stores.d.ts.map +1 -0
  65. package/dist/connection-stores.js +127 -0
  66. package/dist/connection-stores.js.map +1 -0
  67. package/dist/connection-util.d.ts +27 -0
  68. package/dist/connection-util.d.ts.map +1 -0
  69. package/dist/connection-util.js +118 -0
  70. package/dist/connection-util.js.map +1 -0
  71. package/dist/connections.d.ts +152 -0
  72. package/dist/connections.d.ts.map +1 -0
  73. package/dist/connections.js +838 -0
  74. package/dist/connections.js.map +1 -0
  75. package/dist/content-capture.d.ts +32 -0
  76. package/dist/content-capture.d.ts.map +1 -0
  77. package/dist/content-capture.js +77 -0
  78. package/dist/content-capture.js.map +1 -0
  79. package/dist/context-policy-util.d.ts +45 -0
  80. package/dist/context-policy-util.d.ts.map +1 -0
  81. package/dist/context-policy-util.js +120 -0
  82. package/dist/context-policy-util.js.map +1 -0
  83. package/dist/conversation-context.d.ts +55 -0
  84. package/dist/conversation-context.d.ts.map +1 -0
  85. package/dist/conversation-context.js +179 -0
  86. package/dist/conversation-context.js.map +1 -0
  87. package/dist/conversation-transcript.d.ts +49 -0
  88. package/dist/conversation-transcript.d.ts.map +1 -0
  89. package/dist/conversation-transcript.js +77 -0
  90. package/dist/conversation-transcript.js.map +1 -0
  91. package/dist/conversation-turn-dispatch.d.ts +46 -0
  92. package/dist/conversation-turn-dispatch.d.ts.map +1 -0
  93. package/dist/conversation-turn-dispatch.js +201 -0
  94. package/dist/conversation-turn-dispatch.js.map +1 -0
  95. package/dist/delivery-queue.d.ts +89 -0
  96. package/dist/delivery-queue.d.ts.map +1 -0
  97. package/dist/delivery-queue.js +244 -0
  98. package/dist/delivery-queue.js.map +1 -0
  99. package/dist/delivery-send.d.ts +44 -0
  100. package/dist/delivery-send.d.ts.map +1 -0
  101. package/dist/delivery-send.js +494 -0
  102. package/dist/delivery-send.js.map +1 -0
  103. package/dist/env-config.d.ts +6 -0
  104. package/dist/env-config.d.ts.map +1 -0
  105. package/dist/env-config.js +27 -0
  106. package/dist/env-config.js.map +1 -0
  107. package/dist/eval-run-options.d.ts +37 -0
  108. package/dist/eval-run-options.d.ts.map +1 -0
  109. package/dist/eval-run-options.js +106 -0
  110. package/dist/eval-run-options.js.map +1 -0
  111. package/dist/harness-continuation.d.ts +54 -0
  112. package/dist/harness-continuation.d.ts.map +1 -0
  113. package/dist/harness-continuation.js +58 -0
  114. package/dist/harness-continuation.js.map +1 -0
  115. package/dist/harness-loop.d.ts +110 -0
  116. package/dist/harness-loop.d.ts.map +1 -0
  117. package/dist/harness-loop.js +575 -0
  118. package/dist/harness-loop.js.map +1 -0
  119. package/dist/hook-dispatch.d.ts +19 -0
  120. package/dist/hook-dispatch.d.ts.map +1 -0
  121. package/dist/hook-dispatch.js +37 -0
  122. package/dist/hook-dispatch.js.map +1 -0
  123. package/dist/http.d.ts +26 -0
  124. package/dist/http.d.ts.map +1 -0
  125. package/dist/http.js +114 -0
  126. package/dist/http.js.map +1 -0
  127. package/dist/in-memory-conversation-turn-store.d.ts +29 -0
  128. package/dist/in-memory-conversation-turn-store.d.ts.map +1 -0
  129. package/dist/in-memory-conversation-turn-store.js +166 -0
  130. package/dist/in-memory-conversation-turn-store.js.map +1 -0
  131. package/dist/in-memory-run-store.d.ts +39 -0
  132. package/dist/in-memory-run-store.d.ts.map +1 -0
  133. package/dist/in-memory-run-store.js +332 -0
  134. package/dist/in-memory-run-store.js.map +1 -0
  135. package/dist/in-memory-sandbox-session-store.d.ts +33 -0
  136. package/dist/in-memory-sandbox-session-store.d.ts.map +1 -0
  137. package/dist/in-memory-sandbox-session-store.js +186 -0
  138. package/dist/in-memory-sandbox-session-store.js.map +1 -0
  139. package/dist/in-memory-small-stores.d.ts +117 -0
  140. package/dist/in-memory-small-stores.d.ts.map +1 -0
  141. package/dist/in-memory-small-stores.js +340 -0
  142. package/dist/in-memory-small-stores.js.map +1 -0
  143. package/dist/in-memory-usage-store.d.ts +15 -0
  144. package/dist/in-memory-usage-store.d.ts.map +1 -0
  145. package/dist/in-memory-usage-store.js +49 -0
  146. package/dist/in-memory-usage-store.js.map +1 -0
  147. package/dist/index.d.ts +467 -0
  148. package/dist/index.d.ts.map +1 -0
  149. package/dist/index.js +2046 -0
  150. package/dist/index.js.map +1 -0
  151. package/dist/json-util.d.ts +19 -0
  152. package/dist/json-util.d.ts.map +1 -0
  153. package/dist/json-util.js +59 -0
  154. package/dist/json-util.js.map +1 -0
  155. package/dist/local-sandbox.d.ts +16 -0
  156. package/dist/local-sandbox.d.ts.map +1 -0
  157. package/dist/local-sandbox.js +327 -0
  158. package/dist/local-sandbox.js.map +1 -0
  159. package/dist/logging.d.ts +16 -0
  160. package/dist/logging.d.ts.map +1 -0
  161. package/dist/logging.js +41 -0
  162. package/dist/logging.js.map +1 -0
  163. package/dist/mcp-client.d.ts +27 -0
  164. package/dist/mcp-client.d.ts.map +1 -0
  165. package/dist/mcp-client.js +169 -0
  166. package/dist/mcp-client.js.map +1 -0
  167. package/dist/model-harnesses.d.ts +4 -0
  168. package/dist/model-harnesses.d.ts.map +1 -0
  169. package/dist/model-harnesses.js +6 -0
  170. package/dist/model-harnesses.js.map +1 -0
  171. package/dist/model-media.d.ts +36 -0
  172. package/dist/model-media.d.ts.map +1 -0
  173. package/dist/model-media.js +234 -0
  174. package/dist/model-media.js.map +1 -0
  175. package/dist/operator-controls.d.ts +31 -0
  176. package/dist/operator-controls.d.ts.map +1 -0
  177. package/dist/operator-controls.js +71 -0
  178. package/dist/operator-controls.js.map +1 -0
  179. package/dist/pi-agent-harness.d.ts +40 -0
  180. package/dist/pi-agent-harness.d.ts.map +1 -0
  181. package/dist/pi-agent-harness.js +222 -0
  182. package/dist/pi-agent-harness.js.map +1 -0
  183. package/dist/pi-harness.d.ts +42 -0
  184. package/dist/pi-harness.d.ts.map +1 -0
  185. package/dist/pi-harness.js +142 -0
  186. package/dist/pi-harness.js.map +1 -0
  187. package/dist/relay-mcp-client.d.ts +17 -0
  188. package/dist/relay-mcp-client.d.ts.map +1 -0
  189. package/dist/relay-mcp-client.js +119 -0
  190. package/dist/relay-mcp-client.js.map +1 -0
  191. package/dist/replay-resume.d.ts +107 -0
  192. package/dist/replay-resume.d.ts.map +1 -0
  193. package/dist/replay-resume.js +495 -0
  194. package/dist/replay-resume.js.map +1 -0
  195. package/dist/resource-projection-util.d.ts +58 -0
  196. package/dist/resource-projection-util.d.ts.map +1 -0
  197. package/dist/resource-projection-util.js +298 -0
  198. package/dist/resource-projection-util.js.map +1 -0
  199. package/dist/resource-projection.d.ts +45 -0
  200. package/dist/resource-projection.d.ts.map +1 -0
  201. package/dist/resource-projection.js +481 -0
  202. package/dist/resource-projection.js.map +1 -0
  203. package/dist/run-heartbeat.d.ts +34 -0
  204. package/dist/run-heartbeat.d.ts.map +1 -0
  205. package/dist/run-heartbeat.js +62 -0
  206. package/dist/run-heartbeat.js.map +1 -0
  207. package/dist/run-outcome.d.ts +47 -0
  208. package/dist/run-outcome.d.ts.map +1 -0
  209. package/dist/run-outcome.js +92 -0
  210. package/dist/run-outcome.js.map +1 -0
  211. package/dist/run-recovery.d.ts +117 -0
  212. package/dist/run-recovery.d.ts.map +1 -0
  213. package/dist/run-recovery.js +466 -0
  214. package/dist/run-recovery.js.map +1 -0
  215. package/dist/run-resolver.d.ts +35 -0
  216. package/dist/run-resolver.d.ts.map +1 -0
  217. package/dist/run-resolver.js +84 -0
  218. package/dist/run-resolver.js.map +1 -0
  219. package/dist/run-stream.d.ts +37 -0
  220. package/dist/run-stream.d.ts.map +1 -0
  221. package/dist/run-stream.js +78 -0
  222. package/dist/run-stream.js.map +1 -0
  223. package/dist/runs-query.d.ts +29 -0
  224. package/dist/runs-query.d.ts.map +1 -0
  225. package/dist/runs-query.js +142 -0
  226. package/dist/runs-query.js.map +1 -0
  227. package/dist/runtime-bundle.d.ts +3 -0
  228. package/dist/runtime-bundle.d.ts.map +1 -0
  229. package/dist/runtime-bundle.js +18 -0
  230. package/dist/runtime-bundle.js.map +1 -0
  231. package/dist/runtime-types.d.ts +397 -0
  232. package/dist/runtime-types.d.ts.map +1 -0
  233. package/dist/runtime-types.js +2 -0
  234. package/dist/runtime-types.js.map +1 -0
  235. package/dist/runtime-util.d.ts +45 -0
  236. package/dist/runtime-util.d.ts.map +1 -0
  237. package/dist/runtime-util.js +116 -0
  238. package/dist/runtime-util.js.map +1 -0
  239. package/dist/sandbox-cli-client.d.ts +20 -0
  240. package/dist/sandbox-cli-client.d.ts.map +1 -0
  241. package/dist/sandbox-cli-client.js +97 -0
  242. package/dist/sandbox-cli-client.js.map +1 -0
  243. package/dist/sandbox-fs.d.ts +56 -0
  244. package/dist/sandbox-fs.d.ts.map +1 -0
  245. package/dist/sandbox-fs.js +276 -0
  246. package/dist/sandbox-fs.js.map +1 -0
  247. package/dist/sandbox-session-key.d.ts +8 -0
  248. package/dist/sandbox-session-key.d.ts.map +1 -0
  249. package/dist/sandbox-session-key.js +32 -0
  250. package/dist/sandbox-session-key.js.map +1 -0
  251. package/dist/sandbox-snapshot-policy.d.ts +27 -0
  252. package/dist/sandbox-snapshot-policy.d.ts.map +1 -0
  253. package/dist/sandbox-snapshot-policy.js +94 -0
  254. package/dist/sandbox-snapshot-policy.js.map +1 -0
  255. package/dist/sandbox-sync-worker.d.ts +63 -0
  256. package/dist/sandbox-sync-worker.d.ts.map +1 -0
  257. package/dist/sandbox-sync-worker.js +448 -0
  258. package/dist/sandbox-sync-worker.js.map +1 -0
  259. package/dist/schedule-execution.d.ts +59 -0
  260. package/dist/schedule-execution.d.ts.map +1 -0
  261. package/dist/schedule-execution.js +951 -0
  262. package/dist/schedule-execution.js.map +1 -0
  263. package/dist/scheduler.d.ts +19 -0
  264. package/dist/scheduler.d.ts.map +1 -0
  265. package/dist/scheduler.js +38 -0
  266. package/dist/scheduler.js.map +1 -0
  267. package/dist/security.d.ts +20 -0
  268. package/dist/security.d.ts.map +1 -0
  269. package/dist/security.js +145 -0
  270. package/dist/security.js.map +1 -0
  271. package/dist/self-improvement.d.ts +245 -0
  272. package/dist/self-improvement.d.ts.map +1 -0
  273. package/dist/self-improvement.js +554 -0
  274. package/dist/self-improvement.js.map +1 -0
  275. package/dist/source-loader.d.ts +19 -0
  276. package/dist/source-loader.d.ts.map +1 -0
  277. package/dist/source-loader.js +160 -0
  278. package/dist/source-loader.js.map +1 -0
  279. package/dist/state-store-helpers.d.ts +28 -0
  280. package/dist/state-store-helpers.d.ts.map +1 -0
  281. package/dist/state-store-helpers.js +164 -0
  282. package/dist/state-store-helpers.js.map +1 -0
  283. package/dist/state-stores.d.ts +25 -0
  284. package/dist/state-stores.d.ts.map +1 -0
  285. package/dist/state-stores.js +157 -0
  286. package/dist/state-stores.js.map +1 -0
  287. package/dist/state.d.ts +736 -0
  288. package/dist/state.d.ts.map +1 -0
  289. package/dist/state.js +46 -0
  290. package/dist/state.js.map +1 -0
  291. package/dist/storage.d.ts +188 -0
  292. package/dist/storage.d.ts.map +1 -0
  293. package/dist/storage.js +752 -0
  294. package/dist/storage.js.map +1 -0
  295. package/dist/subagent-execution.d.ts +80 -0
  296. package/dist/subagent-execution.d.ts.map +1 -0
  297. package/dist/subagent-execution.js +135 -0
  298. package/dist/subagent-execution.js.map +1 -0
  299. package/dist/telemetry.d.ts +49 -0
  300. package/dist/telemetry.d.ts.map +1 -0
  301. package/dist/telemetry.js +60 -0
  302. package/dist/telemetry.js.map +1 -0
  303. package/dist/tool-apis.d.ts +22 -0
  304. package/dist/tool-apis.d.ts.map +1 -0
  305. package/dist/tool-apis.js +344 -0
  306. package/dist/tool-apis.js.map +1 -0
  307. package/dist/tool-execution.d.ts +143 -0
  308. package/dist/tool-execution.d.ts.map +1 -0
  309. package/dist/tool-execution.js +573 -0
  310. package/dist/tool-execution.js.map +1 -0
  311. package/dist/trusted-attachments.d.ts +20 -0
  312. package/dist/trusted-attachments.d.ts.map +1 -0
  313. package/dist/trusted-attachments.js +276 -0
  314. package/dist/trusted-attachments.js.map +1 -0
  315. package/dist/usage-accounting.d.ts +37 -0
  316. package/dist/usage-accounting.d.ts.map +1 -0
  317. package/dist/usage-accounting.js +181 -0
  318. package/dist/usage-accounting.js.map +1 -0
  319. package/dist/usage-budget.d.ts +29 -0
  320. package/dist/usage-budget.d.ts.map +1 -0
  321. package/dist/usage-budget.js +145 -0
  322. package/dist/usage-budget.js.map +1 -0
  323. package/dist/usage-reconciliation.d.ts +76 -0
  324. package/dist/usage-reconciliation.d.ts.map +1 -0
  325. package/dist/usage-reconciliation.js +463 -0
  326. package/dist/usage-reconciliation.js.map +1 -0
  327. package/dist/usage-records.d.ts +24 -0
  328. package/dist/usage-records.d.ts.map +1 -0
  329. package/dist/usage-records.js +445 -0
  330. package/dist/usage-records.js.map +1 -0
  331. package/package.json +51 -0
@@ -0,0 +1,31 @@
1
+ import type { JsonObject } from "@assemblyline-agents/core";
2
+ import type { ResolvedStateAdapter, RunControlIntent, RunRecord } from "./state.js";
3
+ export interface OperatorControlsHost {
4
+ state: ResolvedStateAdapter;
5
+ boot(): void;
6
+ durableAgentScope(): string;
7
+ cacheRunControl(runId: string, intent?: RunControlIntent): void;
8
+ }
9
+ export declare function getAgentControlImpl(host: OperatorControlsHost): Promise<{
10
+ enabled: boolean;
11
+ maintenance: boolean;
12
+ acceptingWork: boolean;
13
+ inFlightRuns: number;
14
+ scope: string;
15
+ updatedAt?: string;
16
+ }>;
17
+ export declare function setAgentMaintenanceImpl(host: OperatorControlsHost, maintenance: boolean, metadata?: JsonObject): Promise<{
18
+ enabled: boolean;
19
+ maintenance: boolean;
20
+ acceptingWork: boolean;
21
+ inFlightRuns: number;
22
+ scope: string;
23
+ updatedAt?: string;
24
+ }>;
25
+ export declare function setAgentEnabledImpl(host: OperatorControlsHost, enabled: boolean, metadata?: JsonObject): Promise<{
26
+ enabled: boolean;
27
+ scope: string;
28
+ updatedAt: string;
29
+ }>;
30
+ export declare function requestRunControlImpl(host: OperatorControlsHost, runId: string, intent: RunControlIntent): Promise<RunRecord>;
31
+ //# sourceMappingURL=operator-controls.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"operator-controls.d.ts","sourceRoot":"","sources":["../src/operator-controls.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,KAAK,EAAE,oBAAoB,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEpF,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,oBAAoB,CAAC;IAC5B,IAAI,IAAI,IAAI,CAAC;IACb,iBAAiB,IAAI,MAAM,CAAC;IAC5B,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,gBAAgB,GAAG,IAAI,CAAC;CACjE;AAED,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,oBAAoB,GACzB,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,WAAW,EAAE,OAAO,CAAC;IAAC,aAAa,EAAE,OAAO,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAoBtI;AAED,wBAAsB,uBAAuB,CAC3C,IAAI,EAAE,oBAAoB,EAC1B,WAAW,EAAE,OAAO,EACpB,QAAQ,GAAE,UAAe,GACxB,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,WAAW,EAAE,OAAO,CAAC;IAAC,aAAa,EAAE,OAAO,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAMtI;AAED,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,oBAAoB,EAC1B,OAAO,EAAE,OAAO,EAChB,QAAQ,GAAE,UAAe,GACxB,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,CAAC,CAMjE;AAED,wBAAsB,qBAAqB,CACzC,IAAI,EAAE,oBAAoB,EAC1B,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,gBAAgB,GACvB,OAAO,CAAC,SAAS,CAAC,CAgCpB"}
@@ -0,0 +1,71 @@
1
+ export async function getAgentControlImpl(host) {
2
+ host.boot();
3
+ const scope = host.durableAgentScope();
4
+ const setting = await host.state.getSetting(scope, "enabled");
5
+ const maintenanceSetting = await host.state.getSetting(scope, "maintenance");
6
+ const enabled = setting?.value !== false;
7
+ const maintenance = maintenanceSetting?.value === true;
8
+ const running = host.state.listRunsByStatus
9
+ ? await host.state.listRunsByStatus(["created", "running"])
10
+ : (await host.state.listRuns()).filter((run) => run.status === "created" || run.status === "running");
11
+ const result = {
12
+ enabled,
13
+ maintenance,
14
+ acceptingWork: enabled && !maintenance,
15
+ inFlightRuns: running.length,
16
+ scope
17
+ };
18
+ const updatedAt = [setting?.updatedAt, maintenanceSetting?.updatedAt].filter((value) => Boolean(value)).sort().at(-1);
19
+ if (updatedAt)
20
+ result.updatedAt = updatedAt;
21
+ return result;
22
+ }
23
+ export async function setAgentMaintenanceImpl(host, maintenance, metadata = {}) {
24
+ host.boot();
25
+ const scope = host.durableAgentScope();
26
+ await host.state.setSetting(scope, "maintenance", maintenance);
27
+ await host.state.appendRuntimeControlEvent(scope, maintenance ? "agent.quiesced" : "agent.resumed", metadata);
28
+ return getAgentControlImpl(host);
29
+ }
30
+ export async function setAgentEnabledImpl(host, enabled, metadata = {}) {
31
+ host.boot();
32
+ const scope = host.durableAgentScope();
33
+ const setting = await host.state.setSetting(scope, "enabled", enabled);
34
+ await host.state.appendRuntimeControlEvent(scope, enabled ? "agent.enabled" : "agent.disabled", metadata);
35
+ return { enabled, scope, updatedAt: setting.updatedAt };
36
+ }
37
+ export async function requestRunControlImpl(host, runId, intent) {
38
+ host.boot();
39
+ const before = await host.state.getRun(runId);
40
+ if (!before)
41
+ throw new Error(`Run not found: ${runId}`);
42
+ if (intent === "suspend" && before.status !== "running") {
43
+ throw new Error(`Run ${runId} is not running.`);
44
+ }
45
+ if (intent === "cancel" && ["completed", "failed", "cancelled"].includes(before.status)) {
46
+ throw new Error(`Run ${runId} is already terminal.`);
47
+ }
48
+ if (!host.state.requestRunControl) {
49
+ throw new Error("The configured state adapter does not support atomic run controls.");
50
+ }
51
+ const controlled = await host.state.requestRunControl(runId, intent);
52
+ await host.state.appendEvent(runId, intent === "cancel" ? "run.cancel_requested" : "run.suspend_requested", {
53
+ previousStatus: before.status,
54
+ cooperative: controlled.status === "running"
55
+ });
56
+ if (controlled.status === "running" && controlled.controlIntent) {
57
+ host.cacheRunControl(runId, controlled.controlIntent);
58
+ return controlled;
59
+ }
60
+ host.cacheRunControl(runId);
61
+ if (controlled.status === "cancelled") {
62
+ for (const toolCall of await host.state.listToolCalls(runId)) {
63
+ if (["requested", "waiting_for_approval", "running"].includes(toolCall.status)) {
64
+ await host.state.updateToolCall(toolCall.id, { status: "cancelled" });
65
+ }
66
+ }
67
+ await host.state.appendEvent(runId, "run.cancelled", { cooperative: false, previousStatus: before.status });
68
+ }
69
+ return controlled;
70
+ }
71
+ //# sourceMappingURL=operator-controls.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"operator-controls.js","sourceRoot":"","sources":["../src/operator-controls.ts"],"names":[],"mappings":"AAUA,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,IAA0B;IAE1B,IAAI,CAAC,IAAI,EAAE,CAAC;IACZ,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;IACvC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAC9D,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IAC7E,MAAM,OAAO,GAAG,OAAO,EAAE,KAAK,KAAK,KAAK,CAAC;IACzC,MAAM,WAAW,GAAG,kBAAkB,EAAE,KAAK,KAAK,IAAI,CAAC;IACvD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAgB;QACzC,CAAC,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,KAAK,SAAS,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;IACxG,MAAM,MAAM,GAAgI;QAC1I,OAAO;QACP,WAAW;QACX,aAAa,EAAE,OAAO,IAAI,CAAC,WAAW;QACtC,YAAY,EAAE,OAAO,CAAC,MAAM;QAC5B,KAAK;KACN,CAAC;IACF,MAAM,SAAS,GAAG,CAAC,OAAO,EAAE,SAAS,EAAE,kBAAkB,EAAE,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACvI,IAAI,SAAS;QAAE,MAAM,CAAC,SAAS,GAAG,SAAS,CAAC;IAC5C,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,IAA0B,EAC1B,WAAoB,EACpB,WAAuB,EAAE;IAEzB,IAAI,CAAC,IAAI,EAAE,CAAC;IACZ,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;IACvC,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,aAAa,EAAE,WAAW,CAAC,CAAC;IAC/D,MAAM,IAAI,CAAC,KAAK,CAAC,yBAAyB,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;IAC9G,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC;AACnC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,IAA0B,EAC1B,OAAgB,EAChB,WAAuB,EAAE;IAEzB,IAAI,CAAC,IAAI,EAAE,CAAC;IACZ,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;IACvC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IACvE,MAAM,IAAI,CAAC,KAAK,CAAC,yBAAyB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAC;IAC1G,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;AAC1D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,IAA0B,EAC1B,KAAa,EACb,MAAwB;IAExB,IAAI,CAAC,IAAI,EAAE,CAAC;IACZ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9C,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;IACxD,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QACxD,MAAM,IAAI,KAAK,CAAC,OAAO,KAAK,kBAAkB,CAAC,CAAC;IAClD,CAAC;IACD,IAAI,MAAM,KAAK,QAAQ,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QACxF,MAAM,IAAI,KAAK,CAAC,OAAO,KAAK,uBAAuB,CAAC,CAAC;IACvD,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,oEAAoE,CAAC,CAAC;IACxF,CAAC;IACD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACrE,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,uBAAuB,EAAE;QAC1G,cAAc,EAAE,MAAM,CAAC,MAAM;QAC7B,WAAW,EAAE,UAAU,CAAC,MAAM,KAAK,SAAS;KAC7C,CAAC,CAAC;IACH,IAAI,UAAU,CAAC,MAAM,KAAK,SAAS,IAAI,UAAU,CAAC,aAAa,EAAE,CAAC;QAChE,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC;QACtD,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC5B,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;QACtC,KAAK,MAAM,QAAQ,IAAI,MAAM,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7D,IAAI,CAAC,WAAW,EAAE,sBAAsB,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/E,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;YACxE,CAAC;QACH,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,eAAe,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9G,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC"}
@@ -0,0 +1,40 @@
1
+ import type { AgentMessage } from "@earendil-works/pi-agent-core";
2
+ import type { JsonObject, ReasoningLevel } from "@openeve/core";
3
+ import type { AgentHarness, HarnessContinuation } from "./agent-harness.js";
4
+ import type { ConnectionToolMetadata } from "./connections.js";
5
+ export declare const PI_HARNESS_ID = "pi";
6
+ export declare const PI_CONTINUATION_VERSION = 1;
7
+ /**
8
+ * @deprecated Legacy flat continuation shape persisted under the
9
+ * `pi.continuation` checkpoint name. New code persists the engine-neutral
10
+ * `harness.continuation` envelope; this type remains exported for one release
11
+ * for direct importers.
12
+ */
13
+ export interface PiHarnessContinuation {
14
+ modelSpec: string;
15
+ reasoning?: ReasoningLevel | undefined;
16
+ systemPrompt: string;
17
+ sessionId: string;
18
+ channel: string;
19
+ conversationId?: string;
20
+ activeSkills: string[];
21
+ defaultToolNames: string[];
22
+ discoveredConnectionTools: ConnectionToolMetadata[];
23
+ messages: AgentMessage[];
24
+ pendingToolCallId?: string;
25
+ pendingToolName?: string;
26
+ }
27
+ /**
28
+ * In-tree default {@link AgentHarness} wrapping the Pi agent loop. All
29
+ * Pi-specific behavior lives here: AgentTool adaptation, the terminate
30
+ * stop-signal on pause, usage mapping, `mode: "continue"` resume, and
31
+ * continuation transcript splicing. `@earendil-works` code is loaded lazily.
32
+ */
33
+ export declare function piAgentHarness(options?: {
34
+ models?: unknown;
35
+ }): AgentHarness;
36
+ /** Up-convert a legacy flat `pi.continuation` payload into the opaque envelope form. */
37
+ export declare function legacyPiContinuation(data: JsonObject): HarnessContinuation;
38
+ /** Bound a tool output for model consumption (large payloads become previews). */
39
+ export declare function boundModelOutput(value: unknown): unknown;
40
+ //# sourceMappingURL=pi-agent-harness.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"pi-agent-harness.d.ts","sourceRoot":"","sources":["../src/pi-agent-harness.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAA0C,MAAM,+BAA+B,CAAC;AAE1G,OAAO,KAAK,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAChE,OAAO,KAAK,EACV,YAAY,EAIZ,mBAAmB,EAEpB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAM/D,eAAO,MAAM,aAAa,OAAO,CAAC;AAClC,eAAO,MAAM,uBAAuB,IAAI,CAAC;AAEzC;;;;;GAKG;AACH,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACvC,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,yBAAyB,EAAE,sBAAsB,EAAE,CAAC;IACpD,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED;;;;;GAKG;AACH,wBAAgB,cAAc,CAAC,OAAO,GAAE;IAAE,MAAM,CAAC,EAAE,OAAO,CAAA;CAAO,GAAG,YAAY,CAkF/E;AAED,wFAAwF;AACxF,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,UAAU,GAAG,mBAAmB,CAI1E;AAwDD,kFAAkF;AAClF,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAOxD"}
@@ -0,0 +1,222 @@
1
+ export const PI_HARNESS_ID = "pi";
2
+ export const PI_CONTINUATION_VERSION = 1;
3
+ /**
4
+ * In-tree default {@link AgentHarness} wrapping the Pi agent loop. All
5
+ * Pi-specific behavior lives here: AgentTool adaptation, the terminate
6
+ * stop-signal on pause, usage mapping, `mode: "continue"` resume, and
7
+ * continuation transcript splicing. `@earendil-works` code is loaded lazily.
8
+ */
9
+ export function piAgentHarness(options = {}) {
10
+ let defaultModels;
11
+ const resolveModels = async (env) => {
12
+ if (options.models)
13
+ return options.models;
14
+ if (!defaultModels) {
15
+ const { createDefaultPiModels } = await import("./pi-harness.js");
16
+ defaultModels = await createDefaultPiModels(env);
17
+ }
18
+ return defaultModels;
19
+ };
20
+ return {
21
+ id: PI_HARNESS_ID,
22
+ continuationVersion: PI_CONTINUATION_VERSION,
23
+ async runTurn(input, ctx) {
24
+ const { runPiAgentLoop, piUsageToRuntime } = await import("./pi-harness.js");
25
+ const models = await resolveModels(ctx.env);
26
+ const trackedMessages = [];
27
+ let pendingToolCallId;
28
+ let paused = false;
29
+ const buildTools = async () => (await ctx.tools()).map((tool) => piAgentTool(tool, ctx, (toolCallId) => {
30
+ paused = true;
31
+ pendingToolCallId = toolCallId;
32
+ }));
33
+ const loopOptions = {
34
+ models,
35
+ modelSpec: ctx.modelSpec,
36
+ systemPrompt: ctx.systemPrompt,
37
+ recentHistory: [],
38
+ tools: buildTools,
39
+ env: ctx.env,
40
+ sessionId: ctx.sessionId,
41
+ onRequestStarted: async ({ context }, state) => {
42
+ await ctx.emit({
43
+ type: "request_started",
44
+ iteration: state.requestIteration,
45
+ toolNames: context.tools?.map((tool) => tool.name) ?? []
46
+ });
47
+ },
48
+ onEvent: async (event, state) => {
49
+ if (event.type === "message_end") {
50
+ trackedMessages.push(event.message);
51
+ return;
52
+ }
53
+ if (event.type !== "turn_end" || event.message.role !== "assistant")
54
+ return;
55
+ await ctx.emit(responseCompletedEvent(event.message, state.requestIteration, piUsageToRuntime));
56
+ if (ctx.onIterationCheckpoint) {
57
+ await ctx.onIterationCheckpoint(piContinuation(trackedMessages, pendingToolCallId));
58
+ }
59
+ }
60
+ };
61
+ if (ctx.reasoning)
62
+ loopOptions.reasoning = ctx.reasoning;
63
+ if (input.kind === "resume") {
64
+ const state = piContinuationState(input.continuation.state);
65
+ let messages = state.messages;
66
+ if (input.toolResult) {
67
+ messages = replacePiToolResult(messages, state.pendingToolCallId, input.toolResult.toolName, input.toolResult.output);
68
+ }
69
+ trackedMessages.push(...messages);
70
+ loopOptions.messages = messages;
71
+ loopOptions.mode = "continue";
72
+ }
73
+ else {
74
+ const initialMessages = input.recentHistory.map((content) => ({
75
+ role: "user",
76
+ content: [{ type: "text", text: content }],
77
+ timestamp: Date.now()
78
+ }));
79
+ trackedMessages.push(...initialMessages);
80
+ loopOptions.messages = initialMessages;
81
+ loopOptions.message = input.message;
82
+ }
83
+ const loop = await runPiAgentLoop(loopOptions);
84
+ const result = { response: loop.response };
85
+ if (paused) {
86
+ result.paused = true;
87
+ result.continuation = piContinuation(loop.messages, pendingToolCallId);
88
+ }
89
+ return result;
90
+ }
91
+ };
92
+ }
93
+ /** Up-convert a legacy flat `pi.continuation` payload into the opaque envelope form. */
94
+ export function legacyPiContinuation(data) {
95
+ const state = { messages: Array.isArray(data.messages) ? data.messages : [] };
96
+ if (typeof data.pendingToolCallId === "string")
97
+ state.pendingToolCallId = data.pendingToolCallId;
98
+ return { harness: PI_HARNESS_ID, version: PI_CONTINUATION_VERSION, state };
99
+ }
100
+ function piContinuation(messages, pendingToolCallId) {
101
+ const state = jsonClone({ messages });
102
+ if (pendingToolCallId)
103
+ state.pendingToolCallId = pendingToolCallId;
104
+ return { harness: PI_HARNESS_ID, version: PI_CONTINUATION_VERSION, state };
105
+ }
106
+ function piContinuationState(state) {
107
+ const messages = Array.isArray(state.messages) ? state.messages : [];
108
+ const out = { messages };
109
+ if (typeof state.pendingToolCallId === "string")
110
+ out.pendingToolCallId = state.pendingToolCallId;
111
+ return out;
112
+ }
113
+ function piAgentTool(tool, ctx, onPause) {
114
+ return {
115
+ name: tool.name,
116
+ label: tool.name,
117
+ description: tool.description,
118
+ parameters: tool.inputSchema,
119
+ executionMode: "sequential",
120
+ execute: async (toolCallId, params) => {
121
+ const outcome = await ctx.executeTool({ toolCallId, name: tool.name, input: params });
122
+ if (outcome.kind === "pause") {
123
+ onPause(outcome.toolCallId);
124
+ return piToolResult(outcome.output, true);
125
+ }
126
+ return piToolResult(outcome.output);
127
+ }
128
+ };
129
+ }
130
+ function responseCompletedEvent(message, iteration, usageToRuntime) {
131
+ const usage = usageToRuntime(message.usage);
132
+ const event = {
133
+ type: "response_completed",
134
+ iteration,
135
+ toolCallCount: message.content.filter((block) => block.type === "toolCall").length,
136
+ usage
137
+ };
138
+ if (message.stopReason !== undefined)
139
+ event.finishReason = message.stopReason;
140
+ if (message.provider !== undefined)
141
+ event.provider = message.provider;
142
+ const responseModel = message.responseModel ?? message.model;
143
+ if (responseModel !== undefined)
144
+ event.responseModel = responseModel;
145
+ return event;
146
+ }
147
+ /** Bound a tool output for model consumption (large payloads become previews). */
148
+ export function boundModelOutput(value) {
149
+ const json = JSON.stringify(safeJsonValue(value));
150
+ if (json.length <= 8_000)
151
+ return safeJsonValue(value);
152
+ return {
153
+ truncated: true,
154
+ preview: json.slice(0, 8_000)
155
+ };
156
+ }
157
+ function piToolResult(value, terminate = false) {
158
+ const details = boundModelOutput(value);
159
+ const text = typeof details === "string" ? details : JSON.stringify(details);
160
+ const result = {
161
+ content: [{ type: "text", text }],
162
+ details
163
+ };
164
+ if (terminate)
165
+ result.terminate = true;
166
+ return result;
167
+ }
168
+ function replacePiToolResult(messages, toolCallId, toolName, value) {
169
+ const details = boundModelOutput(value);
170
+ const text = typeof details === "string" ? details : JSON.stringify(details);
171
+ const next = messages.map((message) => ({ ...message }));
172
+ const targetIndex = findToolResultIndex(next, toolCallId);
173
+ const resolvedToolCallId = toolCallId ?? inferLastAssistantToolCallId(next, toolName) ?? `connection:${toolName}`;
174
+ const replacement = {
175
+ role: "toolResult",
176
+ toolCallId: resolvedToolCallId,
177
+ toolName,
178
+ content: [{ type: "text", text }],
179
+ details,
180
+ isError: false,
181
+ timestamp: Date.now()
182
+ };
183
+ if (targetIndex >= 0) {
184
+ next[targetIndex] = replacement;
185
+ return next.slice(0, targetIndex + 1);
186
+ }
187
+ next.push(replacement);
188
+ return next;
189
+ }
190
+ function findToolResultIndex(messages, toolCallId) {
191
+ for (let index = messages.length - 1; index >= 0; index -= 1) {
192
+ const message = messages[index];
193
+ if (!isToolResultMessage(message))
194
+ continue;
195
+ if (!toolCallId || message.toolCallId === toolCallId)
196
+ return index;
197
+ }
198
+ return -1;
199
+ }
200
+ function inferLastAssistantToolCallId(messages, toolName) {
201
+ for (let index = messages.length - 1; index >= 0; index -= 1) {
202
+ const message = messages[index];
203
+ if (!message || message.role !== "assistant")
204
+ continue;
205
+ for (let blockIndex = message.content.length - 1; blockIndex >= 0; blockIndex -= 1) {
206
+ const block = message.content[blockIndex];
207
+ if (block && block.type === "toolCall" && block.name === toolName)
208
+ return block.id;
209
+ }
210
+ }
211
+ return undefined;
212
+ }
213
+ function isToolResultMessage(message) {
214
+ return Boolean(message && typeof message === "object" && "role" in message && message.role === "toolResult");
215
+ }
216
+ function jsonClone(value) {
217
+ return JSON.parse(JSON.stringify(value ?? null));
218
+ }
219
+ function safeJsonValue(value) {
220
+ return JSON.parse(JSON.stringify(value ?? null));
221
+ }
222
+ //# sourceMappingURL=pi-agent-harness.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"pi-agent-harness.js","sourceRoot":"","sources":["../src/pi-agent-harness.ts"],"names":[],"mappings":"AAiBA,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,CAAC;AAClC,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC;AAuBzC;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAAC,UAAgC,EAAE;IAC/D,IAAI,aAAiC,CAAC;IACtC,MAAM,aAAa,GAAG,KAAK,EAAE,GAAuC,EAAmB,EAAE;QACvF,IAAI,OAAO,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,MAAgB,CAAC;QACpD,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,EAAE,qBAAqB,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;YAClE,aAAa,GAAG,MAAM,qBAAqB,CAAC,GAAG,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,aAAa,CAAC;IACvB,CAAC,CAAC;IACF,OAAO;QACL,EAAE,EAAE,aAAa;QACjB,mBAAmB,EAAE,uBAAuB;QAC5C,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG;YACtB,MAAM,EAAE,cAAc,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;YAC7E,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,eAAe,GAAmB,EAAE,CAAC;YAC3C,IAAI,iBAAqC,CAAC;YAC1C,IAAI,MAAM,GAAG,KAAK,CAAC;YACnB,MAAM,UAAU,GAAG,KAAK,IAA+B,EAAE,CACvD,CAAC,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC/B,WAAW,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,EAAE;gBACpC,MAAM,GAAG,IAAI,CAAC;gBACd,iBAAiB,GAAG,UAAU,CAAC;YACjC,CAAC,CAAC,CACH,CAAC;YACJ,MAAM,WAAW,GAA0B;gBACzC,MAAM;gBACN,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,aAAa,EAAE,EAAE;gBACjB,KAAK,EAAE,UAAU;gBACjB,GAAG,EAAE,GAAG,CAAC,GAAG;gBACZ,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,gBAAgB,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE;oBAC7C,MAAM,GAAG,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,iBAAiB;wBACvB,SAAS,EAAE,KAAK,CAAC,gBAAgB;wBACjC,SAAS,EAAE,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;qBACzD,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,EAAE,KAAK,EAAE,KAAiB,EAAE,KAAK,EAAE,EAAE;oBAC1C,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;wBACjC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBACpC,OAAO;oBACT,CAAC;oBACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW;wBAAE,OAAO;oBAC5E,MAAM,GAAG,CAAC,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,CAAC,CAAC;oBAChG,IAAI,GAAG,CAAC,qBAAqB,EAAE,CAAC;wBAC9B,MAAM,GAAG,CAAC,qBAAqB,CAAC,cAAc,CAAC,eAAe,EAAE,iBAAiB,CAAC,CAAC,CAAC;oBACtF,CAAC;gBACH,CAAC;aACF,CAAC;YACF,IAAI,GAAG,CAAC,SAAS;gBAAE,WAAW,CAAC,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC;YACzD,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC5B,MAAM,KAAK,GAAG,mBAAmB,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBAC5D,IAAI,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;gBAC9B,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,QAAQ,GAAG,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBACxH,CAAC;gBACD,eAAe,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;gBAClC,WAAW,CAAC,QAAQ,GAAG,QAAQ,CAAC;gBAChC,WAAW,CAAC,IAAI,GAAG,UAAU,CAAC;YAChC,CAAC;iBAAM,CAAC;gBACN,MAAM,eAAe,GAAG,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAgB,EAAE,CAAC,CAAC;oBAC1E,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;oBAC1C,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC,CAAC;gBACJ,eAAe,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;gBACzC,WAAW,CAAC,QAAQ,GAAG,eAAe,CAAC;gBACvC,WAAW,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;YACtC,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,WAAW,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAoB,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC5D,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC;gBACrB,MAAM,CAAC,YAAY,GAAG,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YACzE,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,wFAAwF;AACxF,MAAM,UAAU,oBAAoB,CAAC,IAAgB;IACnD,MAAM,KAAK,GAAe,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IAC1F,IAAI,OAAO,IAAI,CAAC,iBAAiB,KAAK,QAAQ;QAAE,KAAK,CAAC,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,CAAC;IACjG,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,uBAAuB,EAAE,KAAK,EAAE,CAAC;AAC7E,CAAC;AAED,SAAS,cAAc,CAAC,QAAwB,EAAE,iBAAqC;IACrF,MAAM,KAAK,GAAG,SAAS,CAAC,EAAE,QAAQ,EAAa,CAAe,CAAC;IAC/D,IAAI,iBAAiB;QAAE,KAAK,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IACnE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,uBAAuB,EAAE,KAAK,EAAE,CAAC;AAC7E,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAiB;IAC5C,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAE,KAAK,CAAC,QAAsC,CAAC,CAAC,CAAC,EAAE,CAAC;IACpG,MAAM,GAAG,GAA6D,EAAE,QAAQ,EAAE,CAAC;IACnF,IAAI,OAAO,KAAK,CAAC,iBAAiB,KAAK,QAAQ;QAAE,GAAG,CAAC,iBAAiB,GAAG,KAAK,CAAC,iBAAiB,CAAC;IACjG,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,WAAW,CAClB,IAA2B,EAC3B,GAAwB,EACxB,OAAiD;IAEjD,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,KAAK,EAAE,IAAI,CAAC,IAAI;QAChB,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,UAAU,EAAE,IAAI,CAAC,WAAoB;QACrC,aAAa,EAAE,YAAY;QAC3B,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAqC,EAAE;YACvE,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,WAAW,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YACtF,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAC7B,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBAC5B,OAAO,YAAY,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAC5C,CAAC;YACD,OAAO,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAC7B,OAA2B,EAC3B,SAAiB,EACjB,cAAoE;IAEpE,MAAM,KAAK,GAAG,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAsB;QAC/B,IAAI,EAAE,oBAAoB;QAC1B,SAAS;QACT,aAAa,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,MAAM;QAClF,KAAK;KACN,CAAC;IACF,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS;QAAE,KAAK,CAAC,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC;IAC9E,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS;QAAE,KAAK,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IACtE,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,KAAK,CAAC;IAC7D,IAAI,aAAa,KAAK,SAAS;QAAE,KAAK,CAAC,aAAa,GAAG,aAAa,CAAC;IACrE,OAAO,KAAK,CAAC;AACf,CAAC;AAED,kFAAkF;AAClF,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IAClD,IAAI,IAAI,CAAC,MAAM,IAAI,KAAK;QAAE,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC;IACtD,OAAO;QACL,SAAS,EAAE,IAAI;QACf,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;KAC9B,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAc,EAAE,SAAS,GAAG,KAAK;IACrD,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACxC,MAAM,IAAI,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC7E,MAAM,MAAM,GAA6B;QACvC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACjC,OAAO;KACR,CAAC;IACF,IAAI,SAAS;QAAE,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;IACvC,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,mBAAmB,CAC1B,QAAwB,EACxB,UAA8B,EAC9B,QAAgB,EAChB,KAAc;IAEd,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACxC,MAAM,IAAI,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC7E,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAmB,CAAC;IAC3E,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAC1D,MAAM,kBAAkB,GAAG,UAAU,IAAI,4BAA4B,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,cAAc,QAAQ,EAAE,CAAC;IAClH,MAAM,WAAW,GAAiB;QAChC,IAAI,EAAE,YAAY;QAClB,UAAU,EAAE,kBAAkB;QAC9B,QAAQ;QACR,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACjC,OAAO;QACP,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACtB,CAAC;IACF,IAAI,WAAW,IAAI,CAAC,EAAE,CAAC;QACrB,IAAI,CAAC,WAAW,CAAC,GAAG,WAAW,CAAC;QAChC,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,GAAG,CAAC,CAAC,CAAC;IACxC,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACvB,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAwB,EAAE,UAA8B;IACnF,KAAK,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC7D,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC;YAAE,SAAS;QAC5C,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;YAAE,OAAO,KAAK,CAAC;IACrE,CAAC;IACD,OAAO,CAAC,CAAC,CAAC;AACZ,CAAC;AAED,SAAS,4BAA4B,CAAC,QAAwB,EAAE,QAAgB;IAC9E,KAAK,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC7D,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;YAAE,SAAS;QACvD,KAAK,IAAI,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,UAAU,IAAI,CAAC,EAAE,UAAU,IAAI,CAAC,EAAE,CAAC;YACnF,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC1C,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO,KAAK,CAAC,EAAE,CAAC;QACrF,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAiC;IAC5D,OAAO,OAAO,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,MAAM,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;AAC/G,CAAC;AAED,SAAS,SAAS,CAAI,KAAQ;IAC5B,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAM,CAAC;AACxD,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC;AACnD,CAAC"}
@@ -0,0 +1,42 @@
1
+ import { type AgentEvent, type AgentMessage, type AgentTool, type ThinkingLevel } from "@earendil-works/pi-agent-core";
2
+ import { type Api, type AssistantMessage, type Context, type Model, type Models, type SimpleStreamOptions, type Usage } from "@earendil-works/pi-ai";
3
+ export interface RunPiAgentLoopOptions {
4
+ models: Models;
5
+ modelSpec: string;
6
+ reasoning?: ThinkingLevel | undefined;
7
+ systemPrompt: string;
8
+ message?: string;
9
+ recentHistory: string[];
10
+ messages?: AgentMessage[];
11
+ mode?: "prompt" | "continue";
12
+ tools: () => Promise<AgentTool<any>[]>;
13
+ env?: Record<string, string | undefined>;
14
+ sessionId?: string;
15
+ onEvent?: (event: AgentEvent, state: {
16
+ requestIteration: number;
17
+ }) => Promise<void> | void;
18
+ onRequestStarted?: (input: {
19
+ model: Model<Api>;
20
+ context: Context;
21
+ options?: SimpleStreamOptions;
22
+ }, state: {
23
+ requestIteration: number;
24
+ }) => Promise<void> | void;
25
+ }
26
+ export interface RunPiAgentLoopResult {
27
+ response: string;
28
+ messages: AgentMessage[];
29
+ finalAssistant?: AssistantMessage;
30
+ }
31
+ export declare function createDefaultPiModels(env?: Record<string, string | undefined>): Promise<Models>;
32
+ export declare function listPiModelSpecs(models?: Models): Promise<string[]>;
33
+ export declare function resolvePiModel(models: Models, modelSpec: string): Model<Api>;
34
+ export declare function runPiAgentLoop(options: RunPiAgentLoopOptions): Promise<RunPiAgentLoopResult>;
35
+ export declare function piUsageToRuntime(usage: Usage | undefined): {
36
+ inputTokens?: number;
37
+ outputTokens?: number;
38
+ cachedTokens?: number;
39
+ costUsd?: number;
40
+ };
41
+ export declare function assistantText(message: AssistantMessage): string;
42
+ //# sourceMappingURL=pi-harness.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"pi-harness.d.ts","sourceRoot":"","sources":["../src/pi-harness.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,KAAK,UAAU,EAAE,KAAK,YAAY,EAAE,KAAK,SAAS,EAAE,KAAK,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAC9H,OAAO,EAEL,KAAK,GAAG,EACR,KAAK,gBAAgB,EAErB,KAAK,OAAO,EACZ,KAAK,KAAK,EACV,KAAK,MAAM,EACX,KAAK,mBAAmB,EACxB,KAAK,KAAK,EACX,MAAM,uBAAuB,CAAC;AAE/B,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;IACtC,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,QAAQ,CAAC,EAAE,YAAY,EAAE,CAAC;IAC1B,IAAI,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;IAC7B,KAAK,EAAE,MAAM,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACvC,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IACzC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE;QAAE,gBAAgB,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC3F,gBAAgB,CAAC,EAAE,CACjB,KAAK,EAAE;QAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QAAC,OAAO,EAAE,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,mBAAmB,CAAA;KAAE,EAC7E,KAAK,EAAE;QAAE,gBAAgB,EAAE,MAAM,CAAA;KAAE,KAChC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CAC3B;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,cAAc,CAAC,EAAE,gBAAgB,CAAC;CACnC;AAED,wBAAsB,qBAAqB,CAAC,GAAG,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAe,GAAG,OAAO,CAAC,MAAM,CAAC,CAalH;AAED,wBAAsB,gBAAgB,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAWzE;AAED,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAY5E;AAED,wBAAsB,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAyDlG;AAED,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,KAAK,GAAG,SAAS,GAAG;IAC1D,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAQA;AAED,wBAAgB,aAAa,CAAC,OAAO,EAAE,gBAAgB,GAAG,MAAM,CAK/D"}
@@ -0,0 +1,142 @@
1
+ import { Agent } from "@earendil-works/pi-agent-core";
2
+ import { defaultProviderAuthContext } from "@earendil-works/pi-ai";
3
+ export async function createDefaultPiModels(env = process.env) {
4
+ const { builtinModels } = await import("@earendil-works/pi-ai/providers/all");
5
+ const fallback = defaultProviderAuthContext();
6
+ return builtinModels({
7
+ authContext: {
8
+ env: async (name) => {
9
+ const value = env[name];
10
+ if (typeof value === "string" && value.trim().length > 0)
11
+ return value;
12
+ return fallback.env(name);
13
+ },
14
+ fileExists: (path) => fallback.fileExists(path)
15
+ }
16
+ });
17
+ }
18
+ export async function listPiModelSpecs(models) {
19
+ try {
20
+ const catalog = models ?? await createDefaultPiModels();
21
+ const specs = new Set();
22
+ for (const model of catalog.getModels()) {
23
+ specs.add(`${model.provider}/${model.id}`);
24
+ }
25
+ return [...specs].sort();
26
+ }
27
+ catch {
28
+ return [];
29
+ }
30
+ }
31
+ export function resolvePiModel(models, modelSpec) {
32
+ const slash = modelSpec.indexOf("/");
33
+ if (slash <= 0 || slash === modelSpec.length - 1) {
34
+ throw new Error(`Invalid model specifier "${modelSpec}". Use "provider/model".`);
35
+ }
36
+ const provider = modelSpec.slice(0, slash);
37
+ const modelId = modelSpec.slice(slash + 1);
38
+ const model = models.getModel(provider, modelId);
39
+ if (!model) {
40
+ throw new Error(`Unknown Pi model "${modelSpec}". Register the provider/model with @earendil-works/pi-ai or choose a catalog model.`);
41
+ }
42
+ return model;
43
+ }
44
+ export async function runPiAgentLoop(options) {
45
+ const model = resolvePiModel(options.models, options.modelSpec);
46
+ let requestIteration = 0;
47
+ let agent;
48
+ const initialMessages = options.messages ?? options.recentHistory.map((content) => ({
49
+ role: "user",
50
+ content: [{ type: "text", text: content }],
51
+ timestamp: Date.now()
52
+ }));
53
+ const agentOptions = {
54
+ initialState: {
55
+ systemPrompt: options.systemPrompt,
56
+ model,
57
+ messages: initialMessages,
58
+ tools: await options.tools(),
59
+ thinkingLevel: options.reasoning ?? "medium"
60
+ },
61
+ toolExecution: "sequential",
62
+ streamFn: (requestModel, context, streamOptions) => {
63
+ requestIteration += 1;
64
+ const request = {
65
+ model: requestModel,
66
+ context
67
+ };
68
+ if (streamOptions)
69
+ request.options = streamOptions;
70
+ void options.onRequestStarted?.(request, { requestIteration });
71
+ return options.models.streamSimple(requestModel, context, {
72
+ ...streamOptions,
73
+ env: { ...compactEnv(streamOptions?.env), ...compactEnv(options.env) }
74
+ });
75
+ },
76
+ prepareNextTurn: async () => ({
77
+ context: {
78
+ systemPrompt: options.systemPrompt,
79
+ messages: agent.state.messages.slice(),
80
+ tools: await options.tools()
81
+ }
82
+ })
83
+ };
84
+ if (options.sessionId) {
85
+ agent = new Agent({ ...agentOptions, sessionId: options.sessionId });
86
+ }
87
+ else {
88
+ agent = new Agent(agentOptions);
89
+ }
90
+ agent.subscribe((event) => options.onEvent?.(event, { requestIteration }));
91
+ if (options.mode === "continue") {
92
+ await agent.continue();
93
+ }
94
+ else {
95
+ await agent.prompt(options.message ?? "");
96
+ }
97
+ const finalAssistant = lastAssistantMessage(agent.state.messages);
98
+ const result = {
99
+ response: finalAssistant ? assistantText(finalAssistant) : "",
100
+ messages: agent.state.messages.slice()
101
+ };
102
+ if (finalAssistant)
103
+ result.finalAssistant = finalAssistant;
104
+ return result;
105
+ }
106
+ export function piUsageToRuntime(usage) {
107
+ if (!usage)
108
+ return {};
109
+ const out = {};
110
+ if (usage.input !== undefined)
111
+ out.inputTokens = usage.input;
112
+ if (usage.output !== undefined)
113
+ out.outputTokens = usage.output;
114
+ if (usage.cacheRead !== undefined)
115
+ out.cachedTokens = usage.cacheRead;
116
+ if (usage.cost?.total !== undefined)
117
+ out.costUsd = usage.cost.total;
118
+ return out;
119
+ }
120
+ export function assistantText(message) {
121
+ return message.content
122
+ .filter((block) => block.type === "text")
123
+ .map((block) => block.text)
124
+ .join("");
125
+ }
126
+ function lastAssistantMessage(messages) {
127
+ for (let index = messages.length - 1; index >= 0; index -= 1) {
128
+ const message = messages[index];
129
+ if (message?.role === "assistant")
130
+ return message;
131
+ }
132
+ return undefined;
133
+ }
134
+ function compactEnv(env) {
135
+ const out = {};
136
+ for (const [key, value] of Object.entries(env ?? {})) {
137
+ if (typeof value === "string")
138
+ out[key] = value;
139
+ }
140
+ return out;
141
+ }
142
+ //# sourceMappingURL=pi-harness.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"pi-harness.js","sourceRoot":"","sources":["../src/pi-harness.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAA0E,MAAM,+BAA+B,CAAC;AAC9H,OAAO,EACL,0BAA0B,EAS3B,MAAM,uBAAuB,CAAC;AA2B/B,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,MAA0C,OAAO,CAAC,GAAG;IAC/F,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,qCAAqC,CAAC,CAAC;IAC9E,MAAM,QAAQ,GAAG,0BAA0B,EAAE,CAAC;IAC9C,OAAO,aAAa,CAAC;QACnB,WAAW,EAAE;YACX,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;gBAClB,MAAM,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC;gBACxB,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;oBAAE,OAAO,KAAK,CAAC;gBACvE,OAAO,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;YACD,UAAU,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC;SAChD;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,MAAe;IACpD,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,IAAI,MAAM,qBAAqB,EAAE,CAAC;QACxD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;QAChC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;YACxC,KAAK,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAAc,EAAE,SAAiB;IAC9D,MAAM,KAAK,GAAG,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,KAAK,IAAI,CAAC,IAAI,KAAK,KAAK,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,4BAA4B,SAAS,0BAA0B,CAAC,CAAC;IACnF,CAAC;IACD,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,SAAS,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;IAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACjD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,qBAAqB,SAAS,sFAAsF,CAAC,CAAC;IACxI,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,OAA8B;IACjE,MAAM,KAAK,GAAG,cAAc,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;IAChE,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,IAAI,KAAY,CAAC;IACjB,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAgB,EAAE,CAAC,CAAC;QAChG,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QAC1C,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACtB,CAAC,CAAC,CAAC;IACJ,MAAM,YAAY,GAAG;QACnB,YAAY,EAAE;YACZ,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,KAAK;YACL,QAAQ,EAAE,eAAe;YACzB,KAAK,EAAE,MAAM,OAAO,CAAC,KAAK,EAAE;YAC5B,aAAa,EAAE,OAAO,CAAC,SAAS,IAAI,QAAQ;SAC7C;QACD,aAAa,EAAE,YAAY;QAC3B,QAAQ,EAAE,CAAC,YAAY,EAAE,OAAO,EAAE,aAAa,EAA+B,EAAE;YAC9E,gBAAgB,IAAI,CAAC,CAAC;YACtB,MAAM,OAAO,GAA2E;gBACtF,KAAK,EAAE,YAA0B;gBACjC,OAAO;aACR,CAAC;YACF,IAAI,aAAa;gBAAE,OAAO,CAAC,OAAO,GAAG,aAAa,CAAC;YACnD,KAAK,OAAO,CAAC,gBAAgB,EAAE,CAAC,OAAO,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAC/D,OAAO,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,YAA0B,EAAE,OAAO,EAAE;gBACtE,GAAG,aAAa;gBAChB,GAAG,EAAE,EAAE,GAAG,UAAU,CAAC,aAAa,EAAE,GAAG,CAAC,EAAE,GAAG,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;aACvE,CAAC,CAAC;QACL,CAAC;QACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YAC5B,OAAO,EAAE;gBACP,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE;gBACtC,KAAK,EAAE,MAAM,OAAO,CAAC,KAAK,EAAE;aAC7B;SACF,CAAC;KAC8C,CAAC;IACnD,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QACtB,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,YAAY,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IACvE,CAAC;SAAM,CAAC;QACN,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;IAClC,CAAC;IACD,KAAK,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC;IAC3E,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAChC,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC;SAAM,CAAC;QACN,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC;IACD,MAAM,cAAc,GAAG,oBAAoB,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAClE,MAAM,MAAM,GAAyB;QACnC,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE;QAC7D,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE;KACvC,CAAC;IACF,IAAI,cAAc;QAAE,MAAM,CAAC,cAAc,GAAG,cAAc,CAAC;IAC3D,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAwB;IAMvD,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IACtB,MAAM,GAAG,GAA6F,EAAE,CAAC;IACzG,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;QAAE,GAAG,CAAC,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC;IAC7D,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC;IAChE,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS;QAAE,GAAG,CAAC,YAAY,GAAG,KAAK,CAAC,SAAS,CAAC;IACtE,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,KAAK,SAAS;QAAE,GAAG,CAAC,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;IACpE,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,OAAyB;IACrD,OAAO,OAAO,CAAC,OAAO;SACnB,MAAM,CAAC,CAAC,KAAK,EAA2C,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;SACjF,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;SAC1B,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC;AAED,SAAS,oBAAoB,CAAC,QAAwB;IACpD,KAAK,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC7D,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,OAAO,EAAE,IAAI,KAAK,WAAW;YAAE,OAAO,OAA2B,CAAC;IACxE,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,UAAU,CAAC,GAAmD;IACrE,MAAM,GAAG,GAA2B,EAAE,CAAC;IACvC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,CAAC;QACrD,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IAClD,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}
@@ -0,0 +1,17 @@
1
+ import type { CompiledConnection } from "@assemblyline-agents/core";
2
+ import type { ConnectionClient, ResolvedConnection } from "./connection-clients.js";
3
+ import type { ConnectionToolMetadata } from "./connection-stores.js";
4
+ import type { RuntimeConnectionRegistry } from "./connections.js";
5
+ export declare class RelayMcpConnectionClient implements ConnectionClient {
6
+ private readonly compiled;
7
+ private readonly resolve;
8
+ private readonly registry;
9
+ private metadata;
10
+ constructor(compiled: CompiledConnection, resolve: () => Promise<ResolvedConnection>, registry: RuntimeConnectionRegistry);
11
+ getToolMetadata(): Promise<ConnectionToolMetadata[]>;
12
+ executeTool(toolName: string, args: unknown): Promise<unknown>;
13
+ close(): Promise<void>;
14
+ private invoke;
15
+ private binding;
16
+ }
17
+ //# sourceMappingURL=relay-mcp-client.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"relay-mcp-client.d.ts","sourceRoot":"","sources":["../src/relay-mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,kBAAkB,EAGnB,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AACpF,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AASrE,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,kBAAkB,CAAC;AAclE,qBAAa,wBAAyB,YAAW,gBAAgB;IAI7D,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAL3B,OAAO,CAAC,QAAQ,CAAuC;gBAGpC,QAAQ,EAAE,kBAAkB,EAC5B,OAAO,EAAE,MAAM,OAAO,CAAC,kBAAkB,CAAC,EAC1C,QAAQ,EAAE,yBAAyB;IAGhD,eAAe,IAAI,OAAO,CAAC,sBAAsB,EAAE,CAAC;IAuBpD,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAc9D,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAId,MAAM;IAgCpB,OAAO,CAAC,OAAO;CAIhB"}