@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.
- package/LICENSE +201 -0
- package/NOTICE +5 -0
- package/dist/agent-loop.d.ts +2 -0
- package/dist/agent-loop.d.ts.map +1 -0
- package/dist/agent-loop.js +2 -0
- package/dist/agent-loop.js.map +1 -0
- package/dist/agent.d.ts +125 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +427 -0
- package/dist/agent.js.map +1 -0
- package/dist/harness/agent-harness.d.ts +114 -0
- package/dist/harness/agent-harness.d.ts.map +1 -0
- package/dist/harness/agent-harness.js +474 -0
- package/dist/harness/agent-harness.js.map +1 -0
- package/dist/harness/compaction/branch-summarization.d.ts +54 -0
- package/dist/harness/compaction/branch-summarization.d.ts.map +1 -0
- package/dist/harness/compaction/branch-summarization.js +183 -0
- package/dist/harness/compaction/branch-summarization.js.map +1 -0
- package/dist/harness/compaction/compaction.d.ts +19 -0
- package/dist/harness/compaction/compaction.d.ts.map +1 -0
- package/dist/harness/compaction/compaction.js +335 -0
- package/dist/harness/compaction/compaction.js.map +1 -0
- package/dist/harness/compaction/cut-points.d.ts +7 -0
- package/dist/harness/compaction/cut-points.d.ts.map +1 -0
- package/dist/harness/compaction/cut-points.js +106 -0
- package/dist/harness/compaction/cut-points.js.map +1 -0
- package/dist/harness/compaction/overflow.d.ts +3 -0
- package/dist/harness/compaction/overflow.d.ts.map +1 -0
- package/dist/harness/compaction/overflow.js +54 -0
- package/dist/harness/compaction/overflow.js.map +1 -0
- package/dist/harness/compaction/policy.d.ts +8 -0
- package/dist/harness/compaction/policy.d.ts.map +1 -0
- package/dist/harness/compaction/policy.js +57 -0
- package/dist/harness/compaction/policy.js.map +1 -0
- package/dist/harness/compaction/types.d.ts +79 -0
- package/dist/harness/compaction/types.d.ts.map +1 -0
- package/dist/harness/compaction/types.js +2 -0
- package/dist/harness/compaction/types.js.map +1 -0
- package/dist/harness/compaction/usage.d.ts +30 -0
- package/dist/harness/compaction/usage.d.ts.map +1 -0
- package/dist/harness/compaction/usage.js +177 -0
- package/dist/harness/compaction/usage.js.map +1 -0
- package/dist/harness/compaction/utils.d.ts +25 -0
- package/dist/harness/compaction/utils.d.ts.map +1 -0
- package/dist/harness/compaction/utils.js +131 -0
- package/dist/harness/compaction/utils.js.map +1 -0
- package/dist/harness/configuration.d.ts +34 -0
- package/dist/harness/configuration.d.ts.map +1 -0
- package/dist/harness/configuration.js +75 -0
- package/dist/harness/configuration.js.map +1 -0
- package/dist/harness/conversation/context.d.ts +24 -0
- package/dist/harness/conversation/context.d.ts.map +1 -0
- package/dist/harness/conversation/context.js +76 -0
- package/dist/harness/conversation/context.js.map +1 -0
- package/dist/harness/conversation/conversation.d.ts +36 -0
- package/dist/harness/conversation/conversation.d.ts.map +1 -0
- package/dist/harness/conversation/conversation.js +163 -0
- package/dist/harness/conversation/conversation.js.map +1 -0
- package/dist/harness/conversation/entries.d.ts +117 -0
- package/dist/harness/conversation/entries.d.ts.map +1 -0
- package/dist/harness/conversation/entries.js +2 -0
- package/dist/harness/conversation/entries.js.map +1 -0
- package/dist/harness/conversation/jsonl-repo.d.ts +27 -0
- package/dist/harness/conversation/jsonl-repo.d.ts.map +1 -0
- package/dist/harness/conversation/jsonl-repo.js +110 -0
- package/dist/harness/conversation/jsonl-repo.js.map +1 -0
- package/dist/harness/conversation/jsonl-storage.d.ts +34 -0
- package/dist/harness/conversation/jsonl-storage.d.ts.map +1 -0
- package/dist/harness/conversation/jsonl-storage.js +213 -0
- package/dist/harness/conversation/jsonl-storage.js.map +1 -0
- package/dist/harness/conversation/memory-repo.d.ts +18 -0
- package/dist/harness/conversation/memory-repo.d.ts.map +1 -0
- package/dist/harness/conversation/memory-repo.js +45 -0
- package/dist/harness/conversation/memory-repo.js.map +1 -0
- package/dist/harness/conversation/memory-storage.d.ts +25 -0
- package/dist/harness/conversation/memory-storage.d.ts.map +1 -0
- package/dist/harness/conversation/memory-storage.js +88 -0
- package/dist/harness/conversation/memory-storage.js.map +1 -0
- package/dist/harness/conversation/repo-utils.d.ts +13 -0
- package/dist/harness/conversation/repo-utils.d.ts.map +1 -0
- package/dist/harness/conversation/repo-utils.js +39 -0
- package/dist/harness/conversation/repo-utils.js.map +1 -0
- package/dist/harness/conversation/storage-utils.d.ts +11 -0
- package/dist/harness/conversation/storage-utils.d.ts.map +1 -0
- package/dist/harness/conversation/storage-utils.js +32 -0
- package/dist/harness/conversation/storage-utils.js.map +1 -0
- package/dist/harness/conversation/text-extraction.d.ts +8 -0
- package/dist/harness/conversation/text-extraction.d.ts.map +1 -0
- package/dist/harness/conversation/text-extraction.js +24 -0
- package/dist/harness/conversation/text-extraction.js.map +1 -0
- package/dist/harness/conversation/uuid.d.ts +2 -0
- package/dist/harness/conversation/uuid.d.ts.map +1 -0
- package/dist/harness/conversation/uuid.js +54 -0
- package/dist/harness/conversation/uuid.js.map +1 -0
- package/dist/harness/conversation-writes.d.ts +9 -0
- package/dist/harness/conversation-writes.d.ts.map +1 -0
- package/dist/harness/conversation-writes.js +54 -0
- package/dist/harness/conversation-writes.js.map +1 -0
- package/dist/harness/env/nodejs.d.ts +53 -0
- package/dist/harness/env/nodejs.d.ts.map +1 -0
- package/dist/harness/env/nodejs.js +497 -0
- package/dist/harness/env/nodejs.js.map +1 -0
- package/dist/harness/env/types.d.ts +106 -0
- package/dist/harness/env/types.d.ts.map +1 -0
- package/dist/harness/env/types.js +2 -0
- package/dist/harness/env/types.js.map +1 -0
- package/dist/harness/errors.d.ts +48 -0
- package/dist/harness/errors.d.ts.map +1 -0
- package/dist/harness/errors.js +63 -0
- package/dist/harness/errors.js.map +1 -0
- package/dist/harness/events.d.ts +238 -0
- package/dist/harness/events.d.ts.map +1 -0
- package/dist/harness/events.js +2 -0
- package/dist/harness/events.js.map +1 -0
- package/dist/harness/harness-configuration-methods.d.ts +36 -0
- package/dist/harness/harness-configuration-methods.d.ts.map +1 -0
- package/dist/harness/harness-configuration-methods.js +145 -0
- package/dist/harness/harness-configuration-methods.js.map +1 -0
- package/dist/harness/harness-continuation.d.ts +18 -0
- package/dist/harness/harness-continuation.d.ts.map +1 -0
- package/dist/harness/harness-continuation.js +56 -0
- package/dist/harness/harness-continuation.js.map +1 -0
- package/dist/harness/harness-events.d.ts +23 -0
- package/dist/harness/harness-events.d.ts.map +1 -0
- package/dist/harness/harness-events.js +129 -0
- package/dist/harness/harness-events.js.map +1 -0
- package/dist/harness/harness-invocations.d.ts +20 -0
- package/dist/harness/harness-invocations.d.ts.map +1 -0
- package/dist/harness/harness-invocations.js +22 -0
- package/dist/harness/harness-invocations.js.map +1 -0
- package/dist/harness/harness-maintenance.d.ts +30 -0
- package/dist/harness/harness-maintenance.d.ts.map +1 -0
- package/dist/harness/harness-maintenance.js +155 -0
- package/dist/harness/harness-maintenance.js.map +1 -0
- package/dist/harness/harness-queue-methods.d.ts +45 -0
- package/dist/harness/harness-queue-methods.d.ts.map +1 -0
- package/dist/harness/harness-queue-methods.js +51 -0
- package/dist/harness/harness-queue-methods.js.map +1 -0
- package/dist/harness/messages.d.ts +76 -0
- package/dist/harness/messages.d.ts.map +1 -0
- package/dist/harness/messages.js +147 -0
- package/dist/harness/messages.js.map +1 -0
- package/dist/harness/options.d.ts +95 -0
- package/dist/harness/options.d.ts.map +1 -0
- package/dist/harness/options.js +2 -0
- package/dist/harness/options.js.map +1 -0
- package/dist/harness/prompt-templates.d.ts +49 -0
- package/dist/harness/prompt-templates.d.ts.map +1 -0
- package/dist/harness/prompt-templates.js +230 -0
- package/dist/harness/prompt-templates.js.map +1 -0
- package/dist/harness/result.d.ts +19 -0
- package/dist/harness/result.d.ts.map +1 -0
- package/dist/harness/result.js +32 -0
- package/dist/harness/result.js.map +1 -0
- package/dist/harness/run/messages.d.ts +5 -0
- package/dist/harness/run/messages.d.ts.map +1 -0
- package/dist/harness/run/messages.js +37 -0
- package/dist/harness/run/messages.js.map +1 -0
- package/dist/harness/skills/format.d.ts +6 -0
- package/dist/harness/skills/format.d.ts.map +1 -0
- package/dist/harness/skills/format.js +38 -0
- package/dist/harness/skills/format.js.map +1 -0
- package/dist/harness/skills/index.d.ts +5 -0
- package/dist/harness/skills/index.d.ts.map +1 -0
- package/dist/harness/skills/index.js +5 -0
- package/dist/harness/skills/index.js.map +1 -0
- package/dist/harness/skills/loader.d.ts +43 -0
- package/dist/harness/skills/loader.d.ts.map +1 -0
- package/dist/harness/skills/loader.js +279 -0
- package/dist/harness/skills/loader.js.map +1 -0
- package/dist/harness/skills/parser.d.ts +12 -0
- package/dist/harness/skills/parser.d.ts.map +1 -0
- package/dist/harness/skills/parser.js +22 -0
- package/dist/harness/skills/parser.js.map +1 -0
- package/dist/harness/skills/validation.d.ts +3 -0
- package/dist/harness/skills/validation.d.ts.map +1 -0
- package/dist/harness/skills/validation.js +28 -0
- package/dist/harness/skills/validation.js.map +1 -0
- package/dist/harness/skills.d.ts +2 -0
- package/dist/harness/skills.d.ts.map +1 -0
- package/dist/harness/skills.js +2 -0
- package/dist/harness/skills.js.map +1 -0
- package/dist/harness/stream-options.d.ts +5 -0
- package/dist/harness/stream-options.d.ts.map +1 -0
- package/dist/harness/stream-options.js +67 -0
- package/dist/harness/stream-options.js.map +1 -0
- package/dist/harness/system-prompt.d.ts +2 -0
- package/dist/harness/system-prompt.d.ts.map +1 -0
- package/dist/harness/system-prompt.js +2 -0
- package/dist/harness/system-prompt.js.map +1 -0
- package/dist/harness/turn-state.d.ts +27 -0
- package/dist/harness/turn-state.d.ts.map +1 -0
- package/dist/harness/turn-state.js +35 -0
- package/dist/harness/turn-state.js.map +1 -0
- package/dist/harness/types.d.ts +9 -0
- package/dist/harness/types.d.ts.map +1 -0
- package/dist/harness/types.js +8 -0
- package/dist/harness/types.js.map +1 -0
- package/dist/harness/utils/env-path.d.ts +14 -0
- package/dist/harness/utils/env-path.d.ts.map +1 -0
- package/dist/harness/utils/env-path.js +99 -0
- package/dist/harness/utils/env-path.js.map +1 -0
- package/dist/harness/utils/shell-output.d.ts +16 -0
- package/dist/harness/utils/shell-output.d.ts.map +1 -0
- package/dist/harness/utils/shell-output.js +137 -0
- package/dist/harness/utils/shell-output.js.map +1 -0
- package/dist/harness/utils/truncate.d.ts +70 -0
- package/dist/harness/utils/truncate.d.ts.map +1 -0
- package/dist/harness/utils/truncate.js +289 -0
- package/dist/harness/utils/truncate.js.map +1 -0
- package/dist/index.d.ts +24 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +25 -0
- package/dist/index.js.map +1 -0
- package/dist/node.d.ts +3 -0
- package/dist/node.d.ts.map +1 -0
- package/dist/node.js +3 -0
- package/dist/node.js.map +1 -0
- package/dist/proxy.d.ts +67 -0
- package/dist/proxy.d.ts.map +1 -0
- package/dist/proxy.js +299 -0
- package/dist/proxy.js.map +1 -0
- package/dist/runtime/loop/agent-loop.d.ts +24 -0
- package/dist/runtime/loop/agent-loop.d.ts.map +1 -0
- package/dist/runtime/loop/agent-loop.js +72 -0
- package/dist/runtime/loop/agent-loop.js.map +1 -0
- package/dist/runtime/loop/events.d.ts +5 -0
- package/dist/runtime/loop/events.d.ts.map +1 -0
- package/dist/runtime/loop/events.js +5 -0
- package/dist/runtime/loop/events.js.map +1 -0
- package/dist/runtime/loop/index.d.ts +6 -0
- package/dist/runtime/loop/index.d.ts.map +1 -0
- package/dist/runtime/loop/index.js +6 -0
- package/dist/runtime/loop/index.js.map +1 -0
- package/dist/runtime/loop/stream-assistant.d.ts +9 -0
- package/dist/runtime/loop/stream-assistant.d.ts.map +1 -0
- package/dist/runtime/loop/stream-assistant.js +87 -0
- package/dist/runtime/loop/stream-assistant.js.map +1 -0
- package/dist/runtime/loop/tool-execution.d.ts +15 -0
- package/dist/runtime/loop/tool-execution.d.ts.map +1 -0
- package/dist/runtime/loop/tool-execution.js +271 -0
- package/dist/runtime/loop/tool-execution.js.map +1 -0
- package/dist/runtime/loop/turn-loop.d.ts +7 -0
- package/dist/runtime/loop/turn-loop.d.ts.map +1 -0
- package/dist/runtime/loop/turn-loop.js +84 -0
- package/dist/runtime/loop/turn-loop.js.map +1 -0
- package/dist/runtime.d.ts +63 -0
- package/dist/runtime.d.ts.map +1 -0
- package/dist/runtime.js +170 -0
- package/dist/runtime.js.map +1 -0
- package/dist/suspension.d.ts +18 -0
- package/dist/suspension.d.ts.map +1 -0
- package/dist/suspension.js +19 -0
- package/dist/suspension.js.map +1 -0
- package/dist/types.d.ts +395 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/worker.d.ts +2 -0
- package/dist/worker.d.ts.map +1 -0
- package/dist/worker.js +88 -0
- package/dist/worker.js.map +1 -0
- 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"}
|