@langgraph-js/pure-graph 2.4.2 → 2.5.1

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.
@@ -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;;;;"}