@langgraph-js/pure-graph 2.4.1 → 2.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/README.md +2 -2
- package/dist/adapter/hono/index.js +3 -3
- package/dist/adapter/hono/index.js.map +1 -1
- package/dist/adapter/nextjs/index.js +2 -2
- package/dist/checkpoint-D1UR3Ika.js +374 -0
- package/dist/checkpoint-D1UR3Ika.js.map +1 -0
- package/dist/{createEndpoint-Bxzbc61h.js → createEndpoint-pEWA0kt2.js} +2 -2
- package/dist/{createEndpoint-Bxzbc61h.js.map → createEndpoint-pEWA0kt2.js.map} +1 -1
- package/dist/index.js +2 -2
- package/dist/{queue-D3gPPSDe.js → queue-5gRqr4RI.js} +2 -2
- package/dist/{queue-D3gPPSDe.js.map → queue-5gRqr4RI.js.map} +1 -1
- package/dist/{router-DOVN2of2.js → router-DIyq7O-Q.js} +3 -3
- package/dist/{router-DOVN2of2.js.map → router-DIyq7O-Q.js.map} +1 -1
- package/dist/{sqlite-adapter-BNYk10Gl.js → sqlite-adapter-5PeLHaxe.js} +5 -10
- package/dist/sqlite-adapter-5PeLHaxe.js.map +1 -0
- package/dist/storage/kysely/sqlite-adapter.d.ts +1 -2
- package/dist/storage/sqlite/checkpoint.d.ts +29 -7
- package/dist/{stream-CAZ9j0yn.js → stream-Bwuk4b6F.js} +5 -5
- package/dist/stream-Bwuk4b6F.js.map +1 -0
- package/package.json +6 -6
- package/dist/checkpoint-DdL-Wo1x.js +0 -377
- package/dist/checkpoint-DdL-Wo1x.js.map +0 -1
- package/dist/sqlite-adapter-BNYk10Gl.js.map +0 -1
- package/dist/stream-CAZ9j0yn.js.map +0 -1
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@langgraph-js/pure-graph",
|
|
3
|
-
"version": "2.
|
|
3
|
+
"version": "2.5.0",
|
|
4
4
|
"description": "A library that provides a standard LangGraph endpoint for integrating into various frameworks like Next.js and Hono.js, with support for multiple storage backends (SQLite, PostgreSQL, Redis) and message queues.",
|
|
5
5
|
"main": "dist/index.js",
|
|
6
6
|
"types": "dist/index.d.ts",
|
|
@@ -26,11 +26,11 @@
|
|
|
26
26
|
],
|
|
27
27
|
"repository": {
|
|
28
28
|
"type": "git",
|
|
29
|
-
"url": "https://github.com/KonghaYao/
|
|
29
|
+
"url": "https://github.com/KonghaYao/open-langgraph-server.git"
|
|
30
30
|
},
|
|
31
|
-
"homepage": "https://github.com/KonghaYao/
|
|
31
|
+
"homepage": "https://github.com/KonghaYao/open-langgraph-server",
|
|
32
32
|
"bugs": {
|
|
33
|
-
"url": "https://github.com/KonghaYao/
|
|
33
|
+
"url": "https://github.com/KonghaYao/open-langgraph-server/issues"
|
|
34
34
|
},
|
|
35
35
|
"engines": {
|
|
36
36
|
"node": ">=18.0.0"
|
|
@@ -41,7 +41,6 @@
|
|
|
41
41
|
"@langchain/core": "^1.0.1",
|
|
42
42
|
"@langchain/langgraph": "^1.0.0",
|
|
43
43
|
"@langchain/langgraph-checkpoint": "^1.0.0",
|
|
44
|
-
"@langchain/langgraph-checkpoint-sqlite": "^1.0.0",
|
|
45
44
|
"@langgraph-js/pro": "^2.0.0",
|
|
46
45
|
"camelcase-keys": "^10.0.0",
|
|
47
46
|
"eventemitter3": "^5.0.1",
|
|
@@ -53,6 +52,7 @@
|
|
|
53
52
|
"@hono/zod-validator": "^0.7.3",
|
|
54
53
|
"@langchain/langgraph-checkpoint-postgres": "^1.0.0",
|
|
55
54
|
"@langchain/langgraph-checkpoint-redis": "^1.0.0",
|
|
55
|
+
"kysely-bun-worker": "^1.2.1",
|
|
56
56
|
"better-sqlite3": "^12",
|
|
57
57
|
"hono": "^4.9.9",
|
|
58
58
|
"pg": "^8.16.3",
|
|
@@ -72,7 +72,7 @@
|
|
|
72
72
|
"dev": "bun run --port 8123 test/hono.ts",
|
|
73
73
|
"dev:docs": "bunx --bun xyd-js",
|
|
74
74
|
"build": " vite build && tsc",
|
|
75
|
-
"build
|
|
75
|
+
"docs:build": "pnpx xyd-js build",
|
|
76
76
|
"prepublish": "pnpm build"
|
|
77
77
|
}
|
|
78
78
|
}
|
|
@@ -1,377 +0,0 @@
|
|
|
1
|
-
import { Database } from 'better-sqlite3';
|
|
2
|
-
import { BaseCheckpointSaver, TASKS, copyCheckpoint, maxChannelVersion } from '@langchain/langgraph-checkpoint';
|
|
3
|
-
|
|
4
|
-
const checkpointMetadataKeys = ["source", "step", "parents"];
|
|
5
|
-
function validateKeys(keys) {
|
|
6
|
-
return keys;
|
|
7
|
-
}
|
|
8
|
-
const validCheckpointMetadataKeys = validateKeys(
|
|
9
|
-
checkpointMetadataKeys
|
|
10
|
-
);
|
|
11
|
-
function prepareSql(db, checkpointId) {
|
|
12
|
-
const sql = `
|
|
13
|
-
SELECT
|
|
14
|
-
thread_id,
|
|
15
|
-
checkpoint_ns,
|
|
16
|
-
checkpoint_id,
|
|
17
|
-
parent_checkpoint_id,
|
|
18
|
-
type,
|
|
19
|
-
checkpoint,
|
|
20
|
-
metadata,
|
|
21
|
-
(
|
|
22
|
-
SELECT
|
|
23
|
-
json_group_array(
|
|
24
|
-
json_object(
|
|
25
|
-
'task_id', pw.task_id,
|
|
26
|
-
'channel', pw.channel,
|
|
27
|
-
'type', pw.type,
|
|
28
|
-
'value', CAST(pw.value AS TEXT)
|
|
29
|
-
)
|
|
30
|
-
)
|
|
31
|
-
FROM writes as pw
|
|
32
|
-
WHERE pw.thread_id = checkpoints.thread_id
|
|
33
|
-
AND pw.checkpoint_ns = checkpoints.checkpoint_ns
|
|
34
|
-
AND pw.checkpoint_id = checkpoints.checkpoint_id
|
|
35
|
-
) as pending_writes,
|
|
36
|
-
(
|
|
37
|
-
SELECT
|
|
38
|
-
json_group_array(
|
|
39
|
-
json_object(
|
|
40
|
-
'type', ps.type,
|
|
41
|
-
'value', CAST(ps.value AS TEXT)
|
|
42
|
-
)
|
|
43
|
-
)
|
|
44
|
-
FROM writes as ps
|
|
45
|
-
WHERE ps.thread_id = checkpoints.thread_id
|
|
46
|
-
AND ps.checkpoint_ns = checkpoints.checkpoint_ns
|
|
47
|
-
AND ps.checkpoint_id = checkpoints.parent_checkpoint_id
|
|
48
|
-
AND ps.channel = '${TASKS}'
|
|
49
|
-
ORDER BY ps.idx
|
|
50
|
-
) as pending_sends
|
|
51
|
-
FROM checkpoints
|
|
52
|
-
WHERE thread_id = ? AND checkpoint_ns = ? ${checkpointId ? "AND checkpoint_id = ?" : "ORDER BY checkpoint_id DESC LIMIT 1"}`;
|
|
53
|
-
return db.prepare(sql);
|
|
54
|
-
}
|
|
55
|
-
class SqliteSaver extends BaseCheckpointSaver {
|
|
56
|
-
db;
|
|
57
|
-
isSetup;
|
|
58
|
-
withoutCheckpoint;
|
|
59
|
-
withCheckpoint;
|
|
60
|
-
constructor(db, serde) {
|
|
61
|
-
super(serde);
|
|
62
|
-
this.db = db;
|
|
63
|
-
this.isSetup = false;
|
|
64
|
-
}
|
|
65
|
-
static fromConnString(connStringOrLocalPath) {
|
|
66
|
-
return new SqliteSaver(new Database(connStringOrLocalPath));
|
|
67
|
-
}
|
|
68
|
-
setup() {
|
|
69
|
-
if (this.isSetup) {
|
|
70
|
-
return;
|
|
71
|
-
}
|
|
72
|
-
this.db.exec("PRAGMA journal_mode = WAL;");
|
|
73
|
-
this.db.exec(`
|
|
74
|
-
CREATE TABLE IF NOT EXISTS checkpoints (
|
|
75
|
-
thread_id TEXT NOT NULL,
|
|
76
|
-
checkpoint_ns TEXT NOT NULL DEFAULT '',
|
|
77
|
-
checkpoint_id TEXT NOT NULL,
|
|
78
|
-
parent_checkpoint_id TEXT,
|
|
79
|
-
type TEXT,
|
|
80
|
-
checkpoint BLOB,
|
|
81
|
-
metadata BLOB,
|
|
82
|
-
PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id)
|
|
83
|
-
);`);
|
|
84
|
-
this.db.exec(`
|
|
85
|
-
CREATE TABLE IF NOT EXISTS writes (
|
|
86
|
-
thread_id TEXT NOT NULL,
|
|
87
|
-
checkpoint_ns TEXT NOT NULL DEFAULT '',
|
|
88
|
-
checkpoint_id TEXT NOT NULL,
|
|
89
|
-
task_id TEXT NOT NULL,
|
|
90
|
-
idx INTEGER NOT NULL,
|
|
91
|
-
channel TEXT NOT NULL,
|
|
92
|
-
type TEXT,
|
|
93
|
-
value BLOB,
|
|
94
|
-
PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id, task_id, idx)
|
|
95
|
-
);`);
|
|
96
|
-
this.withoutCheckpoint = prepareSql(this.db, false);
|
|
97
|
-
this.withCheckpoint = prepareSql(this.db, true);
|
|
98
|
-
this.isSetup = true;
|
|
99
|
-
}
|
|
100
|
-
async getTuple(config) {
|
|
101
|
-
this.setup();
|
|
102
|
-
const { thread_id, checkpoint_ns = "", checkpoint_id } = config.configurable ?? {};
|
|
103
|
-
const args = [thread_id, checkpoint_ns];
|
|
104
|
-
if (checkpoint_id) args.push(checkpoint_id);
|
|
105
|
-
const stm = checkpoint_id ? this.withCheckpoint : this.withoutCheckpoint;
|
|
106
|
-
const row = stm.get(...args);
|
|
107
|
-
if (row === void 0 || row === null) return void 0;
|
|
108
|
-
let finalConfig = config;
|
|
109
|
-
if (!checkpoint_id) {
|
|
110
|
-
finalConfig = {
|
|
111
|
-
configurable: {
|
|
112
|
-
thread_id: row.thread_id,
|
|
113
|
-
checkpoint_ns,
|
|
114
|
-
checkpoint_id: row.checkpoint_id
|
|
115
|
-
}
|
|
116
|
-
};
|
|
117
|
-
}
|
|
118
|
-
if (finalConfig.configurable?.thread_id === void 0 || finalConfig.configurable?.checkpoint_id === void 0) {
|
|
119
|
-
throw new Error("Missing thread_id or checkpoint_id");
|
|
120
|
-
}
|
|
121
|
-
const pendingWrites = await Promise.all(
|
|
122
|
-
JSON.parse(row.pending_writes).map(async (write) => {
|
|
123
|
-
return [
|
|
124
|
-
write.task_id,
|
|
125
|
-
write.channel,
|
|
126
|
-
await this.serde.loadsTyped(write.type ?? "json", write.value ?? "")
|
|
127
|
-
];
|
|
128
|
-
})
|
|
129
|
-
);
|
|
130
|
-
const checkpoint = await this.serde.loadsTyped(row.type ?? "json", row.checkpoint);
|
|
131
|
-
if (checkpoint.v < 4 && row.parent_checkpoint_id != null) {
|
|
132
|
-
await this.migratePendingSends(checkpoint, row.thread_id, row.parent_checkpoint_id);
|
|
133
|
-
}
|
|
134
|
-
return {
|
|
135
|
-
checkpoint,
|
|
136
|
-
config: finalConfig,
|
|
137
|
-
metadata: await this.serde.loadsTyped(row.type ?? "json", row.metadata),
|
|
138
|
-
parentConfig: row.parent_checkpoint_id ? {
|
|
139
|
-
configurable: {
|
|
140
|
-
thread_id: row.thread_id,
|
|
141
|
-
checkpoint_ns,
|
|
142
|
-
checkpoint_id: row.parent_checkpoint_id
|
|
143
|
-
}
|
|
144
|
-
} : void 0,
|
|
145
|
-
pendingWrites
|
|
146
|
-
};
|
|
147
|
-
}
|
|
148
|
-
async *list(config, options) {
|
|
149
|
-
const { limit, before, filter } = options ?? {};
|
|
150
|
-
this.setup();
|
|
151
|
-
const thread_id = config.configurable?.thread_id;
|
|
152
|
-
const checkpoint_ns = config.configurable?.checkpoint_ns;
|
|
153
|
-
let sql = `
|
|
154
|
-
SELECT
|
|
155
|
-
thread_id,
|
|
156
|
-
checkpoint_ns,
|
|
157
|
-
checkpoint_id,
|
|
158
|
-
parent_checkpoint_id,
|
|
159
|
-
type,
|
|
160
|
-
checkpoint,
|
|
161
|
-
metadata,
|
|
162
|
-
(
|
|
163
|
-
SELECT
|
|
164
|
-
json_group_array(
|
|
165
|
-
json_object(
|
|
166
|
-
'task_id', pw.task_id,
|
|
167
|
-
'channel', pw.channel,
|
|
168
|
-
'type', pw.type,
|
|
169
|
-
'value', CAST(pw.value AS TEXT)
|
|
170
|
-
)
|
|
171
|
-
)
|
|
172
|
-
FROM writes as pw
|
|
173
|
-
WHERE pw.thread_id = checkpoints.thread_id
|
|
174
|
-
AND pw.checkpoint_ns = checkpoints.checkpoint_ns
|
|
175
|
-
AND pw.checkpoint_id = checkpoints.checkpoint_id
|
|
176
|
-
) as pending_writes,
|
|
177
|
-
(
|
|
178
|
-
SELECT
|
|
179
|
-
json_group_array(
|
|
180
|
-
json_object(
|
|
181
|
-
'type', ps.type,
|
|
182
|
-
'value', CAST(ps.value AS TEXT)
|
|
183
|
-
)
|
|
184
|
-
)
|
|
185
|
-
FROM writes as ps
|
|
186
|
-
WHERE ps.thread_id = checkpoints.thread_id
|
|
187
|
-
AND ps.checkpoint_ns = checkpoints.checkpoint_ns
|
|
188
|
-
AND ps.checkpoint_id = checkpoints.parent_checkpoint_id
|
|
189
|
-
AND ps.channel = '${TASKS}'
|
|
190
|
-
ORDER BY ps.idx
|
|
191
|
-
) as pending_sends
|
|
192
|
-
FROM checkpoints
|
|
193
|
-
`;
|
|
194
|
-
const whereClause = [];
|
|
195
|
-
if (thread_id) {
|
|
196
|
-
whereClause.push("thread_id = ?");
|
|
197
|
-
}
|
|
198
|
-
if (checkpoint_ns !== void 0 && checkpoint_ns !== null) {
|
|
199
|
-
whereClause.push("checkpoint_ns = ?");
|
|
200
|
-
}
|
|
201
|
-
if (before?.configurable?.checkpoint_id !== void 0) {
|
|
202
|
-
whereClause.push("checkpoint_id < ?");
|
|
203
|
-
}
|
|
204
|
-
const sanitizedFilter = Object.fromEntries(
|
|
205
|
-
Object.entries(filter ?? {}).filter(
|
|
206
|
-
([key, value]) => value !== void 0 && validCheckpointMetadataKeys.includes(key)
|
|
207
|
-
)
|
|
208
|
-
);
|
|
209
|
-
whereClause.push(
|
|
210
|
-
...Object.entries(sanitizedFilter).map(([key]) => `jsonb(CAST(metadata AS TEXT))->'$.${key}' = ?`)
|
|
211
|
-
);
|
|
212
|
-
if (whereClause.length > 0) {
|
|
213
|
-
sql += `WHERE
|
|
214
|
-
${whereClause.join(" AND\n ")}
|
|
215
|
-
`;
|
|
216
|
-
}
|
|
217
|
-
sql += "\nORDER BY checkpoint_id DESC";
|
|
218
|
-
if (limit) {
|
|
219
|
-
sql += ` LIMIT ${parseInt(limit, 10)}`;
|
|
220
|
-
}
|
|
221
|
-
const args = [
|
|
222
|
-
thread_id,
|
|
223
|
-
checkpoint_ns,
|
|
224
|
-
before?.configurable?.checkpoint_id,
|
|
225
|
-
...Object.values(sanitizedFilter).map((value) => JSON.stringify(value))
|
|
226
|
-
].filter((value) => value !== void 0 && value !== null);
|
|
227
|
-
const rows = this.db.prepare(sql).all(...args);
|
|
228
|
-
if (rows) {
|
|
229
|
-
for (const row of rows) {
|
|
230
|
-
const pendingWrites = await Promise.all(
|
|
231
|
-
JSON.parse(row.pending_writes).map(async (write) => {
|
|
232
|
-
return [
|
|
233
|
-
write.task_id,
|
|
234
|
-
write.channel,
|
|
235
|
-
await this.serde.loadsTyped(write.type ?? "json", write.value ?? "")
|
|
236
|
-
];
|
|
237
|
-
})
|
|
238
|
-
);
|
|
239
|
-
const checkpoint = await this.serde.loadsTyped(row.type ?? "json", row.checkpoint);
|
|
240
|
-
if (checkpoint.v < 4 && row.parent_checkpoint_id != null) {
|
|
241
|
-
await this.migratePendingSends(checkpoint, row.thread_id, row.parent_checkpoint_id);
|
|
242
|
-
}
|
|
243
|
-
yield {
|
|
244
|
-
config: {
|
|
245
|
-
configurable: {
|
|
246
|
-
thread_id: row.thread_id,
|
|
247
|
-
checkpoint_ns: row.checkpoint_ns,
|
|
248
|
-
checkpoint_id: row.checkpoint_id
|
|
249
|
-
}
|
|
250
|
-
},
|
|
251
|
-
checkpoint,
|
|
252
|
-
metadata: await this.serde.loadsTyped(row.type ?? "json", row.metadata),
|
|
253
|
-
parentConfig: row.parent_checkpoint_id ? {
|
|
254
|
-
configurable: {
|
|
255
|
-
thread_id: row.thread_id,
|
|
256
|
-
checkpoint_ns: row.checkpoint_ns,
|
|
257
|
-
checkpoint_id: row.parent_checkpoint_id
|
|
258
|
-
}
|
|
259
|
-
} : void 0,
|
|
260
|
-
pendingWrites
|
|
261
|
-
};
|
|
262
|
-
}
|
|
263
|
-
}
|
|
264
|
-
}
|
|
265
|
-
async put(config, checkpoint, metadata) {
|
|
266
|
-
this.setup();
|
|
267
|
-
if (!config.configurable) {
|
|
268
|
-
throw new Error("Empty configuration supplied.");
|
|
269
|
-
}
|
|
270
|
-
const thread_id = config.configurable?.thread_id;
|
|
271
|
-
const checkpoint_ns = config.configurable?.checkpoint_ns ?? "";
|
|
272
|
-
const parent_checkpoint_id = config.configurable?.checkpoint_id;
|
|
273
|
-
if (!thread_id) {
|
|
274
|
-
throw new Error(`Missing "thread_id" field in passed "config.configurable".`);
|
|
275
|
-
}
|
|
276
|
-
const preparedCheckpoint = copyCheckpoint(checkpoint);
|
|
277
|
-
const [[type1, serializedCheckpoint], [type2, serializedMetadata]] = await Promise.all([
|
|
278
|
-
this.serde.dumpsTyped(preparedCheckpoint),
|
|
279
|
-
this.serde.dumpsTyped(metadata)
|
|
280
|
-
]);
|
|
281
|
-
if (type1 !== type2) {
|
|
282
|
-
throw new Error("Failed to serialized checkpoint and metadata to the same type.");
|
|
283
|
-
}
|
|
284
|
-
const row = [
|
|
285
|
-
thread_id,
|
|
286
|
-
checkpoint_ns,
|
|
287
|
-
checkpoint.id,
|
|
288
|
-
parent_checkpoint_id,
|
|
289
|
-
type1,
|
|
290
|
-
serializedCheckpoint,
|
|
291
|
-
serializedMetadata
|
|
292
|
-
];
|
|
293
|
-
this.db.prepare(
|
|
294
|
-
`INSERT OR REPLACE INTO checkpoints (thread_id, checkpoint_ns, checkpoint_id, parent_checkpoint_id, type, checkpoint, metadata) VALUES (?, ?, ?, ?, ?, ?, ?)`
|
|
295
|
-
).run(...row);
|
|
296
|
-
return {
|
|
297
|
-
configurable: {
|
|
298
|
-
thread_id,
|
|
299
|
-
checkpoint_ns,
|
|
300
|
-
checkpoint_id: checkpoint.id
|
|
301
|
-
}
|
|
302
|
-
};
|
|
303
|
-
}
|
|
304
|
-
async putWrites(config, writes, taskId) {
|
|
305
|
-
this.setup();
|
|
306
|
-
if (!config.configurable) {
|
|
307
|
-
throw new Error("Empty configuration supplied.");
|
|
308
|
-
}
|
|
309
|
-
if (!config.configurable?.thread_id) {
|
|
310
|
-
throw new Error("Missing thread_id field in config.configurable.");
|
|
311
|
-
}
|
|
312
|
-
if (!config.configurable?.checkpoint_id) {
|
|
313
|
-
throw new Error("Missing checkpoint_id field in config.configurable.");
|
|
314
|
-
}
|
|
315
|
-
const stmt = this.db.prepare(`
|
|
316
|
-
INSERT OR REPLACE INTO writes
|
|
317
|
-
(thread_id, checkpoint_ns, checkpoint_id, task_id, idx, channel, type, value)
|
|
318
|
-
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
|
319
|
-
`);
|
|
320
|
-
const transaction = this.db.transaction((rows2) => {
|
|
321
|
-
for (const row of rows2) {
|
|
322
|
-
stmt.run(...row);
|
|
323
|
-
}
|
|
324
|
-
});
|
|
325
|
-
const rows = await Promise.all(
|
|
326
|
-
writes.map(async (write, idx) => {
|
|
327
|
-
const [type, serializedWrite] = await this.serde.dumpsTyped(write[1]);
|
|
328
|
-
return [
|
|
329
|
-
config.configurable?.thread_id,
|
|
330
|
-
config.configurable?.checkpoint_ns,
|
|
331
|
-
config.configurable?.checkpoint_id,
|
|
332
|
-
taskId,
|
|
333
|
-
idx,
|
|
334
|
-
write[0],
|
|
335
|
-
type,
|
|
336
|
-
serializedWrite
|
|
337
|
-
];
|
|
338
|
-
})
|
|
339
|
-
);
|
|
340
|
-
transaction(rows);
|
|
341
|
-
}
|
|
342
|
-
async deleteThread(threadId) {
|
|
343
|
-
const transaction = this.db.transaction(() => {
|
|
344
|
-
this.db.prepare(`DELETE FROM checkpoints WHERE thread_id = ?`).run(threadId);
|
|
345
|
-
this.db.prepare(`DELETE FROM writes WHERE thread_id = ?`).run(threadId);
|
|
346
|
-
});
|
|
347
|
-
transaction();
|
|
348
|
-
}
|
|
349
|
-
async migratePendingSends(checkpoint, threadId, parentCheckpointId) {
|
|
350
|
-
const { pending_sends } = this.db.prepare(
|
|
351
|
-
`
|
|
352
|
-
SELECT
|
|
353
|
-
checkpoint_id,
|
|
354
|
-
json_group_array(
|
|
355
|
-
json_object(
|
|
356
|
-
'type', ps.type,
|
|
357
|
-
'value', CAST(ps.value AS TEXT)
|
|
358
|
-
)
|
|
359
|
-
) as pending_sends
|
|
360
|
-
FROM writes as ps
|
|
361
|
-
WHERE ps.thread_id = ?
|
|
362
|
-
AND ps.checkpoint_id = ?
|
|
363
|
-
AND ps.channel = '${TASKS}'
|
|
364
|
-
ORDER BY ps.idx
|
|
365
|
-
`
|
|
366
|
-
).get(threadId, parentCheckpointId);
|
|
367
|
-
const mutableCheckpoint = checkpoint;
|
|
368
|
-
mutableCheckpoint.channel_values ??= {};
|
|
369
|
-
mutableCheckpoint.channel_values[TASKS] = await Promise.all(
|
|
370
|
-
JSON.parse(pending_sends).map(({ type, value }) => this.serde.loadsTyped(type, value))
|
|
371
|
-
);
|
|
372
|
-
mutableCheckpoint.channel_versions[TASKS] = Object.keys(checkpoint.channel_versions).length > 0 ? maxChannelVersion(...Object.values(checkpoint.channel_versions)) : this.getNextVersion(void 0);
|
|
373
|
-
}
|
|
374
|
-
}
|
|
375
|
-
|
|
376
|
-
export { SqliteSaver };
|
|
377
|
-
//# sourceMappingURL=checkpoint-DdL-Wo1x.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"checkpoint-DdL-Wo1x.js","sources":["../src/storage/sqlite/checkpoint.ts"],"sourcesContent":["import { DatabaseType } from './type.js';\n/** @ts-ignore */\nimport { Database } from 'better-sqlite3';\nimport type { RunnableConfig } from '@langchain/core/runnables';\nimport {\n BaseCheckpointSaver,\n type Checkpoint,\n type CheckpointListOptions,\n type CheckpointTuple,\n type SerializerProtocol,\n type PendingWrite,\n type CheckpointMetadata,\n TASKS,\n copyCheckpoint,\n maxChannelVersion,\n} from '@langchain/langgraph-checkpoint';\n\ninterface CheckpointRow {\n checkpoint: string;\n metadata: string;\n parent_checkpoint_id?: string;\n thread_id: string;\n checkpoint_id: string;\n checkpoint_ns?: string;\n type?: string;\n pending_writes: string;\n}\n\ninterface PendingWriteColumn {\n task_id: string;\n channel: string;\n type: string;\n value: string;\n}\n\ninterface PendingSendColumn {\n type: string;\n value: string;\n}\n\n// In the `SqliteSaver.list` method, we need to sanitize the `options.filter` argument to ensure it only contains keys\n// that are part of the `CheckpointMetadata` type. The lines below ensure that we get compile-time errors if the list\n// of keys that we use is out of sync with the `CheckpointMetadata` type.\nconst checkpointMetadataKeys = ['source', 'step', 'parents'] as const;\n\ntype CheckKeys<T, K extends readonly (keyof T)[]> = [K[number]] extends [keyof T]\n ? [keyof T] extends [K[number]]\n ? K\n : never\n : never;\n\nfunction validateKeys<T, K extends readonly (keyof T)[]>(keys: CheckKeys<T, K>): K {\n return keys;\n}\n\n// If this line fails to compile, the list of keys that we use in the `SqliteSaver.list` method is out of sync with the\n// `CheckpointMetadata` type. In that case, just update `checkpointMetadataKeys` to contain all the keys in\n// `CheckpointMetadata`\nconst validCheckpointMetadataKeys = validateKeys<CheckpointMetadata, typeof checkpointMetadataKeys>(\n checkpointMetadataKeys,\n);\n\nfunction prepareSql(db: DatabaseType, checkpointId: boolean) {\n const sql = `\n SELECT\n thread_id,\n checkpoint_ns,\n checkpoint_id,\n parent_checkpoint_id,\n type,\n checkpoint,\n metadata,\n (\n SELECT\n json_group_array(\n json_object(\n 'task_id', pw.task_id,\n 'channel', pw.channel,\n 'type', pw.type,\n 'value', CAST(pw.value AS TEXT)\n )\n )\n FROM writes as pw\n WHERE pw.thread_id = checkpoints.thread_id\n AND pw.checkpoint_ns = checkpoints.checkpoint_ns\n AND pw.checkpoint_id = checkpoints.checkpoint_id\n ) as pending_writes,\n (\n SELECT\n json_group_array(\n json_object(\n 'type', ps.type,\n 'value', CAST(ps.value AS TEXT)\n )\n )\n FROM writes as ps\n WHERE ps.thread_id = checkpoints.thread_id\n AND ps.checkpoint_ns = checkpoints.checkpoint_ns\n AND ps.checkpoint_id = checkpoints.parent_checkpoint_id\n AND ps.channel = '${TASKS}'\n ORDER BY ps.idx\n ) as pending_sends\n FROM checkpoints\n WHERE thread_id = ? AND checkpoint_ns = ? ${\n checkpointId ? 'AND checkpoint_id = ?' : 'ORDER BY checkpoint_id DESC LIMIT 1'\n }`;\n\n return db.prepare(sql);\n}\n\nexport class SqliteSaver extends BaseCheckpointSaver {\n db: DatabaseType;\n\n protected isSetup: boolean;\n\n protected withoutCheckpoint: any;\n\n protected withCheckpoint: any;\n\n constructor(db: DatabaseType, serde?: SerializerProtocol) {\n super(serde);\n this.db = db;\n this.isSetup = false;\n }\n\n static fromConnString(connStringOrLocalPath: string): SqliteSaver {\n return new SqliteSaver(new Database(connStringOrLocalPath));\n }\n\n protected setup(): void {\n if (this.isSetup) {\n return;\n }\n\n // this.db.pragma('journal_mode=WAL');\n this.db.exec('PRAGMA journal_mode = WAL;');\n this.db.exec(`\nCREATE TABLE IF NOT EXISTS checkpoints (\n thread_id TEXT NOT NULL,\n checkpoint_ns TEXT NOT NULL DEFAULT '',\n checkpoint_id TEXT NOT NULL,\n parent_checkpoint_id TEXT,\n type TEXT,\n checkpoint BLOB,\n metadata BLOB,\n PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id)\n);`);\n this.db.exec(`\nCREATE TABLE IF NOT EXISTS writes (\n thread_id TEXT NOT NULL,\n checkpoint_ns TEXT NOT NULL DEFAULT '',\n checkpoint_id TEXT NOT NULL,\n task_id TEXT NOT NULL,\n idx INTEGER NOT NULL,\n channel TEXT NOT NULL,\n type TEXT,\n value BLOB,\n PRIMARY KEY (thread_id, checkpoint_ns, checkpoint_id, task_id, idx)\n);`);\n\n this.withoutCheckpoint = prepareSql(this.db, false);\n this.withCheckpoint = prepareSql(this.db, true);\n\n this.isSetup = true;\n }\n\n async getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined> {\n this.setup();\n const { thread_id, checkpoint_ns = '', checkpoint_id } = config.configurable ?? {};\n\n const args = [thread_id, checkpoint_ns];\n if (checkpoint_id) args.push(checkpoint_id);\n\n const stm = checkpoint_id ? this.withCheckpoint : this.withoutCheckpoint;\n const row = stm.get(...args) as CheckpointRow;\n if (row === undefined || row === null) return undefined;\n\n let finalConfig = config;\n\n if (!checkpoint_id) {\n finalConfig = {\n configurable: {\n thread_id: row.thread_id,\n checkpoint_ns,\n checkpoint_id: row.checkpoint_id,\n },\n };\n }\n\n if (\n finalConfig.configurable?.thread_id === undefined ||\n finalConfig.configurable?.checkpoint_id === undefined\n ) {\n throw new Error('Missing thread_id or checkpoint_id');\n }\n\n const pendingWrites = await Promise.all(\n (JSON.parse(row.pending_writes) as PendingWriteColumn[]).map(async (write) => {\n return [\n write.task_id,\n write.channel,\n await this.serde.loadsTyped(write.type ?? 'json', write.value ?? ''),\n ] as [string, string, unknown];\n }),\n );\n\n const checkpoint = (await this.serde.loadsTyped(row.type ?? 'json', row.checkpoint)) as Checkpoint;\n\n if (checkpoint.v < 4 && row.parent_checkpoint_id != null) {\n await this.migratePendingSends(checkpoint, row.thread_id, row.parent_checkpoint_id);\n }\n\n return {\n checkpoint,\n config: finalConfig,\n metadata: (await this.serde.loadsTyped(row.type ?? 'json', row.metadata)) as CheckpointMetadata,\n parentConfig: row.parent_checkpoint_id\n ? {\n configurable: {\n thread_id: row.thread_id,\n checkpoint_ns,\n checkpoint_id: row.parent_checkpoint_id,\n },\n }\n : undefined,\n pendingWrites,\n };\n }\n\n async *list(config: RunnableConfig, options?: CheckpointListOptions): AsyncGenerator<CheckpointTuple> {\n const { limit, before, filter } = options ?? {};\n this.setup();\n const thread_id = config.configurable?.thread_id;\n const checkpoint_ns = config.configurable?.checkpoint_ns;\n let sql = `\n SELECT\n thread_id,\n checkpoint_ns,\n checkpoint_id,\n parent_checkpoint_id,\n type,\n checkpoint,\n metadata,\n (\n SELECT\n json_group_array(\n json_object(\n 'task_id', pw.task_id,\n 'channel', pw.channel,\n 'type', pw.type,\n 'value', CAST(pw.value AS TEXT)\n )\n )\n FROM writes as pw\n WHERE pw.thread_id = checkpoints.thread_id\n AND pw.checkpoint_ns = checkpoints.checkpoint_ns\n AND pw.checkpoint_id = checkpoints.checkpoint_id\n ) as pending_writes,\n (\n SELECT\n json_group_array(\n json_object(\n 'type', ps.type,\n 'value', CAST(ps.value AS TEXT)\n )\n )\n FROM writes as ps\n WHERE ps.thread_id = checkpoints.thread_id\n AND ps.checkpoint_ns = checkpoints.checkpoint_ns\n AND ps.checkpoint_id = checkpoints.parent_checkpoint_id\n AND ps.channel = '${TASKS}'\n ORDER BY ps.idx\n ) as pending_sends\n FROM checkpoints\\n`;\n\n const whereClause: string[] = [];\n\n if (thread_id) {\n whereClause.push('thread_id = ?');\n }\n\n if (checkpoint_ns !== undefined && checkpoint_ns !== null) {\n whereClause.push('checkpoint_ns = ?');\n }\n\n if (before?.configurable?.checkpoint_id !== undefined) {\n whereClause.push('checkpoint_id < ?');\n }\n\n const sanitizedFilter = Object.fromEntries(\n Object.entries(filter ?? {}).filter(\n ([key, value]) =>\n value !== undefined && validCheckpointMetadataKeys.includes(key as keyof CheckpointMetadata),\n ),\n );\n\n whereClause.push(\n ...Object.entries(sanitizedFilter).map(([key]) => `jsonb(CAST(metadata AS TEXT))->'$.${key}' = ?`),\n );\n if (whereClause.length > 0) {\n sql += `WHERE\\n ${whereClause.join(' AND\\n ')}\\n`;\n }\n\n sql += '\\nORDER BY checkpoint_id DESC';\n\n if (limit) {\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n sql += ` LIMIT ${parseInt(limit as any, 10)}`; // parseInt here (with cast to make TS happy) to sanitize input, as limit may be user-provided\n }\n\n const args = [\n thread_id,\n checkpoint_ns,\n before?.configurable?.checkpoint_id,\n ...Object.values(sanitizedFilter).map((value) => JSON.stringify(value)),\n ].filter((value) => value !== undefined && value !== null);\n\n const rows: CheckpointRow[] = this.db.prepare(sql).all(...args) as CheckpointRow[];\n\n if (rows) {\n for (const row of rows) {\n const pendingWrites = await Promise.all(\n (JSON.parse(row.pending_writes) as PendingWriteColumn[]).map(async (write) => {\n return [\n write.task_id,\n write.channel,\n await this.serde.loadsTyped(write.type ?? 'json', write.value ?? ''),\n ] as [string, string, unknown];\n }),\n );\n\n const checkpoint = (await this.serde.loadsTyped(row.type ?? 'json', row.checkpoint)) as Checkpoint;\n\n if (checkpoint.v < 4 && row.parent_checkpoint_id != null) {\n await this.migratePendingSends(checkpoint, row.thread_id, row.parent_checkpoint_id);\n }\n\n yield {\n config: {\n configurable: {\n thread_id: row.thread_id,\n checkpoint_ns: row.checkpoint_ns,\n checkpoint_id: row.checkpoint_id,\n },\n },\n checkpoint,\n metadata: (await this.serde.loadsTyped(row.type ?? 'json', row.metadata)) as CheckpointMetadata,\n parentConfig: row.parent_checkpoint_id\n ? {\n configurable: {\n thread_id: row.thread_id,\n checkpoint_ns: row.checkpoint_ns,\n checkpoint_id: row.parent_checkpoint_id,\n },\n }\n : undefined,\n pendingWrites,\n };\n }\n }\n }\n\n async put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata): Promise<RunnableConfig> {\n this.setup();\n\n if (!config.configurable) {\n throw new Error('Empty configuration supplied.');\n }\n\n const thread_id = config.configurable?.thread_id;\n const checkpoint_ns = config.configurable?.checkpoint_ns ?? '';\n const parent_checkpoint_id = config.configurable?.checkpoint_id;\n\n if (!thread_id) {\n throw new Error(`Missing \"thread_id\" field in passed \"config.configurable\".`);\n }\n\n const preparedCheckpoint: Partial<Checkpoint> = copyCheckpoint(checkpoint);\n\n const [[type1, serializedCheckpoint], [type2, serializedMetadata]] = await Promise.all([\n this.serde.dumpsTyped(preparedCheckpoint),\n this.serde.dumpsTyped(metadata),\n ]);\n\n if (type1 !== type2) {\n throw new Error('Failed to serialized checkpoint and metadata to the same type.');\n }\n const row = [\n thread_id,\n checkpoint_ns,\n checkpoint.id,\n parent_checkpoint_id,\n type1,\n serializedCheckpoint,\n serializedMetadata,\n ];\n\n this.db\n .prepare(\n `INSERT OR REPLACE INTO checkpoints (thread_id, checkpoint_ns, checkpoint_id, parent_checkpoint_id, type, checkpoint, metadata) VALUES (?, ?, ?, ?, ?, ?, ?)`,\n )\n .run(...row);\n\n return {\n configurable: {\n thread_id,\n checkpoint_ns,\n checkpoint_id: checkpoint.id,\n },\n };\n }\n\n async putWrites(config: RunnableConfig, writes: PendingWrite[], taskId: string): Promise<void> {\n this.setup();\n\n if (!config.configurable) {\n throw new Error('Empty configuration supplied.');\n }\n\n if (!config.configurable?.thread_id) {\n throw new Error('Missing thread_id field in config.configurable.');\n }\n\n if (!config.configurable?.checkpoint_id) {\n throw new Error('Missing checkpoint_id field in config.configurable.');\n }\n\n const stmt = this.db.prepare(`\n INSERT OR REPLACE INTO writes \n (thread_id, checkpoint_ns, checkpoint_id, task_id, idx, channel, type, value) \n VALUES (?, ?, ?, ?, ?, ?, ?, ?)\n `);\n\n const transaction = this.db.transaction((rows) => {\n for (const row of rows) {\n stmt.run(...row);\n }\n });\n\n const rows = await Promise.all(\n writes.map(async (write, idx) => {\n const [type, serializedWrite] = await this.serde.dumpsTyped(write[1]);\n return [\n config.configurable?.thread_id,\n config.configurable?.checkpoint_ns,\n config.configurable?.checkpoint_id,\n taskId,\n idx,\n write[0],\n type,\n serializedWrite,\n ];\n }),\n );\n\n transaction(rows);\n }\n\n async deleteThread(threadId: string) {\n const transaction = this.db.transaction(() => {\n this.db.prepare(`DELETE FROM checkpoints WHERE thread_id = ?`).run(threadId);\n this.db.prepare(`DELETE FROM writes WHERE thread_id = ?`).run(threadId);\n });\n\n transaction();\n }\n\n protected async migratePendingSends(checkpoint: Checkpoint, threadId: string, parentCheckpointId: string) {\n const { pending_sends } = this.db\n .prepare(\n `\n SELECT\n checkpoint_id,\n json_group_array(\n json_object(\n 'type', ps.type,\n 'value', CAST(ps.value AS TEXT)\n )\n ) as pending_sends\n FROM writes as ps\n WHERE ps.thread_id = ?\n AND ps.checkpoint_id = ?\n AND ps.channel = '${TASKS}'\n ORDER BY ps.idx\n `,\n )\n .get(threadId, parentCheckpointId) as { pending_sends: string };\n\n const mutableCheckpoint = checkpoint;\n\n // add pending sends to checkpoint\n mutableCheckpoint.channel_values ??= {};\n mutableCheckpoint.channel_values[TASKS] = await Promise.all(\n JSON.parse(pending_sends).map(({ type, value }: PendingSendColumn) => this.serde.loadsTyped(type, value)),\n );\n\n // add to versions\n mutableCheckpoint.channel_versions[TASKS] =\n Object.keys(checkpoint.channel_versions).length > 0\n ? maxChannelVersion(...Object.values(checkpoint.channel_versions))\n : this.getNextVersion(undefined);\n }\n}\n"],"names":["rows"],"mappings":";;;AA2CA,MAAM,sBAAA,GAAyB,CAAC,QAAA,EAAU,MAAA,EAAQ,SAAS,CAAA;AAQ3D,SAAS,aAAgD,IAAA,EAA0B;AAC/E,EAAA,OAAO,IAAA;AACX;AAKA,MAAM,2BAAA,GAA8B,YAAA;AAAA,EAChC;AACJ,CAAA;AAEA,SAAS,UAAA,CAAW,IAAkB,YAAA,EAAuB;AACzD,EAAA,MAAM,GAAA,GAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,0BAAA,EAoCY,KAAK,CAAA;AAAA;AAAA;AAAA;AAAA,4CAAA,EAK3B,YAAA,GAAe,0BAA0B,qCAC7C,CAAA,CAAA;AAEE,EAAA,OAAO,EAAA,CAAG,QAAQ,GAAG,CAAA;AACzB;AAEO,MAAM,oBAAoB,mBAAA,CAAoB;AAAA,EACjD,EAAA;AAAA,EAEU,OAAA;AAAA,EAEA,iBAAA;AAAA,EAEA,cAAA;AAAA,EAEV,WAAA,CAAY,IAAkB,KAAA,EAA4B;AACtD,IAAA,KAAA,CAAM,KAAK,CAAA;AACX,IAAA,IAAA,CAAK,EAAA,GAAK,EAAA;AACV,IAAA,IAAA,CAAK,OAAA,GAAU,KAAA;AAAA,EACnB;AAAA,EAEA,OAAO,eAAe,qBAAA,EAA4C;AAC9D,IAAA,OAAO,IAAI,WAAA,CAAY,IAAI,QAAA,CAAS,qBAAqB,CAAC,CAAA;AAAA,EAC9D;AAAA,EAEU,KAAA,GAAc;AACpB,IAAA,IAAI,KAAK,OAAA,EAAS;AACd,MAAA;AAAA,IACJ;AAGA,IAAA,IAAA,CAAK,EAAA,CAAG,KAAK,4BAA4B,CAAA;AACzC,IAAA,IAAA,CAAK,GAAG,IAAA,CAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA,CAUlB,CAAA;AACK,IAAA,IAAA,CAAK,GAAG,IAAA,CAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA,CAWlB,CAAA;AAEK,IAAA,IAAA,CAAK,iBAAA,GAAoB,UAAA,CAAW,IAAA,CAAK,EAAA,EAAI,KAAK,CAAA;AAClD,IAAA,IAAA,CAAK,cAAA,GAAiB,UAAA,CAAW,IAAA,CAAK,EAAA,EAAI,IAAI,CAAA;AAE9C,IAAA,IAAA,CAAK,OAAA,GAAU,IAAA;AAAA,EACnB;AAAA,EAEA,MAAM,SAAS,MAAA,EAA8D;AACzE,IAAA,IAAA,CAAK,KAAA,EAAM;AACX,IAAA,MAAM,EAAE,WAAW,aAAA,GAAgB,EAAA,EAAI,eAAc,GAAI,MAAA,CAAO,gBAAgB,EAAC;AAEjF,IAAA,MAAM,IAAA,GAAO,CAAC,SAAA,EAAW,aAAa,CAAA;AACtC,IAAA,IAAI,aAAA,EAAe,IAAA,CAAK,IAAA,CAAK,aAAa,CAAA;AAE1C,IAAA,MAAM,GAAA,GAAM,aAAA,GAAgB,IAAA,CAAK,cAAA,GAAiB,IAAA,CAAK,iBAAA;AACvD,IAAA,MAAM,GAAA,GAAM,GAAA,CAAI,GAAA,CAAI,GAAG,IAAI,CAAA;AAC3B,IAAA,IAAI,GAAA,KAAQ,MAAA,IAAa,GAAA,KAAQ,IAAA,EAAM,OAAO,MAAA;AAE9C,IAAA,IAAI,WAAA,GAAc,MAAA;AAElB,IAAA,IAAI,CAAC,aAAA,EAAe;AAChB,MAAA,WAAA,GAAc;AAAA,QACV,YAAA,EAAc;AAAA,UACV,WAAW,GAAA,CAAI,SAAA;AAAA,UACf,aAAA;AAAA,UACA,eAAe,GAAA,CAAI;AAAA;AACvB,OACJ;AAAA,IACJ;AAEA,IAAA,IACI,YAAY,YAAA,EAAc,SAAA,KAAc,UACxC,WAAA,CAAY,YAAA,EAAc,kBAAkB,MAAA,EAC9C;AACE,MAAA,MAAM,IAAI,MAAM,oCAAoC,CAAA;AAAA,IACxD;AAEA,IAAA,MAAM,aAAA,GAAgB,MAAM,OAAA,CAAQ,GAAA;AAAA,MAC/B,KAAK,KAAA,CAAM,GAAA,CAAI,cAAc,CAAA,CAA2B,GAAA,CAAI,OAAO,KAAA,KAAU;AAC1E,QAAA,OAAO;AAAA,UACH,KAAA,CAAM,OAAA;AAAA,UACN,KAAA,CAAM,OAAA;AAAA,UACN,MAAM,KAAK,KAAA,CAAM,UAAA,CAAW,MAAM,IAAA,IAAQ,MAAA,EAAQ,KAAA,CAAM,KAAA,IAAS,EAAE;AAAA,SACvE;AAAA,MACJ,CAAC;AAAA,KACL;AAEA,IAAA,MAAM,UAAA,GAAc,MAAM,IAAA,CAAK,KAAA,CAAM,WAAW,GAAA,CAAI,IAAA,IAAQ,MAAA,EAAQ,GAAA,CAAI,UAAU,CAAA;AAElF,IAAA,IAAI,UAAA,CAAW,CAAA,GAAI,CAAA,IAAK,GAAA,CAAI,wBAAwB,IAAA,EAAM;AACtD,MAAA,MAAM,KAAK,mBAAA,CAAoB,UAAA,EAAY,GAAA,CAAI,SAAA,EAAW,IAAI,oBAAoB,CAAA;AAAA,IACtF;AAEA,IAAA,OAAO;AAAA,MACH,UAAA;AAAA,MACA,MAAA,EAAQ,WAAA;AAAA,MACR,QAAA,EAAW,MAAM,IAAA,CAAK,KAAA,CAAM,WAAW,GAAA,CAAI,IAAA,IAAQ,MAAA,EAAQ,GAAA,CAAI,QAAQ,CAAA;AAAA,MACvE,YAAA,EAAc,IAAI,oBAAA,GACZ;AAAA,QACI,YAAA,EAAc;AAAA,UACV,WAAW,GAAA,CAAI,SAAA;AAAA,UACf,aAAA;AAAA,UACA,eAAe,GAAA,CAAI;AAAA;AACvB,OACJ,GACA,MAAA;AAAA,MACN;AAAA,KACJ;AAAA,EACJ;AAAA,EAEA,OAAO,IAAA,CAAK,MAAA,EAAwB,OAAA,EAAkE;AAClG,IAAA,MAAM,EAAE,KAAA,EAAO,MAAA,EAAQ,MAAA,EAAO,GAAI,WAAW,EAAC;AAC9C,IAAA,IAAA,CAAK,KAAA,EAAM;AACX,IAAA,MAAM,SAAA,GAAY,OAAO,YAAA,EAAc,SAAA;AACvC,IAAA,MAAM,aAAA,GAAgB,OAAO,YAAA,EAAc,aAAA;AAC3C,IAAA,IAAI,GAAA,GAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,8BAAA,EAoCc,KAAK,CAAA;AAAA;AAAA;AAAA;AAAA,CAAA;AAK7B,IAAA,MAAM,cAAwB,EAAC;AAE/B,IAAA,IAAI,SAAA,EAAW;AACX,MAAA,WAAA,CAAY,KAAK,eAAe,CAAA;AAAA,IACpC;AAEA,IAAA,IAAI,aAAA,KAAkB,MAAA,IAAa,aAAA,KAAkB,IAAA,EAAM;AACvD,MAAA,WAAA,CAAY,KAAK,mBAAmB,CAAA;AAAA,IACxC;AAEA,IAAA,IAAI,MAAA,EAAQ,YAAA,EAAc,aAAA,KAAkB,MAAA,EAAW;AACnD,MAAA,WAAA,CAAY,KAAK,mBAAmB,CAAA;AAAA,IACxC;AAEA,IAAA,MAAM,kBAAkB,MAAA,CAAO,WAAA;AAAA,MAC3B,MAAA,CAAO,OAAA,CAAQ,MAAA,IAAU,EAAE,CAAA,CAAE,MAAA;AAAA,QACzB,CAAC,CAAC,GAAA,EAAK,KAAK,MACR,KAAA,KAAU,MAAA,IAAa,2BAAA,CAA4B,QAAA,CAAS,GAA+B;AAAA;AACnG,KACJ;AAEA,IAAA,WAAA,CAAY,IAAA;AAAA,MACR,GAAG,MAAA,CAAO,OAAA,CAAQ,eAAe,CAAA,CAAE,GAAA,CAAI,CAAC,CAAC,GAAG,CAAA,KAAM,CAAA,kCAAA,EAAqC,GAAG,CAAA,KAAA,CAAO;AAAA,KACrG;AACA,IAAA,IAAI,WAAA,CAAY,SAAS,CAAA,EAAG;AACxB,MAAA,GAAA,IAAO,CAAA;AAAA,EAAA,EAAY,WAAA,CAAY,IAAA,CAAK,UAAU,CAAC;AAAA,CAAA;AAAA,IACnD;AAEA,IAAA,GAAA,IAAO,+BAAA;AAEP,IAAA,IAAI,KAAA,EAAO;AAEP,MAAA,GAAA,IAAO,CAAA,OAAA,EAAU,QAAA,CAAS,KAAA,EAAc,EAAE,CAAC,CAAA,CAAA;AAAA,IAC/C;AAEA,IAAA,MAAM,IAAA,GAAO;AAAA,MACT,SAAA;AAAA,MACA,aAAA;AAAA,MACA,QAAQ,YAAA,EAAc,aAAA;AAAA,MACtB,GAAG,MAAA,CAAO,MAAA,CAAO,eAAe,CAAA,CAAE,GAAA,CAAI,CAAC,KAAA,KAAU,IAAA,CAAK,SAAA,CAAU,KAAK,CAAC;AAAA,MACxE,MAAA,CAAO,CAAC,UAAU,KAAA,KAAU,MAAA,IAAa,UAAU,IAAI,CAAA;AAEzD,IAAA,MAAM,IAAA,GAAwB,KAAK,EAAA,CAAG,OAAA,CAAQ,GAAG,CAAA,CAAE,GAAA,CAAI,GAAG,IAAI,CAAA;AAE9D,IAAA,IAAI,IAAA,EAAM;AACN,MAAA,KAAA,MAAW,OAAO,IAAA,EAAM;AACpB,QAAA,MAAM,aAAA,GAAgB,MAAM,OAAA,CAAQ,GAAA;AAAA,UAC/B,KAAK,KAAA,CAAM,GAAA,CAAI,cAAc,CAAA,CAA2B,GAAA,CAAI,OAAO,KAAA,KAAU;AAC1E,YAAA,OAAO;AAAA,cACH,KAAA,CAAM,OAAA;AAAA,cACN,KAAA,CAAM,OAAA;AAAA,cACN,MAAM,KAAK,KAAA,CAAM,UAAA,CAAW,MAAM,IAAA,IAAQ,MAAA,EAAQ,KAAA,CAAM,KAAA,IAAS,EAAE;AAAA,aACvE;AAAA,UACJ,CAAC;AAAA,SACL;AAEA,QAAA,MAAM,UAAA,GAAc,MAAM,IAAA,CAAK,KAAA,CAAM,WAAW,GAAA,CAAI,IAAA,IAAQ,MAAA,EAAQ,GAAA,CAAI,UAAU,CAAA;AAElF,QAAA,IAAI,UAAA,CAAW,CAAA,GAAI,CAAA,IAAK,GAAA,CAAI,wBAAwB,IAAA,EAAM;AACtD,UAAA,MAAM,KAAK,mBAAA,CAAoB,UAAA,EAAY,GAAA,CAAI,SAAA,EAAW,IAAI,oBAAoB,CAAA;AAAA,QACtF;AAEA,QAAA,MAAM;AAAA,UACF,MAAA,EAAQ;AAAA,YACJ,YAAA,EAAc;AAAA,cACV,WAAW,GAAA,CAAI,SAAA;AAAA,cACf,eAAe,GAAA,CAAI,aAAA;AAAA,cACnB,eAAe,GAAA,CAAI;AAAA;AACvB,WACJ;AAAA,UACA,UAAA;AAAA,UACA,QAAA,EAAW,MAAM,IAAA,CAAK,KAAA,CAAM,WAAW,GAAA,CAAI,IAAA,IAAQ,MAAA,EAAQ,GAAA,CAAI,QAAQ,CAAA;AAAA,UACvE,YAAA,EAAc,IAAI,oBAAA,GACZ;AAAA,YACI,YAAA,EAAc;AAAA,cACV,WAAW,GAAA,CAAI,SAAA;AAAA,cACf,eAAe,GAAA,CAAI,aAAA;AAAA,cACnB,eAAe,GAAA,CAAI;AAAA;AACvB,WACJ,GACA,MAAA;AAAA,UACN;AAAA,SACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA,EAEA,MAAM,GAAA,CAAI,MAAA,EAAwB,UAAA,EAAwB,QAAA,EAAuD;AAC7G,IAAA,IAAA,CAAK,KAAA,EAAM;AAEX,IAAA,IAAI,CAAC,OAAO,YAAA,EAAc;AACtB,MAAA,MAAM,IAAI,MAAM,+BAA+B,CAAA;AAAA,IACnD;AAEA,IAAA,MAAM,SAAA,GAAY,OAAO,YAAA,EAAc,SAAA;AACvC,IAAA,MAAM,aAAA,GAAgB,MAAA,CAAO,YAAA,EAAc,aAAA,IAAiB,EAAA;AAC5D,IAAA,MAAM,oBAAA,GAAuB,OAAO,YAAA,EAAc,aAAA;AAElD,IAAA,IAAI,CAAC,SAAA,EAAW;AACZ,MAAA,MAAM,IAAI,MAAM,CAAA,0DAAA,CAA4D,CAAA;AAAA,IAChF;AAEA,IAAA,MAAM,kBAAA,GAA0C,eAAe,UAAU,CAAA;AAEzE,IAAA,MAAM,CAAC,CAAC,KAAA,EAAO,oBAAoB,CAAA,EAAG,CAAC,KAAA,EAAO,kBAAkB,CAAC,CAAA,GAAI,MAAM,OAAA,CAAQ,GAAA,CAAI;AAAA,MACnF,IAAA,CAAK,KAAA,CAAM,UAAA,CAAW,kBAAkB,CAAA;AAAA,MACxC,IAAA,CAAK,KAAA,CAAM,UAAA,CAAW,QAAQ;AAAA,KACjC,CAAA;AAED,IAAA,IAAI,UAAU,KAAA,EAAO;AACjB,MAAA,MAAM,IAAI,MAAM,gEAAgE,CAAA;AAAA,IACpF;AACA,IAAA,MAAM,GAAA,GAAM;AAAA,MACR,SAAA;AAAA,MACA,aAAA;AAAA,MACA,UAAA,CAAW,EAAA;AAAA,MACX,oBAAA;AAAA,MACA,KAAA;AAAA,MACA,oBAAA;AAAA,MACA;AAAA,KACJ;AAEA,IAAA,IAAA,CAAK,EAAA,CACA,OAAA;AAAA,MACG,CAAA,2JAAA;AAAA,KACJ,CACC,GAAA,CAAI,GAAG,GAAG,CAAA;AAEf,IAAA,OAAO;AAAA,MACH,YAAA,EAAc;AAAA,QACV,SAAA;AAAA,QACA,aAAA;AAAA,QACA,eAAe,UAAA,CAAW;AAAA;AAC9B,KACJ;AAAA,EACJ;AAAA,EAEA,MAAM,SAAA,CAAU,MAAA,EAAwB,MAAA,EAAwB,MAAA,EAA+B;AAC3F,IAAA,IAAA,CAAK,KAAA,EAAM;AAEX,IAAA,IAAI,CAAC,OAAO,YAAA,EAAc;AACtB,MAAA,MAAM,IAAI,MAAM,+BAA+B,CAAA;AAAA,IACnD;AAEA,IAAA,IAAI,CAAC,MAAA,CAAO,YAAA,EAAc,SAAA,EAAW;AACjC,MAAA,MAAM,IAAI,MAAM,iDAAiD,CAAA;AAAA,IACrE;AAEA,IAAA,IAAI,CAAC,MAAA,CAAO,YAAA,EAAc,aAAA,EAAe;AACrC,MAAA,MAAM,IAAI,MAAM,qDAAqD,CAAA;AAAA,IACzE;AAEA,IAAA,MAAM,IAAA,GAAO,IAAA,CAAK,EAAA,CAAG,OAAA,CAAQ;AAAA;AAAA;AAAA;AAAA,IAAA,CAIhC,CAAA;AAEG,IAAA,MAAM,WAAA,GAAc,IAAA,CAAK,EAAA,CAAG,WAAA,CAAY,CAACA,KAAAA,KAAS;AAC9C,MAAA,KAAA,MAAW,OAAOA,KAAAA,EAAM;AACpB,QAAA,IAAA,CAAK,GAAA,CAAI,GAAG,GAAG,CAAA;AAAA,MACnB;AAAA,IACJ,CAAC,CAAA;AAED,IAAA,MAAM,IAAA,GAAO,MAAM,OAAA,CAAQ,GAAA;AAAA,MACvB,MAAA,CAAO,GAAA,CAAI,OAAO,KAAA,EAAO,GAAA,KAAQ;AAC7B,QAAA,MAAM,CAAC,IAAA,EAAM,eAAe,CAAA,GAAI,MAAM,KAAK,KAAA,CAAM,UAAA,CAAW,KAAA,CAAM,CAAC,CAAC,CAAA;AACpE,QAAA,OAAO;AAAA,UACH,OAAO,YAAA,EAAc,SAAA;AAAA,UACrB,OAAO,YAAA,EAAc,aAAA;AAAA,UACrB,OAAO,YAAA,EAAc,aAAA;AAAA,UACrB,MAAA;AAAA,UACA,GAAA;AAAA,UACA,MAAM,CAAC,CAAA;AAAA,UACP,IAAA;AAAA,UACA;AAAA,SACJ;AAAA,MACJ,CAAC;AAAA,KACL;AAEA,IAAA,WAAA,CAAY,IAAI,CAAA;AAAA,EACpB;AAAA,EAEA,MAAM,aAAa,QAAA,EAAkB;AACjC,IAAA,MAAM,WAAA,GAAc,IAAA,CAAK,EAAA,CAAG,WAAA,CAAY,MAAM;AAC1C,MAAA,IAAA,CAAK,EAAA,CAAG,OAAA,CAAQ,CAAA,2CAAA,CAA6C,CAAA,CAAE,IAAI,QAAQ,CAAA;AAC3E,MAAA,IAAA,CAAK,EAAA,CAAG,OAAA,CAAQ,CAAA,sCAAA,CAAwC,CAAA,CAAE,IAAI,QAAQ,CAAA;AAAA,IAC1E,CAAC,CAAA;AAED,IAAA,WAAA,EAAY;AAAA,EAChB;AAAA,EAEA,MAAgB,mBAAA,CAAoB,UAAA,EAAwB,QAAA,EAAkB,kBAAA,EAA4B;AACtG,IAAA,MAAM,EAAE,aAAA,EAAc,GAAI,IAAA,CAAK,EAAA,CAC1B,OAAA;AAAA,MACG;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,8BAAA,EAYgB,KAAK,CAAA;AAAA;AAAA,QAAA;AAAA,KAGzB,CACC,GAAA,CAAI,QAAA,EAAU,kBAAkB,CAAA;AAErC,IAAA,MAAM,iBAAA,GAAoB,UAAA;AAG1B,IAAA,iBAAA,CAAkB,mBAAmB,EAAC;AACtC,IAAA,iBAAA,CAAkB,cAAA,CAAe,KAAK,CAAA,GAAI,MAAM,OAAA,CAAQ,GAAA;AAAA,MACpD,IAAA,CAAK,KAAA,CAAM,aAAa,CAAA,CAAE,IAAI,CAAC,EAAE,IAAA,EAAM,KAAA,OAA+B,IAAA,CAAK,KAAA,CAAM,UAAA,CAAW,IAAA,EAAM,KAAK,CAAC;AAAA,KAC5G;AAGA,IAAA,iBAAA,CAAkB,gBAAA,CAAiB,KAAK,CAAA,GACpC,MAAA,CAAO,KAAK,UAAA,CAAW,gBAAgB,EAAE,MAAA,GAAS,CAAA,GAC5C,kBAAkB,GAAG,MAAA,CAAO,OAAO,UAAA,CAAW,gBAAgB,CAAC,CAAA,GAC/D,IAAA,CAAK,eAAe,MAAS,CAAA;AAAA,EAC3C;AACJ;;;;"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"sqlite-adapter-BNYk10Gl.js","sources":["../src/storage/kysely/sqlite-adapter.ts"],"sourcesContent":["import { Kysely, sql, SqlBool, Expression, SqliteDialect } from 'kysely';\nimport { DatabaseAdapter } from './adapter';\nimport { Database } from './types';\n\n/**\n * SQLite 适配器\n * - 时间类型:TEXT (ISO 8601 字符串)\n * - JSON 类型:TEXT\n * - JSON 查询:使用 json_extract 函数\n */\nexport class SQLiteAdapter implements DatabaseAdapter {\n db: Kysely<Database>;\n constructor(public database: any) {\n this.db = new Kysely<Database>({\n dialect: new SqliteDialect({\n database: database,\n }),\n });\n }\n dateToDb(date: Date): string {\n // SQLite 存储为 ISO 8601 字符串\n return date.toISOString();\n }\n\n dbToDate(dbValue: any): Date {\n // SQLite 返回字符串,需要转换为 Date\n return new Date(dbValue);\n }\n\n jsonToDb(obj: any): string {\n // SQLite 存储为 JSON 字符串\n return JSON.stringify(obj);\n }\n\n dbToJson(dbValue: any): any {\n // SQLite 返回字符串,需要解析\n if (typeof dbValue === 'string') {\n try {\n return JSON.parse(dbValue);\n } catch {\n return dbValue;\n }\n }\n return dbValue;\n }\n\n buildJsonQuery(\n db: Kysely<Database>,\n field: 'metadata' | 'interrupts',\n key: string,\n value: any,\n ): Expression<SqlBool> {\n // SQLite 使用 json_extract 函数\n return sql<boolean>`json_extract(${sql.ref(field)}, ${sql.lit('$.' + key)}) = ${sql.lit(\n JSON.stringify(value),\n )}`;\n }\n\n now(): string {\n return new Date().toISOString();\n }\n\n async createTables(db: Kysely<Database>): Promise<void> {\n // 创建 threads 表\n await sql`\n CREATE TABLE IF NOT EXISTS threads (\n thread_id TEXT PRIMARY KEY,\n created_at TEXT NOT NULL,\n updated_at TEXT NOT NULL,\n metadata TEXT NOT NULL DEFAULT '{}',\n status TEXT NOT NULL DEFAULT 'idle',\n values TEXT,\n interrupts TEXT NOT NULL DEFAULT '{}'\n )\n `.execute(db);\n\n // 创建 runs 表\n await sql`\n CREATE TABLE IF NOT EXISTS runs (\n run_id TEXT PRIMARY KEY,\n thread_id TEXT NOT NULL,\n assistant_id TEXT NOT NULL,\n created_at TEXT NOT NULL,\n updated_at TEXT NOT NULL,\n status TEXT NOT NULL DEFAULT 'pending',\n metadata TEXT NOT NULL DEFAULT '{}',\n multitask_strategy TEXT NOT NULL DEFAULT 'reject',\n FOREIGN KEY (thread_id) REFERENCES threads(thread_id) ON DELETE CASCADE\n )\n `.execute(db);\n }\n\n async createIndexes(db: Kysely<Database>): Promise<void> {\n await sql`CREATE INDEX IF NOT EXISTS idx_threads_status ON threads(status)`.execute(db);\n await sql`CREATE INDEX IF NOT EXISTS idx_threads_created_at ON threads(created_at)`.execute(db);\n await sql`CREATE INDEX IF NOT EXISTS idx_threads_updated_at ON threads(updated_at)`.execute(db);\n await sql`CREATE INDEX IF NOT EXISTS idx_runs_thread_id ON runs(thread_id)`.execute(db);\n await sql`CREATE INDEX IF NOT EXISTS idx_runs_status ON runs(status)`.execute(db);\n }\n}\n"],"names":[],"mappings":";;AAUO,MAAM,aAAA,CAAyC;AAAA,EAElD,YAAmB,QAAA,EAAe;AAAf,IAAA,IAAA,CAAA,QAAA,GAAA,QAAA;AACf,IAAA,IAAA,CAAK,EAAA,GAAK,IAAI,MAAA,CAAiB;AAAA,MAC3B,OAAA,EAAS,IAAI,aAAA,CAAc;AAAA,QACvB;AAAA,OACH;AAAA,KACJ,CAAA;AAAA,EACL;AAAA,EAPA,EAAA;AAAA,EAQA,SAAS,IAAA,EAAoB;AAEzB,IAAA,OAAO,KAAK,WAAA,EAAY;AAAA,EAC5B;AAAA,EAEA,SAAS,OAAA,EAAoB;AAEzB,IAAA,OAAO,IAAI,KAAK,OAAO,CAAA;AAAA,EAC3B;AAAA,EAEA,SAAS,GAAA,EAAkB;AAEvB,IAAA,OAAO,IAAA,CAAK,UAAU,GAAG,CAAA;AAAA,EAC7B;AAAA,EAEA,SAAS,OAAA,EAAmB;AAExB,IAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC7B,MAAA,IAAI;AACA,QAAA,OAAO,IAAA,CAAK,MAAM,OAAO,CAAA;AAAA,MAC7B,CAAA,CAAA,MAAQ;AACJ,QAAA,OAAO,OAAA;AAAA,MACX;AAAA,IACJ;AACA,IAAA,OAAO,OAAA;AAAA,EACX;AAAA,EAEA,cAAA,CACI,EAAA,EACA,KAAA,EACA,GAAA,EACA,KAAA,EACmB;AAEnB,IAAA,OAAO,GAAA,CAAA,aAAA,EAA4B,GAAA,CAAI,GAAA,CAAI,KAAK,CAAC,CAAA,EAAA,EAAK,GAAA,CAAI,GAAA,CAAI,IAAA,GAAO,GAAG,CAAC,CAAA,IAAA,EAAO,GAAA,CAAI,GAAA;AAAA,MAChF,IAAA,CAAK,UAAU,KAAK;AAAA,KACvB,CAAA,CAAA;AAAA,EACL;AAAA,EAEA,GAAA,GAAc;AACV,IAAA,OAAA,iBAAO,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,EAClC;AAAA,EAEA,MAAM,aAAa,EAAA,EAAqC;AAEpD,IAAA,MAAM,GAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QAAA,CAAA,CAUJ,QAAQ,EAAE,CAAA;AAGZ,IAAA,MAAM,GAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QAAA,CAAA,CAYJ,QAAQ,EAAE,CAAA;AAAA,EAChB;AAAA,EAEA,MAAM,cAAc,EAAA,EAAqC;AACrD,IAAA,MAAM,GAAA,CAAA,gEAAA,CAAA,CAAsE,QAAQ,EAAE,CAAA;AACtF,IAAA,MAAM,GAAA,CAAA,wEAAA,CAAA,CAA8E,QAAQ,EAAE,CAAA;AAC9F,IAAA,MAAM,GAAA,CAAA,wEAAA,CAAA,CAA8E,QAAQ,EAAE,CAAA;AAC9F,IAAA,MAAM,GAAA,CAAA,gEAAA,CAAA,CAAsE,QAAQ,EAAE,CAAA;AACtF,IAAA,MAAM,GAAA,CAAA,0DAAA,CAAA,CAAgE,QAAQ,EAAE,CAAA;AAAA,EACpF;AACJ;;;;"}
|