grok-agent 0.5.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/CHANGELOG.md +113 -0
- package/LICENSE +190 -0
- package/README.md +449 -0
- package/bin/grok-cli.js +26 -0
- package/dist/agent.d.ts +4 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +898 -0
- package/dist/agent.js.map +1 -0
- package/dist/approvals.d.ts +4 -0
- package/dist/approvals.d.ts.map +1 -0
- package/dist/approvals.js +90 -0
- package/dist/approvals.js.map +1 -0
- package/dist/batch-api.d.ts +11 -0
- package/dist/batch-api.d.ts.map +1 -0
- package/dist/batch-api.js +101 -0
- package/dist/batch-api.js.map +1 -0
- package/dist/cli-errors.d.ts +6 -0
- package/dist/cli-errors.d.ts.map +1 -0
- package/dist/cli-errors.js +80 -0
- package/dist/cli-errors.js.map +1 -0
- package/dist/client.d.ts +6 -0
- package/dist/client.d.ts.map +1 -0
- package/dist/client.js +41 -0
- package/dist/client.js.map +1 -0
- package/dist/collections-api.d.ts +11 -0
- package/dist/collections-api.d.ts.map +1 -0
- package/dist/collections-api.js +144 -0
- package/dist/collections-api.js.map +1 -0
- package/dist/compaction.d.ts +11 -0
- package/dist/compaction.d.ts.map +1 -0
- package/dist/compaction.js +94 -0
- package/dist/compaction.js.map +1 -0
- package/dist/config.d.ts +12 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +142 -0
- package/dist/config.js.map +1 -0
- package/dist/diff.d.ts +5 -0
- package/dist/diff.d.ts.map +1 -0
- package/dist/diff.js +80 -0
- package/dist/diff.js.map +1 -0
- package/dist/hooks.d.ts +11 -0
- package/dist/hooks.d.ts.map +1 -0
- package/dist/hooks.js +30 -0
- package/dist/hooks.js.map +1 -0
- package/dist/image.d.ts +8 -0
- package/dist/image.d.ts.map +1 -0
- package/dist/image.js +58 -0
- package/dist/image.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +1264 -0
- package/dist/index.js.map +1 -0
- package/dist/json-output.d.ts +27 -0
- package/dist/json-output.d.ts.map +1 -0
- package/dist/json-output.js +62 -0
- package/dist/json-output.js.map +1 -0
- package/dist/notifications.d.ts +6 -0
- package/dist/notifications.d.ts.map +1 -0
- package/dist/notifications.js +25 -0
- package/dist/notifications.js.map +1 -0
- package/dist/project-context.d.ts +6 -0
- package/dist/project-context.d.ts.map +1 -0
- package/dist/project-context.js +34 -0
- package/dist/project-context.js.map +1 -0
- package/dist/response-utils.d.ts +10 -0
- package/dist/response-utils.d.ts.map +1 -0
- package/dist/response-utils.js +123 -0
- package/dist/response-utils.js.map +1 -0
- package/dist/server-tools.d.ts +7 -0
- package/dist/server-tools.d.ts.map +1 -0
- package/dist/server-tools.js +205 -0
- package/dist/server-tools.js.map +1 -0
- package/dist/session.d.ts +46 -0
- package/dist/session.d.ts.map +1 -0
- package/dist/session.js +305 -0
- package/dist/session.js.map +1 -0
- package/dist/stream.d.ts +22 -0
- package/dist/stream.d.ts.map +1 -0
- package/dist/stream.js +102 -0
- package/dist/stream.js.map +1 -0
- package/dist/system-prompt.d.ts +3 -0
- package/dist/system-prompt.d.ts.map +1 -0
- package/dist/system-prompt.js +64 -0
- package/dist/system-prompt.js.map +1 -0
- package/dist/tools/bash.d.ts +8 -0
- package/dist/tools/bash.d.ts.map +1 -0
- package/dist/tools/bash.js +49 -0
- package/dist/tools/bash.js.map +1 -0
- package/dist/tools/definitions.d.ts +3 -0
- package/dist/tools/definitions.d.ts.map +1 -0
- package/dist/tools/definitions.js +190 -0
- package/dist/tools/definitions.js.map +1 -0
- package/dist/tools/edit-file.d.ts +10 -0
- package/dist/tools/edit-file.d.ts.map +1 -0
- package/dist/tools/edit-file.js +73 -0
- package/dist/tools/edit-file.js.map +1 -0
- package/dist/tools/glob.d.ts +7 -0
- package/dist/tools/glob.d.ts.map +1 -0
- package/dist/tools/glob.js +44 -0
- package/dist/tools/glob.js.map +1 -0
- package/dist/tools/grep.d.ts +10 -0
- package/dist/tools/grep.d.ts.map +1 -0
- package/dist/tools/grep.js +102 -0
- package/dist/tools/grep.js.map +1 -0
- package/dist/tools/index.d.ts +10 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +41 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/list-dir.d.ts +7 -0
- package/dist/tools/list-dir.d.ts.map +1 -0
- package/dist/tools/list-dir.js +68 -0
- package/dist/tools/list-dir.js.map +1 -0
- package/dist/tools/policy.d.ts +4 -0
- package/dist/tools/policy.d.ts.map +1 -0
- package/dist/tools/policy.js +36 -0
- package/dist/tools/policy.js.map +1 -0
- package/dist/tools/read-file.d.ts +8 -0
- package/dist/tools/read-file.d.ts.map +1 -0
- package/dist/tools/read-file.js +45 -0
- package/dist/tools/read-file.js.map +1 -0
- package/dist/tools/write-file.d.ts +7 -0
- package/dist/tools/write-file.d.ts.map +1 -0
- package/dist/tools/write-file.js +35 -0
- package/dist/tools/write-file.js.map +1 -0
- package/dist/truncation.d.ts +13 -0
- package/dist/truncation.d.ts.map +1 -0
- package/dist/truncation.js +55 -0
- package/dist/truncation.js.map +1 -0
- package/dist/types.d.ts +159 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/usage.d.ts +17 -0
- package/dist/usage.d.ts.map +1 -0
- package/dist/usage.js +80 -0
- package/dist/usage.js.map +1 -0
- package/dist/voice-api.d.ts +16 -0
- package/dist/voice-api.d.ts.map +1 -0
- package/dist/voice-api.js +84 -0
- package/dist/voice-api.js.map +1 -0
- package/package.json +64 -0
|
@@ -0,0 +1,205 @@
|
|
|
1
|
+
export function normalizeServerTools(tools) {
|
|
2
|
+
if (!tools)
|
|
3
|
+
return [];
|
|
4
|
+
const normalized = [];
|
|
5
|
+
for (const tool of tools) {
|
|
6
|
+
const candidate = typeof tool === "string"
|
|
7
|
+
? normalizeLegacyTool(tool)
|
|
8
|
+
: tool;
|
|
9
|
+
if (!candidate)
|
|
10
|
+
continue;
|
|
11
|
+
const merged = mergeServerTool(normalized, candidate);
|
|
12
|
+
if (!merged)
|
|
13
|
+
normalized.push(candidate);
|
|
14
|
+
}
|
|
15
|
+
return normalized;
|
|
16
|
+
}
|
|
17
|
+
function normalizeLegacyTool(tool) {
|
|
18
|
+
if (tool === "file_search")
|
|
19
|
+
return null;
|
|
20
|
+
return { type: tool };
|
|
21
|
+
}
|
|
22
|
+
function mergeServerTool(tools, candidate) {
|
|
23
|
+
const existing = tools.find((tool) => sameServerTool(tool, candidate));
|
|
24
|
+
if (!existing)
|
|
25
|
+
return false;
|
|
26
|
+
if (existing.type === "web_search" && candidate.type === "web_search") {
|
|
27
|
+
existing.filters = {
|
|
28
|
+
allowedDomains: uniqueStrings([
|
|
29
|
+
...(existing.filters?.allowedDomains || []),
|
|
30
|
+
...(candidate.filters?.allowedDomains || []),
|
|
31
|
+
]),
|
|
32
|
+
excludedDomains: uniqueStrings([
|
|
33
|
+
...(existing.filters?.excludedDomains || []),
|
|
34
|
+
...(candidate.filters?.excludedDomains || []),
|
|
35
|
+
]),
|
|
36
|
+
};
|
|
37
|
+
existing.enableImageUnderstanding =
|
|
38
|
+
existing.enableImageUnderstanding || candidate.enableImageUnderstanding;
|
|
39
|
+
existing.includeSources = existing.includeSources || candidate.includeSources;
|
|
40
|
+
return true;
|
|
41
|
+
}
|
|
42
|
+
if (existing.type === "x_search" && candidate.type === "x_search") {
|
|
43
|
+
existing.allowedXHandles = uniqueStrings([
|
|
44
|
+
...(existing.allowedXHandles || []),
|
|
45
|
+
...(candidate.allowedXHandles || []),
|
|
46
|
+
]);
|
|
47
|
+
existing.excludedXHandles = uniqueStrings([
|
|
48
|
+
...(existing.excludedXHandles || []),
|
|
49
|
+
...(candidate.excludedXHandles || []),
|
|
50
|
+
]);
|
|
51
|
+
existing.fromDate = existing.fromDate || candidate.fromDate;
|
|
52
|
+
existing.toDate = existing.toDate || candidate.toDate;
|
|
53
|
+
existing.enableImageUnderstanding =
|
|
54
|
+
existing.enableImageUnderstanding || candidate.enableImageUnderstanding;
|
|
55
|
+
existing.enableVideoUnderstanding =
|
|
56
|
+
existing.enableVideoUnderstanding || candidate.enableVideoUnderstanding;
|
|
57
|
+
return true;
|
|
58
|
+
}
|
|
59
|
+
if (existing.type === "code_execution" && candidate.type === "code_execution") {
|
|
60
|
+
existing.includeOutputs = existing.includeOutputs || candidate.includeOutputs;
|
|
61
|
+
return true;
|
|
62
|
+
}
|
|
63
|
+
if (existing.type === "file_search" && candidate.type === "file_search") {
|
|
64
|
+
existing.collectionIds = uniqueStrings([
|
|
65
|
+
...existing.collectionIds,
|
|
66
|
+
...candidate.collectionIds,
|
|
67
|
+
]);
|
|
68
|
+
existing.retrievalMode = candidate.retrievalMode || existing.retrievalMode;
|
|
69
|
+
existing.maxNumResults = candidate.maxNumResults || existing.maxNumResults;
|
|
70
|
+
existing.includeResults = existing.includeResults || candidate.includeResults;
|
|
71
|
+
return true;
|
|
72
|
+
}
|
|
73
|
+
return false;
|
|
74
|
+
}
|
|
75
|
+
function sameServerTool(a, b) {
|
|
76
|
+
if (a.type !== b.type)
|
|
77
|
+
return false;
|
|
78
|
+
if (a.type !== "file_search" || b.type !== "file_search")
|
|
79
|
+
return true;
|
|
80
|
+
return true;
|
|
81
|
+
}
|
|
82
|
+
function uniqueStrings(values) {
|
|
83
|
+
return [...new Set(values.filter(Boolean))];
|
|
84
|
+
}
|
|
85
|
+
export function hasServerTool(tools, kind) {
|
|
86
|
+
return tools.some((tool) => tool.type === kind);
|
|
87
|
+
}
|
|
88
|
+
export function describeServerTools(tools) {
|
|
89
|
+
return tools.map((tool) => {
|
|
90
|
+
switch (tool.type) {
|
|
91
|
+
case "web_search":
|
|
92
|
+
return "web search";
|
|
93
|
+
case "x_search":
|
|
94
|
+
return "X/Twitter search";
|
|
95
|
+
case "code_execution":
|
|
96
|
+
return "Python code execution (sandbox)";
|
|
97
|
+
case "file_search":
|
|
98
|
+
return `file search (${tool.collectionIds.length} collection${tool.collectionIds.length === 1 ? "" : "s"})`;
|
|
99
|
+
}
|
|
100
|
+
});
|
|
101
|
+
}
|
|
102
|
+
export function serializeServerTools(tools, mcpServers) {
|
|
103
|
+
const serialized = [];
|
|
104
|
+
for (const tool of tools) {
|
|
105
|
+
switch (tool.type) {
|
|
106
|
+
case "web_search":
|
|
107
|
+
serialized.push(serializeWebSearch(tool));
|
|
108
|
+
break;
|
|
109
|
+
case "x_search":
|
|
110
|
+
serialized.push(serializeXSearch(tool));
|
|
111
|
+
break;
|
|
112
|
+
case "code_execution":
|
|
113
|
+
serialized.push({ type: "code_interpreter" });
|
|
114
|
+
break;
|
|
115
|
+
case "file_search":
|
|
116
|
+
serialized.push(serializeFileSearch(tool));
|
|
117
|
+
break;
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
for (const mcp of mcpServers) {
|
|
121
|
+
const entry = {
|
|
122
|
+
type: "mcp",
|
|
123
|
+
server_url: mcp.url,
|
|
124
|
+
server_label: mcp.label,
|
|
125
|
+
};
|
|
126
|
+
if (mcp.description)
|
|
127
|
+
entry.server_description = mcp.description;
|
|
128
|
+
if (mcp.allowedTools && mcp.allowedTools.length > 0) {
|
|
129
|
+
entry.allowed_tools = mcp.allowedTools;
|
|
130
|
+
}
|
|
131
|
+
serialized.push(entry);
|
|
132
|
+
}
|
|
133
|
+
return serialized;
|
|
134
|
+
}
|
|
135
|
+
function serializeWebSearch(tool) {
|
|
136
|
+
const entry = { type: "web_search" };
|
|
137
|
+
const filters = {};
|
|
138
|
+
if (tool.filters?.allowedDomains && tool.filters.allowedDomains.length > 0) {
|
|
139
|
+
filters.allowed_domains = tool.filters.allowedDomains;
|
|
140
|
+
}
|
|
141
|
+
if (tool.filters?.excludedDomains && tool.filters.excludedDomains.length > 0) {
|
|
142
|
+
filters.excluded_domains = tool.filters.excludedDomains;
|
|
143
|
+
}
|
|
144
|
+
if (Object.keys(filters).length > 0)
|
|
145
|
+
entry.filters = filters;
|
|
146
|
+
if (tool.enableImageUnderstanding)
|
|
147
|
+
entry.enable_image_understanding = true;
|
|
148
|
+
return entry;
|
|
149
|
+
}
|
|
150
|
+
function serializeXSearch(tool) {
|
|
151
|
+
const entry = { type: "x_search" };
|
|
152
|
+
if (tool.allowedXHandles && tool.allowedXHandles.length > 0) {
|
|
153
|
+
entry.allowed_x_handles = tool.allowedXHandles;
|
|
154
|
+
}
|
|
155
|
+
if (tool.excludedXHandles && tool.excludedXHandles.length > 0) {
|
|
156
|
+
entry.excluded_x_handles = tool.excludedXHandles;
|
|
157
|
+
}
|
|
158
|
+
if (tool.fromDate)
|
|
159
|
+
entry.from_date = tool.fromDate;
|
|
160
|
+
if (tool.toDate)
|
|
161
|
+
entry.to_date = tool.toDate;
|
|
162
|
+
if (tool.enableImageUnderstanding)
|
|
163
|
+
entry.enable_image_understanding = true;
|
|
164
|
+
if (tool.enableVideoUnderstanding)
|
|
165
|
+
entry.enable_video_understanding = true;
|
|
166
|
+
return entry;
|
|
167
|
+
}
|
|
168
|
+
function serializeFileSearch(tool) {
|
|
169
|
+
const entry = {
|
|
170
|
+
type: "file_search",
|
|
171
|
+
vector_store_ids: tool.collectionIds,
|
|
172
|
+
};
|
|
173
|
+
if (tool.maxNumResults)
|
|
174
|
+
entry.max_num_results = tool.maxNumResults;
|
|
175
|
+
if (tool.retrievalMode) {
|
|
176
|
+
entry.retrieval_mode = { type: tool.retrievalMode };
|
|
177
|
+
}
|
|
178
|
+
return entry;
|
|
179
|
+
}
|
|
180
|
+
export function collectResponseIncludes(tools, includeToolOutputs) {
|
|
181
|
+
const includes = [];
|
|
182
|
+
for (const tool of tools) {
|
|
183
|
+
switch (tool.type) {
|
|
184
|
+
case "web_search":
|
|
185
|
+
if (includeToolOutputs || tool.includeSources) {
|
|
186
|
+
includes.push("web_search_call.action.sources");
|
|
187
|
+
}
|
|
188
|
+
break;
|
|
189
|
+
case "code_execution":
|
|
190
|
+
if (includeToolOutputs || tool.includeOutputs) {
|
|
191
|
+
includes.push("code_interpreter_call.outputs");
|
|
192
|
+
}
|
|
193
|
+
break;
|
|
194
|
+
case "file_search":
|
|
195
|
+
if (includeToolOutputs || tool.includeResults) {
|
|
196
|
+
includes.push("file_search_call.results");
|
|
197
|
+
}
|
|
198
|
+
break;
|
|
199
|
+
default:
|
|
200
|
+
break;
|
|
201
|
+
}
|
|
202
|
+
}
|
|
203
|
+
return uniqueStrings(includes);
|
|
204
|
+
}
|
|
205
|
+
//# sourceMappingURL=server-tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server-tools.js","sourceRoot":"","sources":["../src/server-tools.ts"],"names":[],"mappings":"AAUA,MAAM,UAAU,oBAAoB,CAClC,KAA2D;IAE3D,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IAEtB,MAAM,UAAU,GAAuB,EAAE,CAAC;IAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,SAAS,GACb,OAAO,IAAI,KAAK,QAAQ;YACtB,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC3B,CAAC,CAAC,IAAI,CAAC;QACX,IAAI,CAAC,SAAS;YAAE,SAAS;QACzB,MAAM,MAAM,GAAG,eAAe,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,MAAM;YAAE,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAoB;IAC/C,IAAI,IAAI,KAAK,aAAa;QAAE,OAAO,IAAI,CAAC;IACxC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;AACxB,CAAC;AAED,SAAS,eAAe,CACtB,KAAyB,EACzB,SAA2B;IAE3B,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC;IACvE,IAAI,CAAC,QAAQ;QAAE,OAAO,KAAK,CAAC;IAE5B,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY,IAAI,SAAS,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QACtE,QAAQ,CAAC,OAAO,GAAG;YACjB,cAAc,EAAE,aAAa,CAAC;gBAC5B,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE,cAAc,IAAI,EAAE,CAAC;gBAC3C,GAAG,CAAC,SAAS,CAAC,OAAO,EAAE,cAAc,IAAI,EAAE,CAAC;aAC7C,CAAC;YACF,eAAe,EAAE,aAAa,CAAC;gBAC7B,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE,eAAe,IAAI,EAAE,CAAC;gBAC5C,GAAG,CAAC,SAAS,CAAC,OAAO,EAAE,eAAe,IAAI,EAAE,CAAC;aAC9C,CAAC;SACH,CAAC;QACF,QAAQ,CAAC,wBAAwB;YAC/B,QAAQ,CAAC,wBAAwB,IAAI,SAAS,CAAC,wBAAwB,CAAC;QAC1E,QAAQ,CAAC,cAAc,GAAG,QAAQ,CAAC,cAAc,IAAI,SAAS,CAAC,cAAc,CAAC;QAC9E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,QAAQ,CAAC,IAAI,KAAK,UAAU,IAAI,SAAS,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAClE,QAAQ,CAAC,eAAe,GAAG,aAAa,CAAC;YACvC,GAAG,CAAC,QAAQ,CAAC,eAAe,IAAI,EAAE,CAAC;YACnC,GAAG,CAAC,SAAS,CAAC,eAAe,IAAI,EAAE,CAAC;SACrC,CAAC,CAAC;QACH,QAAQ,CAAC,gBAAgB,GAAG,aAAa,CAAC;YACxC,GAAG,CAAC,QAAQ,CAAC,gBAAgB,IAAI,EAAE,CAAC;YACpC,GAAG,CAAC,SAAS,CAAC,gBAAgB,IAAI,EAAE,CAAC;SACtC,CAAC,CAAC;QACH,QAAQ,CAAC,QAAQ,GAAG,QAAQ,CAAC,QAAQ,IAAI,SAAS,CAAC,QAAQ,CAAC;QAC5D,QAAQ,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM,IAAI,SAAS,CAAC,MAAM,CAAC;QACtD,QAAQ,CAAC,wBAAwB;YAC/B,QAAQ,CAAC,wBAAwB,IAAI,SAAS,CAAC,wBAAwB,CAAC;QAC1E,QAAQ,CAAC,wBAAwB;YAC/B,QAAQ,CAAC,wBAAwB,IAAI,SAAS,CAAC,wBAAwB,CAAC;QAC1E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,QAAQ,CAAC,IAAI,KAAK,gBAAgB,IAAI,SAAS,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;QAC9E,QAAQ,CAAC,cAAc,GAAG,QAAQ,CAAC,cAAc,IAAI,SAAS,CAAC,cAAc,CAAC;QAC9E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,QAAQ,CAAC,IAAI,KAAK,aAAa,IAAI,SAAS,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;QACxE,QAAQ,CAAC,aAAa,GAAG,aAAa,CAAC;YACrC,GAAG,QAAQ,CAAC,aAAa;YACzB,GAAG,SAAS,CAAC,aAAa;SAC3B,CAAC,CAAC;QACH,QAAQ,CAAC,aAAa,GAAG,SAAS,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC;QAC3E,QAAQ,CAAC,aAAa,GAAG,SAAS,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC;QAC3E,QAAQ,CAAC,cAAc,GAAG,QAAQ,CAAC,cAAc,IAAI,SAAS,CAAC,cAAc,CAAC;QAC9E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,cAAc,CAAC,CAAmB,EAAE,CAAmB;IAC9D,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;QAAE,OAAO,KAAK,CAAC;IACpC,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa;QAAE,OAAO,IAAI,CAAC;IACtE,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,aAAa,CAAC,MAAgB;IACrC,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,KAAyB,EACzB,IAAoB;IAEpB,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAyB;IAC3D,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACxB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,YAAY;gBACf,OAAO,YAAY,CAAC;YACtB,KAAK,UAAU;gBACb,OAAO,kBAAkB,CAAC;YAC5B,KAAK,gBAAgB;gBACnB,OAAO,iCAAiC,CAAC;YAC3C,KAAK,aAAa;gBAChB,OAAO,gBAAgB,IAAI,CAAC,aAAa,CAAC,MAAM,cAAc,IAAI,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;QAChH,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,KAAyB,EACzB,UAAuB;IAEvB,MAAM,UAAU,GAAU,EAAE,CAAC;IAE7B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,YAAY;gBACf,UAAU,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC1C,MAAM;YACR,KAAK,UAAU;gBACb,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,gBAAgB;gBACnB,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;gBAC9C,MAAM;YACR,KAAK,aAAa;gBAChB,UAAU,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC3C,MAAM;QACV,CAAC;IACH,CAAC;IAED,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAQ;YACjB,IAAI,EAAE,KAAK;YACX,UAAU,EAAE,GAAG,CAAC,GAAG;YACnB,YAAY,EAAE,GAAG,CAAC,KAAK;SACxB,CAAC;QACF,IAAI,GAAG,CAAC,WAAW;YAAE,KAAK,CAAC,kBAAkB,GAAG,GAAG,CAAC,WAAW,CAAC;QAChE,IAAI,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,KAAK,CAAC,aAAa,GAAG,GAAG,CAAC,YAAY,CAAC;QACzC,CAAC;QACD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAyB;IACnD,MAAM,KAAK,GAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;IAC1C,MAAM,OAAO,GAA4B,EAAE,CAAC;IAC5C,IAAI,IAAI,CAAC,OAAO,EAAE,cAAc,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3E,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC;IACxD,CAAC;IACD,IAAI,IAAI,CAAC,OAAO,EAAE,eAAe,IAAI,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7E,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;IAC1D,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC;QAAE,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;IAC7D,IAAI,IAAI,CAAC,wBAAwB;QAAE,KAAK,CAAC,0BAA0B,GAAG,IAAI,CAAC;IAC3E,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAuB;IAC/C,MAAM,KAAK,GAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;IACxC,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5D,KAAK,CAAC,iBAAiB,GAAG,IAAI,CAAC,eAAe,CAAC;IACjD,CAAC;IACD,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9D,KAAK,CAAC,kBAAkB,GAAG,IAAI,CAAC,gBAAgB,CAAC;IACnD,CAAC;IACD,IAAI,IAAI,CAAC,QAAQ;QAAE,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;IACnD,IAAI,IAAI,CAAC,MAAM;QAAE,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;IAC7C,IAAI,IAAI,CAAC,wBAAwB;QAAE,KAAK,CAAC,0BAA0B,GAAG,IAAI,CAAC;IAC3E,IAAI,IAAI,CAAC,wBAAwB;QAAE,KAAK,CAAC,0BAA0B,GAAG,IAAI,CAAC;IAC3E,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,mBAAmB,CAAC,IAA0B;IACrD,MAAM,KAAK,GAAQ;QACjB,IAAI,EAAE,aAAa;QACnB,gBAAgB,EAAE,IAAI,CAAC,aAAa;KACrC,CAAC;IACF,IAAI,IAAI,CAAC,aAAa;QAAE,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,aAAa,CAAC;IACnE,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,KAAK,CAAC,cAAc,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC;IACtD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,KAAyB,EACzB,kBAA2B;IAE3B,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,YAAY;gBACf,IAAI,kBAAkB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;oBAC9C,QAAQ,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;gBAClD,CAAC;gBACD,MAAM;YACR,KAAK,gBAAgB;gBACnB,IAAI,kBAAkB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;oBAC9C,QAAQ,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;gBACjD,CAAC;gBACD,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,kBAAkB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;oBAC9C,QAAQ,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;gBAC5C,CAAC;gBACD,MAAM;YACR;gBACE,MAAM;QACV,CAAC;IACH,CAAC;IAED,OAAO,aAAa,CAAC,QAAQ,CAAC,CAAC;AACjC,CAAC"}
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
import type { ChatMessage, SerializedToolCall, SessionEvent, SessionMeta } from "./types.js";
|
|
2
|
+
export declare class SessionManager {
|
|
3
|
+
private sessionDir;
|
|
4
|
+
private archivedDir;
|
|
5
|
+
constructor(baseDir: string);
|
|
6
|
+
private ensureDir;
|
|
7
|
+
private sessionPath;
|
|
8
|
+
private resolveSessionPath;
|
|
9
|
+
createSession(opts: {
|
|
10
|
+
model: string;
|
|
11
|
+
cwd: string;
|
|
12
|
+
name?: string;
|
|
13
|
+
}): SessionMeta;
|
|
14
|
+
appendMessage(sessionId: string, role: "system" | "user" | "assistant" | "tool", content: string | null, extra?: {
|
|
15
|
+
toolCalls?: SerializedToolCall[];
|
|
16
|
+
toolCallId?: string;
|
|
17
|
+
turn?: number;
|
|
18
|
+
}): void;
|
|
19
|
+
appendToolExec(sessionId: string, toolName: string, toolArgs: string, toolOutput: string, toolError: boolean, turn?: number): void;
|
|
20
|
+
updateMeta(sessionId: string, updates: Partial<SessionMeta>): void;
|
|
21
|
+
renameSession(sessionId: string, name: string): boolean;
|
|
22
|
+
loadSession(sessionId: string): {
|
|
23
|
+
meta: SessionMeta;
|
|
24
|
+
messages: ChatMessage[];
|
|
25
|
+
} | null;
|
|
26
|
+
readSessionEvents(sessionId: string): SessionEvent[];
|
|
27
|
+
rewriteSession(sessionId: string, events: SessionEvent[]): boolean;
|
|
28
|
+
rollbackTurns(sessionId: string, numTurns: number): boolean;
|
|
29
|
+
archiveSession(sessionId: string): boolean;
|
|
30
|
+
unarchiveSession(sessionId: string): boolean;
|
|
31
|
+
listSessions(opts?: {
|
|
32
|
+
archived?: boolean;
|
|
33
|
+
includeArchived?: boolean;
|
|
34
|
+
}): SessionMeta[];
|
|
35
|
+
deleteSession(sessionId: string): boolean;
|
|
36
|
+
clearSessions(opts?: {
|
|
37
|
+
archived?: boolean;
|
|
38
|
+
}): number;
|
|
39
|
+
private loadMetasFromDir;
|
|
40
|
+
private eventToMessage;
|
|
41
|
+
private groupTurnEvents;
|
|
42
|
+
private generateId;
|
|
43
|
+
sessionExists(sessionId: string): boolean;
|
|
44
|
+
autoName(prompt: string): string;
|
|
45
|
+
}
|
|
46
|
+
//# sourceMappingURL=session.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../src/session.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,WAAW,EACX,kBAAkB,EAClB,YAAY,EACZ,WAAW,EACZ,MAAM,YAAY,CAAC;AAKpB,qBAAa,cAAc;IACzB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,WAAW,CAAS;gBAEhB,OAAO,EAAE,MAAM;IAM3B,OAAO,CAAC,SAAS;IASjB,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,kBAAkB;IAQ1B,aAAa,CAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,WAAW;IAqB/E,aAAa,CACX,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,EAC9C,OAAO,EAAE,MAAM,GAAG,IAAI,EACtB,KAAK,CAAC,EAAE;QACN,SAAS,CAAC,EAAE,kBAAkB,EAAE,CAAC;QACjC,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,GACA,IAAI;IAaP,cAAc,CACZ,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,OAAO,EAClB,IAAI,CAAC,EAAE,MAAM,GACZ,IAAI;IAeP,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,IAAI;IAiBlE,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO;IAMvD,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG;QAAE,IAAI,EAAE,WAAW,CAAC;QAAC,QAAQ,EAAE,WAAW,EAAE,CAAA;KAAE,GAAG,IAAI;IAmBrF,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,YAAY,EAAE;IAoBpD,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,OAAO;IAQlE,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO;IAwB3D,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAS1C,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAS5C,YAAY,CAAC,IAAI,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,OAAO,CAAC;QAAC,eAAe,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,WAAW,EAAE;IAkBrF,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAOzC,aAAa,CAAC,IAAI,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,MAAM;IAepD,OAAO,CAAC,gBAAgB;IAqBxB,OAAO,CAAC,cAAc;IA8BtB,OAAO,CAAC,eAAe;IAiBvB,OAAO,CAAC,UAAU;IAMlB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAIzC,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;CAKjC"}
|
package/dist/session.js
ADDED
|
@@ -0,0 +1,305 @@
|
|
|
1
|
+
import fs from "node:fs";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import crypto from "node:crypto";
|
|
4
|
+
const SESSION_DIR_NAME = "sessions";
|
|
5
|
+
const ARCHIVED_DIR_NAME = "archived";
|
|
6
|
+
export class SessionManager {
|
|
7
|
+
sessionDir;
|
|
8
|
+
archivedDir;
|
|
9
|
+
constructor(baseDir) {
|
|
10
|
+
this.sessionDir = path.join(baseDir, SESSION_DIR_NAME);
|
|
11
|
+
this.archivedDir = path.join(this.sessionDir, ARCHIVED_DIR_NAME);
|
|
12
|
+
this.ensureDir();
|
|
13
|
+
}
|
|
14
|
+
ensureDir() {
|
|
15
|
+
try {
|
|
16
|
+
fs.mkdirSync(this.sessionDir, { recursive: true });
|
|
17
|
+
fs.mkdirSync(this.archivedDir, { recursive: true });
|
|
18
|
+
}
|
|
19
|
+
catch (err) {
|
|
20
|
+
throw new Error(`Unable to initialize session storage at ${this.sessionDir}: ${err.message}`);
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
sessionPath(id, archived = false) {
|
|
24
|
+
return path.join(archived ? this.archivedDir : this.sessionDir, `${id}.jsonl`);
|
|
25
|
+
}
|
|
26
|
+
resolveSessionPath(id) {
|
|
27
|
+
const active = this.sessionPath(id, false);
|
|
28
|
+
if (fs.existsSync(active))
|
|
29
|
+
return { filePath: active, archived: false };
|
|
30
|
+
const archived = this.sessionPath(id, true);
|
|
31
|
+
if (fs.existsSync(archived))
|
|
32
|
+
return { filePath: archived, archived: true };
|
|
33
|
+
return null;
|
|
34
|
+
}
|
|
35
|
+
createSession(opts) {
|
|
36
|
+
const id = this.generateId();
|
|
37
|
+
const now = new Date().toISOString();
|
|
38
|
+
const meta = {
|
|
39
|
+
id,
|
|
40
|
+
name: opts.name || "New session",
|
|
41
|
+
model: opts.model,
|
|
42
|
+
cwd: opts.cwd,
|
|
43
|
+
archived: false,
|
|
44
|
+
created: now,
|
|
45
|
+
updated: now,
|
|
46
|
+
turns: 0,
|
|
47
|
+
lastResponseId: null,
|
|
48
|
+
};
|
|
49
|
+
const event = { ts: now, type: "meta", meta };
|
|
50
|
+
fs.writeFileSync(this.sessionPath(id), JSON.stringify(event) + "\n", "utf-8");
|
|
51
|
+
return meta;
|
|
52
|
+
}
|
|
53
|
+
appendMessage(sessionId, role, content, extra) {
|
|
54
|
+
const resolved = this.resolveSessionPath(sessionId);
|
|
55
|
+
if (!resolved)
|
|
56
|
+
return;
|
|
57
|
+
const event = {
|
|
58
|
+
ts: new Date().toISOString(),
|
|
59
|
+
type: "msg",
|
|
60
|
+
role,
|
|
61
|
+
content,
|
|
62
|
+
...extra,
|
|
63
|
+
};
|
|
64
|
+
fs.appendFileSync(resolved.filePath, JSON.stringify(event) + "\n", "utf-8");
|
|
65
|
+
}
|
|
66
|
+
appendToolExec(sessionId, toolName, toolArgs, toolOutput, toolError, turn) {
|
|
67
|
+
const resolved = this.resolveSessionPath(sessionId);
|
|
68
|
+
if (!resolved)
|
|
69
|
+
return;
|
|
70
|
+
const event = {
|
|
71
|
+
ts: new Date().toISOString(),
|
|
72
|
+
type: "tool_exec",
|
|
73
|
+
toolName,
|
|
74
|
+
toolArgs,
|
|
75
|
+
toolOutput: toolOutput.slice(0, 10000),
|
|
76
|
+
toolError,
|
|
77
|
+
turn,
|
|
78
|
+
};
|
|
79
|
+
fs.appendFileSync(resolved.filePath, JSON.stringify(event) + "\n", "utf-8");
|
|
80
|
+
}
|
|
81
|
+
updateMeta(sessionId, updates) {
|
|
82
|
+
const resolved = this.resolveSessionPath(sessionId);
|
|
83
|
+
if (!resolved)
|
|
84
|
+
return;
|
|
85
|
+
const lines = fs.readFileSync(resolved.filePath, "utf-8").split("\n").filter(Boolean);
|
|
86
|
+
if (lines.length === 0)
|
|
87
|
+
return;
|
|
88
|
+
const firstEvent = JSON.parse(lines[0]);
|
|
89
|
+
if (firstEvent.type === "meta" && firstEvent.meta) {
|
|
90
|
+
Object.assign(firstEvent.meta, updates, {
|
|
91
|
+
archived: resolved.archived,
|
|
92
|
+
updated: new Date().toISOString(),
|
|
93
|
+
});
|
|
94
|
+
lines[0] = JSON.stringify(firstEvent);
|
|
95
|
+
fs.writeFileSync(resolved.filePath, lines.join("\n") + "\n", "utf-8");
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
renameSession(sessionId, name) {
|
|
99
|
+
if (!this.sessionExists(sessionId))
|
|
100
|
+
return false;
|
|
101
|
+
this.updateMeta(sessionId, { name });
|
|
102
|
+
return true;
|
|
103
|
+
}
|
|
104
|
+
loadSession(sessionId) {
|
|
105
|
+
const events = this.readSessionEvents(sessionId);
|
|
106
|
+
if (events.length === 0)
|
|
107
|
+
return null;
|
|
108
|
+
let meta = null;
|
|
109
|
+
const messages = [];
|
|
110
|
+
for (const event of events) {
|
|
111
|
+
if (event.type === "meta" && event.meta) {
|
|
112
|
+
meta = event.meta;
|
|
113
|
+
}
|
|
114
|
+
else if (event.type === "msg") {
|
|
115
|
+
const msg = this.eventToMessage(event);
|
|
116
|
+
if (msg)
|
|
117
|
+
messages.push(msg);
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
if (!meta)
|
|
121
|
+
return null;
|
|
122
|
+
return { meta, messages };
|
|
123
|
+
}
|
|
124
|
+
readSessionEvents(sessionId) {
|
|
125
|
+
const resolved = this.resolveSessionPath(sessionId);
|
|
126
|
+
if (!resolved)
|
|
127
|
+
return [];
|
|
128
|
+
const lines = fs.readFileSync(resolved.filePath, "utf-8").split("\n").filter(Boolean);
|
|
129
|
+
const events = [];
|
|
130
|
+
for (const line of lines) {
|
|
131
|
+
try {
|
|
132
|
+
const event = JSON.parse(line);
|
|
133
|
+
if (event.type === "meta" && event.meta) {
|
|
134
|
+
event.meta.archived = resolved.archived;
|
|
135
|
+
}
|
|
136
|
+
events.push(event);
|
|
137
|
+
}
|
|
138
|
+
catch {
|
|
139
|
+
// Skip malformed lines.
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
return events;
|
|
143
|
+
}
|
|
144
|
+
rewriteSession(sessionId, events) {
|
|
145
|
+
const resolved = this.resolveSessionPath(sessionId);
|
|
146
|
+
if (!resolved)
|
|
147
|
+
return false;
|
|
148
|
+
const serialized = events.map((event) => JSON.stringify(event)).join("\n");
|
|
149
|
+
fs.writeFileSync(resolved.filePath, serialized + (serialized ? "\n" : ""), "utf-8");
|
|
150
|
+
return true;
|
|
151
|
+
}
|
|
152
|
+
rollbackTurns(sessionId, numTurns) {
|
|
153
|
+
if (numTurns <= 0)
|
|
154
|
+
return true;
|
|
155
|
+
const events = this.readSessionEvents(sessionId);
|
|
156
|
+
if (events.length === 0)
|
|
157
|
+
return false;
|
|
158
|
+
const metaEvent = events.find((event) => event.type === "meta" && event.meta);
|
|
159
|
+
if (!metaEvent?.meta)
|
|
160
|
+
return false;
|
|
161
|
+
const bodyEvents = events.slice(1);
|
|
162
|
+
const firstTurnIndex = bodyEvents.findIndex((event) => event.type === "msg" && event.role === "user");
|
|
163
|
+
const prefixEvents = firstTurnIndex >= 0 ? bodyEvents.slice(0, firstTurnIndex) : bodyEvents;
|
|
164
|
+
const turnEvents = firstTurnIndex >= 0 ? this.groupTurnEvents(bodyEvents.slice(firstTurnIndex)) : [];
|
|
165
|
+
if (turnEvents.length === 0)
|
|
166
|
+
return true;
|
|
167
|
+
const keptTurns = Math.max(0, turnEvents.length - numTurns);
|
|
168
|
+
const keptEvents = [...prefixEvents, ...turnEvents.slice(0, keptTurns).flat()];
|
|
169
|
+
metaEvent.meta.turns = keptTurns;
|
|
170
|
+
metaEvent.meta.updated = new Date().toISOString();
|
|
171
|
+
return this.rewriteSession(sessionId, [metaEvent, ...keptEvents]);
|
|
172
|
+
}
|
|
173
|
+
archiveSession(sessionId) {
|
|
174
|
+
const active = this.sessionPath(sessionId);
|
|
175
|
+
if (!fs.existsSync(active))
|
|
176
|
+
return false;
|
|
177
|
+
const archived = this.sessionPath(sessionId, true);
|
|
178
|
+
fs.renameSync(active, archived);
|
|
179
|
+
this.updateMeta(sessionId, { archived: true });
|
|
180
|
+
return true;
|
|
181
|
+
}
|
|
182
|
+
unarchiveSession(sessionId) {
|
|
183
|
+
const archived = this.sessionPath(sessionId, true);
|
|
184
|
+
if (!fs.existsSync(archived))
|
|
185
|
+
return false;
|
|
186
|
+
const active = this.sessionPath(sessionId);
|
|
187
|
+
fs.renameSync(archived, active);
|
|
188
|
+
this.updateMeta(sessionId, { archived: false });
|
|
189
|
+
return true;
|
|
190
|
+
}
|
|
191
|
+
listSessions(opts) {
|
|
192
|
+
this.ensureDir();
|
|
193
|
+
const sessions = [];
|
|
194
|
+
const includeActive = opts?.archived !== true;
|
|
195
|
+
const includeArchived = opts?.archived === true || opts?.includeArchived;
|
|
196
|
+
if (includeActive) {
|
|
197
|
+
sessions.push(...this.loadMetasFromDir(this.sessionDir, false));
|
|
198
|
+
}
|
|
199
|
+
if (includeArchived) {
|
|
200
|
+
sessions.push(...this.loadMetasFromDir(this.archivedDir, true));
|
|
201
|
+
}
|
|
202
|
+
sessions.sort((a, b) => new Date(b.updated).getTime() - new Date(a.updated).getTime());
|
|
203
|
+
return sessions;
|
|
204
|
+
}
|
|
205
|
+
deleteSession(sessionId) {
|
|
206
|
+
const resolved = this.resolveSessionPath(sessionId);
|
|
207
|
+
if (!resolved)
|
|
208
|
+
return false;
|
|
209
|
+
fs.unlinkSync(resolved.filePath);
|
|
210
|
+
return true;
|
|
211
|
+
}
|
|
212
|
+
clearSessions(opts) {
|
|
213
|
+
const dirs = opts?.archived ? [this.archivedDir] : [this.sessionDir, this.archivedDir];
|
|
214
|
+
let deleted = 0;
|
|
215
|
+
for (const dir of dirs) {
|
|
216
|
+
const files = fs.readdirSync(dir).filter((file) => file.endsWith(".jsonl"));
|
|
217
|
+
for (const file of files) {
|
|
218
|
+
fs.unlinkSync(path.join(dir, file));
|
|
219
|
+
deleted++;
|
|
220
|
+
}
|
|
221
|
+
}
|
|
222
|
+
return deleted;
|
|
223
|
+
}
|
|
224
|
+
loadMetasFromDir(dir, archived) {
|
|
225
|
+
const files = fs.readdirSync(dir).filter((file) => file.endsWith(".jsonl"));
|
|
226
|
+
const sessions = [];
|
|
227
|
+
for (const file of files) {
|
|
228
|
+
try {
|
|
229
|
+
const firstLine = fs.readFileSync(path.join(dir, file), "utf-8").split("\n")[0];
|
|
230
|
+
if (!firstLine)
|
|
231
|
+
continue;
|
|
232
|
+
const event = JSON.parse(firstLine);
|
|
233
|
+
if (event.type === "meta" && event.meta) {
|
|
234
|
+
event.meta.archived = archived;
|
|
235
|
+
sessions.push(event.meta);
|
|
236
|
+
}
|
|
237
|
+
}
|
|
238
|
+
catch {
|
|
239
|
+
// Skip corrupted session files.
|
|
240
|
+
}
|
|
241
|
+
}
|
|
242
|
+
return sessions;
|
|
243
|
+
}
|
|
244
|
+
eventToMessage(event) {
|
|
245
|
+
if (!event.role)
|
|
246
|
+
return null;
|
|
247
|
+
switch (event.role) {
|
|
248
|
+
case "system":
|
|
249
|
+
return { role: "system", content: event.content || "" };
|
|
250
|
+
case "user":
|
|
251
|
+
return { role: "user", content: event.content || "" };
|
|
252
|
+
case "assistant": {
|
|
253
|
+
const msg = { role: "assistant", content: event.content || null };
|
|
254
|
+
if (event.toolCalls && event.toolCalls.length > 0) {
|
|
255
|
+
msg.tool_calls = event.toolCalls.map((tc) => ({
|
|
256
|
+
id: tc.id,
|
|
257
|
+
type: "function",
|
|
258
|
+
function: { name: tc.name, arguments: tc.arguments },
|
|
259
|
+
}));
|
|
260
|
+
}
|
|
261
|
+
return msg;
|
|
262
|
+
}
|
|
263
|
+
case "tool":
|
|
264
|
+
return {
|
|
265
|
+
role: "tool",
|
|
266
|
+
tool_call_id: event.toolCallId || "",
|
|
267
|
+
content: event.content || "",
|
|
268
|
+
};
|
|
269
|
+
default:
|
|
270
|
+
return null;
|
|
271
|
+
}
|
|
272
|
+
}
|
|
273
|
+
groupTurnEvents(events) {
|
|
274
|
+
const turns = [];
|
|
275
|
+
let current = [];
|
|
276
|
+
for (const event of events) {
|
|
277
|
+
if (event.type === "msg" && event.role === "user") {
|
|
278
|
+
if (current.length > 0)
|
|
279
|
+
turns.push(current);
|
|
280
|
+
current = [event];
|
|
281
|
+
}
|
|
282
|
+
else if (current.length > 0) {
|
|
283
|
+
current.push(event);
|
|
284
|
+
}
|
|
285
|
+
}
|
|
286
|
+
if (current.length > 0)
|
|
287
|
+
turns.push(current);
|
|
288
|
+
return turns;
|
|
289
|
+
}
|
|
290
|
+
generateId() {
|
|
291
|
+
const timestamp = Date.now().toString(36);
|
|
292
|
+
const random = crypto.randomBytes(4).toString("hex");
|
|
293
|
+
return `${timestamp}-${random}`;
|
|
294
|
+
}
|
|
295
|
+
sessionExists(sessionId) {
|
|
296
|
+
return this.resolveSessionPath(sessionId) !== null;
|
|
297
|
+
}
|
|
298
|
+
autoName(prompt) {
|
|
299
|
+
const cleaned = prompt.replace(/\s+/g, " ").trim();
|
|
300
|
+
if (cleaned.length <= 60)
|
|
301
|
+
return cleaned;
|
|
302
|
+
return cleaned.slice(0, 57) + "...";
|
|
303
|
+
}
|
|
304
|
+
}
|
|
305
|
+
//# sourceMappingURL=session.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.js","sourceRoot":"","sources":["../src/session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,MAAM,MAAM,aAAa,CAAC;AAQjC,MAAM,gBAAgB,GAAG,UAAU,CAAC;AACpC,MAAM,iBAAiB,GAAG,UAAU,CAAC;AAErC,MAAM,OAAO,cAAc;IACjB,UAAU,CAAS;IACnB,WAAW,CAAS;IAE5B,YAAY,OAAe;QACzB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QACvD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC;QACjE,IAAI,CAAC,SAAS,EAAE,CAAC;IACnB,CAAC;IAEO,SAAS;QACf,IAAI,CAAC;YACH,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACnD,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,2CAA2C,IAAI,CAAC,UAAU,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAChG,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,EAAU,EAAE,QAAQ,GAAG,KAAK;QAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;IACjF,CAAC;IAEO,kBAAkB,CAAC,EAAU;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAC3C,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;YAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACxE,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC5C,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;QAC3E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa,CAAC,IAAmD;QAC/D,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAErC,MAAM,IAAI,GAAgB;YACxB,EAAE;YACF,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,aAAa;YAChC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE,GAAG;YACZ,OAAO,EAAE,GAAG;YACZ,KAAK,EAAE,CAAC;YACR,cAAc,EAAE,IAAI;SACrB,CAAC;QAEF,MAAM,KAAK,GAAiB,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QAC5D,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;QAC9E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa,CACX,SAAiB,EACjB,IAA8C,EAC9C,OAAsB,EACtB,KAIC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ;YAAE,OAAO;QACtB,MAAM,KAAK,GAAiB;YAC1B,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YAC5B,IAAI,EAAE,KAAK;YACX,IAAI;YACJ,OAAO;YACP,GAAG,KAAK;SACT,CAAC;QACF,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;IAC9E,CAAC;IAED,cAAc,CACZ,SAAiB,EACjB,QAAgB,EAChB,QAAgB,EAChB,UAAkB,EAClB,SAAkB,EAClB,IAAa;QAEb,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ;YAAE,OAAO;QACtB,MAAM,KAAK,GAAiB;YAC1B,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YAC5B,IAAI,EAAE,WAAW;YACjB,QAAQ;YACR,QAAQ;YACR,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;YACtC,SAAS;YACT,IAAI;SACL,CAAC;QACF,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;IAC9E,CAAC;IAED,UAAU,CAAC,SAAiB,EAAE,OAA6B;QACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ;YAAE,OAAO;QACtB,MAAM,KAAK,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACtF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE/B,MAAM,UAAU,GAAiB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACtD,IAAI,UAAU,CAAC,IAAI,KAAK,MAAM,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,OAAO,EAAE;gBACtC,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aAClC,CAAC,CAAC;YACH,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACtC,EAAE,CAAC,aAAa,CAAC,QAAQ,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,IAAY;QAC3C,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC;YAAE,OAAO,KAAK,CAAC;QACjD,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,WAAW,CAAC,SAAiB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAErC,IAAI,IAAI,GAAuB,IAAI,CAAC;QACpC,MAAM,QAAQ,GAAkB,EAAE,CAAC;QACnC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;gBACxC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;YACpB,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;gBAChC,MAAM,GAAG,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;gBACvC,IAAI,GAAG;oBAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QACvB,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;IAC5B,CAAC;IAED,iBAAiB,CAAC,SAAiB;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ;YAAE,OAAO,EAAE,CAAC;QAEzB,MAAM,KAAK,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACtF,MAAM,MAAM,GAAmB,EAAE,CAAC;QAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,KAAK,GAAiB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC7C,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;oBACxC,KAAK,CAAC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;gBAC1C,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,cAAc,CAAC,SAAiB,EAAE,MAAsB;QACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC5B,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3E,EAAE,CAAC,aAAa,CAAC,QAAQ,CAAC,QAAQ,EAAE,UAAU,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;QACpF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,QAAgB;QAC/C,IAAI,QAAQ,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAEtC,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC;QAC9E,IAAI,CAAC,SAAS,EAAE,IAAI;YAAE,OAAO,KAAK,CAAC;QAEnC,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,cAAc,GAAG,UAAU,CAAC,SAAS,CACzC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,CACzD,CAAC;QACF,MAAM,YAAY,GAAG,cAAc,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QAC5F,MAAM,UAAU,GAAG,cAAc,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACrG,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAEzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC;QAC5D,MAAM,UAAU,GAAG,CAAC,GAAG,YAAY,EAAE,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QAC/E,SAAS,CAAC,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;QACjC,SAAS,CAAC,IAAI,CAAC,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAElD,OAAO,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC,SAAS,EAAE,GAAG,UAAU,CAAC,CAAC,CAAC;IACpE,CAAC;IAED,cAAc,CAAC,SAAiB;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;YAAE,OAAO,KAAK,CAAC;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QACnD,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAChC,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB,CAAC,SAAiB;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAO,KAAK,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC3C,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,YAAY,CAAC,IAAwD;QACnE,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,MAAM,QAAQ,GAAkB,EAAE,CAAC;QAEnC,MAAM,aAAa,GAAG,IAAI,EAAE,QAAQ,KAAK,IAAI,CAAC;QAC9C,MAAM,eAAe,GAAG,IAAI,EAAE,QAAQ,KAAK,IAAI,IAAI,IAAI,EAAE,eAAe,CAAC;QAEzE,IAAI,aAAa,EAAE,CAAC;YAClB,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,eAAe,EAAE,CAAC;YACpB,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC;QAClE,CAAC;QAED,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QACvF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,aAAa,CAAC,SAAiB;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC5B,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACjC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa,CAAC,IAA6B;QACzC,MAAM,IAAI,GAAG,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACvF,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC5E,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;gBACpC,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,gBAAgB,CAAC,GAAW,EAAE,QAAiB;QACrD,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC5E,MAAM,QAAQ,GAAkB,EAAE,CAAC;QAEnC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAChF,IAAI,CAAC,SAAS;oBAAE,SAAS;gBACzB,MAAM,KAAK,GAAiB,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAClD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;oBACxC,KAAK,CAAC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;oBAC/B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,gCAAgC;YAClC,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,cAAc,CAAC,KAAmB;QACxC,IAAI,CAAC,KAAK,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QAE7B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,QAAQ;gBACX,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YAC1D,KAAK,MAAM;gBACT,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YACxD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,MAAM,GAAG,GAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;gBACvE,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAClD,GAAG,CAAC,UAAU,GAAG,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC5C,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,IAAI,EAAE,UAAmB;wBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE;qBACrD,CAAC,CAAC,CAAC;gBACN,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC;YACD,KAAK,MAAM;gBACT,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,KAAK,CAAC,UAAU,IAAI,EAAE;oBACpC,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,EAAE;iBAC7B,CAAC;YACJ;gBACE,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,MAAsB;QAC5C,MAAM,KAAK,GAAqB,EAAE,CAAC;QACnC,IAAI,OAAO,GAAmB,EAAE,CAAC;QAEjC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAClD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;oBAAE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC5C,OAAO,GAAG,CAAC,KAAK,CAAC,CAAC;YACpB,CAAC;iBAAM,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;YAAE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5C,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,UAAU;QAChB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACrD,OAAO,GAAG,SAAS,IAAI,MAAM,EAAE,CAAC;IAClC,CAAC;IAED,aAAa,CAAC,SAAiB;QAC7B,OAAO,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,KAAK,IAAI,CAAC;IACrD,CAAC;IAED,QAAQ,CAAC,MAAc;QACrB,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACnD,IAAI,OAAO,CAAC,MAAM,IAAI,EAAE;YAAE,OAAO,OAAO,CAAC;QACzC,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC;IACtC,CAAC;CACF"}
|
package/dist/stream.d.ts
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
export interface StreamAccumulator {
|
|
2
|
+
content: string;
|
|
3
|
+
reasoningContent: string;
|
|
4
|
+
toolCalls: AccumulatedToolCall[];
|
|
5
|
+
finishReason: string | null;
|
|
6
|
+
}
|
|
7
|
+
export interface AccumulatedToolCall {
|
|
8
|
+
index: number;
|
|
9
|
+
id: string;
|
|
10
|
+
function: {
|
|
11
|
+
name: string;
|
|
12
|
+
arguments: string;
|
|
13
|
+
};
|
|
14
|
+
}
|
|
15
|
+
export declare function createAccumulator(): StreamAccumulator;
|
|
16
|
+
export declare function processChunk(acc: StreamAccumulator, chunk: any, options: {
|
|
17
|
+
showReasoning: boolean;
|
|
18
|
+
showOutput: boolean;
|
|
19
|
+
}): void;
|
|
20
|
+
export declare function formatToolCall(name: string, args: string, verbose: boolean): string;
|
|
21
|
+
export declare function formatToolResult(output: string, isError: boolean): string;
|
|
22
|
+
//# sourceMappingURL=stream.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../src/stream.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,MAAM,CAAC;IACzB,SAAS,EAAE,mBAAmB,EAAE,CAAC;IACjC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B;AAED,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,MAAM,CAAC;IACd,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,wBAAgB,iBAAiB,IAAI,iBAAiB,CAOrD;AAED,wBAAgB,YAAY,CAC1B,GAAG,EAAE,iBAAiB,EACtB,KAAK,EAAE,GAAG,EACV,OAAO,EAAE;IAAE,aAAa,EAAE,OAAO,CAAC;IAAC,UAAU,EAAE,OAAO,CAAA;CAAE,GACvD,IAAI,CA4CN;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,MAAM,CAWnF;AA4BD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,MAAM,CAQzE"}
|