npm - @undefineds.co/xpod - Versions diffs - 0.2.0 → 0.2.2-preview.0 - Mend

@undefineds.co/xpod 0.2.0 → 0.2.2-preview.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.

Files changed (1545) hide show

package/dist/storage/sqlite/types.d.ts ADDED Viewed

@@ -0,0 +1,29 @@
+export type SqliteRuntimeKind = 'node-sqlite' | 'bun-sqlite';
+export interface SqliteOpenOptions {
+    readonly?: boolean;
+}
+export interface SqliteRunResult {
+    changes: number;
+    lastInsertRowid: number | bigint;
+}
+export interface SqliteStatement<TResult = unknown> {
+    run(...params: unknown[]): SqliteRunResult;
+    get(...params: unknown[]): TResult | undefined;
+    all(...params: unknown[]): TResult[];
+    iterate(...params: unknown[]): IterableIterator<TResult>;
+}
+export interface SqliteDatabase {
+    readonly kind: SqliteRuntimeKind;
+    readonly native: unknown;
+    exec(sql: string): unknown;
+    pragma(pragma: string): unknown;
+    prepare<TResult = unknown>(sql: string): SqliteStatement<TResult>;
+    loadExtension(path: string): void;
+    transaction<TArgs extends unknown[], TResult>(fn: (...args: TArgs) => TResult): (...args: TArgs) => TResult;
+    close(): void;
+}
+export interface SqliteRuntime {
+    readonly kind: SqliteRuntimeKind;
+    openDatabase(path: string, options?: SqliteOpenOptions): SqliteDatabase;
+    createDrizzleDatabase(database: SqliteDatabase): any;
+}

package/dist/storage/sqlite/types.js ADDED Viewed

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/storage/sqlite/types.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/storage/sqlite/types.ts"],"names":[],"mappings":"","sourcesContent":["export type SqliteRuntimeKind = 'node-sqlite' | 'bun-sqlite';\n\nexport interface SqliteOpenOptions {\n readonly?: boolean;\n}\n\nexport interface SqliteRunResult {\n changes: number;\n lastInsertRowid: number | bigint;\n}\n\nexport interface SqliteStatement<TResult = unknown> {\n run(...params: unknown[]): SqliteRunResult;\n get(...params: unknown[]): TResult | undefined;\n all(...params: unknown[]): TResult[];\n iterate(...params: unknown[]): IterableIterator<TResult>;\n}\n\nexport interface SqliteDatabase {\n readonly kind: SqliteRuntimeKind;\n readonly native: unknown;\n exec(sql: string): unknown;\n pragma(pragma: string): unknown;\n prepare<TResult = unknown>(sql: string): SqliteStatement<TResult>;\n loadExtension(path: string): void;\n transaction<TArgs extends unknown[], TResult>(fn: (...args: TArgs) => TResult): (...args: TArgs) => TResult;\n close(): void;\n}\n\nexport interface SqliteRuntime {\n readonly kind: SqliteRuntimeKind;\n openDatabase(path: string, options?: SqliteOpenOptions): SqliteDatabase;\n createDrizzleDatabase(database: SqliteDatabase): any;\n}\n"]}

package/dist/storage/vector/PostgresVectorStore.d.ts CHANGED Viewed

@@ -23,6 +23,7 @@ export declare class PostgresVectorStore extends VectorStore implements Initiali
     private ensureOpen;
     private getDb;
     protected getTableName(modelId: string): string;
+    private getCountTableCandidates;
     ensureVectorTable(modelId: string): Promise<void>;
     dropVectorTable(modelId: string): Promise<void>;
     hasVectorTable(modelId: string): Promise<boolean>;

package/dist/storage/vector/PostgresVectorStore.js CHANGED Viewed

@@ -60,6 +60,14 @@ class PostgresVectorStore extends VectorStore_1.VectorStore {
     getTableName(modelId) {
         return `quint_vec_${(0, VectorStore_1.hashModelId)(modelId)}`;
     }
+    getCountTableCandidates(modelIdOrTableName) {
+        const candidates = [];
+        if (modelIdOrTableName.startsWith('quint_vec_')) {
+            candidates.push(modelIdOrTableName);
+        }
+        candidates.push(this.getTableName(modelIdOrTableName));
+        return [...new Set(candidates)];
+    }
     // ============================================
     // 向量表管理
     // ============================================
@@ -219,16 +227,18 @@ class PostgresVectorStore extends VectorStore_1.VectorStore {
     // ============================================
     async countVectors(modelId) {
         await this.ensureOpen();
-        const tableName = this.getTableName(modelId);
-        try {
-            const result = await this.getDb().execute((0, drizzle_orm_1.sql) `
-        SELECT COUNT(*) as count FROM ${drizzle_orm_1.sql.identifier(tableName)}
-      `);
-            return parseInt(result.rows[0]?.count ?? '0', 10);
-        }
-        catch {
-            return 0;
+        for (const tableName of this.getCountTableCandidates(modelId)) {
+            try {
+                const result = await this.getDb().execute((0, drizzle_orm_1.sql) `
+          SELECT COUNT(*) as count FROM ${drizzle_orm_1.sql.identifier(tableName)}
+        `);
+                return parseInt(result.rows[0]?.count ?? '0', 10);
+            }
+            catch {
+                continue;
+            }
         }
+        return 0;
     }
     // ============================================
     // 迁移支持

package/dist/storage/vector/PostgresVectorStore.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"PostgresVectorStore.js","sourceRoot":"","sources":["../../../src/storage/vector/PostgresVectorStore.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;AAEH,2BAA0B;AAC1B,6DAAiF;AACjF,6CAAkC;AAElC,+CAAyD;AAGzD,MAAa,mBAAoB,SAAQ,yBAAW;IAOlD,YAAmB,OAA2B;QAC5C,KAAK,EAAE,CAAC;QANV,eAAe;QACP,SAAI,GAAgB,IAAI,CAAC;QACjC,eAAe;QACP,OAAE,GAA0B,IAAI,CAAC;QAIvC,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IACnD,CAAC;IAED,+CAA+C;IAC/C,0CAA0C;IAC1C,+CAA+C;IAExC,KAAK,CAAC,UAAU;QACrB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;IACpB,CAAC;IAEM,KAAK,CAAC,QAAQ;QACnB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEM,KAAK,CAAC,IAAI;QACf,IAAI,IAAI,CAAC,IAAI;YAAE,OAAO;QAEtB,IAAI,CAAC,IAAI,GAAG,IAAI,SAAI,CAAC,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,EAAE,GAAG,IAAA,uBAAS,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE/B,oBAAoB;QACpB,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA,uCAAuC,CAAC,CAAC;IACpE,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;YACjB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAEO,KAAK;QACX,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,qEAAqE,CAAC,CAAC;QACzF,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAES,YAAY,CAAC,OAAe;QACpC,OAAO,aAAa,IAAA,yBAAW,EAAC,OAAO,CAAC,EAAE,CAAC;IAC7C,CAAC;IAED,+CAA+C;IAC/C,QAAQ;IACR,+CAA+C;IAExC,KAAK,CAAC,iBAAiB,CAAC,OAAe;QAC5C,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QAExB,MAAM;QACN,MAAM,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;mCACW,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;;2BAEjC,iBAAG,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC;;;KAG/D,CAAC,CAAC;QAEH,qBAAqB;QACrB,MAAM,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;mCACW,iBAAG,CAAC,UAAU,CAAC,GAAG,SAAS,gBAAgB,CAAC;WACpE,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;;KAE/B,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,eAAe,CAAC,OAAe;QAC1C,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA,wBAAwB,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAC7F,CAAC;IAEM,KAAK,CAAC,cAAc,CAAC,OAAe;QACzC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;;;+BAGpB,SAAS;;OAEjC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,KAAK,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,gBAAgB;QAC3B,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;;;OAG5C,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACnD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAExC,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,EAAU,EAAE,SAAmB;QACxE,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAEhD,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;oBACd,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;gBAC7B,EAAE,KAAK,YAAY;mDACgB,YAAY;KAC1D,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,OAA8C;QAC7F,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEjC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,EAAE,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5B,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;gBAC1B,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;gBACpD,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;wBACd,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;oBAC7B,GAAG,CAAC,EAAE,KAAK,YAAY;uDACY,YAAY;SAC1D,CAAC,CAAC;YACL,CAAC;YACD,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,OAAe,EAAE,EAAU;QAChD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;gDACH,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC,eAAe,EAAE;OACnF,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC1E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,EAAU;QACnD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA,eAAe,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;IAC7F,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,GAAa;QAC5D,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE7B,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE7B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;oBACd,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;qBACxB,iBAAG,CAAC,GAAG,CAAC,MAAM,CAAC;KAC/B,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,cAAwB,EAAE,UAA+B,EAAE;QAC9F,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;QACtD,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAErD,SAAS;QACT,IAAI,WAAW,GAAG,6BAA6B,CAAC;QAEhD,mBAAmB;QACnB,IAAI,UAAU,IAAI,UAAU,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrD,WAAW,IAAI,mBAAmB,WAAW,GAAG,CAAC;QACnD,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;iCAChB,YAAY;aAChC,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAC9B,iBAAG,CAAC,GAAG,CAAC,WAAW,CAAC;;cAEd,KAAK,GAAG,CAAC;KAClB,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,MAAM,CAAC,IAA0C,CAAC;QAC/D,MAAM,OAAO,GAAyB,EAAE,CAAC;QAEzC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;YAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YAE7C,IAAI,SAAS,KAAK,SAAS,IAAI,KAAK,GAAG,SAAS;gBAAE,SAAS;YAE3D,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,+CAA+C;IAC/C,KAAK;IACL,+CAA+C;IAExC,KAAK,CAAC,YAAY,CAAC,OAAe;QACvC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;wCACX,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;OAC1D,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;QACpD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAExC,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,UAAgD,EAAE;QAC3F,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,EAAE,KAAK,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QAE1C,IAAI,CAAC;YACH,IAAI,MAAM,CAAC;YACX,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBAC1B,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;2BACpB,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;uBAC7B,OAAO;;kBAEZ,KAAK;SACd,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;2BACpB,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;;kBAElC,KAAK;SACd,CAAC,CAAC;YACL,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAEvC,WAAW,CAAC,GAAQ;QAC1B,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC;YAC7C,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC;SAC5C,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,SAAc;QACnC,wBAAwB;QACxB,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7B,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAClC,kBAAkB;YAClB,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC/B,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AAzTD,kDAyTC","sourcesContent":["/*\n PostgresVectorStore - PostgreSQL 向量存储实现（使用 pgvector 扩展）\n \n 使用 pgvector 扩展进行高性能向量搜索：\n * - 向量表命名：quint_vec_{hash(modelId)}\n * - 利用 <=> 运算符（余弦距离）进行近似最近邻搜索\n * - 支持 IVFFlat 和 HNSW 索引\n /\n\nimport { Pool } from 'pg';\nimport { drizzle as drizzlePg, NodePgDatabase } from 'drizzle-orm/node-postgres';\nimport { sql } from 'drizzle-orm';\nimport type { Finalizable, Initializable } from '@solid/community-server';\nimport { VectorStore, hashModelId } from './VectorStore';\nimport type { VectorRecord, VectorSearchOptions, VectorSearchResult, VectorStoreOptions } from './types';\n\nexport class PostgresVectorStore extends VectorStore implements Initializable, Finalizable {\n protected readonly connectionString: string;\n /* @ignored /\n private pool: Pool \| null = null;\n /* @ignored /\n private db: NodePgDatabase \| null = null;\n\n public constructor(options: VectorStoreOptions) {\n super();\n this.connectionString = options.connectionString;\n }\n\n // ============================================\n // Lifecycle (Initializable / Finalizable)\n // ============================================\n\n public async initialize(): Promise<void> {\n await this.open();\n }\n\n public async finalize(): Promise<void> {\n await this.close();\n }\n\n public async open(): Promise<void> {\n if (this.pool) return;\n\n this.pool = new Pool({ connectionString: this.connectionString });\n this.db = drizzlePg(this.pool);\n\n // 确保 pgvector 扩展已安装\n await this.db.execute(sql`CREATE EXTENSION IF NOT EXISTS vector`);\n }\n\n public async close(): Promise<void> {\n if (this.pool) {\n await this.pool.end();\n this.pool = null;\n this.db = null;\n }\n }\n\n private async ensureOpen(): Promise<void> {\n if (!this.pool) {\n await this.open();\n }\n }\n\n private getDb(): NodePgDatabase {\n if (!this.db) {\n throw new Error('PostgresVectorStore is not open. Call open() or initialize() first.');\n }\n return this.db;\n }\n\n protected getTableName(modelId: string): string {\n return `quint_vec_${hashModelId(modelId)}`;\n }\n\n // ============================================\n // 向量表管理\n // ============================================\n\n public async ensureVectorTable(modelId: string): Promise<void> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const db = this.getDb();\n\n // 创建表\n await db.execute(sql`\n CREATE TABLE IF NOT EXISTS ${sql.identifier(tableName)} (\n id INTEGER PRIMARY KEY,\n embedding vector(${sql.raw(this.defaultDimension.toString())}),\n created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()\n )\n `);\n\n // 创建 HNSW 索引（高维向量推荐）\n await db.execute(sql`\n CREATE INDEX IF NOT EXISTS ${sql.identifier(`${tableName}_embedding_idx`)}\n ON ${sql.identifier(tableName)}\n USING hnsw (embedding vector_cosine_ops)\n `);\n }\n\n public async dropVectorTable(modelId: string): Promise<void> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n await this.getDb().execute(sql`DROP TABLE IF EXISTS ${sql.identifier(tableName)} CASCADE`);\n }\n\n public async hasVectorTable(modelId: string): Promise<boolean> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const result = await this.getDb().execute(sql`\n SELECT EXISTS (\n SELECT FROM information_schema.tables\n WHERE table_name = ${tableName}\n ) as exists\n `);\n return result.rows[0]?.exists ?? false;\n } catch {\n return false;\n }\n }\n\n public async listVectorTables(): Promise<string[]> {\n await this.ensureOpen();\n\n try {\n const result = await this.getDb().execute(sql`\n SELECT table_name FROM information_schema.tables\n WHERE table_name LIKE 'quint_vec_%'\n `);\n return result.rows.map((r: any) => r.table_name);\n } catch {\n return [];\n }\n }\n\n // ============================================\n // 向量操作\n // ============================================\n\n public async upsertVector(modelId: string, id: number, embedding: number[]): Promise<void> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const embeddingStr = `[${embedding.join(',')}]`;\n\n await this.getDb().execute(sql`\n INSERT INTO ${sql.identifier(tableName)} (id, embedding)\n VALUES (${id}, ${embeddingStr}::vector)\n ON CONFLICT (id) DO UPDATE SET embedding = ${embeddingStr}::vector, created_at = NOW()\n `);\n }\n\n public async batchUpsertVectors(modelId: string, records: { id: number; embedding: number[] }[]): Promise<void> {\n if (records.length === 0) return;\n\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n const client = await this.pool!.connect();\n try {\n await client.query('BEGIN');\n for (const rec of records) {\n const embeddingStr = `[${rec.embedding.join(',')}]`;\n await this.getDb().execute(sql`\n INSERT INTO ${sql.identifier(tableName)} (id, embedding)\n VALUES (${rec.id}, ${embeddingStr}::vector)\n ON CONFLICT (id) DO UPDATE SET embedding = ${embeddingStr}::vector, created_at = NOW()\n `);\n }\n await client.query('COMMIT');\n } catch (e) {\n await client.query('ROLLBACK');\n throw e;\n } finally {\n client.release();\n }\n }\n\n public async getVector(modelId: string, id: number): Promise<VectorRecord \| null> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const result = await this.getDb().execute(sql`\n SELECT id, embedding, created_at FROM ${sql.identifier(tableName)} WHERE id = ${id}\n `);\n return result.rows.length > 0 ? this.rowToVector(result.rows[0]) : null;\n } catch {\n return null;\n }\n }\n\n public async deleteVector(modelId: string, id: number): Promise<void> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n await this.getDb().execute(sql`DELETE FROM ${sql.identifier(tableName)} WHERE id = ${id}`);\n }\n\n public async batchDeleteVectors(modelId: string, ids: number[]): Promise<void> {\n if (ids.length === 0) return;\n\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const idList = ids.join(',');\n\n await this.getDb().execute(sql`\n DELETE FROM ${sql.identifier(tableName)}\n WHERE id IN (${sql.raw(idList)})\n `);\n }\n\n public async search(modelId: string, queryEmbedding: number[], options: VectorSearchOptions = {}): Promise<VectorSearchResult[]> {\n await this.ensureOpen();\n const { limit = 10, threshold, excludeIds } = options;\n const tableName = this.getTableName(modelId);\n const embeddingStr = `[${queryEmbedding.join(',')}]`;\n\n // 构建基础查询\n let whereClause = 'WHERE embedding IS NOT NULL';\n \n // 添加 excludeIds 过滤\n if (excludeIds && excludeIds.size > 0) {\n const excludeList = Array.from(excludeIds).join(',');\n whereClause += ` AND id NOT IN (${excludeList})`;\n }\n\n const result = await this.getDb().execute(sql`\n SELECT id, embedding <=> ${embeddingStr}::vector as distance\n FROM ${sql.identifier(tableName)}\n ${sql.raw(whereClause)}\n ORDER BY distance\n LIMIT ${limit 2}\n `);\n\n const rows = result.rows as { id: number; distance: number }[];\n const results: VectorSearchResult[] = [];\n\n for (const row of rows) {\n const distance = row.distance;\n const score = this.distanceToScore(distance);\n\n if (threshold !== undefined && score < threshold) continue;\n\n results.push({ id: row.id, distance, score });\n }\n\n return results.slice(0, limit);\n }\n\n // ============================================\n // 统计\n // ============================================\n\n public async countVectors(modelId: string): Promise<number> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const result = await this.getDb().execute(sql`\n SELECT COUNT(*) as count FROM ${sql.identifier(tableName)}\n `);\n return parseInt(result.rows[0]?.count ?? '0', 10);\n } catch {\n return 0;\n }\n }\n\n // ============================================\n // 迁移支持\n // ============================================\n\n public async getVectorIds(modelId: string, options: { limit?: number; afterId?: number } = {}): Promise<number[]> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const { limit = 1000, afterId } = options;\n\n try {\n let result;\n if (afterId !== undefined) {\n result = await this.getDb().execute(sql`\n SELECT id FROM ${sql.identifier(tableName)}\n WHERE id > ${afterId}\n ORDER BY id\n LIMIT ${limit}\n `);\n } else {\n result = await this.getDb().execute(sql`\n SELECT id FROM ${sql.identifier(tableName)}\n ORDER BY id\n LIMIT ${limit}\n `);\n }\n return result.rows.map((r: any) => r.id);\n } catch {\n return [];\n }\n }\n\n // ============================================\n // 私有方法\n // ============================================\n\n private rowToVector(row: any): VectorRecord {\n return {\n id: row.id,\n embedding: this.parseEmbedding(row.embedding),\n createdAt: this.toTimestamp(row.created_at),\n };\n }\n\n private parseEmbedding(embedding: any): number[] {\n // pgvector 返回的可能是数组或字符串\n if (Array.isArray(embedding)) {\n return embedding;\n }\n if (typeof embedding === 'string') {\n // 解析 \"[1,2,3]\" 格式\n try {\n return JSON.parse(embedding);\n } catch {\n return [];\n }\n }\n return [];\n }\n}\n"]}
1	+ {"version":3,"file":"PostgresVectorStore.js","sourceRoot":"","sources":["../../../src/storage/vector/PostgresVectorStore.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;AAEH,2BAA0B;AAC1B,6DAAiF;AACjF,6CAAkC;AAElC,+CAAyD;AAGzD,MAAa,mBAAoB,SAAQ,yBAAW;IAOlD,YAAmB,OAA2B;QAC5C,KAAK,EAAE,CAAC;QANV,eAAe;QACP,SAAI,GAAgB,IAAI,CAAC;QACjC,eAAe;QACP,OAAE,GAA0B,IAAI,CAAC;QAIvC,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IACnD,CAAC;IAED,+CAA+C;IAC/C,0CAA0C;IAC1C,+CAA+C;IAExC,KAAK,CAAC,UAAU;QACrB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;IACpB,CAAC;IAEM,KAAK,CAAC,QAAQ;QACnB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEM,KAAK,CAAC,IAAI;QACf,IAAI,IAAI,CAAC,IAAI;YAAE,OAAO;QAEtB,IAAI,CAAC,IAAI,GAAG,IAAI,SAAI,CAAC,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,EAAE,GAAG,IAAA,uBAAS,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE/B,oBAAoB;QACpB,MAAM,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA,uCAAuC,CAAC,CAAC;IACpE,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;YACjB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAEO,KAAK;QACX,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,qEAAqE,CAAC,CAAC;QACzF,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAES,YAAY,CAAC,OAAe;QACpC,OAAO,aAAa,IAAA,yBAAW,EAAC,OAAO,CAAC,EAAE,CAAC;IAC7C,CAAC;IAEO,uBAAuB,CAAC,kBAA0B;QACxD,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,IAAI,kBAAkB,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;YAChD,UAAU,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;QACD,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;IAClC,CAAC;IAED,+CAA+C;IAC/C,QAAQ;IACR,+CAA+C;IAExC,KAAK,CAAC,iBAAiB,CAAC,OAAe;QAC5C,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QAExB,MAAM;QACN,MAAM,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;mCACW,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;;2BAEjC,iBAAG,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC;;;KAG/D,CAAC,CAAC;QAEH,qBAAqB;QACrB,MAAM,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;mCACW,iBAAG,CAAC,UAAU,CAAC,GAAG,SAAS,gBAAgB,CAAC;WACpE,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;;KAE/B,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,eAAe,CAAC,OAAe;QAC1C,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA,wBAAwB,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAC7F,CAAC;IAEM,KAAK,CAAC,cAAc,CAAC,OAAe;QACzC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;;;+BAGpB,SAAS;;OAEjC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,KAAK,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,gBAAgB;QAC3B,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;;;OAG5C,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACnD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAExC,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,EAAU,EAAE,SAAmB;QACxE,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAEhD,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;oBACd,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;gBAC7B,EAAE,KAAK,YAAY;mDACgB,YAAY;KAC1D,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,OAA8C;QAC7F,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEjC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,EAAE,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5B,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;gBAC1B,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;gBACpD,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;wBACd,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;oBAC7B,GAAG,CAAC,EAAE,KAAK,YAAY;uDACY,YAAY;SAC1D,CAAC,CAAC;YACL,CAAC;YACD,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,OAAe,EAAE,EAAU;QAChD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;gDACH,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC,eAAe,EAAE;OACnF,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC1E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,EAAU;QACnD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA,eAAe,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;IAC7F,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,GAAa;QAC5D,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE7B,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE7B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;oBACd,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;qBACxB,iBAAG,CAAC,GAAG,CAAC,MAAM,CAAC;KAC/B,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,cAAwB,EAAE,UAA+B,EAAE;QAC9F,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;QACtD,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;QAErD,SAAS;QACT,IAAI,WAAW,GAAG,6BAA6B,CAAC;QAEhD,mBAAmB;QACnB,IAAI,UAAU,IAAI,UAAU,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrD,WAAW,IAAI,mBAAmB,WAAW,GAAG,CAAC;QACnD,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;iCAChB,YAAY;aAChC,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAC9B,iBAAG,CAAC,GAAG,CAAC,WAAW,CAAC;;cAEd,KAAK,GAAG,CAAC;KAClB,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,MAAM,CAAC,IAA0C,CAAC;QAC/D,MAAM,OAAO,GAAyB,EAAE,CAAC;QAEzC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;YAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YAE7C,IAAI,SAAS,KAAK,SAAS,IAAI,KAAK,GAAG,SAAS;gBAAE,SAAS;YAE3D,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,+CAA+C;IAC/C,KAAK;IACL,+CAA+C;IAExC,KAAK,CAAC,YAAY,CAAC,OAAe;QACvC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;0CACX,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;SAC1D,CAAC,CAAC;gBACH,OAAO,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;YACpD,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QAED,OAAO,CAAC,CAAC;IACX,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAExC,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,UAAgD,EAAE;QAC3F,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,EAAE,KAAK,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QAE1C,IAAI,CAAC;YACH,IAAI,MAAM,CAAC;YACX,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBAC1B,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;2BACpB,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;uBAC7B,OAAO;;kBAEZ,KAAK;SACd,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,IAAA,iBAAG,EAAA;2BACpB,iBAAG,CAAC,UAAU,CAAC,SAAS,CAAC;;kBAElC,KAAK;SACd,CAAC,CAAC;YACL,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAEvC,WAAW,CAAC,GAAQ;QAC1B,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC;YAC7C,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC;SAC5C,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,SAAc;QACnC,wBAAwB;QACxB,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7B,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAClC,kBAAkB;YAClB,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC/B,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AApUD,kDAoUC","sourcesContent":["/*\n PostgresVectorStore - PostgreSQL 向量存储实现（使用 pgvector 扩展）\n \n 使用 pgvector 扩展进行高性能向量搜索：\n * - 向量表命名：quint_vec_{hash(modelId)}\n * - 利用 <=> 运算符（余弦距离）进行近似最近邻搜索\n * - 支持 IVFFlat 和 HNSW 索引\n /\n\nimport { Pool } from 'pg';\nimport { drizzle as drizzlePg, NodePgDatabase } from 'drizzle-orm/node-postgres';\nimport { sql } from 'drizzle-orm';\nimport type { Finalizable, Initializable } from '@solid/community-server';\nimport { VectorStore, hashModelId } from './VectorStore';\nimport type { VectorRecord, VectorSearchOptions, VectorSearchResult, VectorStoreOptions } from './types';\n\nexport class PostgresVectorStore extends VectorStore implements Initializable, Finalizable {\n protected readonly connectionString: string;\n /* @ignored /\n private pool: Pool \| null = null;\n /* @ignored /\n private db: NodePgDatabase \| null = null;\n\n public constructor(options: VectorStoreOptions) {\n super();\n this.connectionString = options.connectionString;\n }\n\n // ============================================\n // Lifecycle (Initializable / Finalizable)\n // ============================================\n\n public async initialize(): Promise<void> {\n await this.open();\n }\n\n public async finalize(): Promise<void> {\n await this.close();\n }\n\n public async open(): Promise<void> {\n if (this.pool) return;\n\n this.pool = new Pool({ connectionString: this.connectionString });\n this.db = drizzlePg(this.pool);\n\n // 确保 pgvector 扩展已安装\n await this.db.execute(sql`CREATE EXTENSION IF NOT EXISTS vector`);\n }\n\n public async close(): Promise<void> {\n if (this.pool) {\n await this.pool.end();\n this.pool = null;\n this.db = null;\n }\n }\n\n private async ensureOpen(): Promise<void> {\n if (!this.pool) {\n await this.open();\n }\n }\n\n private getDb(): NodePgDatabase {\n if (!this.db) {\n throw new Error('PostgresVectorStore is not open. Call open() or initialize() first.');\n }\n return this.db;\n }\n\n protected getTableName(modelId: string): string {\n return `quint_vec_${hashModelId(modelId)}`;\n }\n\n private getCountTableCandidates(modelIdOrTableName: string): string[] {\n const candidates: string[] = [];\n if (modelIdOrTableName.startsWith('quint_vec_')) {\n candidates.push(modelIdOrTableName);\n }\n candidates.push(this.getTableName(modelIdOrTableName));\n return [...new Set(candidates)];\n }\n\n // ============================================\n // 向量表管理\n // ============================================\n\n public async ensureVectorTable(modelId: string): Promise<void> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const db = this.getDb();\n\n // 创建表\n await db.execute(sql`\n CREATE TABLE IF NOT EXISTS ${sql.identifier(tableName)} (\n id INTEGER PRIMARY KEY,\n embedding vector(${sql.raw(this.defaultDimension.toString())}),\n created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()\n )\n `);\n\n // 创建 HNSW 索引（高维向量推荐）\n await db.execute(sql`\n CREATE INDEX IF NOT EXISTS ${sql.identifier(`${tableName}_embedding_idx`)}\n ON ${sql.identifier(tableName)}\n USING hnsw (embedding vector_cosine_ops)\n `);\n }\n\n public async dropVectorTable(modelId: string): Promise<void> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n await this.getDb().execute(sql`DROP TABLE IF EXISTS ${sql.identifier(tableName)} CASCADE`);\n }\n\n public async hasVectorTable(modelId: string): Promise<boolean> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const result = await this.getDb().execute(sql`\n SELECT EXISTS (\n SELECT FROM information_schema.tables\n WHERE table_name = ${tableName}\n ) as exists\n `);\n return result.rows[0]?.exists ?? false;\n } catch {\n return false;\n }\n }\n\n public async listVectorTables(): Promise<string[]> {\n await this.ensureOpen();\n\n try {\n const result = await this.getDb().execute(sql`\n SELECT table_name FROM information_schema.tables\n WHERE table_name LIKE 'quint_vec_%'\n `);\n return result.rows.map((r: any) => r.table_name);\n } catch {\n return [];\n }\n }\n\n // ============================================\n // 向量操作\n // ============================================\n\n public async upsertVector(modelId: string, id: number, embedding: number[]): Promise<void> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const embeddingStr = `[${embedding.join(',')}]`;\n\n await this.getDb().execute(sql`\n INSERT INTO ${sql.identifier(tableName)} (id, embedding)\n VALUES (${id}, ${embeddingStr}::vector)\n ON CONFLICT (id) DO UPDATE SET embedding = ${embeddingStr}::vector, created_at = NOW()\n `);\n }\n\n public async batchUpsertVectors(modelId: string, records: { id: number; embedding: number[] }[]): Promise<void> {\n if (records.length === 0) return;\n\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n const client = await this.pool!.connect();\n try {\n await client.query('BEGIN');\n for (const rec of records) {\n const embeddingStr = `[${rec.embedding.join(',')}]`;\n await this.getDb().execute(sql`\n INSERT INTO ${sql.identifier(tableName)} (id, embedding)\n VALUES (${rec.id}, ${embeddingStr}::vector)\n ON CONFLICT (id) DO UPDATE SET embedding = ${embeddingStr}::vector, created_at = NOW()\n `);\n }\n await client.query('COMMIT');\n } catch (e) {\n await client.query('ROLLBACK');\n throw e;\n } finally {\n client.release();\n }\n }\n\n public async getVector(modelId: string, id: number): Promise<VectorRecord \| null> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const result = await this.getDb().execute(sql`\n SELECT id, embedding, created_at FROM ${sql.identifier(tableName)} WHERE id = ${id}\n `);\n return result.rows.length > 0 ? this.rowToVector(result.rows[0]) : null;\n } catch {\n return null;\n }\n }\n\n public async deleteVector(modelId: string, id: number): Promise<void> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n await this.getDb().execute(sql`DELETE FROM ${sql.identifier(tableName)} WHERE id = ${id}`);\n }\n\n public async batchDeleteVectors(modelId: string, ids: number[]): Promise<void> {\n if (ids.length === 0) return;\n\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const idList = ids.join(',');\n\n await this.getDb().execute(sql`\n DELETE FROM ${sql.identifier(tableName)}\n WHERE id IN (${sql.raw(idList)})\n `);\n }\n\n public async search(modelId: string, queryEmbedding: number[], options: VectorSearchOptions = {}): Promise<VectorSearchResult[]> {\n await this.ensureOpen();\n const { limit = 10, threshold, excludeIds } = options;\n const tableName = this.getTableName(modelId);\n const embeddingStr = `[${queryEmbedding.join(',')}]`;\n\n // 构建基础查询\n let whereClause = 'WHERE embedding IS NOT NULL';\n \n // 添加 excludeIds 过滤\n if (excludeIds && excludeIds.size > 0) {\n const excludeList = Array.from(excludeIds).join(',');\n whereClause += ` AND id NOT IN (${excludeList})`;\n }\n\n const result = await this.getDb().execute(sql`\n SELECT id, embedding <=> ${embeddingStr}::vector as distance\n FROM ${sql.identifier(tableName)}\n ${sql.raw(whereClause)}\n ORDER BY distance\n LIMIT ${limit 2}\n `);\n\n const rows = result.rows as { id: number; distance: number }[];\n const results: VectorSearchResult[] = [];\n\n for (const row of rows) {\n const distance = row.distance;\n const score = this.distanceToScore(distance);\n\n if (threshold !== undefined && score < threshold) continue;\n\n results.push({ id: row.id, distance, score });\n }\n\n return results.slice(0, limit);\n }\n\n // ============================================\n // 统计\n // ============================================\n\n public async countVectors(modelId: string): Promise<number> {\n await this.ensureOpen();\n for (const tableName of this.getCountTableCandidates(modelId)) {\n try {\n const result = await this.getDb().execute(sql`\n SELECT COUNT(*) as count FROM ${sql.identifier(tableName)}\n `);\n return parseInt(result.rows[0]?.count ?? '0', 10);\n } catch {\n continue;\n }\n }\n\n return 0;\n }\n\n // ============================================\n // 迁移支持\n // ============================================\n\n public async getVectorIds(modelId: string, options: { limit?: number; afterId?: number } = {}): Promise<number[]> {\n await this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const { limit = 1000, afterId } = options;\n\n try {\n let result;\n if (afterId !== undefined) {\n result = await this.getDb().execute(sql`\n SELECT id FROM ${sql.identifier(tableName)}\n WHERE id > ${afterId}\n ORDER BY id\n LIMIT ${limit}\n `);\n } else {\n result = await this.getDb().execute(sql`\n SELECT id FROM ${sql.identifier(tableName)}\n ORDER BY id\n LIMIT ${limit}\n `);\n }\n return result.rows.map((r: any) => r.id);\n } catch {\n return [];\n }\n }\n\n // ============================================\n // 私有方法\n // ============================================\n\n private rowToVector(row: any): VectorRecord {\n return {\n id: row.id,\n embedding: this.parseEmbedding(row.embedding),\n createdAt: this.toTimestamp(row.created_at),\n };\n }\n\n private parseEmbedding(embedding: any): number[] {\n // pgvector 返回的可能是数组或字符串\n if (Array.isArray(embedding)) {\n return embedding;\n }\n if (typeof embedding === 'string') {\n // 解析 \"[1,2,3]\" 格式\n try {\n return JSON.parse(embedding);\n } catch {\n return [];\n }\n }\n return [];\n }\n}\n"]}

package/dist/storage/vector/PostgresVectorStore.jsonld CHANGED Viewed

@@ -67,6 +67,10 @@
           "@id": "undefineds:dist/storage/vector/PostgresVectorStore.jsonld#PostgresVectorStore__member_getTableName",
           "memberFieldName": "getTableName"
         },
+        {
+          "@id": "undefineds:dist/storage/vector/PostgresVectorStore.jsonld#PostgresVectorStore__member_getCountTableCandidates",
+          "memberFieldName": "getCountTableCandidates"
+        },
         {
           "@id": "undefineds:dist/storage/vector/PostgresVectorStore.jsonld#PostgresVectorStore__member_ensureVectorTable",
           "memberFieldName": "ensureVectorTable"

package/dist/storage/vector/SqliteVecExtension.d.ts ADDED Viewed

@@ -0,0 +1,3 @@
+import type { SqliteDatabase } from '../SqliteRuntime';
+export declare function getSqliteVecExtensionPath(): string;
+export declare function loadSqliteVecExtension(database: SqliteDatabase): void;

package/dist/storage/vector/SqliteVecExtension.js ADDED Viewed

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getSqliteVecExtensionPath = getSqliteVecExtensionPath;
+exports.loadSqliteVecExtension = loadSqliteVecExtension;
+function getSqliteVecModule() {
+    return require('sqlite-vec');
+}
+function getSqliteVecExtensionPath() {
+    return getSqliteVecModule().getLoadablePath();
+}
+function loadSqliteVecExtension(database) {
+    const sqliteVec = getSqliteVecModule();
+    const extensionPath = sqliteVec.getLoadablePath();
+    database.loadExtension(extensionPath);
+}
+//# sourceMappingURL=SqliteVecExtension.js.map

package/dist/storage/vector/SqliteVecExtension.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"SqliteVecExtension.js","sourceRoot":"","sources":["../../../src/storage/vector/SqliteVecExtension.ts"],"names":[],"mappings":";;AAWA,8DAEC;AAED,wDAIC;AAZD,SAAS,kBAAkB;IACzB,OAAO,OAAO,CAAC,YAAY,CAAoB,CAAC;AAClD,CAAC;AAED,SAAgB,yBAAyB;IACvC,OAAO,kBAAkB,EAAE,CAAC,eAAe,EAAE,CAAC;AAChD,CAAC;AAED,SAAgB,sBAAsB,CAAC,QAAwB;IAC7D,MAAM,SAAS,GAAG,kBAAkB,EAAE,CAAC;IACvC,MAAM,aAAa,GAAG,SAAS,CAAC,eAAe,EAAE,CAAC;IAClD,QAAQ,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;AACxC,CAAC","sourcesContent":["import type { SqliteDatabase } from '../SqliteRuntime';\n\ninterface SqliteVecModule {\n load?: (database: unknown, extensionPath: string) => void;\n getLoadablePath: () => string;\n}\n\nfunction getSqliteVecModule(): SqliteVecModule {\n return require('sqlite-vec') as SqliteVecModule;\n}\n\nexport function getSqliteVecExtensionPath(): string {\n return getSqliteVecModule().getLoadablePath();\n}\n\nexport function loadSqliteVecExtension(database: SqliteDatabase): void {\n const sqliteVec = getSqliteVecModule();\n const extensionPath = sqliteVec.getLoadablePath();\n database.loadExtension(extensionPath);\n}\n"]}

package/dist/storage/vector/SqliteVectorStore.d.ts CHANGED Viewed

@@ -13,6 +13,7 @@ export declare class SqliteVectorStore extends VectorStore implements Initializa
     /** @ignored */
     private db;
     private readonly filename;
+    private readonly sqliteRuntime;
     constructor(options: VectorStoreOptions);
     initialize(): Promise<void>;
     finalize(): Promise<void>;
@@ -20,6 +21,7 @@ export declare class SqliteVectorStore extends VectorStore implements Initializa
     close(): Promise<void>;
     private ensureOpen;
     private getTableName;
+    private getCountTableCandidates;
     private getDimension;
     ensureVectorTable(modelId: string): Promise<void>;
     dropVectorTable(modelId: string): Promise<void>;

package/dist/storage/vector/SqliteVectorStore.js CHANGED Viewed

@@ -12,15 +12,17 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SqliteVectorStore = void 0;
-const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const VectorStore_1 = require("./VectorStore");
+const SqliteRuntime_1 = require("../SqliteRuntime");
+const SqliteVecExtension_1 = require("./SqliteVecExtension");
 class SqliteVectorStore extends VectorStore_1.VectorStore {
     constructor(options) {
         super();
         /** @ignored */
         this.db = null;
+        this.sqliteRuntime = (0, SqliteRuntime_1.getSqliteRuntime)();
         let connStr = options.connectionString;
         if (connStr.startsWith('sqlite:')) {
             connStr = connStr.slice(7);
@@ -46,11 +48,8 @@ class SqliteVectorStore extends VectorStore_1.VectorStore {
                 fs_1.default.mkdirSync(dir, { recursive: true });
             }
         }
-        this.db = new better_sqlite3_1.default(this.filename);
-        // 加载 sqlite-vec 扩展
-        const { load } = require('sqlite-vec');
-        const vecPath = require('sqlite-vec').getLoadablePath();
-        load(this.db, vecPath);
+        this.db = this.sqliteRuntime.openDatabase(this.filename);
+        (0, SqliteVecExtension_1.loadSqliteVecExtension)(this.db);
         if (this.filename !== ':memory:') {
             this.db.pragma('journal_mode = WAL');
             this.db.pragma('busy_timeout = 5000');
@@ -72,11 +71,8 @@ class SqliteVectorStore extends VectorStore_1.VectorStore {
                     fs_1.default.mkdirSync(dir, { recursive: true });
                 }
             }
-            this.db = new better_sqlite3_1.default(this.filename);
-            // 加载 sqlite-vec 扩展
-            const { load } = require('sqlite-vec');
-            const vecPath = require('sqlite-vec').getLoadablePath();
-            load(this.db, vecPath);
+            this.db = this.sqliteRuntime.openDatabase(this.filename);
+            (0, SqliteVecExtension_1.loadSqliteVecExtension)(this.db);
             if (this.filename !== ':memory:') {
                 this.db.pragma('journal_mode = WAL');
                 this.db.pragma('busy_timeout = 5000');
@@ -87,6 +83,14 @@ class SqliteVectorStore extends VectorStore_1.VectorStore {
     getTableName(modelId) {
         return `vec_${(0, VectorStore_1.hashModelId)(modelId)}`;
     }
+    getCountTableCandidates(modelIdOrTableName) {
+        const candidates = [];
+        if (modelIdOrTableName.startsWith('vec_')) {
+            candidates.push(modelIdOrTableName);
+        }
+        candidates.push(this.getTableName(modelIdOrTableName));
+        return [...new Set(candidates)];
+    }
     getDimension() {
         return this.defaultDimension;
     }
@@ -126,7 +130,13 @@ class SqliteVectorStore extends VectorStore_1.VectorStore {
         const db = this.ensureOpen();
         try {
             const rows = db
-                .prepare(`SELECT name FROM sqlite_master WHERE type='table' AND name LIKE 'vec_%'`)
+                .prepare(`
+          SELECT name
+          FROM sqlite_master
+          WHERE type = 'table'
+            AND name LIKE 'vec_%'
+            AND sql LIKE '%USING vec0%'
+        `)
                 .all();
             return rows.map((r) => r.name);
         }
@@ -230,14 +240,16 @@ class SqliteVectorStore extends VectorStore_1.VectorStore {
     // ============================================
     async countVectors(modelId) {
         const db = this.ensureOpen();
-        const tableName = this.getTableName(modelId);
-        try {
-            const row = db.prepare(`SELECT COUNT(*) as count FROM ${tableName}`).get();
-            return row?.count ?? 0;
-        }
-        catch {
-            return 0;
+        for (const tableName of this.getCountTableCandidates(modelId)) {
+            try {
+                const row = db.prepare(`SELECT COUNT(*) as count FROM ${tableName}`).get();
+                return row?.count ?? 0;
+            }
+            catch {
+                continue;
+            }
         }
+        return 0;
     }
     // ============================================
     // 迁移支持

package/dist/storage/vector/SqliteVectorStore.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"SqliteVectorStore.js","sourceRoot":"","sources":["../../../src/storage/vector/SqliteVectorStore.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;;;;AAEH,oEAAsC;AACtC,4CAAoB;AACpB,gDAAwB;AAExB,+CAAyD;AAGzD,MAAa,iBAAkB,SAAQ,yBAAW;IAKhD,YAAmB,OAA2B;QAC5C,KAAK,EAAE,CAAC;QALV,eAAe;QACP,OAAE,GAA6B,IAAI,CAAC;QAK1C,IAAI,OAAO,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACvC,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAClC,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED,+CAA+C;IAC/C,0CAA0C;IAC1C,+CAA+C;IAExC,KAAK,CAAC,UAAU;QACrB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;IACpB,CAAC;IAEM,KAAK,CAAC,QAAQ;QACnB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEe,KAAK,CAAC,IAAI;QACxB,IAAI,IAAI,CAAC,EAAE;YAAE,OAAO;QAEpB,SAAS;QACT,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;YACjC,MAAM,GAAG,GAAG,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACxC,IAAI,GAAG,IAAI,GAAG,KAAK,GAAG,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC9C,YAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAED,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEtC,mBAAmB;QACnB,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;QACvC,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,eAAe,EAAE,CAAC;QACxD,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAEvB,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;YACjC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;YACrC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAEe,KAAK,CAAC,KAAK;QACzB,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,oBAAoB;YACpB,SAAS;YACT,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACjC,MAAM,GAAG,GAAG,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACxC,IAAI,GAAG,IAAI,GAAG,KAAK,GAAG,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC9C,YAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YAED,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAEtC,mBAAmB;YACnB,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;YACvC,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,eAAe,EAAE,CAAC;YACxD,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAEvB,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACjC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;gBACrC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAEO,YAAY,CAAC,OAAe;QAClC,OAAO,OAAO,IAAA,yBAAW,EAAC,OAAO,CAAC,EAAE,CAAC;IACvC,CAAC;IAEO,YAAY;QAClB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED,+CAA+C;IAC/C,QAAQ;IACR,+CAA+C;IAE/B,KAAK,CAAC,iBAAiB,CAAC,OAAe;QACrD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,2BAA2B;QAC3B,kCAAkC;QAClC,EAAE,CAAC,IAAI,CAAC;2CAC+B,SAAS;0BAC1B,IAAI,CAAC,YAAY,EAAE;;KAExC,CAAC,CAAC;IACL,CAAC;IAEe,KAAK,CAAC,eAAe,CAAC,OAAe;QACnD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,EAAE,CAAC,IAAI,CAAC,wBAAwB,SAAS,EAAE,CAAC,CAAC;IAC/C,CAAC;IAEe,KAAK,CAAC,cAAc,CAAC,OAAe;QAClD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,EAAE;iBACX,OAAO,CAAC,8DAA8D,CAAC;iBACvE,GAAG,CAAC,SAAS,CAAiC,CAAC;YAClD,OAAO,CAAC,CAAC,GAAG,CAAC;QACf,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAEe,KAAK,CAAC,gBAAgB;QACpC,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,EAAE;iBACZ,OAAO,CAAC,yEAAyE,CAAC;iBAClF,GAAG,EAAwB,CAAC;YAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAE/B,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,EAAU,EAAE,SAAmB;QACjF,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,kCAAkC;QAClC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QAErC,iCAAiC;QACjC,EAAE,CAAC,OAAO,CAAC,eAAe,SAAS,kBAAkB,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAElE,oBAAoB;QACpB,EAAE,CAAC,OAAO,CACR,eAAe,SAAS,mCAAmC,CAC5D,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1C,CAAC;IAEe,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,OAA8C;QACtG,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEjC,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,WAAW,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,IAAoB,EAAE,EAAE;YAC1D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,kCAAkC;gBAClC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;gBAEzC,iCAAiC;gBACjC,EAAE,CAAC,OAAO,CAAC,eAAe,SAAS,kBAAkB,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAClE,EAAE,CAAC,OAAO,CACR,eAAe,SAAS,mCAAmC,CAC5D,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,WAAW,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAEe,KAAK,CAAC,SAAS,CAAC,OAAe,EAAE,EAAU;QACzD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,gCAAgC,SAAS,kBAAkB,CAAC,CAAC,GAAG,CAAC,EAAE,CAAQ,CAAC;YACnG,OAAO,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC5C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEe,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,EAAU;QAC5D,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,EAAE,CAAC,OAAO,CAAC,eAAe,SAAS,kBAAkB,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAEe,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,GAAa;QACrE,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE7B,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,YAAY,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,EAAE,CAAC,OAAO,CAAC,eAAe,SAAS,oBAAoB,YAAY,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;IACtF,CAAC;IAEe,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,cAAwB,EAAE,UAA+B,EAAE;QACvG,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAE7B,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;QACtD,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,yBAAyB;YACzB,qCAAqC;YACrC,MAAM,GAAG,GAAG;;;;eAIH,SAAS;;;;OAIjB,CAAC;YAEF,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,EAAE,KAAK,GAAG,CAAC,CAAuC,CAAC;YAElH,MAAM,OAAO,GAAyB,EAAE,CAAC;YAEzC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,IAAI,UAAU,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBAAE,SAAS;gBAEtC,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;gBAC9B,wCAAwC;gBACxC,oCAAoC;gBACpC,kCAAkC;gBAClC,uBAAuB;gBACvB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAEtE,IAAI,SAAS,KAAK,SAAS,IAAI,KAAK,GAAG,SAAS;oBAAE,SAAS;gBAE3D,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAChD,CAAC;YAED,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,wBAAwB;YACxB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,KAAK;IACL,+CAA+C;IAE/B,KAAK,CAAC,YAAY,CAAC,OAAe;QAChD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,iCAAiC,SAAS,EAAE,CAAC,CAAC,GAAG,EAAS,CAAC;YAClF,OAAO,GAAG,EAAE,KAAK,IAAI,CAAC,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAE/B,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,UAAgD,EAAE;QACpG,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,EAAE,KAAK,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QAE1C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,OAAO,KAAK,SAAS;gBAC/B,CAAC,CAAC,qBAAqB,SAAS,yCAAyC;gBACzE,CAAC,CAAC,qBAAqB,SAAS,yBAAyB,CAAC;YAC5D,MAAM,MAAM,GAAG,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YAClE,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;YACrD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,MAAM;IACN,+CAA+C;IAEvC,WAAW,CAAC,GAAQ;QAC1B,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,KAAK;YACb,SAAS,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC;YACnD,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;SACzC,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAAC,IAAyB;QACpD,iCAAiC;QACjC,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChE,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1C,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AA5TD,8CA4TC","sourcesContent":["/*\n SqliteVectorStore - SQLite 向量存储实现（使用 sqlite-vec 扩展）\n \n 设计思路：\n * - sqlite-vec 的 vec0 表使用 rowid 作为主键\n * - rowid 直接与 quints 表的 rowid 对应，无需额外映射\n * - 通过 rowid JOIN quints 表即可关联业务数据\n /\n\nimport Database from 'better-sqlite3';\nimport fs from 'fs';\nimport path from 'path';\nimport type { Finalizable, Initializable } from '@solid/community-server';\nimport { VectorStore, hashModelId } from './VectorStore';\nimport type { VectorRecord, VectorSearchOptions, VectorSearchResult, VectorStoreOptions } from './types';\n\nexport class SqliteVectorStore extends VectorStore implements Initializable, Finalizable {\n /* @ignored /\n private db: Database.Database \| null = null;\n private readonly filename: string;\n\n public constructor(options: VectorStoreOptions) {\n super();\n let connStr = options.connectionString;\n if (connStr.startsWith('sqlite:')) {\n connStr = connStr.slice(7);\n }\n this.filename = connStr;\n }\n\n // ============================================\n // Lifecycle (Initializable / Finalizable)\n // ============================================\n\n public async initialize(): Promise<void> {\n await this.open();\n }\n\n public async finalize(): Promise<void> {\n await this.close();\n }\n\n public override async open(): Promise<void> {\n if (this.db) return;\n\n // 确保目录存在\n if (this.filename !== ':memory:') {\n const dir = path.dirname(this.filename);\n if (dir && dir !== '.' && !fs.existsSync(dir)) {\n fs.mkdirSync(dir, { recursive: true });\n }\n }\n\n this.db = new Database(this.filename);\n\n // 加载 sqlite-vec 扩展\n const { load } = require('sqlite-vec');\n const vecPath = require('sqlite-vec').getLoadablePath();\n load(this.db, vecPath);\n\n if (this.filename !== ':memory:') {\n this.db.pragma('journal_mode = WAL');\n this.db.pragma('busy_timeout = 5000');\n }\n }\n\n public override async close(): Promise<void> {\n if (this.db) {\n this.db.close();\n this.db = null;\n }\n }\n\n private ensureOpen(): Database.Database {\n if (!this.db) {\n // 懒初始化：如果尚未打开，则自动打开\n // 确保目录存在\n if (this.filename !== ':memory:') {\n const dir = path.dirname(this.filename);\n if (dir && dir !== '.' && !fs.existsSync(dir)) {\n fs.mkdirSync(dir, { recursive: true });\n }\n }\n\n this.db = new Database(this.filename);\n\n // 加载 sqlite-vec 扩展\n const { load } = require('sqlite-vec');\n const vecPath = require('sqlite-vec').getLoadablePath();\n load(this.db, vecPath);\n\n if (this.filename !== ':memory:') {\n this.db.pragma('journal_mode = WAL');\n this.db.pragma('busy_timeout = 5000');\n }\n }\n return this.db;\n }\n\n private getTableName(modelId: string): string {\n return `vec_${hashModelId(modelId)}`;\n }\n\n private getDimension(): number {\n return this.defaultDimension;\n }\n\n // ============================================\n // 向量表管理\n // ============================================\n\n public override async ensureVectorTable(modelId: string): Promise<void> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n // 使用 sqlite-vec 的 vec0 虚拟表\n // rowid 作为主键，与 quints 表的 rowid 对应\n db.exec(`\n CREATE VIRTUAL TABLE IF NOT EXISTS ${tableName} USING vec0(\n embedding float[${this.getDimension()}]\n )\n `);\n }\n\n public override async dropVectorTable(modelId: string): Promise<void> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n db.exec(`DROP TABLE IF EXISTS ${tableName}`);\n }\n\n public override async hasVectorTable(modelId: string): Promise<boolean> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const row = db\n .prepare(`SELECT name FROM sqlite_master WHERE type='table' AND name=?`)\n .get(tableName) as { name: string } \| undefined;\n return !!row;\n } catch {\n return false;\n }\n }\n\n public override async listVectorTables(): Promise<string[]> {\n const db = this.ensureOpen();\n\n try {\n const rows = db\n .prepare(`SELECT name FROM sqlite_master WHERE type='table' AND name LIKE 'vec_%'`)\n .all() as { name: string }[];\n return rows.map((r) => r.name);\n } catch {\n return [];\n }\n }\n\n // ============================================\n // 向量操作\n // ============================================\n\n public override async upsertVector(modelId: string, id: number, embedding: number[]): Promise<void> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n // 转换为 BigInt 以确保 sqlite-vec 识别为整数\n const bigId = BigInt(Math.floor(id));\n\n // sqlite-vec 不支持 UPDATE，需要先删除再插入\n db.prepare(`DELETE FROM ${tableName} WHERE rowid = ?`).run(bigId);\n\n // 使用 rowid 作为主键插入向量\n db.prepare(\n `INSERT INTO ${tableName} (rowid, embedding) VALUES (?, ?)`\n ).run(bigId, JSON.stringify(embedding));\n }\n\n public override async batchUpsertVectors(modelId: string, records: { id: number; embedding: number[] }[]): Promise<void> {\n if (records.length === 0) return;\n\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n const transaction = db.transaction((recs: typeof records) => {\n for (const rec of recs) {\n // 转换为 BigInt 以确保 sqlite-vec 识别为整数\n const bigId = BigInt(Math.floor(rec.id));\n\n // sqlite-vec 不支持 UPDATE，需要先删除再插入\n db.prepare(`DELETE FROM ${tableName} WHERE rowid = ?`).run(bigId);\n db.prepare(\n `INSERT INTO ${tableName} (rowid, embedding) VALUES (?, ?)`\n ).run(bigId, JSON.stringify(rec.embedding));\n }\n });\n\n transaction(records);\n }\n\n public override async getVector(modelId: string, id: number): Promise<VectorRecord \| null> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const row = db.prepare(`SELECT rowid, embedding FROM ${tableName} WHERE rowid = ?`).get(id) as any;\n return row ? this.rowToVector(row) : null;\n } catch {\n return null;\n }\n }\n\n public override async deleteVector(modelId: string, id: number): Promise<void> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n db.prepare(`DELETE FROM ${tableName} WHERE rowid = ?`).run(id);\n }\n\n public override async batchDeleteVectors(modelId: string, ids: number[]): Promise<void> {\n if (ids.length === 0) return;\n\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n const placeholders = ids.map(() => '?').join(',');\n db.prepare(`DELETE FROM ${tableName} WHERE rowid IN (${placeholders})`).run(...ids);\n }\n\n public override async search(modelId: string, queryEmbedding: number[], options: VectorSearchOptions = {}): Promise<VectorSearchResult[]> {\n const db = this.ensureOpen();\n\n const { limit = 10, threshold, excludeIds } = options;\n const tableName = this.getTableName(modelId);\n\n try {\n // 使用 sqlite-vec 的近似最近邻搜索\n // 注意：sqlite-vec 要求在 MATCH 中使用 k=? 参数\n const sql = `\n SELECT\n rowid as id,\n distance\n FROM ${tableName}\n WHERE embedding MATCH ?\n AND k = ?\n ORDER BY distance\n `;\n\n const rows = db.prepare(sql).all(JSON.stringify(queryEmbedding), limit 2) as { id: number; distance: number }[];\n\n const results: VectorSearchResult[] = [];\n\n for (const row of rows) {\n if (excludeIds?.has(row.id)) continue;\n\n const distance = row.distance;\n // sqlite-vec 使用 L2 距离，对于归一化向量，转换为余弦相似度：\n // L2² = 2 * (1 - cosine_similarity)\n // cosine_similarity = 1 - L2² / 2\n // score 限制在 [0, 1] 范围内\n const score = Math.max(0, Math.min(1, 1 - (distance * distance) / 2));\n\n if (threshold !== undefined && score < threshold) continue;\n\n results.push({ id: row.id, distance, score });\n }\n\n return results.slice(0, limit);\n } catch (e) {\n // sqlite-vec 扩展不可用时抛出错误\n throw new Error(`sqlite-vec search failed: ${e}`);\n }\n }\n\n // ============================================\n // 统计\n // ============================================\n\n public override async countVectors(modelId: string): Promise<number> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const row = db.prepare(`SELECT COUNT(*) as count FROM ${tableName}`).get() as any;\n return row?.count ?? 0;\n } catch {\n return 0;\n }\n }\n\n // ============================================\n // 迁移支持\n // ============================================\n\n public override async getVectorIds(modelId: string, options: { limit?: number; afterId?: number } = {}): Promise<number[]> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const { limit = 1000, afterId } = options;\n\n try {\n const sql = afterId !== undefined\n ? `SELECT rowid FROM ${tableName} WHERE rowid > ? ORDER BY rowid LIMIT ?`\n : `SELECT rowid FROM ${tableName} ORDER BY rowid LIMIT ?`;\n const params = afterId !== undefined ? [afterId, limit] : [limit];\n const rows = db.prepare(sql).all(...params) as any[];\n return rows.map((r) => r.rowid);\n } catch {\n return [];\n }\n }\n\n // ============================================\n // 行转换\n // ============================================\n\n private rowToVector(row: any): VectorRecord {\n return {\n id: row.rowid,\n embedding: this.deserializeEmbedding(row.embedding),\n createdAt: Math.floor(Date.now() / 1000),\n };\n }\n\n private deserializeEmbedding(blob: Buffer \| Uint8Array): number[] {\n // sqlite-vec 返回的是 float32 buffer\n const buffer = Buffer.isBuffer(blob) ? blob : Buffer.from(blob);\n const embedding: number[] = [];\n for (let i = 0; i < buffer.length; i += 4) {\n embedding.push(buffer.readFloatLE(i));\n }\n return embedding;\n }\n}\n"]}
1	+ {"version":3,"file":"SqliteVectorStore.js","sourceRoot":"","sources":["../../../src/storage/vector/SqliteVectorStore.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;;;;AAEH,4CAAoB;AACpB,gDAAwB;AAExB,+CAAyD;AAEzD,oDAAyE;AACzE,6DAA8D;AAE9D,MAAa,iBAAkB,SAAQ,yBAAW;IAMhD,YAAmB,OAA2B;QAC5C,KAAK,EAAE,CAAC;QANV,eAAe;QACP,OAAE,GAA0B,IAAI,CAAC;QAExB,kBAAa,GAAG,IAAA,gCAAgB,GAAE,CAAC;QAIlD,IAAI,OAAO,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACvC,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAClC,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED,+CAA+C;IAC/C,0CAA0C;IAC1C,+CAA+C;IAExC,KAAK,CAAC,UAAU;QACrB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;IACpB,CAAC;IAEM,KAAK,CAAC,QAAQ;QACnB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEe,KAAK,CAAC,IAAI;QACxB,IAAI,IAAI,CAAC,EAAE;YAAE,OAAO;QAEpB,SAAS;QACT,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;YACjC,MAAM,GAAG,GAAG,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACxC,IAAI,GAAG,IAAI,GAAG,KAAK,GAAG,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC9C,YAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAED,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAA,2CAAsB,EAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhC,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;YACjC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;YACrC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAEe,KAAK,CAAC,KAAK;QACzB,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YACb,oBAAoB;YACpB,SAAS;YACT,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACjC,MAAM,GAAG,GAAG,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACxC,IAAI,GAAG,IAAI,GAAG,KAAK,GAAG,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC9C,YAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YAED,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzD,IAAA,2CAAsB,EAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAEhC,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACjC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;gBACrC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAEO,YAAY,CAAC,OAAe;QAClC,OAAO,OAAO,IAAA,yBAAW,EAAC,OAAO,CAAC,EAAE,CAAC;IACvC,CAAC;IAEO,uBAAuB,CAAC,kBAA0B;QACxD,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,IAAI,kBAAkB,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1C,UAAU,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;QACD,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;IAClC,CAAC;IAEO,YAAY;QAClB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED,+CAA+C;IAC/C,QAAQ;IACR,+CAA+C;IAE/B,KAAK,CAAC,iBAAiB,CAAC,OAAe;QACrD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,2BAA2B;QAC3B,kCAAkC;QAClC,EAAE,CAAC,IAAI,CAAC;2CAC+B,SAAS;0BAC1B,IAAI,CAAC,YAAY,EAAE;;KAExC,CAAC,CAAC;IACL,CAAC;IAEe,KAAK,CAAC,eAAe,CAAC,OAAe;QACnD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,EAAE,CAAC,IAAI,CAAC,wBAAwB,SAAS,EAAE,CAAC,CAAC;IAC/C,CAAC;IAEe,KAAK,CAAC,cAAc,CAAC,OAAe;QAClD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,EAAE;iBACX,OAAO,CAAC,8DAA8D,CAAC;iBACvE,GAAG,CAAC,SAAS,CAAiC,CAAC;YAClD,OAAO,CAAC,CAAC,GAAG,CAAC;QACf,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAEe,KAAK,CAAC,gBAAgB;QACpC,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,EAAE;iBACZ,OAAO,CAAC;;;;;;SAMR,CAAC;iBACD,GAAG,EAAwB,CAAC;YAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAE/B,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,EAAU,EAAE,SAAmB;QACjF,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,kCAAkC;QAClC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QAErC,iCAAiC;QACjC,EAAE,CAAC,OAAO,CAAC,eAAe,SAAS,kBAAkB,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAElE,oBAAoB;QACpB,EAAE,CAAC,OAAO,CACR,eAAe,SAAS,mCAAmC,CAC5D,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1C,CAAC;IAEe,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,OAA8C;QACtG,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEjC,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,WAAW,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,IAAoB,EAAE,EAAE;YAC1D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,kCAAkC;gBAClC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;gBAEzC,iCAAiC;gBACjC,EAAE,CAAC,OAAO,CAAC,eAAe,SAAS,kBAAkB,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAClE,EAAE,CAAC,OAAO,CACR,eAAe,SAAS,mCAAmC,CAC5D,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,WAAW,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAEe,KAAK,CAAC,SAAS,CAAC,OAAe,EAAE,EAAU;QACzD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,gCAAgC,SAAS,kBAAkB,CAAC,CAAC,GAAG,CAAC,EAAE,CAAQ,CAAC;YACnG,OAAO,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC5C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEe,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,EAAU;QAC5D,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,EAAE,CAAC,OAAO,CAAC,eAAe,SAAS,kBAAkB,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAEe,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,GAAa;QACrE,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE7B,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,YAAY,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,EAAE,CAAC,OAAO,CAAC,eAAe,SAAS,oBAAoB,YAAY,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;IACtF,CAAC;IAEe,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,cAAwB,EAAE,UAA+B,EAAE;QACvG,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAE7B,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;QACtD,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,yBAAyB;YACzB,qCAAqC;YACrC,MAAM,GAAG,GAAG;;;;eAIH,SAAS;;;;OAIjB,CAAC;YAEF,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,EAAE,KAAK,GAAG,CAAC,CAAuC,CAAC;YAElH,MAAM,OAAO,GAAyB,EAAE,CAAC;YAEzC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,IAAI,UAAU,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBAAE,SAAS;gBAEtC,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;gBAC9B,wCAAwC;gBACxC,oCAAoC;gBACpC,kCAAkC;gBAClC,uBAAuB;gBACvB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAEtE,IAAI,SAAS,KAAK,SAAS,IAAI,KAAK,GAAG,SAAS;oBAAE,SAAS;gBAE3D,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAChD,CAAC;YAED,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,wBAAwB;YACxB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,KAAK;IACL,+CAA+C;IAE/B,KAAK,CAAC,YAAY,CAAC,OAAe;QAChD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAE7B,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9D,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,iCAAiC,SAAS,EAAE,CAAC,CAAC,GAAG,EAAS,CAAC;gBAClF,OAAO,GAAG,EAAE,KAAK,IAAI,CAAC,CAAC;YACzB,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QAED,OAAO,CAAC,CAAC;IACX,CAAC;IAED,+CAA+C;IAC/C,OAAO;IACP,+CAA+C;IAE/B,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,UAAgD,EAAE;QACpG,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,EAAE,KAAK,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QAE1C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,OAAO,KAAK,SAAS;gBAC/B,CAAC,CAAC,qBAAqB,SAAS,yCAAyC;gBACzE,CAAC,CAAC,qBAAqB,SAAS,yBAAyB,CAAC;YAC5D,MAAM,MAAM,GAAG,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YAClE,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;YACrD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,MAAM;IACN,+CAA+C;IAEvC,WAAW,CAAC,GAAQ;QAC1B,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,KAAK;YACb,SAAS,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC;YACnD,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;SACzC,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAAC,IAAyB;QACpD,iCAAiC;QACjC,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChE,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1C,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAvUD,8CAuUC","sourcesContent":["/*\n SqliteVectorStore - SQLite 向量存储实现（使用 sqlite-vec 扩展）\n \n 设计思路：\n * - sqlite-vec 的 vec0 表使用 rowid 作为主键\n * - rowid 直接与 quints 表的 rowid 对应，无需额外映射\n * - 通过 rowid JOIN quints 表即可关联业务数据\n /\n\nimport fs from 'fs';\nimport path from 'path';\nimport type { Finalizable, Initializable } from '@solid/community-server';\nimport { VectorStore, hashModelId } from './VectorStore';\nimport type { VectorRecord, VectorSearchOptions, VectorSearchResult, VectorStoreOptions } from './types';\nimport { getSqliteRuntime, type SqliteDatabase } from '../SqliteRuntime';\nimport { loadSqliteVecExtension } from './SqliteVecExtension';\n\nexport class SqliteVectorStore extends VectorStore implements Initializable, Finalizable {\n /* @ignored /\n private db: SqliteDatabase \| null = null;\n private readonly filename: string;\n private readonly sqliteRuntime = getSqliteRuntime();\n\n public constructor(options: VectorStoreOptions) {\n super();\n let connStr = options.connectionString;\n if (connStr.startsWith('sqlite:')) {\n connStr = connStr.slice(7);\n }\n this.filename = connStr;\n }\n\n // ============================================\n // Lifecycle (Initializable / Finalizable)\n // ============================================\n\n public async initialize(): Promise<void> {\n await this.open();\n }\n\n public async finalize(): Promise<void> {\n await this.close();\n }\n\n public override async open(): Promise<void> {\n if (this.db) return;\n\n // 确保目录存在\n if (this.filename !== ':memory:') {\n const dir = path.dirname(this.filename);\n if (dir && dir !== '.' && !fs.existsSync(dir)) {\n fs.mkdirSync(dir, { recursive: true });\n }\n }\n\n this.db = this.sqliteRuntime.openDatabase(this.filename);\n loadSqliteVecExtension(this.db);\n\n if (this.filename !== ':memory:') {\n this.db.pragma('journal_mode = WAL');\n this.db.pragma('busy_timeout = 5000');\n }\n }\n\n public override async close(): Promise<void> {\n if (this.db) {\n this.db.close();\n this.db = null;\n }\n }\n\n private ensureOpen(): SqliteDatabase {\n if (!this.db) {\n // 懒初始化：如果尚未打开，则自动打开\n // 确保目录存在\n if (this.filename !== ':memory:') {\n const dir = path.dirname(this.filename);\n if (dir && dir !== '.' && !fs.existsSync(dir)) {\n fs.mkdirSync(dir, { recursive: true });\n }\n }\n\n this.db = this.sqliteRuntime.openDatabase(this.filename);\n loadSqliteVecExtension(this.db);\n\n if (this.filename !== ':memory:') {\n this.db.pragma('journal_mode = WAL');\n this.db.pragma('busy_timeout = 5000');\n }\n }\n return this.db;\n }\n\n private getTableName(modelId: string): string {\n return `vec_${hashModelId(modelId)}`;\n }\n\n private getCountTableCandidates(modelIdOrTableName: string): string[] {\n const candidates: string[] = [];\n if (modelIdOrTableName.startsWith('vec_')) {\n candidates.push(modelIdOrTableName);\n }\n candidates.push(this.getTableName(modelIdOrTableName));\n return [...new Set(candidates)];\n }\n\n private getDimension(): number {\n return this.defaultDimension;\n }\n\n // ============================================\n // 向量表管理\n // ============================================\n\n public override async ensureVectorTable(modelId: string): Promise<void> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n // 使用 sqlite-vec 的 vec0 虚拟表\n // rowid 作为主键，与 quints 表的 rowid 对应\n db.exec(`\n CREATE VIRTUAL TABLE IF NOT EXISTS ${tableName} USING vec0(\n embedding float[${this.getDimension()}]\n )\n `);\n }\n\n public override async dropVectorTable(modelId: string): Promise<void> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n db.exec(`DROP TABLE IF EXISTS ${tableName}`);\n }\n\n public override async hasVectorTable(modelId: string): Promise<boolean> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const row = db\n .prepare(`SELECT name FROM sqlite_master WHERE type='table' AND name=?`)\n .get(tableName) as { name: string } \| undefined;\n return !!row;\n } catch {\n return false;\n }\n }\n\n public override async listVectorTables(): Promise<string[]> {\n const db = this.ensureOpen();\n\n try {\n const rows = db\n .prepare(`\n SELECT name\n FROM sqlite_master\n WHERE type = 'table'\n AND name LIKE 'vec_%'\n AND sql LIKE '%USING vec0%'\n `)\n .all() as { name: string }[];\n return rows.map((r) => r.name);\n } catch {\n return [];\n }\n }\n\n // ============================================\n // 向量操作\n // ============================================\n\n public override async upsertVector(modelId: string, id: number, embedding: number[]): Promise<void> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n // 转换为 BigInt 以确保 sqlite-vec 识别为整数\n const bigId = BigInt(Math.floor(id));\n\n // sqlite-vec 不支持 UPDATE，需要先删除再插入\n db.prepare(`DELETE FROM ${tableName} WHERE rowid = ?`).run(bigId);\n\n // 使用 rowid 作为主键插入向量\n db.prepare(\n `INSERT INTO ${tableName} (rowid, embedding) VALUES (?, ?)`\n ).run(bigId, JSON.stringify(embedding));\n }\n\n public override async batchUpsertVectors(modelId: string, records: { id: number; embedding: number[] }[]): Promise<void> {\n if (records.length === 0) return;\n\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n const transaction = db.transaction((recs: typeof records) => {\n for (const rec of recs) {\n // 转换为 BigInt 以确保 sqlite-vec 识别为整数\n const bigId = BigInt(Math.floor(rec.id));\n\n // sqlite-vec 不支持 UPDATE，需要先删除再插入\n db.prepare(`DELETE FROM ${tableName} WHERE rowid = ?`).run(bigId);\n db.prepare(\n `INSERT INTO ${tableName} (rowid, embedding) VALUES (?, ?)`\n ).run(bigId, JSON.stringify(rec.embedding));\n }\n });\n\n transaction(records);\n }\n\n public override async getVector(modelId: string, id: number): Promise<VectorRecord \| null> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n try {\n const row = db.prepare(`SELECT rowid, embedding FROM ${tableName} WHERE rowid = ?`).get(id) as any;\n return row ? this.rowToVector(row) : null;\n } catch {\n return null;\n }\n }\n\n public override async deleteVector(modelId: string, id: number): Promise<void> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n db.prepare(`DELETE FROM ${tableName} WHERE rowid = ?`).run(id);\n }\n\n public override async batchDeleteVectors(modelId: string, ids: number[]): Promise<void> {\n if (ids.length === 0) return;\n\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n\n const placeholders = ids.map(() => '?').join(',');\n db.prepare(`DELETE FROM ${tableName} WHERE rowid IN (${placeholders})`).run(...ids);\n }\n\n public override async search(modelId: string, queryEmbedding: number[], options: VectorSearchOptions = {}): Promise<VectorSearchResult[]> {\n const db = this.ensureOpen();\n\n const { limit = 10, threshold, excludeIds } = options;\n const tableName = this.getTableName(modelId);\n\n try {\n // 使用 sqlite-vec 的近似最近邻搜索\n // 注意：sqlite-vec 要求在 MATCH 中使用 k=? 参数\n const sql = `\n SELECT\n rowid as id,\n distance\n FROM ${tableName}\n WHERE embedding MATCH ?\n AND k = ?\n ORDER BY distance\n `;\n\n const rows = db.prepare(sql).all(JSON.stringify(queryEmbedding), limit 2) as { id: number; distance: number }[];\n\n const results: VectorSearchResult[] = [];\n\n for (const row of rows) {\n if (excludeIds?.has(row.id)) continue;\n\n const distance = row.distance;\n // sqlite-vec 使用 L2 距离，对于归一化向量，转换为余弦相似度：\n // L2² = 2 * (1 - cosine_similarity)\n // cosine_similarity = 1 - L2² / 2\n // score 限制在 [0, 1] 范围内\n const score = Math.max(0, Math.min(1, 1 - (distance * distance) / 2));\n\n if (threshold !== undefined && score < threshold) continue;\n\n results.push({ id: row.id, distance, score });\n }\n\n return results.slice(0, limit);\n } catch (e) {\n // sqlite-vec 扩展不可用时抛出错误\n throw new Error(`sqlite-vec search failed: ${e}`);\n }\n }\n\n // ============================================\n // 统计\n // ============================================\n\n public override async countVectors(modelId: string): Promise<number> {\n const db = this.ensureOpen();\n\n for (const tableName of this.getCountTableCandidates(modelId)) {\n try {\n const row = db.prepare(`SELECT COUNT(*) as count FROM ${tableName}`).get() as any;\n return row?.count ?? 0;\n } catch {\n continue;\n }\n }\n\n return 0;\n }\n\n // ============================================\n // 迁移支持\n // ============================================\n\n public override async getVectorIds(modelId: string, options: { limit?: number; afterId?: number } = {}): Promise<number[]> {\n const db = this.ensureOpen();\n const tableName = this.getTableName(modelId);\n const { limit = 1000, afterId } = options;\n\n try {\n const sql = afterId !== undefined\n ? `SELECT rowid FROM ${tableName} WHERE rowid > ? ORDER BY rowid LIMIT ?`\n : `SELECT rowid FROM ${tableName} ORDER BY rowid LIMIT ?`;\n const params = afterId !== undefined ? [afterId, limit] : [limit];\n const rows = db.prepare(sql).all(...params) as any[];\n return rows.map((r) => r.rowid);\n } catch {\n return [];\n }\n }\n\n // ============================================\n // 行转换\n // ============================================\n\n private rowToVector(row: any): VectorRecord {\n return {\n id: row.rowid,\n embedding: this.deserializeEmbedding(row.embedding),\n createdAt: Math.floor(Date.now() / 1000),\n };\n }\n\n private deserializeEmbedding(blob: Buffer \| Uint8Array): number[] {\n // sqlite-vec 返回的是 float32 buffer\n const buffer = Buffer.isBuffer(blob) ? blob : Buffer.from(blob);\n const embedding: number[] = [];\n for (let i = 0; i < buffer.length; i += 4) {\n embedding.push(buffer.readFloatLE(i));\n }\n return embedding;\n }\n}\n"]}

package/dist/storage/vector/SqliteVectorStore.jsonld CHANGED Viewed

@@ -30,6 +30,10 @@
           "@id": "undefineds:dist/storage/vector/SqliteVectorStore.jsonld#SqliteVectorStore__member_filename",
           "memberFieldName": "filename"
         },
+        {
+          "@id": "undefineds:dist/storage/vector/SqliteVectorStore.jsonld#SqliteVectorStore__member_sqliteRuntime",
+          "memberFieldName": "sqliteRuntime"
+        },
         {
           "@id": "undefineds:dist/storage/vector/SqliteVectorStore.jsonld#SqliteVectorStore__member_constructor",
           "memberFieldName": "constructor"
@@ -58,6 +62,10 @@
           "@id": "undefineds:dist/storage/vector/SqliteVectorStore.jsonld#SqliteVectorStore__member_getTableName",
           "memberFieldName": "getTableName"
         },
+        {
+          "@id": "undefineds:dist/storage/vector/SqliteVectorStore.jsonld#SqliteVectorStore__member_getCountTableCandidates",
+          "memberFieldName": "getCountTableCandidates"
+        },
         {
           "@id": "undefineds:dist/storage/vector/SqliteVectorStore.jsonld#SqliteVectorStore__member_getDimension",
           "memberFieldName": "getDimension"

package/dist/task/DrizzleTaskQueue.d.ts CHANGED Viewed

@@ -5,8 +5,6 @@
  *
  * 任务存储在 Pod 的 /tasks/{YYYY-MM-DD}.ttl 中，按天分片
  */
-import type { SolidDatabase } from '@undefineds.co/drizzle-solid';
-import { Task as taskTable } from './schema';
 import type { Task, TaskStatus, CreateTaskInput, TaskQueue, TaskQueueStats, ListTasksOptions } from './types';
 export interface DrizzleTaskQueueOptions {
     /**
@@ -16,9 +14,7 @@ export interface DrizzleTaskQueueOptions {
     /**
      * drizzle-solid 数据库实例
      */
-    db: SolidDatabase<{
-        task: typeof taskTable;
-    }>;
+    db: any;
 }
 /**
  * 基于 drizzle-solid 的任务队列实现

package/dist/task/DrizzleTaskQueue.js CHANGED Viewed

@@ -48,11 +48,11 @@ class DrizzleTaskQueue {
      * 获取任务
      */
     async getTask(taskId) {
-        const tasks = await this.db.select().from(schema_1.Task).where((0, drizzle_solid_1.eq)(schema_1.Task.id, taskId));
-        if (tasks.length === 0) {
+        const task = await this.db.findByLocator(schema_1.Task, { id: taskId });
+        if (!task) {
             return null;
         }
-        return this.dbTaskToTask(tasks[0]);
+        return this.dbTaskToTask(task);
     }
     /**
      * 获取任务列表
@@ -109,14 +109,14 @@ class DrizzleTaskQueue {
             if (updates.error !== undefined)
                 updateData.error = updates.error;
         }
-        await this.db.update(schema_1.Task).set(updateData).where((0, drizzle_solid_1.eq)(schema_1.Task.id, taskId));
+        await this.db.updateByLocator(schema_1.Task, { id: taskId }, updateData);
         this.logger.info(`Task ${taskId} status changed to ${status}`);
     }
     /**
      * 删除任务
      */
     async deleteTask(taskId) {
-        await this.db.delete(schema_1.Task).where((0, drizzle_solid_1.eq)(schema_1.Task.id, taskId));
+        await this.db.deleteByLocator(schema_1.Task, { id: taskId });
         this.logger.info(`Deleted task ${taskId}`);
     }
     /**

package/dist/task/DrizzleTaskQueue.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"DrizzleTaskQueue.js","sourceRoot":"","sources":["../../src/task/DrizzleTaskQueue.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAEH,iEAAqD;AACrD,mCAAqC;AACrC,gEAAkD;AAGlD,qCAA4E;AAqB5E;;GAEG;AACH,MAAa,gBAAgB;IAK3B,YAAmB,OAAgC;QAJhC,WAAM,GAAG,IAAA,oCAAY,EAAC,IAAI,CAAC,CAAC;QAK7C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,UAAU,GAAG,CAAC;QACnG,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,UAAU,CAAC,KAAsB;QAC5C,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,MAAM,QAAQ,GAAG;YACf,EAAE;YACF,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,mBAAe,CAAC,OAAO;YAC/B,SAAS,EAAE,GAAG;SACf,CAAC;QAEF,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAS,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAEjD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,eAAe,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;QAElE,OAAO;YACL,EAAE;YACF,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,GAAG;SACf,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,OAAO,CAAC,MAAc;QACjC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAS,CAAC,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAErF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,SAAS,CAAC,OAA0B;QAC/C,IAAI,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAS,CAAC,CAAC;QAE7C,OAAO;QACP,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;YACpB,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC;QAE1B,cAAc;QACd,IAAI,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAA0B,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE7E,WAAW;QACX,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;YACnB,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAO,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,CAAC,CAAC;QACjE,CAAC;QAED,KAAK;QACL,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAO,EAAE,CAAO,EAAE,EAAE;gBAC/B,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,OAAQ,CAAC,CAAC;gBACjC,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,OAAQ,CAAC,CAAC;gBACjC,IAAI,IAAI,YAAY,IAAI,IAAI,IAAI,YAAY,IAAI,EAAE,CAAC;oBACjD,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,KAAK,CAAC;gBACnD,CAAC;gBACD,OAAO,CAAC,CAAC;YACX,CAAC,CAAC,CAAC;QACL,CAAC;QAED,KAAK;QACL,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;QACpC,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC;QAC9C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,KAAK,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB,CAC3B,MAAc,EACd,MAAkB,EAClB,OAAuB;QAEvB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,MAAM,UAAU,GAA4B;YAC1C,MAAM;SACP,CAAC;QAEF,aAAa;QACb,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,UAAU,CAAC,SAAS,GAAG,GAAG,CAAC;QAC7B,CAAC;aAAM,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACzD,UAAU,CAAC,WAAW,GAAG,GAAG,CAAC;QAC/B,CAAC;QAED,SAAS;QACT,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS;gBAAE,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YACrE,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS;gBAAE,UAAU,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QACpE,CAAC;QAED,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAS,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAEhF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,MAAM,sBAAsB,MAAM,EAAE,CAAC,CAAC;IACjE,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,UAAU,CAAC,MAAc;QACpC,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAS,CAAC,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,QAAQ;QACnB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAS,CAAC,CAAC;QAErD,MAAM,KAAK,GAAmB;YAC5B,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,CAAC;YACT,KAAK,EAAE,KAAK,CAAC,MAAM;YACnB,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,IAA+B,CAAC;YACnD,MAAM,MAAM,GAAI,UAAU,CAAC,MAAqB,IAAI,SAAS,CAAC;YAC9D,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;YAEhB,aAAa;YACb,MAAM,KAAK,GAAG,UAAU,CAAC,KAAe,IAAI,SAAS,CAAC;YACtD,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,+CAA+C;IAC/C,kBAAkB;IAClB,+CAA+C;IAEvC,cAAc;QACpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC9C,OAAO,QAAQ,SAAS,IAAI,MAAM,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,MAA+B;QAClD,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,EAAY;YACvB,KAAK,EAAE,MAAM,CAAC,KAAe;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAiB;YACjC,MAAM,EAAG,MAAM,CAAC,MAAqB,IAAI,SAAS;YAClD,SAAS,EAAE,MAAM,CAAC,SAAiB;YACnC,SAAS,EAAE,MAAM,CAAC,SAA6B;YAC/C,WAAW,EAAE,MAAM,CAAC,WAA+B;YACnD,MAAM,EAAE,MAAM,CAAC,MAAiB;YAChC,KAAK,EAAE,MAAM,CAAC,KAA2B;SAC1C,CAAC;IACJ,CAAC;CACF;AAzLD,4CAyLC","sourcesContent":["/*\n DrizzleTaskQueue - 基于 drizzle-solid 的任务队列\n \n 简化模型：消息 + Agent\n \n 任务存储在 Pod 的 /tasks/{YYYY-MM-DD}.ttl 中，按天分片\n /\n\nimport { getLoggerFor } from 'global-logger-factory';\nimport { randomBytes } from 'crypto';\nimport { eq } from '@undefineds.co/drizzle-solid';\nimport type { SolidDatabase } from '@undefineds.co/drizzle-solid';\n\nimport { Task as taskTable, TaskStatus as TaskStatusConst } from './schema';\nimport type {\n Task,\n TaskStatus,\n CreateTaskInput,\n TaskQueue,\n TaskQueueStats,\n ListTasksOptions,\n} from './types';\n\nexport interface DrizzleTaskQueueOptions {\n /\n Pod base URL\n /\n podBaseUrl: string;\n /\n drizzle-solid 数据库实例\n /\n db: SolidDatabase<{ task: typeof taskTable }>;\n}\n\n/\n 基于 drizzle-solid 的任务队列实现\n /\nexport class DrizzleTaskQueue implements TaskQueue {\n protected readonly logger = getLoggerFor(this);\n private readonly podBaseUrl: string;\n private readonly db: SolidDatabase<{ task: typeof taskTable }>;\n\n public constructor(options: DrizzleTaskQueueOptions) {\n this.podBaseUrl = options.podBaseUrl.endsWith('/') ? options.podBaseUrl : `${options.podBaseUrl}/`;\n this.db = options.db;\n }\n\n /\n 创建任务\n /\n public async createTask(input: CreateTaskInput): Promise<Task> {\n const id = this.generateTaskId();\n const now = new Date();\n\n const taskData = {\n id,\n agent: input.agent,\n message: input.message,\n status: TaskStatusConst.PENDING,\n createdAt: now,\n };\n\n await this.db.insert(taskTable).values(taskData);\n\n this.logger.info(`Created task ${id} for agent \"${input.agent}\"`);\n\n return {\n id,\n agent: input.agent,\n message: input.message,\n status: 'pending',\n createdAt: now,\n };\n }\n\n /\n 获取任务\n /\n public async getTask(taskId: string): Promise<Task \| null> {\n const tasks = await this.db.select().from(taskTable).where(eq(taskTable.id, taskId));\n\n if (tasks.length === 0) {\n return null;\n }\n\n return this.dbTaskToTask(tasks[0]);\n }\n\n /\n 获取任务列表\n /\n public async listTasks(options?: ListTasksOptions): Promise<Task[]> {\n let query = this.db.select().from(taskTable);\n\n // 状态过滤\n if (options?.status) {\n query = query.where(eq(taskTable.status, options.status));\n }\n\n const tasks = await query;\n\n // 转换为 Task 接口\n let result = tasks.map((t: Record<string, unknown>) => this.dbTaskToTask(t));\n\n // Agent 过滤\n if (options?.agent) {\n result = result.filter((t: Task) => t.agent === options.agent);\n }\n\n // 排序\n if (options?.orderBy) {\n const order = options.order === 'asc' ? 1 : -1;\n result.sort((a: Task, b: Task) => {\n const aVal = a[options.orderBy!];\n const bVal = b[options.orderBy!];\n if (aVal instanceof Date && bVal instanceof Date) {\n return (aVal.getTime() - bVal.getTime()) order;\n }\n return 0;\n });\n }\n\n // 分页\n const offset = options?.offset ?? 0;\n const limit = options?.limit ?? result.length;\n return result.slice(offset, offset + limit);\n }\n\n /*\n 更新任务状态\n /\n public async updateTaskStatus(\n taskId: string,\n status: TaskStatus,\n updates?: Partial<Task>,\n ): Promise<void> {\n const now = new Date();\n\n const updateData: Record<string, unknown> = {\n status,\n };\n\n // 根据状态设置时间字段\n if (status === 'running') {\n updateData.startedAt = now;\n } else if (status === 'completed' \|\| status === 'failed') {\n updateData.completedAt = now;\n }\n\n // 合并其他更新\n if (updates) {\n if (updates.result !== undefined) updateData.result = updates.result;\n if (updates.error !== undefined) updateData.error = updates.error;\n }\n\n await this.db.update(taskTable).set(updateData).where(eq(taskTable.id, taskId));\n\n this.logger.info(`Task ${taskId} status changed to ${status}`);\n }\n\n /\n 删除任务\n /\n public async deleteTask(taskId: string): Promise<void> {\n await this.db.delete(taskTable).where(eq(taskTable.id, taskId));\n this.logger.info(`Deleted task ${taskId}`);\n }\n\n /\n 获取队列统计\n /\n public async getStats(): Promise<TaskQueueStats> {\n const tasks = await this.db.select().from(taskTable);\n\n const stats: TaskQueueStats = {\n pending: 0,\n running: 0,\n completed: 0,\n failed: 0,\n total: tasks.length,\n byAgent: {},\n };\n\n for (const task of tasks) {\n const taskRecord = task as Record<string, unknown>;\n const status = (taskRecord.status as TaskStatus) ?? 'pending';\n stats[status]++;\n\n // 按 Agent 统计\n const agent = taskRecord.agent as string ?? 'unknown';\n stats.byAgent[agent] = (stats.byAgent[agent] ?? 0) + 1;\n }\n\n return stats;\n }\n\n // ============================================\n // Private Methods\n // ============================================\n\n private generateTaskId(): string {\n const timestamp = Date.now().toString(36);\n const random = randomBytes(4).toString('hex');\n return `task-${timestamp}-${random}`;\n }\n\n /\n 将数据库记录转换为 Task 接口\n */\n private dbTaskToTask(dbTask: Record<string, unknown>): Task {\n return {\n id: dbTask.id as string,\n agent: dbTask.agent as string,\n message: dbTask.message as string,\n status: (dbTask.status as TaskStatus) ?? 'pending',\n createdAt: dbTask.createdAt as Date,\n startedAt: dbTask.startedAt as Date \| undefined,\n completedAt: dbTask.completedAt as Date \| undefined,\n result: dbTask.result as unknown,\n error: dbTask.error as string \| undefined,\n };\n }\n}\n"]}
1	+ {"version":3,"file":"DrizzleTaskQueue.js","sourceRoot":"","sources":["../../src/task/DrizzleTaskQueue.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAEH,iEAAqD;AACrD,mCAAqC;AACrC,gEAAkD;AAElD,qCAA4E;AAqB5E;;GAEG;AACH,MAAa,gBAAgB;IAK3B,YAAmB,OAAgC;QAJhC,WAAM,GAAG,IAAA,oCAAY,EAAC,IAAI,CAAC,CAAC;QAK7C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,UAAU,GAAG,CAAC;QACnG,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,UAAU,CAAC,KAAsB;QAC5C,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,MAAM,QAAQ,GAAG;YACf,EAAE;YACF,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,mBAAe,CAAC,OAAO;YAC/B,SAAS,EAAE,GAAG;SACf,CAAC;QAEF,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,aAAS,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAEjD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,eAAe,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;QAElE,OAAO;YACL,EAAE;YACF,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,GAAG;SACf,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,OAAO,CAAC,MAAc;QACjC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,aAAS,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,SAAS,CAAC,OAA0B;QAC/C,IAAI,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAS,CAAC,CAAC;QAE7C,OAAO;QACP,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;YACpB,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAA,kBAAE,EAAC,aAAS,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC;QAE1B,cAAc;QACd,IAAI,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAA0B,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAE7E,WAAW;QACX,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;YACnB,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAO,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,CAAC,CAAC;QACjE,CAAC;QAED,KAAK;QACL,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAO,EAAE,CAAO,EAAE,EAAE;gBAC/B,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,OAAQ,CAAC,CAAC;gBACjC,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,OAAQ,CAAC,CAAC;gBACjC,IAAI,IAAI,YAAY,IAAI,IAAI,IAAI,YAAY,IAAI,EAAE,CAAC;oBACjD,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,KAAK,CAAC;gBACnD,CAAC;gBACD,OAAO,CAAC,CAAC;YACX,CAAC,CAAC,CAAC;QACL,CAAC;QAED,KAAK;QACL,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;QACpC,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC;QAC9C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,KAAK,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB,CAC3B,MAAc,EACd,MAAkB,EAClB,OAAuB;QAEvB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,MAAM,UAAU,GAA4B;YAC1C,MAAM;SACP,CAAC;QAEF,aAAa;QACb,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,UAAU,CAAC,SAAS,GAAG,GAAG,CAAC;QAC7B,CAAC;aAAM,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACzD,UAAU,CAAC,WAAW,GAAG,GAAG,CAAC;QAC/B,CAAC;QAED,SAAS;QACT,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS;gBAAE,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YACrE,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS;gBAAE,UAAU,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QACpE,CAAC;QAED,MAAM,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,aAAS,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC;QAErE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,MAAM,sBAAsB,MAAM,EAAE,CAAC,CAAC;IACjE,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,UAAU,CAAC,MAAc;QACpC,MAAM,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,aAAS,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QACzD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,QAAQ;QACnB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,aAAS,CAAC,CAAC;QAErD,MAAM,KAAK,GAAmB;YAC5B,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,CAAC;YACT,KAAK,EAAE,KAAK,CAAC,MAAM;YACnB,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,IAA+B,CAAC;YACnD,MAAM,MAAM,GAAI,UAAU,CAAC,MAAqB,IAAI,SAAS,CAAC;YAC9D,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;YAEhB,aAAa;YACb,MAAM,KAAK,GAAG,UAAU,CAAC,KAAe,IAAI,SAAS,CAAC;YACtD,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,+CAA+C;IAC/C,kBAAkB;IAClB,+CAA+C;IAEvC,cAAc;QACpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC9C,OAAO,QAAQ,SAAS,IAAI,MAAM,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,MAA+B;QAClD,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,EAAY;YACvB,KAAK,EAAE,MAAM,CAAC,KAAe;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAiB;YACjC,MAAM,EAAG,MAAM,CAAC,MAAqB,IAAI,SAAS;YAClD,SAAS,EAAE,MAAM,CAAC,SAAiB;YACnC,SAAS,EAAE,MAAM,CAAC,SAA6B;YAC/C,WAAW,EAAE,MAAM,CAAC,WAA+B;YACnD,MAAM,EAAE,MAAM,CAAC,MAAiB;YAChC,KAAK,EAAE,MAAM,CAAC,KAA2B;SAC1C,CAAC;IACJ,CAAC;CACF;AAvLD,4CAuLC","sourcesContent":["/*\n DrizzleTaskQueue - 基于 drizzle-solid 的任务队列\n \n 简化模型：消息 + Agent\n \n 任务存储在 Pod 的 /tasks/{YYYY-MM-DD}.ttl 中，按天分片\n /\n\nimport { getLoggerFor } from 'global-logger-factory';\nimport { randomBytes } from 'crypto';\nimport { eq } from '@undefineds.co/drizzle-solid';\n\nimport { Task as taskTable, TaskStatus as TaskStatusConst } from './schema';\nimport type {\n Task,\n TaskStatus,\n CreateTaskInput,\n TaskQueue,\n TaskQueueStats,\n ListTasksOptions,\n} from './types';\n\nexport interface DrizzleTaskQueueOptions {\n /\n Pod base URL\n /\n podBaseUrl: string;\n /\n drizzle-solid 数据库实例\n /\n db: any;\n}\n\n/\n 基于 drizzle-solid 的任务队列实现\n /\nexport class DrizzleTaskQueue implements TaskQueue {\n protected readonly logger = getLoggerFor(this);\n private readonly podBaseUrl: string;\n private readonly db: any;\n\n public constructor(options: DrizzleTaskQueueOptions) {\n this.podBaseUrl = options.podBaseUrl.endsWith('/') ? options.podBaseUrl : `${options.podBaseUrl}/`;\n this.db = options.db;\n }\n\n /\n 创建任务\n /\n public async createTask(input: CreateTaskInput): Promise<Task> {\n const id = this.generateTaskId();\n const now = new Date();\n\n const taskData = {\n id,\n agent: input.agent,\n message: input.message,\n status: TaskStatusConst.PENDING,\n createdAt: now,\n };\n\n await this.db.insert(taskTable).values(taskData);\n\n this.logger.info(`Created task ${id} for agent \"${input.agent}\"`);\n\n return {\n id,\n agent: input.agent,\n message: input.message,\n status: 'pending',\n createdAt: now,\n };\n }\n\n /\n 获取任务\n /\n public async getTask(taskId: string): Promise<Task \| null> {\n const task = await this.db.findByLocator(taskTable, { id: taskId });\n if (!task) {\n return null;\n }\n return this.dbTaskToTask(task);\n }\n\n /\n 获取任务列表\n /\n public async listTasks(options?: ListTasksOptions): Promise<Task[]> {\n let query = this.db.select().from(taskTable);\n\n // 状态过滤\n if (options?.status) {\n query = query.where(eq(taskTable.status, options.status));\n }\n\n const tasks = await query;\n\n // 转换为 Task 接口\n let result = tasks.map((t: Record<string, unknown>) => this.dbTaskToTask(t));\n\n // Agent 过滤\n if (options?.agent) {\n result = result.filter((t: Task) => t.agent === options.agent);\n }\n\n // 排序\n if (options?.orderBy) {\n const order = options.order === 'asc' ? 1 : -1;\n result.sort((a: Task, b: Task) => {\n const aVal = a[options.orderBy!];\n const bVal = b[options.orderBy!];\n if (aVal instanceof Date && bVal instanceof Date) {\n return (aVal.getTime() - bVal.getTime()) order;\n }\n return 0;\n });\n }\n\n // 分页\n const offset = options?.offset ?? 0;\n const limit = options?.limit ?? result.length;\n return result.slice(offset, offset + limit);\n }\n\n /*\n 更新任务状态\n /\n public async updateTaskStatus(\n taskId: string,\n status: TaskStatus,\n updates?: Partial<Task>,\n ): Promise<void> {\n const now = new Date();\n\n const updateData: Record<string, unknown> = {\n status,\n };\n\n // 根据状态设置时间字段\n if (status === 'running') {\n updateData.startedAt = now;\n } else if (status === 'completed' \|\| status === 'failed') {\n updateData.completedAt = now;\n }\n\n // 合并其他更新\n if (updates) {\n if (updates.result !== undefined) updateData.result = updates.result;\n if (updates.error !== undefined) updateData.error = updates.error;\n }\n\n await this.db.updateByLocator(taskTable, { id: taskId }, updateData);\n\n this.logger.info(`Task ${taskId} status changed to ${status}`);\n }\n\n /\n 删除任务\n /\n public async deleteTask(taskId: string): Promise<void> {\n await this.db.deleteByLocator(taskTable, { id: taskId });\n this.logger.info(`Deleted task ${taskId}`);\n }\n\n /\n 获取队列统计\n /\n public async getStats(): Promise<TaskQueueStats> {\n const tasks = await this.db.select().from(taskTable);\n\n const stats: TaskQueueStats = {\n pending: 0,\n running: 0,\n completed: 0,\n failed: 0,\n total: tasks.length,\n byAgent: {},\n };\n\n for (const task of tasks) {\n const taskRecord = task as Record<string, unknown>;\n const status = (taskRecord.status as TaskStatus) ?? 'pending';\n stats[status]++;\n\n // 按 Agent 统计\n const agent = taskRecord.agent as string ?? 'unknown';\n stats.byAgent[agent] = (stats.byAgent[agent] ?? 0) + 1;\n }\n\n return stats;\n }\n\n // ============================================\n // Private Methods\n // ============================================\n\n private generateTaskId(): string {\n const timestamp = Date.now().toString(36);\n const random = randomBytes(4).toString('hex');\n return `task-${timestamp}-${random}`;\n }\n\n /\n 将数据库记录转换为 Task 接口\n */\n private dbTaskToTask(dbTask: Record<string, unknown>): Task {\n return {\n id: dbTask.id as string,\n agent: dbTask.agent as string,\n message: dbTask.message as string,\n status: (dbTask.status as TaskStatus) ?? 'pending',\n createdAt: dbTask.createdAt as Date,\n startedAt: dbTask.startedAt as Date \| undefined,\n completedAt: dbTask.completedAt as Date \| undefined,\n result: dbTask.result as unknown,\n error: dbTask.error as string \| undefined,\n };\n }\n}\n"]}

package/dist/test-utils/seed-pod.js CHANGED Viewed

@@ -7,15 +7,16 @@ exports.seedPod = seedPod;
 /**
  * Seed 测试 Pod 数据 - 在 SPARQL 数据库插入容器元数据
  */
-const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
 const fs_1 = require("fs");
 const path_1 = __importDefault(require("path"));
+const SqliteRuntime_1 = require("../storage/SqliteRuntime");
 async function seedPod(options) {
     const { rootFilePath, podName = 'test' } = options;
     // 确保目录存在
     await fs_1.promises.mkdir(rootFilePath, { recursive: true });
     const sparqlDbPath = path_1.default.join(rootFilePath, 'sparql.db');
-    const db = new better_sqlite3_1.default(sparqlDbPath);
+    const sqliteRuntime = (0, SqliteRuntime_1.getSqliteRuntime)();
+    const db = sqliteRuntime.openDatabase(sparqlDbPath);
     // 创建 quints 表（如果不存在）
     db.exec(`
     CREATE TABLE IF NOT EXISTS quints (