@nervekit/agent 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 (263) hide show
  1. package/LICENSE +201 -0
  2. package/NOTICE +5 -0
  3. package/dist/agent-loop.d.ts +2 -0
  4. package/dist/agent-loop.d.ts.map +1 -0
  5. package/dist/agent-loop.js +2 -0
  6. package/dist/agent-loop.js.map +1 -0
  7. package/dist/agent.d.ts +125 -0
  8. package/dist/agent.d.ts.map +1 -0
  9. package/dist/agent.js +427 -0
  10. package/dist/agent.js.map +1 -0
  11. package/dist/harness/agent-harness.d.ts +114 -0
  12. package/dist/harness/agent-harness.d.ts.map +1 -0
  13. package/dist/harness/agent-harness.js +474 -0
  14. package/dist/harness/agent-harness.js.map +1 -0
  15. package/dist/harness/compaction/branch-summarization.d.ts +54 -0
  16. package/dist/harness/compaction/branch-summarization.d.ts.map +1 -0
  17. package/dist/harness/compaction/branch-summarization.js +183 -0
  18. package/dist/harness/compaction/branch-summarization.js.map +1 -0
  19. package/dist/harness/compaction/compaction.d.ts +19 -0
  20. package/dist/harness/compaction/compaction.d.ts.map +1 -0
  21. package/dist/harness/compaction/compaction.js +335 -0
  22. package/dist/harness/compaction/compaction.js.map +1 -0
  23. package/dist/harness/compaction/cut-points.d.ts +7 -0
  24. package/dist/harness/compaction/cut-points.d.ts.map +1 -0
  25. package/dist/harness/compaction/cut-points.js +106 -0
  26. package/dist/harness/compaction/cut-points.js.map +1 -0
  27. package/dist/harness/compaction/overflow.d.ts +3 -0
  28. package/dist/harness/compaction/overflow.d.ts.map +1 -0
  29. package/dist/harness/compaction/overflow.js +54 -0
  30. package/dist/harness/compaction/overflow.js.map +1 -0
  31. package/dist/harness/compaction/policy.d.ts +8 -0
  32. package/dist/harness/compaction/policy.d.ts.map +1 -0
  33. package/dist/harness/compaction/policy.js +57 -0
  34. package/dist/harness/compaction/policy.js.map +1 -0
  35. package/dist/harness/compaction/types.d.ts +79 -0
  36. package/dist/harness/compaction/types.d.ts.map +1 -0
  37. package/dist/harness/compaction/types.js +2 -0
  38. package/dist/harness/compaction/types.js.map +1 -0
  39. package/dist/harness/compaction/usage.d.ts +30 -0
  40. package/dist/harness/compaction/usage.d.ts.map +1 -0
  41. package/dist/harness/compaction/usage.js +177 -0
  42. package/dist/harness/compaction/usage.js.map +1 -0
  43. package/dist/harness/compaction/utils.d.ts +25 -0
  44. package/dist/harness/compaction/utils.d.ts.map +1 -0
  45. package/dist/harness/compaction/utils.js +131 -0
  46. package/dist/harness/compaction/utils.js.map +1 -0
  47. package/dist/harness/configuration.d.ts +34 -0
  48. package/dist/harness/configuration.d.ts.map +1 -0
  49. package/dist/harness/configuration.js +75 -0
  50. package/dist/harness/configuration.js.map +1 -0
  51. package/dist/harness/conversation/context.d.ts +24 -0
  52. package/dist/harness/conversation/context.d.ts.map +1 -0
  53. package/dist/harness/conversation/context.js +76 -0
  54. package/dist/harness/conversation/context.js.map +1 -0
  55. package/dist/harness/conversation/conversation.d.ts +36 -0
  56. package/dist/harness/conversation/conversation.d.ts.map +1 -0
  57. package/dist/harness/conversation/conversation.js +163 -0
  58. package/dist/harness/conversation/conversation.js.map +1 -0
  59. package/dist/harness/conversation/entries.d.ts +117 -0
  60. package/dist/harness/conversation/entries.d.ts.map +1 -0
  61. package/dist/harness/conversation/entries.js +2 -0
  62. package/dist/harness/conversation/entries.js.map +1 -0
  63. package/dist/harness/conversation/jsonl-repo.d.ts +27 -0
  64. package/dist/harness/conversation/jsonl-repo.d.ts.map +1 -0
  65. package/dist/harness/conversation/jsonl-repo.js +110 -0
  66. package/dist/harness/conversation/jsonl-repo.js.map +1 -0
  67. package/dist/harness/conversation/jsonl-storage.d.ts +34 -0
  68. package/dist/harness/conversation/jsonl-storage.d.ts.map +1 -0
  69. package/dist/harness/conversation/jsonl-storage.js +213 -0
  70. package/dist/harness/conversation/jsonl-storage.js.map +1 -0
  71. package/dist/harness/conversation/memory-repo.d.ts +18 -0
  72. package/dist/harness/conversation/memory-repo.d.ts.map +1 -0
  73. package/dist/harness/conversation/memory-repo.js +45 -0
  74. package/dist/harness/conversation/memory-repo.js.map +1 -0
  75. package/dist/harness/conversation/memory-storage.d.ts +25 -0
  76. package/dist/harness/conversation/memory-storage.d.ts.map +1 -0
  77. package/dist/harness/conversation/memory-storage.js +88 -0
  78. package/dist/harness/conversation/memory-storage.js.map +1 -0
  79. package/dist/harness/conversation/repo-utils.d.ts +13 -0
  80. package/dist/harness/conversation/repo-utils.d.ts.map +1 -0
  81. package/dist/harness/conversation/repo-utils.js +39 -0
  82. package/dist/harness/conversation/repo-utils.js.map +1 -0
  83. package/dist/harness/conversation/storage-utils.d.ts +11 -0
  84. package/dist/harness/conversation/storage-utils.d.ts.map +1 -0
  85. package/dist/harness/conversation/storage-utils.js +32 -0
  86. package/dist/harness/conversation/storage-utils.js.map +1 -0
  87. package/dist/harness/conversation/text-extraction.d.ts +8 -0
  88. package/dist/harness/conversation/text-extraction.d.ts.map +1 -0
  89. package/dist/harness/conversation/text-extraction.js +24 -0
  90. package/dist/harness/conversation/text-extraction.js.map +1 -0
  91. package/dist/harness/conversation/uuid.d.ts +2 -0
  92. package/dist/harness/conversation/uuid.d.ts.map +1 -0
  93. package/dist/harness/conversation/uuid.js +54 -0
  94. package/dist/harness/conversation/uuid.js.map +1 -0
  95. package/dist/harness/conversation-writes.d.ts +9 -0
  96. package/dist/harness/conversation-writes.d.ts.map +1 -0
  97. package/dist/harness/conversation-writes.js +54 -0
  98. package/dist/harness/conversation-writes.js.map +1 -0
  99. package/dist/harness/env/nodejs.d.ts +53 -0
  100. package/dist/harness/env/nodejs.d.ts.map +1 -0
  101. package/dist/harness/env/nodejs.js +497 -0
  102. package/dist/harness/env/nodejs.js.map +1 -0
  103. package/dist/harness/env/types.d.ts +106 -0
  104. package/dist/harness/env/types.d.ts.map +1 -0
  105. package/dist/harness/env/types.js +2 -0
  106. package/dist/harness/env/types.js.map +1 -0
  107. package/dist/harness/errors.d.ts +48 -0
  108. package/dist/harness/errors.d.ts.map +1 -0
  109. package/dist/harness/errors.js +63 -0
  110. package/dist/harness/errors.js.map +1 -0
  111. package/dist/harness/events.d.ts +238 -0
  112. package/dist/harness/events.d.ts.map +1 -0
  113. package/dist/harness/events.js +2 -0
  114. package/dist/harness/events.js.map +1 -0
  115. package/dist/harness/harness-configuration-methods.d.ts +36 -0
  116. package/dist/harness/harness-configuration-methods.d.ts.map +1 -0
  117. package/dist/harness/harness-configuration-methods.js +145 -0
  118. package/dist/harness/harness-configuration-methods.js.map +1 -0
  119. package/dist/harness/harness-continuation.d.ts +18 -0
  120. package/dist/harness/harness-continuation.d.ts.map +1 -0
  121. package/dist/harness/harness-continuation.js +56 -0
  122. package/dist/harness/harness-continuation.js.map +1 -0
  123. package/dist/harness/harness-events.d.ts +23 -0
  124. package/dist/harness/harness-events.d.ts.map +1 -0
  125. package/dist/harness/harness-events.js +129 -0
  126. package/dist/harness/harness-events.js.map +1 -0
  127. package/dist/harness/harness-invocations.d.ts +20 -0
  128. package/dist/harness/harness-invocations.d.ts.map +1 -0
  129. package/dist/harness/harness-invocations.js +22 -0
  130. package/dist/harness/harness-invocations.js.map +1 -0
  131. package/dist/harness/harness-maintenance.d.ts +30 -0
  132. package/dist/harness/harness-maintenance.d.ts.map +1 -0
  133. package/dist/harness/harness-maintenance.js +155 -0
  134. package/dist/harness/harness-maintenance.js.map +1 -0
  135. package/dist/harness/harness-queue-methods.d.ts +45 -0
  136. package/dist/harness/harness-queue-methods.d.ts.map +1 -0
  137. package/dist/harness/harness-queue-methods.js +51 -0
  138. package/dist/harness/harness-queue-methods.js.map +1 -0
  139. package/dist/harness/messages.d.ts +76 -0
  140. package/dist/harness/messages.d.ts.map +1 -0
  141. package/dist/harness/messages.js +147 -0
  142. package/dist/harness/messages.js.map +1 -0
  143. package/dist/harness/options.d.ts +95 -0
  144. package/dist/harness/options.d.ts.map +1 -0
  145. package/dist/harness/options.js +2 -0
  146. package/dist/harness/options.js.map +1 -0
  147. package/dist/harness/prompt-templates.d.ts +49 -0
  148. package/dist/harness/prompt-templates.d.ts.map +1 -0
  149. package/dist/harness/prompt-templates.js +230 -0
  150. package/dist/harness/prompt-templates.js.map +1 -0
  151. package/dist/harness/result.d.ts +19 -0
  152. package/dist/harness/result.d.ts.map +1 -0
  153. package/dist/harness/result.js +32 -0
  154. package/dist/harness/result.js.map +1 -0
  155. package/dist/harness/run/messages.d.ts +5 -0
  156. package/dist/harness/run/messages.d.ts.map +1 -0
  157. package/dist/harness/run/messages.js +37 -0
  158. package/dist/harness/run/messages.js.map +1 -0
  159. package/dist/harness/skills/format.d.ts +6 -0
  160. package/dist/harness/skills/format.d.ts.map +1 -0
  161. package/dist/harness/skills/format.js +38 -0
  162. package/dist/harness/skills/format.js.map +1 -0
  163. package/dist/harness/skills/index.d.ts +5 -0
  164. package/dist/harness/skills/index.d.ts.map +1 -0
  165. package/dist/harness/skills/index.js +5 -0
  166. package/dist/harness/skills/index.js.map +1 -0
  167. package/dist/harness/skills/loader.d.ts +43 -0
  168. package/dist/harness/skills/loader.d.ts.map +1 -0
  169. package/dist/harness/skills/loader.js +279 -0
  170. package/dist/harness/skills/loader.js.map +1 -0
  171. package/dist/harness/skills/parser.d.ts +12 -0
  172. package/dist/harness/skills/parser.d.ts.map +1 -0
  173. package/dist/harness/skills/parser.js +22 -0
  174. package/dist/harness/skills/parser.js.map +1 -0
  175. package/dist/harness/skills/validation.d.ts +3 -0
  176. package/dist/harness/skills/validation.d.ts.map +1 -0
  177. package/dist/harness/skills/validation.js +28 -0
  178. package/dist/harness/skills/validation.js.map +1 -0
  179. package/dist/harness/skills.d.ts +2 -0
  180. package/dist/harness/skills.d.ts.map +1 -0
  181. package/dist/harness/skills.js +2 -0
  182. package/dist/harness/skills.js.map +1 -0
  183. package/dist/harness/stream-options.d.ts +5 -0
  184. package/dist/harness/stream-options.d.ts.map +1 -0
  185. package/dist/harness/stream-options.js +67 -0
  186. package/dist/harness/stream-options.js.map +1 -0
  187. package/dist/harness/system-prompt.d.ts +2 -0
  188. package/dist/harness/system-prompt.d.ts.map +1 -0
  189. package/dist/harness/system-prompt.js +2 -0
  190. package/dist/harness/system-prompt.js.map +1 -0
  191. package/dist/harness/turn-state.d.ts +27 -0
  192. package/dist/harness/turn-state.d.ts.map +1 -0
  193. package/dist/harness/turn-state.js +35 -0
  194. package/dist/harness/turn-state.js.map +1 -0
  195. package/dist/harness/types.d.ts +9 -0
  196. package/dist/harness/types.d.ts.map +1 -0
  197. package/dist/harness/types.js +8 -0
  198. package/dist/harness/types.js.map +1 -0
  199. package/dist/harness/utils/env-path.d.ts +14 -0
  200. package/dist/harness/utils/env-path.d.ts.map +1 -0
  201. package/dist/harness/utils/env-path.js +99 -0
  202. package/dist/harness/utils/env-path.js.map +1 -0
  203. package/dist/harness/utils/shell-output.d.ts +16 -0
  204. package/dist/harness/utils/shell-output.d.ts.map +1 -0
  205. package/dist/harness/utils/shell-output.js +137 -0
  206. package/dist/harness/utils/shell-output.js.map +1 -0
  207. package/dist/harness/utils/truncate.d.ts +70 -0
  208. package/dist/harness/utils/truncate.d.ts.map +1 -0
  209. package/dist/harness/utils/truncate.js +289 -0
  210. package/dist/harness/utils/truncate.js.map +1 -0
  211. package/dist/index.d.ts +24 -0
  212. package/dist/index.d.ts.map +1 -0
  213. package/dist/index.js +25 -0
  214. package/dist/index.js.map +1 -0
  215. package/dist/node.d.ts +3 -0
  216. package/dist/node.d.ts.map +1 -0
  217. package/dist/node.js +3 -0
  218. package/dist/node.js.map +1 -0
  219. package/dist/proxy.d.ts +67 -0
  220. package/dist/proxy.d.ts.map +1 -0
  221. package/dist/proxy.js +299 -0
  222. package/dist/proxy.js.map +1 -0
  223. package/dist/runtime/loop/agent-loop.d.ts +24 -0
  224. package/dist/runtime/loop/agent-loop.d.ts.map +1 -0
  225. package/dist/runtime/loop/agent-loop.js +72 -0
  226. package/dist/runtime/loop/agent-loop.js.map +1 -0
  227. package/dist/runtime/loop/events.d.ts +5 -0
  228. package/dist/runtime/loop/events.d.ts.map +1 -0
  229. package/dist/runtime/loop/events.js +5 -0
  230. package/dist/runtime/loop/events.js.map +1 -0
  231. package/dist/runtime/loop/index.d.ts +6 -0
  232. package/dist/runtime/loop/index.d.ts.map +1 -0
  233. package/dist/runtime/loop/index.js +6 -0
  234. package/dist/runtime/loop/index.js.map +1 -0
  235. package/dist/runtime/loop/stream-assistant.d.ts +9 -0
  236. package/dist/runtime/loop/stream-assistant.d.ts.map +1 -0
  237. package/dist/runtime/loop/stream-assistant.js +87 -0
  238. package/dist/runtime/loop/stream-assistant.js.map +1 -0
  239. package/dist/runtime/loop/tool-execution.d.ts +15 -0
  240. package/dist/runtime/loop/tool-execution.d.ts.map +1 -0
  241. package/dist/runtime/loop/tool-execution.js +271 -0
  242. package/dist/runtime/loop/tool-execution.js.map +1 -0
  243. package/dist/runtime/loop/turn-loop.d.ts +7 -0
  244. package/dist/runtime/loop/turn-loop.d.ts.map +1 -0
  245. package/dist/runtime/loop/turn-loop.js +84 -0
  246. package/dist/runtime/loop/turn-loop.js.map +1 -0
  247. package/dist/runtime.d.ts +63 -0
  248. package/dist/runtime.d.ts.map +1 -0
  249. package/dist/runtime.js +170 -0
  250. package/dist/runtime.js.map +1 -0
  251. package/dist/suspension.d.ts +18 -0
  252. package/dist/suspension.d.ts.map +1 -0
  253. package/dist/suspension.js +19 -0
  254. package/dist/suspension.js.map +1 -0
  255. package/dist/types.d.ts +395 -0
  256. package/dist/types.d.ts.map +1 -0
  257. package/dist/types.js +2 -0
  258. package/dist/types.js.map +1 -0
  259. package/dist/worker.d.ts +2 -0
  260. package/dist/worker.d.ts.map +1 -0
  261. package/dist/worker.js +88 -0
  262. package/dist/worker.js.map +1 -0
  263. package/package.json +54 -0
@@ -0,0 +1,474 @@
1
+ import { streamSimple, } from "@earendil-works/pi-ai";
2
+ import { runAgentLoop } from "../agent-loop.js";
3
+ import { isAgentToolSuspension } from "../suspension.js";
4
+ import { cloneHarnessResources, createToolMap, validateToolNames, } from "./configuration.js";
5
+ import { flushPendingConversationWrites as flushHarnessPendingConversationWrites } from "./conversation-writes.js";
6
+ import { AgentHarnessError } from "./errors.js";
7
+ import { abortHarnessRun, getHarnessActiveTools, setHarnessActiveTools, setHarnessModel, setHarnessResources, setHarnessThinkingLevel, setHarnessTools, } from "./harness-configuration-methods.js";
8
+ import { continueHarnessRun, } from "./harness-continuation.js";
9
+ import { AgentHarnessEventHub, normalizeHarnessError, normalizeHookError, } from "./harness-events.js";
10
+ import { invokePromptTemplate, invokeSkill, } from "./harness-invocations.js";
11
+ import { compactHarnessConversation, navigateHarnessTree, } from "./harness-maintenance.js";
12
+ import { appendExternalHarnessMessage, appendHarnessMessage, enqueueHarnessMessage as enqueueHarnessQueueMessage, enqueueNextTurn, steerHarness, } from "./harness-queue-methods.js";
13
+ import { convertToLlm } from "./messages.js";
14
+ import { toError } from "./result.js";
15
+ import { createFailureMessage, createUserMessage } from "./run/messages.js";
16
+ import { cloneStreamOptions, mergeHeaders } from "./stream-options.js";
17
+ import { createTurnState as createAgentHarnessTurnState, } from "./turn-state.js";
18
+ export class AgentHarness {
19
+ env;
20
+ conversation;
21
+ phase = "idle";
22
+ runAbortController;
23
+ runPromise;
24
+ pendingConversationWrites = [];
25
+ model;
26
+ thinkingLevel;
27
+ systemPrompt;
28
+ streamOptions;
29
+ getApiKeyAndHeaders;
30
+ resources;
31
+ tools = new Map();
32
+ activeToolNames;
33
+ steerQueue = [];
34
+ steeringQueueMode;
35
+ followUpQueue = [];
36
+ followUpQueueMode;
37
+ nextTurnQueue = [];
38
+ queuedMessageWrites = new WeakMap();
39
+ events = new AgentHarnessEventHub();
40
+ constructor(options) {
41
+ this.env = options.env;
42
+ this.conversation = options.conversation;
43
+ this.resources = options.resources ?? {};
44
+ this.streamOptions = cloneStreamOptions(options.streamOptions);
45
+ this.systemPrompt = options.systemPrompt;
46
+ this.getApiKeyAndHeaders = options.getApiKeyAndHeaders;
47
+ this.tools = createToolMap(options.tools ?? []);
48
+ this.model = options.model;
49
+ this.thinkingLevel = options.thinkingLevel ?? "off";
50
+ this.activeToolNames = options.activeToolNames
51
+ ? [...options.activeToolNames]
52
+ : [...this.tools.keys()];
53
+ this.validateToolNames(this.activeToolNames);
54
+ this.steeringQueueMode = options.steeringMode ?? "one-at-a-time";
55
+ this.followUpQueueMode = options.followUpMode ?? "one-at-a-time";
56
+ }
57
+ async emitOwn(event, signal) {
58
+ await this.events.emitOwn(event, signal);
59
+ }
60
+ async emitAny(event, signal) {
61
+ await this.events.emitAny(event, signal);
62
+ }
63
+ async emitHook(event) {
64
+ return this.events.emitHook(event);
65
+ }
66
+ async emitBeforeProviderRequest(model, conversationId, streamOptions) {
67
+ return this.events.emitBeforeProviderRequest(model, conversationId, streamOptions);
68
+ }
69
+ async emitBeforeProviderPayload(model, payload) {
70
+ return this.events.emitBeforeProviderPayload(model, payload);
71
+ }
72
+ async emitQueueUpdate() {
73
+ await this.emitOwn({
74
+ type: "queue_update",
75
+ steer: this.steerQueue.map((entry) => entry.message),
76
+ followUp: this.followUpQueue.map((entry) => entry.message),
77
+ nextTurn: [...this.nextTurnQueue],
78
+ });
79
+ }
80
+ startRunPromise() {
81
+ let finish = () => { };
82
+ this.runPromise = new Promise((resolve) => {
83
+ finish = resolve;
84
+ });
85
+ return () => {
86
+ this.runPromise = undefined;
87
+ finish();
88
+ };
89
+ }
90
+ async createTurnState() {
91
+ return createAgentHarnessTurnState({
92
+ env: this.env,
93
+ conversation: this.conversation,
94
+ resources: this.getResources(),
95
+ streamOptions: this.streamOptions,
96
+ systemPrompt: this.systemPrompt,
97
+ model: this.model,
98
+ thinkingLevel: this.thinkingLevel,
99
+ tools: this.tools,
100
+ activeToolNames: this.activeToolNames,
101
+ });
102
+ }
103
+ createContext(turnState, systemPrompt) {
104
+ return {
105
+ systemPrompt: systemPrompt ?? turnState.systemPrompt,
106
+ messages: turnState.messages.slice(),
107
+ tools: turnState.activeTools.slice(),
108
+ };
109
+ }
110
+ createStreamFn(getTurnState) {
111
+ return async (model, context, streamOptions) => {
112
+ const turnState = getTurnState();
113
+ const auth = await this.getApiKeyAndHeaders?.(model);
114
+ const snapshotOptions = {
115
+ ...turnState.streamOptions,
116
+ headers: mergeHeaders(turnState.streamOptions.headers, auth?.headers),
117
+ };
118
+ const requestOptions = await this.emitBeforeProviderRequest(model, turnState.conversationId, snapshotOptions);
119
+ return streamSimple(model, context, {
120
+ cacheRetention: requestOptions.cacheRetention,
121
+ headers: requestOptions.headers,
122
+ maxRetries: requestOptions.maxRetries,
123
+ maxRetryDelayMs: requestOptions.maxRetryDelayMs,
124
+ metadata: requestOptions.metadata,
125
+ onPayload: async (payload) => await this.emitBeforeProviderPayload(model, payload),
126
+ onResponse: async (response) => {
127
+ const headers = { ...response.headers };
128
+ await this.emitOwn({
129
+ type: "after_provider_response",
130
+ status: response.status,
131
+ headers,
132
+ }, streamOptions?.signal);
133
+ },
134
+ reasoning: streamOptions?.reasoning,
135
+ signal: streamOptions?.signal,
136
+ sessionId: turnState.conversationId,
137
+ timeoutMs: requestOptions.timeoutMs,
138
+ transport: requestOptions.transport,
139
+ apiKey: auth?.apiKey,
140
+ });
141
+ };
142
+ }
143
+ async drainQueuedMessages(queue, mode) {
144
+ const entries = mode === "all" ? queue.splice(0) : queue.splice(0, 1);
145
+ if (entries.length === 0)
146
+ return [];
147
+ try {
148
+ await this.emitQueueUpdate();
149
+ for (const entry of entries) {
150
+ if (entry.id || entry.timestamp) {
151
+ this.queuedMessageWrites.set(entry.message, {
152
+ id: entry.id,
153
+ timestamp: entry.timestamp,
154
+ });
155
+ }
156
+ }
157
+ return entries.map((entry) => entry.message);
158
+ }
159
+ catch (error) {
160
+ queue.unshift(...entries);
161
+ throw normalizeHookError(error);
162
+ }
163
+ }
164
+ createLoopConfig(getTurnState, setTurnState) {
165
+ const turnState = getTurnState();
166
+ return {
167
+ model: turnState.model,
168
+ reasoning: turnState.thinkingLevel === "off" ? undefined : turnState.thinkingLevel,
169
+ convertToLlm,
170
+ transformContext: async (messages) => {
171
+ const result = await this.emitHook({
172
+ type: "context",
173
+ messages: [...messages],
174
+ });
175
+ return result?.messages ?? messages;
176
+ },
177
+ beforeToolCall: async ({ toolCall, args }) => {
178
+ const result = await this.emitHook({
179
+ type: "tool_call",
180
+ toolCallId: toolCall.id,
181
+ toolName: toolCall.name,
182
+ input: args,
183
+ });
184
+ return result
185
+ ? { block: result.block, reason: result.reason }
186
+ : undefined;
187
+ },
188
+ afterToolCall: async ({ toolCall, args, result, isError }) => {
189
+ const patch = await this.emitHook({
190
+ type: "tool_result",
191
+ toolCallId: toolCall.id,
192
+ toolName: toolCall.name,
193
+ input: args,
194
+ content: result.content,
195
+ details: result.details,
196
+ isError,
197
+ });
198
+ return patch
199
+ ? {
200
+ content: patch.content,
201
+ details: patch.details,
202
+ isError: patch.isError,
203
+ terminate: patch.terminate,
204
+ }
205
+ : undefined;
206
+ },
207
+ prepareNextTurn: async () => {
208
+ await this.flushPendingConversationWrites();
209
+ const nextTurnState = await this.createTurnState();
210
+ setTurnState(nextTurnState);
211
+ return {
212
+ context: this.createContext(nextTurnState),
213
+ model: nextTurnState.model,
214
+ thinkingLevel: nextTurnState.thinkingLevel,
215
+ };
216
+ },
217
+ getSteeringMessages: async () => this.drainQueuedMessages(this.steerQueue, this.steeringQueueMode),
218
+ getFollowUpMessages: async () => [],
219
+ };
220
+ }
221
+ validateToolNames(toolNames, tools = this.tools) {
222
+ validateToolNames(toolNames, tools);
223
+ }
224
+ async flushPendingConversationWrites() {
225
+ await flushHarnessPendingConversationWrites(this.conversation, this.pendingConversationWrites);
226
+ }
227
+ async handleAgentEvent(event, signal) {
228
+ if (event.type === "message_end") {
229
+ const queuedWrite = this.queuedMessageWrites.get(event.message);
230
+ this.queuedMessageWrites.delete(event.message);
231
+ if (queuedWrite?.id) {
232
+ await this.conversation.appendMessageWithId(queuedWrite.id, event.message, queuedWrite.timestamp);
233
+ }
234
+ else {
235
+ await this.conversation.appendMessage(event.message);
236
+ }
237
+ await this.emitAny(event, signal);
238
+ return;
239
+ }
240
+ if (event.type === "turn_end") {
241
+ let eventError;
242
+ try {
243
+ await this.emitAny(event, signal);
244
+ }
245
+ catch (error) {
246
+ eventError = error;
247
+ }
248
+ const hadPendingMutations = this.pendingConversationWrites.length > 0;
249
+ await this.flushPendingConversationWrites();
250
+ if (eventError)
251
+ throw eventError;
252
+ await this.emitOwn({ type: "save_point", hadPendingMutations });
253
+ return;
254
+ }
255
+ if (event.type === "agent_end") {
256
+ await this.flushPendingConversationWrites();
257
+ this.phase = "idle";
258
+ await this.emitAny(event, signal);
259
+ await this.emitOwn({ type: "settled", nextTurnCount: this.nextTurnQueue.length }, signal);
260
+ return;
261
+ }
262
+ await this.emitAny(event, signal);
263
+ }
264
+ async emitRunFailure(model, error, aborted, signal) {
265
+ const failureMessage = createFailureMessage(model, error, aborted);
266
+ await this.handleAgentEvent({ type: "message_start", message: failureMessage }, signal);
267
+ await this.handleAgentEvent({ type: "message_end", message: failureMessage }, signal);
268
+ await this.handleAgentEvent({ type: "turn_end", message: failureMessage, toolResults: [] }, signal);
269
+ await this.handleAgentEvent({ type: "agent_end", messages: [failureMessage] }, signal);
270
+ return [failureMessage];
271
+ }
272
+ async executeTurn(turnState, text, options) {
273
+ let activeTurnState = turnState;
274
+ const promptMessage = createUserMessage(text, options?.images);
275
+ let messages = [promptMessage];
276
+ if (this.nextTurnQueue.length > 0) {
277
+ const queuedMessages = this.nextTurnQueue.splice(0);
278
+ try {
279
+ await this.emitQueueUpdate();
280
+ }
281
+ catch (error) {
282
+ this.nextTurnQueue.unshift(...queuedMessages);
283
+ throw normalizeHookError(error);
284
+ }
285
+ messages = [...queuedMessages, promptMessage];
286
+ }
287
+ const beforeResult = await this.emitHook({
288
+ type: "before_agent_start",
289
+ prompt: text,
290
+ images: options?.images,
291
+ systemPrompt: turnState.systemPrompt,
292
+ resources: turnState.resources,
293
+ });
294
+ if (beforeResult?.messages)
295
+ messages = [...messages, ...beforeResult.messages];
296
+ const abortController = new AbortController();
297
+ const getTurnState = () => activeTurnState;
298
+ const setTurnState = (nextTurnState) => {
299
+ activeTurnState = nextTurnState;
300
+ };
301
+ this.runAbortController = abortController;
302
+ const runResultPromise = (async () => {
303
+ try {
304
+ return await runAgentLoop(messages, this.createContext(turnState, beforeResult?.systemPrompt), this.createLoopConfig(getTurnState, setTurnState), (event) => this.handleAgentEvent(event, abortController.signal), abortController.signal, this.createStreamFn(getTurnState));
305
+ }
306
+ catch (error) {
307
+ if (isAgentToolSuspension(error)) {
308
+ this.phase = "idle";
309
+ throw error;
310
+ }
311
+ try {
312
+ return await this.emitRunFailure(activeTurnState.model, error, abortController.signal.aborted, abortController.signal);
313
+ }
314
+ catch (failureError) {
315
+ const cause = new AggregateError([toError(error), toError(failureError)], "Agent run failed and failure reporting failed");
316
+ throw new AgentHarnessError("unknown", cause.message, cause);
317
+ }
318
+ }
319
+ })();
320
+ try {
321
+ const newMessages = await runResultPromise;
322
+ for (const message of [...newMessages].reverse()) {
323
+ if (message.role === "assistant") {
324
+ return message;
325
+ }
326
+ }
327
+ throw new AgentHarnessError("invalid_state", "AgentHarness prompt completed without an assistant message");
328
+ }
329
+ finally {
330
+ try {
331
+ await this.flushPendingConversationWrites();
332
+ }
333
+ finally {
334
+ this.runAbortController = undefined;
335
+ }
336
+ }
337
+ }
338
+ async runForegroundTurn(resolvePrompt) {
339
+ if (this.phase !== "idle") {
340
+ throw new AgentHarnessError("busy", "AgentHarness is busy");
341
+ }
342
+ this.phase = "turn";
343
+ const finishRunPromise = this.startRunPromise();
344
+ try {
345
+ const turnState = await this.createTurnState();
346
+ const prompt = await resolvePrompt(turnState);
347
+ return await this.executeTurn(turnState, prompt.text, prompt.options);
348
+ }
349
+ catch (error) {
350
+ this.phase = "idle";
351
+ throw normalizeHarnessError(error, "unknown");
352
+ }
353
+ finally {
354
+ finishRunPromise();
355
+ }
356
+ }
357
+ async prompt(text, options) {
358
+ return this.runForegroundTurn(() => ({ text, options }));
359
+ }
360
+ async continue() {
361
+ return continueHarnessRun(this);
362
+ }
363
+ async skill(name, additionalInstructions) {
364
+ return invokeSkill(this, name, additionalInstructions);
365
+ }
366
+ async promptFromTemplate(name, args = []) {
367
+ return invokePromptTemplate(this, name, args);
368
+ }
369
+ queueState() {
370
+ return this;
371
+ }
372
+ async steer(text, options) {
373
+ return steerHarness(this.queueState(), text, options);
374
+ }
375
+ async followUp(text, options) {
376
+ return this.steer(text, options);
377
+ }
378
+ async enqueueHarnessMessage(input) {
379
+ return enqueueHarnessQueueMessage(this.queueState(), input);
380
+ }
381
+ async nextTurn(text, options) {
382
+ return enqueueNextTurn(this.queueState(), text, options);
383
+ }
384
+ async appendMessage(message) {
385
+ return appendHarnessMessage(this.queueState(), message);
386
+ }
387
+ async appendExternalMessage(input) {
388
+ return appendExternalHarnessMessage(this.queueState(), input);
389
+ }
390
+ async compact(customInstructions) {
391
+ return compactHarnessConversation(this.maintenanceContext(), customInstructions);
392
+ }
393
+ async navigateTree(targetId, options) {
394
+ return navigateHarnessTree(this.maintenanceContext(), targetId, options);
395
+ }
396
+ maintenanceContext() {
397
+ return {
398
+ getPhase: () => this.phase,
399
+ setPhase: (phase) => {
400
+ this.phase = phase;
401
+ },
402
+ conversation: this.conversation,
403
+ getModel: () => this.model,
404
+ getThinkingLevel: () => this.thinkingLevel,
405
+ getApiKeyAndHeaders: this.getApiKeyAndHeaders,
406
+ emitHook: (event) => this.emitHook(event),
407
+ emitOwn: (event) => this.emitOwn(event),
408
+ };
409
+ }
410
+ configurationState() {
411
+ return this;
412
+ }
413
+ getModel() {
414
+ return this.model;
415
+ }
416
+ async setModel(model) {
417
+ return setHarnessModel(this.configurationState(), model);
418
+ }
419
+ getThinkingLevel() {
420
+ return this.thinkingLevel;
421
+ }
422
+ async setThinkingLevel(level) {
423
+ return setHarnessThinkingLevel(this.configurationState(), level);
424
+ }
425
+ getTools() {
426
+ return [...this.tools.values()];
427
+ }
428
+ async setTools(tools, activeToolNames) {
429
+ return setHarnessTools(this.configurationState(), tools, activeToolNames);
430
+ }
431
+ getActiveTools() {
432
+ return getHarnessActiveTools(this.configurationState());
433
+ }
434
+ async setActiveTools(toolNames) {
435
+ return setHarnessActiveTools(this.configurationState(), toolNames);
436
+ }
437
+ getSteeringMode() {
438
+ return this.steeringQueueMode;
439
+ }
440
+ async setSteeringMode(mode) {
441
+ this.steeringQueueMode = mode;
442
+ }
443
+ getFollowUpMode() {
444
+ return this.followUpQueueMode;
445
+ }
446
+ async setFollowUpMode(mode) {
447
+ this.followUpQueueMode = mode;
448
+ }
449
+ getResources() {
450
+ return cloneHarnessResources(this.resources);
451
+ }
452
+ async setResources(resources) {
453
+ return setHarnessResources(this.configurationState(), resources);
454
+ }
455
+ getStreamOptions() {
456
+ return cloneStreamOptions(this.streamOptions);
457
+ }
458
+ async setStreamOptions(streamOptions) {
459
+ this.streamOptions = cloneStreamOptions(streamOptions);
460
+ }
461
+ async abort() {
462
+ return abortHarnessRun(this.configurationState());
463
+ }
464
+ async waitForIdle() {
465
+ await this.runPromise;
466
+ }
467
+ subscribe(listener) {
468
+ return this.events.subscribe(listener);
469
+ }
470
+ on(type, handler) {
471
+ return this.events.on(type, handler);
472
+ }
473
+ }
474
+ //# sourceMappingURL=agent-harness.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"agent-harness.js","sourceRoot":"","sources":["../../src/harness/agent-harness.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,YAAY,GACb,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAYzD,OAAO,EACL,qBAAqB,EACrB,aAAa,EACb,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EAAE,8BAA8B,IAAI,qCAAqC,EAAE,MAAM,0BAA0B,CAAC;AAEnH,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAUhD,OAAO,EACL,eAAe,EACf,qBAAqB,EAErB,qBAAqB,EACrB,eAAe,EACf,mBAAmB,EACnB,uBAAuB,EACvB,eAAe,GAChB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EACL,kBAAkB,GAEnB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,oBAAoB,EACpB,qBAAqB,EACrB,kBAAkB,GACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,oBAAoB,EACpB,WAAW,GACZ,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,0BAA0B,EAE1B,mBAAmB,GACpB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,4BAA4B,EAC5B,oBAAoB,EACpB,qBAAqB,IAAI,0BAA0B,EACnD,eAAe,EAGf,YAAY,GACb,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAQ7C,OAAO,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AACtC,OAAO,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAC5E,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAEL,eAAe,IAAI,2BAA2B,GAC/C,MAAM,iBAAiB,CAAC;AACzB,MAAM,OAAO,YAAY;IAKd,GAAG,CAAe;IACnB,YAAY,CAAe;IAC3B,KAAK,GAAsB,MAAM,CAAC;IAC1C,kBAAkB,CAAmB;IAC7B,UAAU,CAAiB;IAC3B,yBAAyB,GAA+B,EAAE,CAAC;IAC3D,KAAK,CAAW;IAChB,aAAa,CAAgB;IAC7B,YAAY,CAIF;IACV,aAAa,CAA4B;IACzC,mBAAmB,CAA8C;IACjE,SAAS,CAAiD;IAC1D,KAAK,GAAG,IAAI,GAAG,EAAiB,CAAC;IACjC,eAAe,CAAW;IAC1B,UAAU,GAA2B,EAAE,CAAC;IACxC,iBAAiB,CAAY;IAC7B,aAAa,GAA2B,EAAE,CAAC;IAC3C,iBAAiB,CAAY;IAC7B,aAAa,GAAmB,EAAE,CAAC;IAC1B,mBAAmB,GAAG,IAAI,OAAO,EAG/C,CAAC;IACI,MAAM,GAAG,IAAI,oBAAoB,EAA2B,CAAC;IACrE,YAAY,OAA4D;QACtE,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC;QACzC,IAAI,CAAC,aAAa,GAAG,kBAAkB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAC/D,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAC;QACvD,IAAI,CAAC,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,KAAK,CAAC;QACpD,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe;YAC5C,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;YAC9B,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3B,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC7C,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,YAAY,IAAI,eAAe,CAAC;QACjE,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,YAAY,IAAI,eAAe,CAAC;IACnE,CAAC;IACO,KAAK,CAAC,OAAO,CACnB,KAAoD,EACpD,MAAoB;QAEpB,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IACO,KAAK,CAAC,OAAO,CACnB,KAAiD,EACjD,MAAoB;QAEpB,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IACO,KAAK,CAAC,QAAQ,CACpB,KAAqD;QAErD,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAEO,KAAK,CAAC,yBAAyB,CACrC,KAAe,EACf,cAAsB,EACtB,aAAwC;QAExC,OAAO,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAC1C,KAAK,EACL,cAAc,EACd,aAAa,CACd,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,yBAAyB,CACrC,KAAe,EACf,OAAgB;QAEhB,OAAO,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAC/D,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,MAAM,IAAI,CAAC,OAAO,CAAC;YACjB,IAAI,EAAE,cAAc;YACpB,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;YACpD,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;YAC1D,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC;SAClC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe;QACrB,IAAI,MAAM,GAAG,GAAG,EAAE,GAAE,CAAC,CAAC;QACtB,IAAI,CAAC,UAAU,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAC9C,MAAM,GAAG,OAAO,CAAC;QACnB,CAAC,CAAC,CAAC;QACH,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;YAC5B,MAAM,EAAE,CAAC;QACX,CAAC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,eAAe;QAG3B,OAAO,2BAA2B,CAAC;YACjC,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,SAAS,EAAE,IAAI,CAAC,YAAY,EAAE;YAC9B,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CACnB,SAAgE,EAChE,YAAqB;QAErB,OAAO;YACL,YAAY,EAAE,YAAY,IAAI,SAAS,CAAC,YAAY;YACpD,QAAQ,EAAE,SAAS,CAAC,QAAQ,CAAC,KAAK,EAAE;YACpC,KAAK,EAAE,SAAS,CAAC,WAAW,CAAC,KAAK,EAAE;SACrC,CAAC;IACJ,CAAC;IAEO,cAAc,CACpB,YAAyE;QAEzE,OAAO,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE;YAC7C,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,CAAC;YACrD,MAAM,eAAe,GAA8B;gBACjD,GAAG,SAAS,CAAC,aAAa;gBAC1B,OAAO,EAAE,YAAY,CAAC,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC;aACtE,CAAC;YACF,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,yBAAyB,CACzD,KAAK,EACL,SAAS,CAAC,cAAc,EACxB,eAAe,CAChB,CAAC;YACF,OAAO,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE;gBAClC,cAAc,EAAE,cAAc,CAAC,cAAc;gBAC7C,OAAO,EAAE,cAAc,CAAC,OAAO;gBAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;gBACrC,eAAe,EAAE,cAAc,CAAC,eAAe;gBAC/C,QAAQ,EAAE,cAAc,CAAC,QAAQ;gBACjC,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAC3B,MAAM,IAAI,CAAC,yBAAyB,CAAC,KAAK,EAAE,OAAO,CAAC;gBACtD,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;oBAC7B,MAAM,OAAO,GAAG,EAAE,GAAI,QAAQ,CAAC,OAAkC,EAAE,CAAC;oBACpE,MAAM,IAAI,CAAC,OAAO,CAChB;wBACE,IAAI,EAAE,yBAAyB;wBAC/B,MAAM,EAAE,QAAQ,CAAC,MAAM;wBACvB,OAAO;qBACR,EACD,aAAa,EAAE,MAAM,CACtB,CAAC;gBACJ,CAAC;gBACD,SAAS,EAAE,aAAa,EAAE,SAAS;gBACnC,MAAM,EAAE,aAAa,EAAE,MAAM;gBAC7B,SAAS,EAAE,SAAS,CAAC,cAAc;gBACnC,SAAS,EAAE,cAAc,CAAC,SAAS;gBACnC,SAAS,EAAE,cAAc,CAAC,SAAS;gBACnC,MAAM,EAAE,IAAI,EAAE,MAAM;aACrB,CAAC,CAAC;QACL,CAAC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,KAA6B,EAC7B,IAAe;QAEf,MAAM,OAAO,GAAG,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACtE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAC7B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;oBAChC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE;wBAC1C,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,SAAS,EAAE,KAAK,CAAC,SAAS;qBAC3B,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC;YAC1B,MAAM,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAEO,gBAAgB,CACtB,YAAyE,EACzE,YAES;QAET,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QACjC,OAAO;YACL,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,SAAS,EACP,SAAS,CAAC,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa;YACzE,YAAY;YACZ,gBAAgB,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;gBACnC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC;oBACjC,IAAI,EAAE,SAAS;oBACf,QAAQ,EAAE,CAAC,GAAG,QAAQ,CAAC;iBACxB,CAAC,CAAC;gBACH,OAAO,MAAM,EAAE,QAAQ,IAAI,QAAQ,CAAC;YACtC,CAAC;YACD,cAAc,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE;gBAC3C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC;oBACjC,IAAI,EAAE,WAAW;oBACjB,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,KAAK,EAAE,IAA+B;iBACvC,CAAC,CAAC;gBACH,OAAO,MAAM;oBACX,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE;oBAChD,CAAC,CAAC,SAAS,CAAC;YAChB,CAAC;YACD,aAAa,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE;gBAC3D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC;oBAChC,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,KAAK,EAAE,IAA+B;oBACtC,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,OAAO;iBACR,CAAC,CAAC;gBACH,OAAO,KAAK;oBACV,CAAC,CAAC;wBACE,OAAO,EAAE,KAAK,CAAC,OAAO;wBACtB,OAAO,EAAE,KAAK,CAAC,OAAO;wBACtB,OAAO,EAAE,KAAK,CAAC,OAAO;wBACtB,SAAS,EAAE,KAAK,CAAC,SAAS;qBAC3B;oBACH,CAAC,CAAC,SAAS,CAAC;YAChB,CAAC;YACD,eAAe,EAAE,KAAK,IAAI,EAAE;gBAC1B,MAAM,IAAI,CAAC,8BAA8B,EAAE,CAAC;gBAC5C,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;gBACnD,YAAY,CAAC,aAAa,CAAC,CAAC;gBAC5B,OAAO;oBACL,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC;oBAC1C,KAAK,EAAE,aAAa,CAAC,KAAK;oBAC1B,aAAa,EAAE,aAAa,CAAC,aAAa;iBAC3C,CAAC;YACJ,CAAC;YACD,mBAAmB,EAAE,KAAK,IAAI,EAAE,CAC9B,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,iBAAiB,CAAC;YACnE,mBAAmB,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;SACpC,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,SAAmB,EACnB,QAA4B,IAAI,CAAC,KAAK;QAEtC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,8BAA8B;QAC1C,MAAM,qCAAqC,CACzC,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,yBAAyB,CAC/B,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,KAAiB,EACjB,MAAoB;QAEpB,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACjC,MAAM,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAChE,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC/C,IAAI,WAAW,EAAE,EAAE,EAAE,CAAC;gBACpB,MAAM,IAAI,CAAC,YAAY,CAAC,mBAAmB,CACzC,WAAW,CAAC,EAAE,EACd,KAAK,CAAC,OAAO,EACb,WAAW,CAAC,SAAS,CACtB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACvD,CAAC;YACD,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAClC,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC9B,IAAI,UAAmB,CAAC;YACxB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACpC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,UAAU,GAAG,KAAK,CAAC;YACrB,CAAC;YACD,MAAM,mBAAmB,GAAG,IAAI,CAAC,yBAAyB,CAAC,MAAM,GAAG,CAAC,CAAC;YACtE,MAAM,IAAI,CAAC,8BAA8B,EAAE,CAAC;YAC5C,IAAI,UAAU;gBAAE,MAAM,UAAU,CAAC;YACjC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,mBAAmB,EAAE,CAAC,CAAC;YAChE,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/B,MAAM,IAAI,CAAC,8BAA8B,EAAE,CAAC;YAC5C,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;YACpB,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAClC,MAAM,IAAI,CAAC,OAAO,CAChB,EAAE,IAAI,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAC7D,MAAM,CACP,CAAC;YACF,OAAO;QACT,CAAC;QACD,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,KAAe,EACf,KAAc,EACd,OAAgB,EAChB,MAAmB;QAEnB,MAAM,cAAc,GAAG,oBAAoB,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACnE,MAAM,IAAI,CAAC,gBAAgB,CACzB,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,cAAc,EAAE,EAClD,MAAM,CACP,CAAC;QACF,MAAM,IAAI,CAAC,gBAAgB,CACzB,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,EAChD,MAAM,CACP,CAAC;QACF,MAAM,IAAI,CAAC,gBAAgB,CACzB,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,EAAE,EAAE,EAC9D,MAAM,CACP,CAAC;QACF,MAAM,IAAI,CAAC,gBAAgB,CACzB,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,cAAc,CAAC,EAAE,EACjD,MAAM,CACP,CAAC;QACF,OAAO,CAAC,cAAc,CAAC,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,SAAgE,EAChE,IAAY,EACZ,OAAqC;QAErC,IAAI,eAAe,GAAG,SAAS,CAAC;QAChC,MAAM,aAAa,GAAG,iBAAiB,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAC/D,IAAI,QAAQ,GAAmB,CAAC,aAAa,CAAC,CAAC;QAC/C,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClC,MAAM,cAAc,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACpD,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAC/B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,cAAc,CAAC,CAAC;gBAC9C,MAAM,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAClC,CAAC;YACD,QAAQ,GAAG,CAAC,GAAG,cAAc,EAAE,aAAa,CAAC,CAAC;QAChD,CAAC;QACD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC;YACvC,IAAI,EAAE,oBAAoB;YAC1B,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,OAAO,EAAE,MAAM;YACvB,YAAY,EAAE,SAAS,CAAC,YAAY;YACpC,SAAS,EAAE,SAAS,CAAC,SAAS;SAC/B,CAAC,CAAC;QACH,IAAI,YAAY,EAAE,QAAQ;YACxB,QAAQ,GAAG,CAAC,GAAG,QAAQ,EAAE,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QAErD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,eAAe,CAAC;QAC3C,MAAM,YAAY,GAAG,CACnB,aAAoE,EACpE,EAAE;YACF,eAAe,GAAG,aAAa,CAAC;QAClC,CAAC,CAAC;QACF,IAAI,CAAC,kBAAkB,GAAG,eAAe,CAAC;QAC1C,MAAM,gBAAgB,GAAG,CAAC,KAAK,IAAI,EAAE;YACnC,IAAI,CAAC;gBACH,OAAO,MAAM,YAAY,CACvB,QAAQ,EACR,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,YAAY,EAAE,YAAY,CAAC,EACzD,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,YAAY,CAAC,EACjD,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,eAAe,CAAC,MAAM,CAAC,EAC/D,eAAe,CAAC,MAAM,EACtB,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAClC,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjC,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;oBACpB,MAAM,KAAK,CAAC;gBACd,CAAC;gBACD,IAAI,CAAC;oBACH,OAAO,MAAM,IAAI,CAAC,cAAc,CAC9B,eAAe,CAAC,KAAK,EACrB,KAAK,EACL,eAAe,CAAC,MAAM,CAAC,OAAO,EAC9B,eAAe,CAAC,MAAM,CACvB,CAAC;gBACJ,CAAC;gBAAC,OAAO,YAAY,EAAE,CAAC;oBACtB,MAAM,KAAK,GAAG,IAAI,cAAc,CAC9B,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC,EACvC,+CAA+C,CAChD,CAAC;oBACF,MAAM,IAAI,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QACL,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,gBAAgB,CAAC;YAC3C,KAAK,MAAM,OAAO,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;gBACjD,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACjC,OAAO,OAAO,CAAC;gBACjB,CAAC;YACH,CAAC;YACD,MAAM,IAAI,iBAAiB,CACzB,eAAe,EACf,4DAA4D,CAC7D,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,8BAA8B,EAAE,CAAC;YAC9C,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,kBAAkB,GAAG,SAAS,CAAC;YACtC,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAC7B,aAIoE;QAEpE,IAAI,IAAI,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,iBAAiB,CAAC,MAAM,EAAE,sBAAsB,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;QACpB,MAAM,gBAAgB,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QAChD,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,SAAS,CAAC,CAAC;YAC9C,OAAO,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACxE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;YACpB,MAAM,qBAAqB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAChD,CAAC;gBAAS,CAAC;YACT,gBAAgB,EAAE,CAAC;QACrB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,IAAY,EACZ,OAAqC;QAErC,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,kBAAkB,CACvB,IAIC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CACT,IAAY,EACZ,sBAA+B;QAE/B,OAAO,WAAW,CAChB,IAAyE,EACzE,IAAI,EACJ,sBAAsB,CACvB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,IAAY,EACZ,OAAiB,EAAE;QAEnB,OAAO,oBAAoB,CACzB,IAAyE,EACzE,IAAI,EACJ,IAAI,CACL,CAAC;IACJ,CAAC;IAEO,UAAU;QAChB,OAAO,IAAoC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,KAAK,CACT,IAAY,EACZ,OAAqC;QAErC,OAAO,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,IAAY,EACZ,OAAqC;QAErC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,KAM3B;QACC,OAAO,0BAA0B,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,IAAY,EACZ,OAAqC;QAErC,OAAO,eAAe,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAqB;QACvC,OAAO,oBAAoB,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,KAI3B;QACC,OAAO,4BAA4B,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,kBAA2B;QAMvC,OAAO,0BAA0B,CAC/B,IAAI,CAAC,kBAAkB,EAAE,EACzB,kBAAkB,CACnB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,QAAgB,EAChB,OAKC;QAED,OAAO,mBAAmB,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC3E,CAAC;IAEO,kBAAkB;QAKxB,OAAO;YACL,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK;YAC1B,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;gBAClB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;YACrB,CAAC;YACD,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK;YAC1B,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa;YAC1C,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;YAC7C,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAc,CAAC;YAClD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAc,CAAC;SACjD,CAAC;IACJ,CAAC;IAEO,kBAAkB;QAKxB,OAAO,IAIN,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAe;QAC5B,OAAO,eAAe,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,KAAK,CAAC,CAAC;IAC3D,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAoB;QACzC,OAAO,uBAAuB,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,KAAK,CAAC,CAAC;IACnE,CAAC;IAED,QAAQ;QACN,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAc,EAAE,eAA0B;QACvD,OAAO,eAAe,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC;IAC5E,CAAC;IAED,cAAc;QACZ,OAAO,qBAAqB,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,SAAmB;QACtC,OAAO,qBAAqB,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,SAAS,CAAC,CAAC;IACrE,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAAe;QACnC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAAe;QACnC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,YAAY;QACV,OAAO,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,SAAyD;QAEzD,OAAO,mBAAmB,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,SAAS,CAAC,CAAC;IACnE,CAAC;IAED,gBAAgB;QACd,OAAO,kBAAkB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,aAAwC;QAExC,IAAI,CAAC,aAAa,GAAG,kBAAkB,CAAC,aAAa,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,eAAe,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,IAAI,CAAC,UAAU,CAAC;IACxB,CAAC;IAED,SAAS,CACP,QAGyB;QAEzB,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED,EAAE,CACA,IAAW,EACX,OAIqC;QAErC,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;CACF"}
@@ -0,0 +1,54 @@
1
+ import type { AgentMessage, AnyModel } from "../../types.js";
2
+ import type { Conversation, ConversationTreeEntry } from "../conversation/entries.js";
3
+ import { BranchSummaryError } from "../errors.js";
4
+ import type { BranchSummaryResult } from "../events.js";
5
+ import { type Result } from "../result.js";
6
+ import { type FileOperations } from "./utils.js";
7
+ /** File-operation details stored on generated branch summary entries. */
8
+ export interface BranchSummaryDetails {
9
+ /** Files read while exploring the summarized branch. */
10
+ readFiles: string[];
11
+ /** Files modified while exploring the summarized branch. */
12
+ modifiedFiles: string[];
13
+ }
14
+ export type { FileOperations } from "./utils.js";
15
+ /** Prepared branch content for summarization. */
16
+ export interface BranchPreparation {
17
+ /** Messages selected for the branch summary. */
18
+ messages: AgentMessage[];
19
+ /** File operations extracted from the branch. */
20
+ fileOps: FileOperations;
21
+ /** Estimated token count for selected messages. */
22
+ totalTokens: number;
23
+ }
24
+ /** Entries selected for branch summarization. */
25
+ export interface CollectEntriesResult {
26
+ /** Entries to summarize in chronological order. */
27
+ entries: ConversationTreeEntry[];
28
+ /** Deepest common ancestor between the previous leaf and target entry. */
29
+ commonAncestorId: string | null;
30
+ }
31
+ /** Options for generating a branch summary. */
32
+ export interface GenerateBranchSummaryOptions {
33
+ /** Model used for summarization. */
34
+ model: AnyModel;
35
+ /** API key forwarded to the provider. */
36
+ apiKey: string;
37
+ /** Optional request headers forwarded to the provider. */
38
+ headers?: Record<string, string>;
39
+ /** Abort signal for the summarization request. */
40
+ signal: AbortSignal;
41
+ /** Optional instructions appended to or replacing the default prompt. */
42
+ customInstructions?: string;
43
+ /** Replace the default prompt with custom instructions instead of appending them. */
44
+ replaceInstructions?: boolean;
45
+ /** Tokens reserved for prompt and model output. Defaults to 16384. */
46
+ reserveTokens?: number;
47
+ }
48
+ /** Collect entries that should be summarized before navigating to a different conversation tree entry. */
49
+ export declare function collectEntriesForBranchSummary(conversation: Conversation, oldLeafId: string | null, targetId: string): Promise<CollectEntriesResult>;
50
+ /** Prepare branch entries for summarization within an optional token budget. */
51
+ export declare function prepareBranchEntries(entries: ConversationTreeEntry[], tokenBudget?: number): BranchPreparation;
52
+ /** Generate a summary for abandoned branch entries. */
53
+ export declare function generateBranchSummary(entries: ConversationTreeEntry[], options: GenerateBranchSummaryOptions): Promise<Result<BranchSummaryResult, BranchSummaryError>>;
54
+ //# sourceMappingURL=branch-summarization.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"branch-summarization.d.ts","sourceRoot":"","sources":["../../../src/harness/compaction/branch-summarization.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,KAAK,EACV,YAAY,EACZ,qBAAqB,EACtB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,kBAAkB,EAAqB,MAAM,cAAc,CAAC;AACrE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAOxD,OAAO,EAAW,KAAK,MAAM,EAAE,MAAM,cAAc,CAAC;AAEpD,OAAO,EAIL,KAAK,cAAc,EAGpB,MAAM,YAAY,CAAC;AAEpB,yEAAyE;AACzE,MAAM,WAAW,oBAAoB;IACnC,wDAAwD;IACxD,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,4DAA4D;IAC5D,aAAa,EAAE,MAAM,EAAE,CAAC;CACzB;AAED,YAAY,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,iDAAiD;AACjD,MAAM,WAAW,iBAAiB;IAChC,gDAAgD;IAChD,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,iDAAiD;IACjD,OAAO,EAAE,cAAc,CAAC;IACxB,mDAAmD;IACnD,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,iDAAiD;AACjD,MAAM,WAAW,oBAAoB;IACnC,mDAAmD;IACnD,OAAO,EAAE,qBAAqB,EAAE,CAAC;IACjC,0EAA0E;IAC1E,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,+CAA+C;AAC/C,MAAM,WAAW,4BAA4B;IAC3C,oCAAoC;IACpC,KAAK,EAAE,QAAQ,CAAC;IAChB,yCAAyC;IACzC,MAAM,EAAE,MAAM,CAAC;IACf,0DAA0D;IAC1D,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,kDAAkD;IAClD,MAAM,EAAE,WAAW,CAAC;IACpB,yEAAyE;IACzE,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,qFAAqF;IACrF,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,sEAAsE;IACtE,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,0GAA0G;AAC1G,wBAAsB,8BAA8B,CAClD,YAAY,EAAE,YAAY,EAC1B,SAAS,EAAE,MAAM,GAAG,IAAI,EACxB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,oBAAoB,CAAC,CA+B/B;AA0CD,gFAAgF;AAChF,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,qBAAqB,EAAE,EAChC,WAAW,GAAE,MAAU,GACtB,iBAAiB,CAuCnB;AAoCD,uDAAuD;AACvD,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,qBAAqB,EAAE,EAChC,OAAO,EAAE,4BAA4B,GACpC,OAAO,CAAC,MAAM,CAAC,mBAAmB,EAAE,kBAAkB,CAAC,CAAC,CA+E1D"}