@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,131 @@
|
|
|
1
|
+
/** Create an empty file-operation accumulator. */
|
|
2
|
+
export function createFileOps() {
|
|
3
|
+
return {
|
|
4
|
+
read: new Set(),
|
|
5
|
+
written: new Set(),
|
|
6
|
+
edited: new Set(),
|
|
7
|
+
};
|
|
8
|
+
}
|
|
9
|
+
/** Add file operations from assistant tool calls to an accumulator. */
|
|
10
|
+
export function extractFileOpsFromMessage(message, fileOps) {
|
|
11
|
+
if (message.role !== "assistant")
|
|
12
|
+
return;
|
|
13
|
+
if (!("content" in message) || !Array.isArray(message.content))
|
|
14
|
+
return;
|
|
15
|
+
for (const block of message.content) {
|
|
16
|
+
if (typeof block !== "object" || block === null)
|
|
17
|
+
continue;
|
|
18
|
+
if (!("type" in block) || block.type !== "toolCall")
|
|
19
|
+
continue;
|
|
20
|
+
if (!("arguments" in block) || !("name" in block))
|
|
21
|
+
continue;
|
|
22
|
+
const args = block.arguments;
|
|
23
|
+
if (!args)
|
|
24
|
+
continue;
|
|
25
|
+
const path = typeof args.path === "string" ? args.path : undefined;
|
|
26
|
+
if (!path)
|
|
27
|
+
continue;
|
|
28
|
+
switch (block.name) {
|
|
29
|
+
case "read":
|
|
30
|
+
fileOps.read.add(path);
|
|
31
|
+
break;
|
|
32
|
+
case "write":
|
|
33
|
+
fileOps.written.add(path);
|
|
34
|
+
break;
|
|
35
|
+
case "edit":
|
|
36
|
+
fileOps.edited.add(path);
|
|
37
|
+
break;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
/** Compute sorted read-only and modified file lists from accumulated operations. */
|
|
42
|
+
export function computeFileLists(fileOps) {
|
|
43
|
+
const modified = new Set([...fileOps.edited, ...fileOps.written]);
|
|
44
|
+
const readOnly = [...fileOps.read].filter((f) => !modified.has(f)).sort();
|
|
45
|
+
const modifiedFiles = [...modified].sort();
|
|
46
|
+
return { readFiles: readOnly, modifiedFiles };
|
|
47
|
+
}
|
|
48
|
+
/** Format file lists as summary metadata tags. */
|
|
49
|
+
export function formatFileOperations(readFiles, modifiedFiles) {
|
|
50
|
+
const sections = [];
|
|
51
|
+
if (readFiles.length > 0) {
|
|
52
|
+
sections.push(`<read-files>\n${readFiles.join("\n")}\n</read-files>`);
|
|
53
|
+
}
|
|
54
|
+
if (modifiedFiles.length > 0) {
|
|
55
|
+
sections.push(`<modified-files>\n${modifiedFiles.join("\n")}\n</modified-files>`);
|
|
56
|
+
}
|
|
57
|
+
if (sections.length === 0)
|
|
58
|
+
return "";
|
|
59
|
+
return `\n\n${sections.join("\n\n")}`;
|
|
60
|
+
}
|
|
61
|
+
const TOOL_RESULT_MAX_CHARS = 2000;
|
|
62
|
+
function safeJsonStringify(value) {
|
|
63
|
+
try {
|
|
64
|
+
return JSON.stringify(value) ?? "undefined";
|
|
65
|
+
}
|
|
66
|
+
catch {
|
|
67
|
+
return "[unserializable]";
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
function truncateForSummary(text, maxChars) {
|
|
71
|
+
if (text.length <= maxChars)
|
|
72
|
+
return text;
|
|
73
|
+
const truncatedChars = text.length - maxChars;
|
|
74
|
+
return `${text.slice(0, maxChars)}\n\n[... ${truncatedChars} more characters truncated]`;
|
|
75
|
+
}
|
|
76
|
+
/** Serialize LLM messages to plain text for summarization prompts. */
|
|
77
|
+
export function serializeConversation(messages) {
|
|
78
|
+
const parts = [];
|
|
79
|
+
for (const msg of messages) {
|
|
80
|
+
if (msg.role === "user") {
|
|
81
|
+
const content = typeof msg.content === "string"
|
|
82
|
+
? msg.content
|
|
83
|
+
: msg.content
|
|
84
|
+
.filter((c) => c.type === "text")
|
|
85
|
+
.map((c) => c.text)
|
|
86
|
+
.join("");
|
|
87
|
+
if (content)
|
|
88
|
+
parts.push(`[User]: ${content}`);
|
|
89
|
+
}
|
|
90
|
+
else if (msg.role === "assistant") {
|
|
91
|
+
const textParts = [];
|
|
92
|
+
const thinkingParts = [];
|
|
93
|
+
const toolCalls = [];
|
|
94
|
+
for (const block of msg.content) {
|
|
95
|
+
if (block.type === "text") {
|
|
96
|
+
textParts.push(block.text);
|
|
97
|
+
}
|
|
98
|
+
else if (block.type === "thinking") {
|
|
99
|
+
thinkingParts.push(block.thinking);
|
|
100
|
+
}
|
|
101
|
+
else if (block.type === "toolCall") {
|
|
102
|
+
const args = block.arguments;
|
|
103
|
+
const argsStr = Object.entries(args)
|
|
104
|
+
.map(([k, v]) => `${k}=${safeJsonStringify(v)}`)
|
|
105
|
+
.join(", ");
|
|
106
|
+
toolCalls.push(`${block.name}(${argsStr})`);
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
if (thinkingParts.length > 0) {
|
|
110
|
+
parts.push(`[Assistant thinking]: ${thinkingParts.join("\n")}`);
|
|
111
|
+
}
|
|
112
|
+
if (textParts.length > 0) {
|
|
113
|
+
parts.push(`[Assistant]: ${textParts.join("\n")}`);
|
|
114
|
+
}
|
|
115
|
+
if (toolCalls.length > 0) {
|
|
116
|
+
parts.push(`[Assistant tool calls]: ${toolCalls.join("; ")}`);
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
else if (msg.role === "toolResult") {
|
|
120
|
+
const content = msg.content
|
|
121
|
+
.filter((c) => c.type === "text")
|
|
122
|
+
.map((c) => c.text)
|
|
123
|
+
.join("");
|
|
124
|
+
if (content) {
|
|
125
|
+
parts.push(`[Tool result]: ${truncateForSummary(content, TOOL_RESULT_MAX_CHARS)}`);
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
}
|
|
129
|
+
return parts.join("\n\n");
|
|
130
|
+
}
|
|
131
|
+
//# sourceMappingURL=utils.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/harness/compaction/utils.ts"],"names":[],"mappings":"AAaA,kDAAkD;AAClD,MAAM,UAAU,aAAa;IAC3B,OAAO;QACL,IAAI,EAAE,IAAI,GAAG,EAAE;QACf,OAAO,EAAE,IAAI,GAAG,EAAE;QAClB,MAAM,EAAE,IAAI,GAAG,EAAE;KAClB,CAAC;AACJ,CAAC;AAED,uEAAuE;AACvE,MAAM,UAAU,yBAAyB,CACvC,OAAqB,EACrB,OAAuB;IAEvB,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;QAAE,OAAO;IACzC,IAAI,CAAC,CAAC,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO;IAEvE,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI;YAAE,SAAS;QAC1D,IAAI,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;YAAE,SAAS;QAC9D,IAAI,CAAC,CAAC,WAAW,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC;YAAE,SAAS;QAE5D,MAAM,IAAI,GAAG,KAAK,CAAC,SAAgD,CAAC;QACpE,IAAI,CAAC,IAAI;YAAE,SAAS;QAEpB,MAAM,IAAI,GAAG,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QACnE,IAAI,CAAC,IAAI;YAAE,SAAS;QAEpB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,MAAM;gBACT,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACvB,MAAM;YACR,KAAK,OAAO;gBACV,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC1B,MAAM;YACR,KAAK,MAAM;gBACT,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACzB,MAAM;QACV,CAAC;IACH,CAAC;AACH,CAAC;AAED,oFAAoF;AACpF,MAAM,UAAU,gBAAgB,CAAC,OAAuB;IAItD,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAClE,MAAM,QAAQ,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC1E,MAAM,aAAa,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;IAC3C,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC;AAChD,CAAC;AAED,kDAAkD;AAClD,MAAM,UAAU,oBAAoB,CAClC,SAAmB,EACnB,aAAuB;IAEvB,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,QAAQ,CAAC,IAAI,CAAC,iBAAiB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACxE,CAAC;IACD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,QAAQ,CAAC,IAAI,CACX,qBAAqB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,qBAAqB,CACnE,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IACrC,OAAO,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;AACxC,CAAC;AAED,MAAM,qBAAqB,GAAG,IAAI,CAAC;AAEnC,SAAS,iBAAiB,CAAC,KAAc;IACvC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,WAAW,CAAC;IAC9C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,kBAAkB,CAAC;IAC5B,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY,EAAE,QAAgB;IACxD,IAAI,IAAI,CAAC,MAAM,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IACzC,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;IAC9C,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,YAAY,cAAc,6BAA6B,CAAC;AAC3F,CAAC;AAED,sEAAsE;AACtE,MAAM,UAAU,qBAAqB,CAAC,QAAmB;IACvD,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,MAAM,OAAO,GACX,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ;gBAC7B,CAAC,CAAC,GAAG,CAAC,OAAO;gBACb,CAAC,CAAC,GAAG,CAAC,OAAO;qBACR,MAAM,CACL,CAAC,CAAC,EAAuC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAC9D;qBACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;qBAClB,IAAI,CAAC,EAAE,CAAC,CAAC;YAClB,IAAI,OAAO;gBAAE,KAAK,CAAC,IAAI,CAAC,WAAW,OAAO,EAAE,CAAC,CAAC;QAChD,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACpC,MAAM,SAAS,GAAa,EAAE,CAAC;YAC/B,MAAM,aAAa,GAAa,EAAE,CAAC;YACnC,MAAM,SAAS,GAAa,EAAE,CAAC;YAE/B,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC7B,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACrC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACrC,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACrC,MAAM,IAAI,GAAG,KAAK,CAAC,SAAoC,CAAC;oBACxD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;yBACjC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,iBAAiB,CAAC,CAAC,CAAC,EAAE,CAAC;yBAC/C,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,SAAS,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,IAAI,IAAI,OAAO,GAAG,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;YAED,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7B,KAAK,CAAC,IAAI,CAAC,yBAAyB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,KAAK,CAAC,IAAI,CAAC,gBAAgB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACrD,CAAC;YACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,KAAK,CAAC,IAAI,CAAC,2BAA2B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACrC,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO;iBACxB,MAAM,CAAC,CAAC,CAAC,EAAuC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBACrE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,EAAE,CAAC,CAAC;YACZ,IAAI,OAAO,EAAE,CAAC;gBACZ,KAAK,CAAC,IAAI,CACR,kBAAkB,kBAAkB,CAAC,OAAO,EAAE,qBAAqB,CAAC,EAAE,CACvE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC5B,CAAC"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import type { AgentTool, ThinkingLevel } from "../types.js";
|
|
2
|
+
import type { ModelUpdateEvent, ThinkingLevelUpdateEvent, ToolsUpdateEvent } from "./events.js";
|
|
3
|
+
import type { AgentHarnessResources, PromptTemplate, Skill } from "./options.js";
|
|
4
|
+
export declare function findDuplicateNames(names: string[]): string[];
|
|
5
|
+
export declare function validateUniqueNames(names: string[], message: string): void;
|
|
6
|
+
export declare function validateToolNames<TTool extends AgentTool>(toolNames: string[], tools: Map<string, TTool>): void;
|
|
7
|
+
export declare function createToolMap<TTool extends AgentTool>(tools: TTool[]): Map<string, TTool>;
|
|
8
|
+
export interface ToolConfiguration<TTool extends AgentTool> {
|
|
9
|
+
tools: Map<string, TTool>;
|
|
10
|
+
activeToolNames: string[];
|
|
11
|
+
}
|
|
12
|
+
export interface PreparedToolConfiguration<TTool extends AgentTool> extends ToolConfiguration<TTool> {
|
|
13
|
+
event: ToolsUpdateEvent;
|
|
14
|
+
}
|
|
15
|
+
export declare function prepareToolConfiguration<TTool extends AgentTool>(options: {
|
|
16
|
+
currentTools: Map<string, TTool>;
|
|
17
|
+
currentActiveToolNames: string[];
|
|
18
|
+
tools: TTool[];
|
|
19
|
+
activeToolNames?: string[];
|
|
20
|
+
source: ToolsUpdateEvent["source"];
|
|
21
|
+
}): PreparedToolConfiguration<TTool>;
|
|
22
|
+
export declare function prepareActiveToolsConfiguration<TTool extends AgentTool>(options: {
|
|
23
|
+
tools: Map<string, TTool>;
|
|
24
|
+
currentActiveToolNames: string[];
|
|
25
|
+
activeToolNames: string[];
|
|
26
|
+
source: ToolsUpdateEvent["source"];
|
|
27
|
+
}): {
|
|
28
|
+
activeToolNames: string[];
|
|
29
|
+
event: ToolsUpdateEvent;
|
|
30
|
+
};
|
|
31
|
+
export declare function createModelUpdateEvent(model: ModelUpdateEvent["model"], previousModel: ModelUpdateEvent["previousModel"], source: ModelUpdateEvent["source"]): ModelUpdateEvent;
|
|
32
|
+
export declare function createThinkingLevelUpdateEvent(level: ThinkingLevel, previousLevel: ThinkingLevel): ThinkingLevelUpdateEvent;
|
|
33
|
+
export declare function cloneHarnessResources<TSkill extends Skill = Skill, TPromptTemplate extends PromptTemplate = PromptTemplate>(resources: AgentHarnessResources<TSkill, TPromptTemplate>): AgentHarnessResources<TSkill, TPromptTemplate>;
|
|
34
|
+
//# sourceMappingURL=configuration.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"configuration.d.ts","sourceRoot":"","sources":["../../src/harness/configuration.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5D,OAAO,KAAK,EACV,gBAAgB,EAChB,wBAAwB,EACxB,gBAAgB,EACjB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EACV,qBAAqB,EACrB,cAAc,EACd,KAAK,EACN,MAAM,cAAc,CAAC;AAEtB,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE,CAQ5D;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAO1E;AAED,wBAAgB,iBAAiB,CAAC,KAAK,SAAS,SAAS,EACvD,SAAS,EAAE,MAAM,EAAE,EACnB,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,GACxB,IAAI,CAQN;AAED,wBAAgB,aAAa,CAAC,KAAK,SAAS,SAAS,EACnD,KAAK,EAAE,KAAK,EAAE,GACb,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAMpB;AAED,MAAM,WAAW,iBAAiB,CAAC,KAAK,SAAS,SAAS;IACxD,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC1B,eAAe,EAAE,MAAM,EAAE,CAAC;CAC3B;AAED,MAAM,WAAW,yBAAyB,CAAC,KAAK,SAAS,SAAS,CAChE,SAAQ,iBAAiB,CAAC,KAAK,CAAC;IAChC,KAAK,EAAE,gBAAgB,CAAC;CACzB;AAED,wBAAgB,wBAAwB,CAAC,KAAK,SAAS,SAAS,EAAE,OAAO,EAAE;IACzE,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACjC,sBAAsB,EAAE,MAAM,EAAE,CAAC;IACjC,KAAK,EAAE,KAAK,EAAE,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC,CAAC;CACpC,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAoBnC;AAED,wBAAgB,+BAA+B,CAC7C,KAAK,SAAS,SAAS,EACvB,OAAO,EAAE;IACT,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC1B,sBAAsB,EAAE,MAAM,EAAE,CAAC;IACjC,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC,CAAC;CACpC,GAAG;IAAE,eAAe,EAAE,MAAM,EAAE,CAAC;IAAC,KAAK,EAAE,gBAAgB,CAAA;CAAE,CAczD;AAED,wBAAgB,sBAAsB,CACpC,KAAK,EAAE,gBAAgB,CAAC,OAAO,CAAC,EAChC,aAAa,EAAE,gBAAgB,CAAC,eAAe,CAAC,EAChD,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC,GACjC,gBAAgB,CAElB;AAED,wBAAgB,8BAA8B,CAC5C,KAAK,EAAE,aAAa,EACpB,aAAa,EAAE,aAAa,GAC3B,wBAAwB,CAE1B;AAED,wBAAgB,qBAAqB,CACnC,MAAM,SAAS,KAAK,GAAG,KAAK,EAC5B,eAAe,SAAS,cAAc,GAAG,cAAc,EAEvD,SAAS,EAAE,qBAAqB,CAAC,MAAM,EAAE,eAAe,CAAC,GACxD,qBAAqB,CAAC,MAAM,EAAE,eAAe,CAAC,CAKhD"}
|
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
import { AgentHarnessError } from "./errors.js";
|
|
2
|
+
export function findDuplicateNames(names) {
|
|
3
|
+
const seen = new Set();
|
|
4
|
+
const duplicates = new Set();
|
|
5
|
+
for (const name of names) {
|
|
6
|
+
if (seen.has(name))
|
|
7
|
+
duplicates.add(name);
|
|
8
|
+
seen.add(name);
|
|
9
|
+
}
|
|
10
|
+
return [...duplicates];
|
|
11
|
+
}
|
|
12
|
+
export function validateUniqueNames(names, message) {
|
|
13
|
+
const duplicates = findDuplicateNames(names);
|
|
14
|
+
if (duplicates.length > 0)
|
|
15
|
+
throw new AgentHarnessError("invalid_argument", `${message}: ${duplicates.join(", ")}`);
|
|
16
|
+
}
|
|
17
|
+
export function validateToolNames(toolNames, tools) {
|
|
18
|
+
validateUniqueNames(toolNames, "Duplicate active tool name(s)");
|
|
19
|
+
const missing = toolNames.filter((name) => !tools.has(name));
|
|
20
|
+
if (missing.length > 0)
|
|
21
|
+
throw new AgentHarnessError("invalid_argument", `Unknown tool(s): ${missing.join(", ")}`);
|
|
22
|
+
}
|
|
23
|
+
export function createToolMap(tools) {
|
|
24
|
+
validateUniqueNames(tools.map((tool) => tool.name), "Duplicate tool name(s)");
|
|
25
|
+
return new Map(tools.map((tool) => [tool.name, tool]));
|
|
26
|
+
}
|
|
27
|
+
export function prepareToolConfiguration(options) {
|
|
28
|
+
const tools = createToolMap(options.tools);
|
|
29
|
+
const activeToolNames = options.activeToolNames
|
|
30
|
+
? [...options.activeToolNames]
|
|
31
|
+
: [...options.currentActiveToolNames];
|
|
32
|
+
validateToolNames(activeToolNames, tools);
|
|
33
|
+
const previousToolNames = [...options.currentTools.keys()];
|
|
34
|
+
const previousActiveToolNames = [...options.currentActiveToolNames];
|
|
35
|
+
return {
|
|
36
|
+
tools,
|
|
37
|
+
activeToolNames,
|
|
38
|
+
event: {
|
|
39
|
+
type: "tools_update",
|
|
40
|
+
toolNames: [...tools.keys()],
|
|
41
|
+
previousToolNames,
|
|
42
|
+
activeToolNames: [...activeToolNames],
|
|
43
|
+
previousActiveToolNames,
|
|
44
|
+
source: options.source,
|
|
45
|
+
},
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
export function prepareActiveToolsConfiguration(options) {
|
|
49
|
+
const activeToolNames = [...options.activeToolNames];
|
|
50
|
+
validateToolNames(activeToolNames, options.tools);
|
|
51
|
+
return {
|
|
52
|
+
activeToolNames,
|
|
53
|
+
event: {
|
|
54
|
+
type: "tools_update",
|
|
55
|
+
toolNames: [...options.tools.keys()],
|
|
56
|
+
previousToolNames: [...options.tools.keys()],
|
|
57
|
+
activeToolNames: [...activeToolNames],
|
|
58
|
+
previousActiveToolNames: [...options.currentActiveToolNames],
|
|
59
|
+
source: options.source,
|
|
60
|
+
},
|
|
61
|
+
};
|
|
62
|
+
}
|
|
63
|
+
export function createModelUpdateEvent(model, previousModel, source) {
|
|
64
|
+
return { type: "model_update", model, previousModel, source };
|
|
65
|
+
}
|
|
66
|
+
export function createThinkingLevelUpdateEvent(level, previousLevel) {
|
|
67
|
+
return { type: "thinking_level_update", level, previousLevel };
|
|
68
|
+
}
|
|
69
|
+
export function cloneHarnessResources(resources) {
|
|
70
|
+
return {
|
|
71
|
+
skills: resources.skills?.slice(),
|
|
72
|
+
promptTemplates: resources.promptTemplates?.slice(),
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
//# sourceMappingURL=configuration.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"configuration.js","sourceRoot":"","sources":["../../src/harness/configuration.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAYhD,MAAM,UAAU,kBAAkB,CAAC,KAAe;IAChD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;IACrC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjB,CAAC;IACD,OAAO,CAAC,GAAG,UAAU,CAAC,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAe,EAAE,OAAe;IAClE,MAAM,UAAU,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAC7C,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC;QACvB,MAAM,IAAI,iBAAiB,CACzB,kBAAkB,EAClB,GAAG,OAAO,KAAK,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACvC,CAAC;AACN,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,SAAmB,EACnB,KAAyB;IAEzB,mBAAmB,CAAC,SAAS,EAAE,+BAA+B,CAAC,CAAC;IAChE,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7D,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;QACpB,MAAM,IAAI,iBAAiB,CACzB,kBAAkB,EAClB,oBAAoB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACzC,CAAC;AACN,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,KAAc;IAEd,mBAAmB,CACjB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAC9B,wBAAwB,CACzB,CAAC;IACF,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;AACzD,CAAC;AAYD,MAAM,UAAU,wBAAwB,CAA0B,OAMjE;IACC,MAAM,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC3C,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe;QAC7C,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;QAC9B,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,sBAAsB,CAAC,CAAC;IACxC,iBAAiB,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IAC1C,MAAM,iBAAiB,GAAG,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3D,MAAM,uBAAuB,GAAG,CAAC,GAAG,OAAO,CAAC,sBAAsB,CAAC,CAAC;IACpE,OAAO;QACL,KAAK;QACL,eAAe;QACf,KAAK,EAAE;YACL,IAAI,EAAE,cAAc;YACpB,SAAS,EAAE,CAAC,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;YAC5B,iBAAiB;YACjB,eAAe,EAAE,CAAC,GAAG,eAAe,CAAC;YACrC,uBAAuB;YACvB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,+BAA+B,CAE7C,OAKD;IACC,MAAM,eAAe,GAAG,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;IACrD,iBAAiB,CAAC,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO;QACL,eAAe;QACf,KAAK,EAAE;YACL,IAAI,EAAE,cAAc;YACpB,SAAS,EAAE,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YACpC,iBAAiB,EAAE,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YAC5C,eAAe,EAAE,CAAC,GAAG,eAAe,CAAC;YACrC,uBAAuB,EAAE,CAAC,GAAG,OAAO,CAAC,sBAAsB,CAAC;YAC5D,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,KAAgC,EAChC,aAAgD,EAChD,MAAkC;IAElC,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;AAChE,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,KAAoB,EACpB,aAA4B;IAE5B,OAAO,EAAE,IAAI,EAAE,uBAAuB,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,qBAAqB,CAInC,SAAyD;IAEzD,OAAO;QACL,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,KAAK,EAAE;QACjC,eAAe,EAAE,SAAS,CAAC,eAAe,EAAE,KAAK,EAAE;KACpD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import type { AgentMessage } from "../../types.js";
|
|
2
|
+
import type { CompactionEntry, ConversationTreeEntry } from "./entries.js";
|
|
3
|
+
export interface ConversationContext {
|
|
4
|
+
messages: AgentMessage[];
|
|
5
|
+
thinkingLevel: string;
|
|
6
|
+
model: {
|
|
7
|
+
provider: string;
|
|
8
|
+
modelId: string;
|
|
9
|
+
} | null;
|
|
10
|
+
activeToolNames: string[] | null;
|
|
11
|
+
}
|
|
12
|
+
export interface ConversationState {
|
|
13
|
+
thinkingLevel: string;
|
|
14
|
+
model: {
|
|
15
|
+
provider: string;
|
|
16
|
+
modelId: string;
|
|
17
|
+
} | null;
|
|
18
|
+
activeToolNames: string[] | null;
|
|
19
|
+
compaction: CompactionEntry | null;
|
|
20
|
+
}
|
|
21
|
+
export declare function extractConversationState(pathEntries: ConversationTreeEntry[]): ConversationState;
|
|
22
|
+
export declare function buildContextMessages(pathEntries: ConversationTreeEntry[], state: ConversationState): AgentMessage[];
|
|
23
|
+
export declare function buildConversationContext(pathEntries: ConversationTreeEntry[]): ConversationContext;
|
|
24
|
+
//# sourceMappingURL=context.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../../src/harness/conversation/context.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAMnD,OAAO,KAAK,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AAE3E,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IACpD,eAAe,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;CAClC;AAED,MAAM,WAAW,iBAAiB;IAChC,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IACpD,eAAe,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IACjC,UAAU,EAAE,eAAe,GAAG,IAAI,CAAC;CACpC;AAED,wBAAgB,wBAAwB,CACtC,WAAW,EAAE,qBAAqB,EAAE,GACnC,iBAAiB,CAwBnB;AAiCD,wBAAgB,oBAAoB,CAClC,WAAW,EAAE,qBAAqB,EAAE,EACpC,KAAK,EAAE,iBAAiB,GACvB,YAAY,EAAE,CA8BhB;AAED,wBAAgB,wBAAwB,CACtC,WAAW,EAAE,qBAAqB,EAAE,GACnC,mBAAmB,CAQrB"}
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
import { createBranchSummaryMessage, createCompactionSummaryMessage, createCustomMessage, } from "../messages.js";
|
|
2
|
+
export function extractConversationState(pathEntries) {
|
|
3
|
+
let thinkingLevel = "off";
|
|
4
|
+
let model = null;
|
|
5
|
+
let activeToolNames = null;
|
|
6
|
+
let compaction = null;
|
|
7
|
+
for (const entry of pathEntries) {
|
|
8
|
+
if (entry.type === "thinking_level_change") {
|
|
9
|
+
thinkingLevel = entry.thinkingLevel;
|
|
10
|
+
}
|
|
11
|
+
else if (entry.type === "model_change") {
|
|
12
|
+
model = { provider: entry.provider, modelId: entry.modelId };
|
|
13
|
+
}
|
|
14
|
+
else if (entry.type === "message" && entry.message.role === "assistant") {
|
|
15
|
+
model = {
|
|
16
|
+
provider: entry.message.provider,
|
|
17
|
+
modelId: entry.message.model,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
else if (entry.type === "active_tools_change") {
|
|
21
|
+
activeToolNames = [...entry.activeToolNames];
|
|
22
|
+
}
|
|
23
|
+
else if (entry.type === "compaction") {
|
|
24
|
+
compaction = entry;
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
return { thinkingLevel, model, activeToolNames, compaction };
|
|
28
|
+
}
|
|
29
|
+
function messageFromEntry(entry) {
|
|
30
|
+
if (entry.type === "message")
|
|
31
|
+
return entry.message;
|
|
32
|
+
if (entry.type === "custom_message") {
|
|
33
|
+
return createCustomMessage(entry.customType, entry.content, entry.display, entry.details, entry.timestamp);
|
|
34
|
+
}
|
|
35
|
+
if (entry.type === "branch_summary" && entry.summary) {
|
|
36
|
+
return createBranchSummaryMessage(entry.summary, entry.fromId, entry.timestamp);
|
|
37
|
+
}
|
|
38
|
+
return undefined;
|
|
39
|
+
}
|
|
40
|
+
function appendContextMessage(messages, entry) {
|
|
41
|
+
const message = messageFromEntry(entry);
|
|
42
|
+
if (message)
|
|
43
|
+
messages.push(message);
|
|
44
|
+
}
|
|
45
|
+
export function buildContextMessages(pathEntries, state) {
|
|
46
|
+
const messages = [];
|
|
47
|
+
if (state.compaction) {
|
|
48
|
+
messages.push(createCompactionSummaryMessage(state.compaction.summary, state.compaction.tokensBefore, state.compaction.timestamp));
|
|
49
|
+
const compactionIdx = pathEntries.findIndex((entry) => entry.type === "compaction" && entry.id === state.compaction?.id);
|
|
50
|
+
let foundFirstKept = false;
|
|
51
|
+
for (const entry of pathEntries.slice(0, compactionIdx)) {
|
|
52
|
+
if (entry.id === state.compaction.firstKeptEntryId)
|
|
53
|
+
foundFirstKept = true;
|
|
54
|
+
if (foundFirstKept)
|
|
55
|
+
appendContextMessage(messages, entry);
|
|
56
|
+
}
|
|
57
|
+
for (const entry of pathEntries.slice(compactionIdx + 1)) {
|
|
58
|
+
appendContextMessage(messages, entry);
|
|
59
|
+
}
|
|
60
|
+
return messages;
|
|
61
|
+
}
|
|
62
|
+
for (const entry of pathEntries) {
|
|
63
|
+
appendContextMessage(messages, entry);
|
|
64
|
+
}
|
|
65
|
+
return messages;
|
|
66
|
+
}
|
|
67
|
+
export function buildConversationContext(pathEntries) {
|
|
68
|
+
const state = extractConversationState(pathEntries);
|
|
69
|
+
return {
|
|
70
|
+
messages: buildContextMessages(pathEntries, state),
|
|
71
|
+
thinkingLevel: state.thinkingLevel,
|
|
72
|
+
model: state.model,
|
|
73
|
+
activeToolNames: state.activeToolNames,
|
|
74
|
+
};
|
|
75
|
+
}
|
|
76
|
+
//# sourceMappingURL=context.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context.js","sourceRoot":"","sources":["../../../src/harness/conversation/context.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,0BAA0B,EAC1B,8BAA8B,EAC9B,mBAAmB,GACpB,MAAM,gBAAgB,CAAC;AAiBxB,MAAM,UAAU,wBAAwB,CACtC,WAAoC;IAEpC,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,IAAI,KAAK,GAAiD,IAAI,CAAC;IAC/D,IAAI,eAAe,GAAoB,IAAI,CAAC;IAC5C,IAAI,UAAU,GAA2B,IAAI,CAAC;IAE9C,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAChC,IAAI,KAAK,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;YAC3C,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC;QACtC,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YACzC,KAAK,GAAG,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;QAC/D,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC1E,KAAK,GAAG;gBACN,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ;gBAChC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;aAC7B,CAAC;QACJ,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;YAChD,eAAe,GAAG,CAAC,GAAG,KAAK,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACvC,UAAU,GAAG,KAAK,CAAC;QACrB,CAAC;IACH,CAAC;IAED,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,CAAC;AAC/D,CAAC;AAED,SAAS,gBAAgB,CACvB,KAA4B;IAE5B,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC,OAAuB,CAAC;IACnE,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;QACpC,OAAO,mBAAmB,CACxB,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,OAAkD,EACxD,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,SAAS,CAChB,CAAC;IACJ,CAAC;IACD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QACrD,OAAO,0BAA0B,CAC/B,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,SAAS,CAChB,CAAC;IACJ,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,oBAAoB,CAC3B,QAAwB,EACxB,KAA4B;IAE5B,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACxC,IAAI,OAAO;QAAE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,WAAoC,EACpC,KAAwB;IAExB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IAEpC,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACrB,QAAQ,CAAC,IAAI,CACX,8BAA8B,CAC5B,KAAK,CAAC,UAAU,CAAC,OAAO,EACxB,KAAK,CAAC,UAAU,CAAC,YAAY,EAC7B,KAAK,CAAC,UAAU,CAAC,SAAS,CAC3B,CACF,CAAC;QACF,MAAM,aAAa,GAAG,WAAW,CAAC,SAAS,CACzC,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,EAAE,EAAE,CACnE,CAAC;QACF,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,EAAE,CAAC;YACxD,IAAI,KAAK,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,CAAC,gBAAgB;gBAAE,cAAc,GAAG,IAAI,CAAC;YAC1E,IAAI,cAAc;gBAAE,oBAAoB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,KAAK,CAAC,aAAa,GAAG,CAAC,CAAC,EAAE,CAAC;YACzD,oBAAoB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAChC,oBAAoB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,WAAoC;IAEpC,MAAM,KAAK,GAAG,wBAAwB,CAAC,WAAW,CAAC,CAAC;IACpD,OAAO;QACL,QAAQ,EAAE,oBAAoB,CAAC,WAAW,EAAE,KAAK,CAAC;QAClD,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,eAAe,EAAE,KAAK,CAAC,eAAe;KACvC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import type { ImageContent, TextContent } from "@earendil-works/pi-ai";
|
|
2
|
+
import type { AgentMessage } from "../../types.js";
|
|
3
|
+
import type { ConversationContext } from "./context.js";
|
|
4
|
+
import type { ConversationMetadata, ConversationStorage, ConversationTreeEntry } from "./entries.js";
|
|
5
|
+
export { buildConversationContext } from "./context.js";
|
|
6
|
+
export declare class Conversation<TMetadata extends ConversationMetadata = ConversationMetadata> {
|
|
7
|
+
private storage;
|
|
8
|
+
constructor(storage: ConversationStorage<TMetadata>);
|
|
9
|
+
getMetadata(): Promise<TMetadata>;
|
|
10
|
+
getStorage(): ConversationStorage<TMetadata>;
|
|
11
|
+
getLeafId(): Promise<string | null>;
|
|
12
|
+
getEntry(id: string): Promise<ConversationTreeEntry | undefined>;
|
|
13
|
+
getEntries(): Promise<ConversationTreeEntry[]>;
|
|
14
|
+
getBranch(fromId?: string): Promise<ConversationTreeEntry[]>;
|
|
15
|
+
buildContext(): Promise<ConversationContext>;
|
|
16
|
+
getLabel(id: string): Promise<string | undefined>;
|
|
17
|
+
getConversationName(): Promise<string | undefined>;
|
|
18
|
+
private appendTypedEntry;
|
|
19
|
+
appendMessage(message: AgentMessage): Promise<string>;
|
|
20
|
+
appendMessageWithId(id: string, message: AgentMessage, timestamp?: string): Promise<string>;
|
|
21
|
+
appendHarnessMessageWithId(id: string, message: AgentMessage, timestamp?: string): Promise<string>;
|
|
22
|
+
appendThinkingLevelChange(thinkingLevel: string): Promise<string>;
|
|
23
|
+
appendModelChange(provider: string, modelId: string): Promise<string>;
|
|
24
|
+
appendActiveToolsChange(activeToolNames: string[]): Promise<string>;
|
|
25
|
+
appendCompaction<T = unknown>(summary: string, firstKeptEntryId: string, tokensBefore: number, details?: T, fromHook?: boolean): Promise<string>;
|
|
26
|
+
appendCustomEntry(customType: string, data?: unknown): Promise<string>;
|
|
27
|
+
appendCustomMessageEntry<T = unknown>(customType: string, content: string | (TextContent | ImageContent)[], display: boolean, details?: T): Promise<string>;
|
|
28
|
+
appendLabel(targetId: string, label: string | undefined): Promise<string>;
|
|
29
|
+
appendConversationName(name: string): Promise<string>;
|
|
30
|
+
moveTo(entryId: string | null, summary?: {
|
|
31
|
+
summary: string;
|
|
32
|
+
details?: unknown;
|
|
33
|
+
fromHook?: boolean;
|
|
34
|
+
}): Promise<string | undefined>;
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=conversation.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"conversation.d.ts","sourceRoot":"","sources":["../../../src/harness/conversation/conversation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAEnD,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAExD,OAAO,KAAK,EAKV,oBAAoB,EACpB,mBAAmB,EACnB,qBAAqB,EAOtB,MAAM,cAAc,CAAC;AAEtB,OAAO,EAAE,wBAAwB,EAAE,MAAM,cAAc,CAAC;AAExD,qBAAa,YAAY,CACvB,SAAS,SAAS,oBAAoB,GAAG,oBAAoB;IAE7D,OAAO,CAAC,OAAO,CAAiC;gBAEpC,OAAO,EAAE,mBAAmB,CAAC,SAAS,CAAC;IAInD,WAAW,IAAI,OAAO,CAAC,SAAS,CAAC;IAIjC,UAAU,IAAI,mBAAmB,CAAC,SAAS,CAAC;IAI5C,SAAS,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAInC,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,GAAG,SAAS,CAAC;IAIhE,UAAU,IAAI,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAIxC,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAK5D,YAAY,IAAI,OAAO,CAAC,mBAAmB,CAAC;IAIlD,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAI3C,mBAAmB,IAAI,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;YAK1C,gBAAgB;IAOxB,aAAa,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC;IAOrD,mBAAmB,CACvB,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,YAAY,EACrB,SAAS,SAA2B,GACnC,OAAO,CAAC,MAAM,CAAC;IAWZ,0BAA0B,CAC9B,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,YAAY,EACrB,SAAS,SAA2B,GACnC,OAAO,CAAC,MAAM,CAAC;IAIZ,yBAAyB,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAUjE,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAWrE,uBAAuB,CAAC,eAAe,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAUnE,gBAAgB,CAAC,CAAC,GAAG,OAAO,EAChC,OAAO,EAAE,MAAM,EACf,gBAAgB,EAAE,MAAM,EACxB,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,CAAC,EACX,QAAQ,CAAC,EAAE,OAAO,GACjB,OAAO,CAAC,MAAM,CAAC;IAcZ,iBAAiB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;IAWtE,wBAAwB,CAAC,CAAC,GAAG,OAAO,EACxC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,GAAG,CAAC,WAAW,GAAG,YAAY,CAAC,EAAE,EAChD,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,CAAC,GACV,OAAO,CAAC,MAAM,CAAC;IAaZ,WAAW,CACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,GAAG,SAAS,GACxB,OAAO,CAAC,MAAM,CAAC;IAcZ,sBAAsB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAUrD,MAAM,CACV,OAAO,EAAE,MAAM,GAAG,IAAI,EACtB,OAAO,CAAC,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GACnE,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;CAiB/B"}
|
|
@@ -0,0 +1,163 @@
|
|
|
1
|
+
import { ConversationError } from "../errors.js";
|
|
2
|
+
import { buildConversationContext } from "./context.js";
|
|
3
|
+
export { buildConversationContext } from "./context.js";
|
|
4
|
+
export class Conversation {
|
|
5
|
+
storage;
|
|
6
|
+
constructor(storage) {
|
|
7
|
+
this.storage = storage;
|
|
8
|
+
}
|
|
9
|
+
getMetadata() {
|
|
10
|
+
return this.storage.getMetadata();
|
|
11
|
+
}
|
|
12
|
+
getStorage() {
|
|
13
|
+
return this.storage;
|
|
14
|
+
}
|
|
15
|
+
getLeafId() {
|
|
16
|
+
return this.storage.getLeafId();
|
|
17
|
+
}
|
|
18
|
+
getEntry(id) {
|
|
19
|
+
return this.storage.getEntry(id);
|
|
20
|
+
}
|
|
21
|
+
getEntries() {
|
|
22
|
+
return this.storage.getEntries();
|
|
23
|
+
}
|
|
24
|
+
async getBranch(fromId) {
|
|
25
|
+
const leafId = fromId ?? (await this.storage.getLeafId());
|
|
26
|
+
return this.storage.getPathToRoot(leafId);
|
|
27
|
+
}
|
|
28
|
+
async buildContext() {
|
|
29
|
+
return buildConversationContext(await this.getBranch());
|
|
30
|
+
}
|
|
31
|
+
getLabel(id) {
|
|
32
|
+
return this.storage.getLabel(id);
|
|
33
|
+
}
|
|
34
|
+
async getConversationName() {
|
|
35
|
+
const entries = await this.storage.findEntries("conversation_info");
|
|
36
|
+
return entries[entries.length - 1]?.name?.trim() || undefined;
|
|
37
|
+
}
|
|
38
|
+
async appendTypedEntry(entry) {
|
|
39
|
+
await this.storage.appendEntry(entry);
|
|
40
|
+
return entry.id;
|
|
41
|
+
}
|
|
42
|
+
async appendMessage(message) {
|
|
43
|
+
return this.appendMessageWithId(await this.storage.createEntryId(), message);
|
|
44
|
+
}
|
|
45
|
+
async appendMessageWithId(id, message, timestamp = new Date().toISOString()) {
|
|
46
|
+
if (await this.storage.getEntry(id))
|
|
47
|
+
return id;
|
|
48
|
+
return this.appendTypedEntry({
|
|
49
|
+
type: "message",
|
|
50
|
+
id,
|
|
51
|
+
parentId: await this.storage.getLeafId(),
|
|
52
|
+
timestamp,
|
|
53
|
+
message,
|
|
54
|
+
});
|
|
55
|
+
}
|
|
56
|
+
async appendHarnessMessageWithId(id, message, timestamp = new Date().toISOString()) {
|
|
57
|
+
return this.appendMessageWithId(id, message, timestamp);
|
|
58
|
+
}
|
|
59
|
+
async appendThinkingLevelChange(thinkingLevel) {
|
|
60
|
+
return this.appendTypedEntry({
|
|
61
|
+
type: "thinking_level_change",
|
|
62
|
+
id: await this.storage.createEntryId(),
|
|
63
|
+
parentId: await this.storage.getLeafId(),
|
|
64
|
+
timestamp: new Date().toISOString(),
|
|
65
|
+
thinkingLevel,
|
|
66
|
+
});
|
|
67
|
+
}
|
|
68
|
+
async appendModelChange(provider, modelId) {
|
|
69
|
+
return this.appendTypedEntry({
|
|
70
|
+
type: "model_change",
|
|
71
|
+
id: await this.storage.createEntryId(),
|
|
72
|
+
parentId: await this.storage.getLeafId(),
|
|
73
|
+
timestamp: new Date().toISOString(),
|
|
74
|
+
provider,
|
|
75
|
+
modelId,
|
|
76
|
+
});
|
|
77
|
+
}
|
|
78
|
+
async appendActiveToolsChange(activeToolNames) {
|
|
79
|
+
return this.appendTypedEntry({
|
|
80
|
+
type: "active_tools_change",
|
|
81
|
+
id: await this.storage.createEntryId(),
|
|
82
|
+
parentId: await this.storage.getLeafId(),
|
|
83
|
+
timestamp: new Date().toISOString(),
|
|
84
|
+
activeToolNames: [...activeToolNames],
|
|
85
|
+
});
|
|
86
|
+
}
|
|
87
|
+
async appendCompaction(summary, firstKeptEntryId, tokensBefore, details, fromHook) {
|
|
88
|
+
return this.appendTypedEntry({
|
|
89
|
+
type: "compaction",
|
|
90
|
+
id: await this.storage.createEntryId(),
|
|
91
|
+
parentId: await this.storage.getLeafId(),
|
|
92
|
+
timestamp: new Date().toISOString(),
|
|
93
|
+
summary,
|
|
94
|
+
firstKeptEntryId,
|
|
95
|
+
tokensBefore,
|
|
96
|
+
details,
|
|
97
|
+
fromHook,
|
|
98
|
+
});
|
|
99
|
+
}
|
|
100
|
+
async appendCustomEntry(customType, data) {
|
|
101
|
+
return this.appendTypedEntry({
|
|
102
|
+
type: "custom",
|
|
103
|
+
id: await this.storage.createEntryId(),
|
|
104
|
+
parentId: await this.storage.getLeafId(),
|
|
105
|
+
timestamp: new Date().toISOString(),
|
|
106
|
+
customType,
|
|
107
|
+
data,
|
|
108
|
+
});
|
|
109
|
+
}
|
|
110
|
+
async appendCustomMessageEntry(customType, content, display, details) {
|
|
111
|
+
return this.appendTypedEntry({
|
|
112
|
+
type: "custom_message",
|
|
113
|
+
id: await this.storage.createEntryId(),
|
|
114
|
+
parentId: await this.storage.getLeafId(),
|
|
115
|
+
timestamp: new Date().toISOString(),
|
|
116
|
+
customType,
|
|
117
|
+
content,
|
|
118
|
+
display,
|
|
119
|
+
details,
|
|
120
|
+
});
|
|
121
|
+
}
|
|
122
|
+
async appendLabel(targetId, label) {
|
|
123
|
+
if (!(await this.storage.getEntry(targetId))) {
|
|
124
|
+
throw new ConversationError("not_found", `Entry ${targetId} not found`);
|
|
125
|
+
}
|
|
126
|
+
return this.appendTypedEntry({
|
|
127
|
+
type: "label",
|
|
128
|
+
id: await this.storage.createEntryId(),
|
|
129
|
+
parentId: await this.storage.getLeafId(),
|
|
130
|
+
timestamp: new Date().toISOString(),
|
|
131
|
+
targetId,
|
|
132
|
+
label,
|
|
133
|
+
});
|
|
134
|
+
}
|
|
135
|
+
async appendConversationName(name) {
|
|
136
|
+
return this.appendTypedEntry({
|
|
137
|
+
type: "conversation_info",
|
|
138
|
+
id: await this.storage.createEntryId(),
|
|
139
|
+
parentId: await this.storage.getLeafId(),
|
|
140
|
+
timestamp: new Date().toISOString(),
|
|
141
|
+
name: name.trim(),
|
|
142
|
+
});
|
|
143
|
+
}
|
|
144
|
+
async moveTo(entryId, summary) {
|
|
145
|
+
if (entryId !== null && !(await this.storage.getEntry(entryId))) {
|
|
146
|
+
throw new ConversationError("not_found", `Entry ${entryId} not found`);
|
|
147
|
+
}
|
|
148
|
+
await this.storage.setLeafId(entryId);
|
|
149
|
+
if (!summary)
|
|
150
|
+
return undefined;
|
|
151
|
+
return this.appendTypedEntry({
|
|
152
|
+
type: "branch_summary",
|
|
153
|
+
id: await this.storage.createEntryId(),
|
|
154
|
+
parentId: entryId,
|
|
155
|
+
timestamp: new Date().toISOString(),
|
|
156
|
+
fromId: entryId ?? "root",
|
|
157
|
+
summary: summary.summary,
|
|
158
|
+
details: summary.details,
|
|
159
|
+
fromHook: summary.fromHook,
|
|
160
|
+
});
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
//# sourceMappingURL=conversation.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"conversation.js","sourceRoot":"","sources":["../../../src/harness/conversation/conversation.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAEjD,OAAO,EAAE,wBAAwB,EAAE,MAAM,cAAc,CAAC;AAiBxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,cAAc,CAAC;AAExD,MAAM,OAAO,YAAY;IAGf,OAAO,CAAiC;IAEhD,YAAY,OAAuC;QACjD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;IACpC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACnC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAe;QAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,OAAO,wBAAwB,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,mBAAmB;QACvB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;QACpE,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;IAChE,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,KAAa;QAEb,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACtC,OAAO,KAAK,CAAC,EAAE,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAqB;QACvC,OAAO,IAAI,CAAC,mBAAmB,CAC7B,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,EAClC,OAAO,CACR,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,EAAU,EACV,OAAqB,EACrB,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QAEpC,IAAI,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAAE,OAAO,EAAE,CAAC;QAC/C,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,SAAS;YACf,EAAE;YACF,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS;YACT,OAAO;SACe,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,0BAA0B,CAC9B,EAAU,EACV,OAAqB,EACrB,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QAEpC,OAAO,IAAI,CAAC,mBAAmB,CAAC,EAAE,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,aAAqB;QACnD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,uBAAuB;YAC7B,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,aAAa;SACqB,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE,OAAe;QACvD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,cAAc;YACpB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,QAAQ;YACR,OAAO;SACmB,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,eAAyB;QACrD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,qBAAqB;YAC3B,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,eAAe,EAAE,CAAC,GAAG,eAAe,CAAC;SACL,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,OAAe,EACf,gBAAwB,EACxB,YAAoB,EACpB,OAAW,EACX,QAAkB;QAElB,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,YAAY;YAClB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,OAAO;YACP,gBAAgB;YAChB,YAAY;YACZ,OAAO;YACP,QAAQ;SACoB,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,UAAkB,EAAE,IAAc;QACxD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,QAAQ;YACd,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,UAAU;YACV,IAAI;SACiB,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,wBAAwB,CAC5B,UAAkB,EAClB,OAAgD,EAChD,OAAgB,EAChB,OAAW;QAEX,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,gBAAgB;YACtB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,UAAU;YACV,OAAO;YACP,OAAO;YACP,OAAO;SACwB,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,WAAW,CACf,QAAgB,EAChB,KAAyB;QAEzB,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,iBAAiB,CAAC,WAAW,EAAE,SAAS,QAAQ,YAAY,CAAC,CAAC;QAC1E,CAAC;QACD,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,OAAO;YACb,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,QAAQ;YACR,KAAK;SACe,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,IAAY;QACvC,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,mBAAmB;YACzB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;SACc,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,MAAM,CACV,OAAsB,EACtB,OAAoE;QAEpE,IAAI,OAAO,KAAK,IAAI,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;YAChE,MAAM,IAAI,iBAAiB,CAAC,WAAW,EAAE,SAAS,OAAO,YAAY,CAAC,CAAC;QACzE,CAAC;QACD,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACtC,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAC/B,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,IAAI,EAAE,gBAAgB;YACtB,EAAE,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE;YACtC,QAAQ,EAAE,OAAO;YACjB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,MAAM,EAAE,OAAO,IAAI,MAAM;YACzB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;SACE,CAAC,CAAC;IAClC,CAAC;CACF"}
|