@nahisaho/katashiro-rag 2.0.0

package/dist/vectordb/FileVectorStore.d.ts

@@ -0,0 +1,93 @@
+/**
+ * File-based Vector Store
+ *
+ * @requirement REQ-RAG-102
+ * @design DES-KATASHIRO-003-RAG §3.3
+ */
+import type { Chunk, SearchResult, Vector, VectorStore } from '../types.js';
+/**
+ * FileVectorStoreの設定
+ */
+export interface FileVectorStoreConfig {
+    /** 保存先ファイルパス */
+    filePath: string;
+    /** 類似度しきい値（デフォルト: 0.7） */
+    similarityThreshold?: number;
+    /** 自動保存（デフォルト: true） */
+    autoSave?: boolean;
+    /** 保存間隔（ミリ秒、デフォルト: 1000） */
+    saveIntervalMs?: number;
+}
+/**
+ * ファイルベースのベクトルストア
+ * データをJSONファイルに永続化
+ */
+export declare class FileVectorStore implements VectorStore {
+    readonly name = "file";
+    private entries;
+    private filePath;
+    private similarityThreshold;
+    private autoSave;
+    private saveIntervalMs;
+    private isDirty;
+    private saveTimer;
+    private createdAt;
+    private isInitialized;
+    constructor(config: FileVectorStoreConfig);
+    /**
+     * 格納エントリ数を取得
+     */
+    get size(): number;
+    /**
+     * ストアを初期化（ファイルから読み込み）
+     */
+    init(): Promise<void>;
+    add(chunk: Chunk, vector: Vector): Promise<void>;
+    addBatch(items: Array<{
+        chunk: Chunk;
+        vector: Vector;
+    }>): Promise<void>;
+    search(vector: Vector, topK: number): Promise<SearchResult[]>;
+    delete(chunkId: string): Promise<boolean>;
+    /**
+     * ストアをクリア
+     */
+    clear(): Promise<void>;
+    /**
+     * 指定IDのチャンクが存在するか確認
+     */
+    has(chunkId: string): Promise<boolean>;
+    /**
+     * 即座にファイルに保存
+     */
+    save(): Promise<void>;
+    /**
+     * 未保存の変更があるか確認
+     */
+    get dirty(): boolean;
+    /**
+     * ストアを閉じる（保留中の保存を実行）
+     */
+    close(): Promise<void>;
+    /**
+     * ファイルパスを取得
+     */
+    getFilePath(): string;
+    /**
+     * 初期化を確認
+     */
+    private ensureInitialized;
+    /**
+     * 変更をマーク
+     */
+    private markDirty;
+    /**
+     * 保存をスケジュール
+     */
+    private scheduleSave;
+    /**
+     * コサイン類似度を計算
+     */
+    private cosineSimilarity;
+}
+//# sourceMappingURL=FileVectorStore.d.ts.map

package/dist/vectordb/FileVectorStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"FileVectorStore.d.ts","sourceRoot":"","sources":["../../src/vectordb/FileVectorStore.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAI5E;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,gBAAgB;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,0BAA0B;IAC1B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,wBAAwB;IACxB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,4BAA4B;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAwBD;;;GAGG;AACH,qBAAa,eAAgB,YAAW,WAAW;IACjD,QAAQ,CAAC,IAAI,UAAU;IAEvB,OAAO,CAAC,OAAO,CAAuC;IACtD,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,mBAAmB,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,SAAS,CAA8C;IAC/D,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,aAAa,CAAS;gBAElB,MAAM,EAAE,qBAAqB;IAQzC;;OAEG;IACH,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IA6BrB,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMhD,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,KAAK,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAQvE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAsB7D,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAS/C;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAM5B;;OAEG;IACG,GAAG,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAK5C;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAwB3B;;OAEG;IACH,IAAI,KAAK,IAAI,OAAO,CAEnB;IAED;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAW5B;;OAEG;IACH,WAAW,IAAI,MAAM;IAIrB;;OAEG;YACW,iBAAiB;IAM/B;;OAEG;IACH,OAAO,CAAC,SAAS;IAQjB;;OAEG;IACH,OAAO,CAAC,YAAY;IAYpB;;OAEG;IACH,OAAO,CAAC,gBAAgB;CAsBzB"}

package/dist/vectordb/FileVectorStore.js

@@ -0,0 +1,218 @@
+/**
+ * File-based Vector Store
+ *
+ * @requirement REQ-RAG-102
+ * @design DES-KATASHIRO-003-RAG §3.3
+ */
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+/**
+ * ファイルベースのベクトルストア
+ * データをJSONファイルに永続化
+ */
+export class FileVectorStore {
+    name = 'file';
+    entries = new Map();
+    filePath;
+    similarityThreshold;
+    autoSave;
+    saveIntervalMs;
+    isDirty = false;
+    saveTimer = null;
+    createdAt;
+    isInitialized = false;
+    constructor(config) {
+        this.filePath = path.resolve(config.filePath);
+        this.similarityThreshold = config.similarityThreshold ?? 0.7;
+        this.autoSave = config.autoSave ?? true;
+        this.saveIntervalMs = config.saveIntervalMs ?? 1000;
+        this.createdAt = new Date().toISOString();
+    }
+    /**
+     * 格納エントリ数を取得
+     */
+    get size() {
+        return this.entries.size;
+    }
+    /**
+     * ストアを初期化（ファイルから読み込み）
+     */
+    async init() {
+        if (this.isInitialized)
+            return;
+        if (fs.existsSync(this.filePath)) {
+            try {
+                const data = await fs.promises.readFile(this.filePath, 'utf-8');
+                const parsed = JSON.parse(data);
+                // データを読み込み
+                for (const entry of parsed.entries) {
+                    this.entries.set(entry.chunkId, {
+                        chunk: entry.chunk,
+                        vector: entry.vector,
+                    });
+                }
+                if (parsed.createdAt) {
+                    this.createdAt = parsed.createdAt;
+                }
+            }
+            catch (error) {
+                // ファイルが壊れている場合は空で開始
+                console.warn(`Failed to load vector store from ${this.filePath}:`, error);
+                this.entries.clear();
+            }
+        }
+        this.isInitialized = true;
+    }
+    async add(chunk, vector) {
+        await this.ensureInitialized();
+        this.entries.set(chunk.id, { chunk, vector });
+        this.markDirty();
+    }
+    async addBatch(items) {
+        await this.ensureInitialized();
+        for (const item of items) {
+            this.entries.set(item.chunk.id, { chunk: item.chunk, vector: item.vector });
+        }
+        this.markDirty();
+    }
+    async search(vector, topK) {
+        await this.ensureInitialized();
+        const results = [];
+        for (const entry of this.entries.values()) {
+            const similarity = this.cosineSimilarity(vector, entry.vector);
+            if (similarity >= this.similarityThreshold) {
+                results.push({ entry, similarity });
+            }
+        }
+        // スコア降順でソート
+        results.sort((a, b) => b.similarity - a.similarity);
+        // topKに制限
+        return results.slice(0, topK).map(({ entry, similarity }) => ({
+            chunk: entry.chunk,
+            score: similarity,
+        }));
+    }
+    async delete(chunkId) {
+        await this.ensureInitialized();
+        const result = this.entries.delete(chunkId);
+        if (result) {
+            this.markDirty();
+        }
+        return result;
+    }
+    /**
+     * ストアをクリア
+     */
+    async clear() {
+        await this.ensureInitialized();
+        this.entries.clear();
+        this.markDirty();
+    }
+    /**
+     * 指定IDのチャンクが存在するか確認
+     */
+    async has(chunkId) {
+        await this.ensureInitialized();
+        return this.entries.has(chunkId);
+    }
+    /**
+     * 即座にファイルに保存
+     */
+    async save() {
+        await this.ensureInitialized();
+        // ディレクトリが存在しない場合は作成
+        const dir = path.dirname(this.filePath);
+        if (!fs.existsSync(dir)) {
+            await fs.promises.mkdir(dir, { recursive: true });
+        }
+        const data = {
+            version: '1.0.0',
+            createdAt: this.createdAt,
+            updatedAt: new Date().toISOString(),
+            entries: Array.from(this.entries.entries()).map(([chunkId, entry]) => ({
+                chunkId,
+                chunk: entry.chunk,
+                vector: entry.vector,
+            })),
+        };
+        await fs.promises.writeFile(this.filePath, JSON.stringify(data, null, 2), 'utf-8');
+        this.isDirty = false;
+    }
+    /**
+     * 未保存の変更があるか確認
+     */
+    get dirty() {
+        return this.isDirty;
+    }
+    /**
+     * ストアを閉じる（保留中の保存を実行）
+     */
+    async close() {
+        if (this.saveTimer) {
+            clearTimeout(this.saveTimer);
+            this.saveTimer = null;
+        }
+        if (this.isDirty) {
+            await this.save();
+        }
+    }
+    /**
+     * ファイルパスを取得
+     */
+    getFilePath() {
+        return this.filePath;
+    }
+    /**
+     * 初期化を確認
+     */
+    async ensureInitialized() {
+        if (!this.isInitialized) {
+            await this.init();
+        }
+    }
+    /**
+     * 変更をマーク
+     */
+    markDirty() {
+        this.isDirty = true;
+        if (this.autoSave) {
+            this.scheduleSave();
+        }
+    }
+    /**
+     * 保存をスケジュール
+     */
+    scheduleSave() {
+        if (this.saveTimer) {
+            clearTimeout(this.saveTimer);
+        }
+        this.saveTimer = setTimeout(() => {
+            this.save().catch((err) => {
+                console.error('Failed to auto-save vector store:', err);
+            });
+        }, this.saveIntervalMs);
+    }
+    /**
+     * コサイン類似度を計算
+     */
+    cosineSimilarity(a, b) {
+        if (a.length !== b.length) {
+            throw new Error(`Vector dimension mismatch: ${a.length} vs ${b.length}`);
+        }
+        let dotProduct = 0;
+        let normA = 0;
+        let normB = 0;
+        for (let i = 0; i < a.length; i++) {
+            const valA = a[i] ?? 0;
+            const valB = b[i] ?? 0;
+            dotProduct += valA * valB;
+            normA += valA * valA;
+            normB += valB * valB;
+        }
+        const denominator = Math.sqrt(normA) * Math.sqrt(normB);
+        if (denominator === 0)
+            return 0;
+        return dotProduct / denominator;
+    }
+}
+//# sourceMappingURL=FileVectorStore.js.map

package/dist/vectordb/FileVectorStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FileVectorStore.js","sourceRoot":"","sources":["../../src/vectordb/FileVectorStore.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAsClC;;;GAGG;AACH,MAAM,OAAO,eAAe;IACjB,IAAI,GAAG,MAAM,CAAC;IAEf,OAAO,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC9C,QAAQ,CAAS;IACjB,mBAAmB,CAAS;IAC5B,QAAQ,CAAU;IAClB,cAAc,CAAS;IACvB,OAAO,GAAG,KAAK,CAAC;IAChB,SAAS,GAAyC,IAAI,CAAC;IACvD,SAAS,CAAS;IAClB,aAAa,GAAG,KAAK,CAAC;IAE9B,YAAY,MAA6B;QACvC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9C,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,mBAAmB,IAAI,GAAG,CAAC;QAC7D,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC;QACxC,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC;QACpD,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,aAAa;YAAE,OAAO;QAE/B,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBAChE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAkB,CAAC;gBAEjD,WAAW;gBACX,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE;wBAC9B,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,MAAM,EAAE,KAAK,CAAC,MAAM;qBACrB,CAAC,CAAC;gBACL,CAAC;gBAED,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;gBACpC,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,oBAAoB;gBACpB,OAAO,CAAC,IAAI,CAAC,oCAAoC,IAAI,CAAC,QAAQ,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC1E,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACvB,CAAC;QACH,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,MAAc;QACpC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAA8C;QAC3D,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9E,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,IAAY;QACvC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAsD,EAAE,CAAC;QAEtE,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;YAE/D,IAAI,UAAU,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC3C,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QAED,YAAY;QACZ,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;QAEpD,UAAU;QACV,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;YAC5D,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,KAAK,EAAE,UAAU;SAClB,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAe;QAC1B,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/B,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,SAAS,EAAE,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,GAAG,CAAC,OAAe;QACvB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI;QACR,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,oBAAoB;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,IAAI,GAAkB;YAC1B,OAAO,EAAE,OAAO;YAChB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;gBACrE,OAAO;gBACP,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,MAAM,EAAE,KAAK,CAAC,MAAM;aACrB,CAAC,CAAC;SACJ,CAAC;QAEF,MAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACnF,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC7B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB;QAC7B,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,SAAS;QACf,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,YAAY;QAClB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;YAC/B,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACxB,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,GAAG,CAAC,CAAC;YAC1D,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,CAAS,EAAE,CAAS;QAC3C,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC,MAAM,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACvB,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACvB,UAAU,IAAI,IAAI,GAAG,IAAI,CAAC;YAC1B,KAAK,IAAI,IAAI,GAAG,IAAI,CAAC;YACrB,KAAK,IAAI,IAAI,GAAG,IAAI,CAAC;QACvB,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,WAAW,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAEhC,OAAO,UAAU,GAAG,WAAW,CAAC;IAClC,CAAC;CACF"}

package/dist/vectordb/InMemoryVectorStore.d.ts

@@ -0,0 +1,48 @@
+/**
+ * In-Memory Vector Store
+ *
+ * @requirement REQ-RAG-002
+ * @design DES-KATASHIRO-003-RAG §3.2
+ */
+import type { Chunk, SearchResult, Vector, VectorStore } from '../types.js';
+/**
+ * インメモリベクトルストアの設定
+ */
+export interface InMemoryVectorStoreConfig {
+    /** 類似度しきい値（デフォルト: 0.7） */
+    similarityThreshold?: number;
+}
+/**
+ * インメモリベクトルストア
+ * テスト用およびプロトタイピング用
+ */
+export declare class InMemoryVectorStore implements VectorStore {
+    readonly name = "in-memory";
+    private entries;
+    private similarityThreshold;
+    constructor(config?: InMemoryVectorStoreConfig);
+    /**
+     * 格納エントリ数を取得
+     */
+    get size(): number;
+    add(chunk: Chunk, vector: Vector): Promise<void>;
+    addBatch(items: Array<{
+        chunk: Chunk;
+        vector: Vector;
+    }>): Promise<void>;
+    search(vector: Vector, topK: number): Promise<SearchResult[]>;
+    delete(chunkId: string): Promise<boolean>;
+    /**
+     * ストアをクリア
+     */
+    clear(): void;
+    /**
+     * 指定IDのチャンクが存在するか確認
+     */
+    has(chunkId: string): boolean;
+    /**
+     * コサイン類似度を計算
+     */
+    private cosineSimilarity;
+}
+//# sourceMappingURL=InMemoryVectorStore.d.ts.map

package/dist/vectordb/InMemoryVectorStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"InMemoryVectorStore.d.ts","sourceRoot":"","sources":["../../src/vectordb/InMemoryVectorStore.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE5E;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACxC,0BAA0B;IAC1B,mBAAmB,CAAC,EAAE,MAAM,CAAC;CAC9B;AAUD;;;GAGG;AACH,qBAAa,mBAAoB,YAAW,WAAW;IACrD,QAAQ,CAAC,IAAI,eAAe;IAE5B,OAAO,CAAC,OAAO,CAAuC;IACtD,OAAO,CAAC,mBAAmB,CAAS;gBAExB,MAAM,GAAE,yBAA8B;IAIlD;;OAEG;IACH,IAAI,IAAI,IAAI,MAAM,CAEjB;IAEK,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIhD,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,KAAK,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAMvE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAqB7D,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAI/C;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;OAEG;IACH,GAAG,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAI7B;;OAEG;IACH,OAAO,CAAC,gBAAgB;CAsBzB"}

package/dist/vectordb/InMemoryVectorStore.js

@@ -0,0 +1,86 @@
+/**
+ * In-Memory Vector Store
+ *
+ * @requirement REQ-RAG-002
+ * @design DES-KATASHIRO-003-RAG §3.2
+ */
+/**
+ * インメモリベクトルストア
+ * テスト用およびプロトタイピング用
+ */
+export class InMemoryVectorStore {
+    name = 'in-memory';
+    entries = new Map();
+    similarityThreshold;
+    constructor(config = {}) {
+        this.similarityThreshold = config.similarityThreshold ?? 0.7;
+    }
+    /**
+     * 格納エントリ数を取得
+     */
+    get size() {
+        return this.entries.size;
+    }
+    async add(chunk, vector) {
+        this.entries.set(chunk.id, { chunk, vector });
+    }
+    async addBatch(items) {
+        for (const item of items) {
+            this.entries.set(item.chunk.id, { chunk: item.chunk, vector: item.vector });
+        }
+    }
+    async search(vector, topK) {
+        const results = [];
+        for (const entry of this.entries.values()) {
+            const similarity = this.cosineSimilarity(vector, entry.vector);
+            if (similarity >= this.similarityThreshold) {
+                results.push({ entry, similarity });
+            }
+        }
+        // スコア降順でソート
+        results.sort((a, b) => b.similarity - a.similarity);
+        // topKに制限
+        return results.slice(0, topK).map(({ entry, similarity }) => ({
+            chunk: entry.chunk,
+            score: similarity,
+        }));
+    }
+    async delete(chunkId) {
+        return this.entries.delete(chunkId);
+    }
+    /**
+     * ストアをクリア
+     */
+    clear() {
+        this.entries.clear();
+    }
+    /**
+     * 指定IDのチャンクが存在するか確認
+     */
+    has(chunkId) {
+        return this.entries.has(chunkId);
+    }
+    /**
+     * コサイン類似度を計算
+     */
+    cosineSimilarity(a, b) {
+        if (a.length !== b.length) {
+            throw new Error(`Vector dimension mismatch: ${a.length} vs ${b.length}`);
+        }
+        let dotProduct = 0;
+        let normA = 0;
+        let normB = 0;
+        for (let i = 0; i < a.length; i++) {
+            const valA = a[i] ?? 0;
+            const valB = b[i] ?? 0;
+            dotProduct += valA * valB;
+            normA += valA * valA;
+            normB += valB * valB;
+        }
+        const denominator = Math.sqrt(normA) * Math.sqrt(normB);
+        if (denominator === 0)
+            return 0;
+        return dotProduct / denominator;
+    }
+}
+//# sourceMappingURL=InMemoryVectorStore.js.map

package/dist/vectordb/InMemoryVectorStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"InMemoryVectorStore.js","sourceRoot":"","sources":["../../src/vectordb/InMemoryVectorStore.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAoBH;;;GAGG;AACH,MAAM,OAAO,mBAAmB;IACrB,IAAI,GAAG,WAAW,CAAC;IAEpB,OAAO,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC9C,mBAAmB,CAAS;IAEpC,YAAY,SAAoC,EAAE;QAChD,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,mBAAmB,IAAI,GAAG,CAAC;IAC/D,CAAC;IAED;;OAEG;IACH,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,MAAc;QACpC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAA8C;QAC3D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,IAAY;QACvC,MAAM,OAAO,GAAsD,EAAE,CAAC;QAEtE,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;YAE/D,IAAI,UAAU,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC3C,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QAED,YAAY;QACZ,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;QAEpD,UAAU;QACV,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;YAC5D,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,KAAK,EAAE,UAAU;SAClB,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAe;QAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,OAAe;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,CAAS,EAAE,CAAS;QAC3C,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC,MAAM,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACvB,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACvB,UAAU,IAAI,IAAI,GAAG,IAAI,CAAC;YAC1B,KAAK,IAAI,IAAI,GAAG,IAAI,CAAC;YACrB,KAAK,IAAI,IAAI,GAAG,IAAI,CAAC;QACvB,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,WAAW,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAEhC,OAAO,UAAU,GAAG,WAAW,CAAC;IAClC,CAAC;CACF"}

package/dist/vectordb/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Vector DB module exports
+ */
+export { InMemoryVectorStore } from './InMemoryVectorStore.js';
+export type { InMemoryVectorStoreConfig } from './InMemoryVectorStore.js';
+export { FileVectorStore } from './FileVectorStore.js';
+export type { FileVectorStoreConfig } from './FileVectorStore.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/vectordb/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vectordb/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/D,YAAY,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AAE1E,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,YAAY,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC"}

package/dist/vectordb/index.js

@@ -0,0 +1,6 @@
+/**
+ * Vector DB module exports
+ */
+export { InMemoryVectorStore } from './InMemoryVectorStore.js';
+export { FileVectorStore } from './FileVectorStore.js';
+//# sourceMappingURL=index.js.map

package/dist/vectordb/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/vectordb/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAG/D,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC"}

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@nahisaho/katashiro-rag",
+  "version": "2.0.0",
+  "description": "RAG and VectorDB integration for KATASHIRO",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "src"
+  ],
+  "dependencies": {
+    "@nahisaho/katashiro-core": "2.0.0",
+    "@nahisaho/katashiro-llm": "2.0.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.6.3",
+    "@types/node": "^22.10.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "author": "nahisaho",
+  "license": "MIT",
+  "scripts": {
+    "build": "tsc",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "clean": "rm -rf dist"
+  }
+}

package/src/RAGEngine.ts

@@ -0,0 +1,127 @@
+/**
+ * RAG Engine - RAGシステムのファサード
+ *
+ * @requirement REQ-RAG-005
+ * @design DES-KATASHIRO-003-RAG §3.5
+ */
+
+import { DocumentChunker } from './chunking/DocumentChunker.js';
+import { Retriever } from './Retriever.js';
+import type {
+  Chunk,
+  ChunkingConfig,
+  Document,
+  EmbeddingProvider,
+  RAGEngineConfig,
+  RetrieverConfig,
+  SearchResult,
+  VectorStore,
+} from './types.js';
+
+/**
+ * RAGエンジン
+ * ドキュメント処理、インデックス作成、検索を統合したファサード
+ */
+export class RAGEngine {
+  private chunker: DocumentChunker;
+  private retriever: Retriever;
+
+  constructor(
+    embeddingProvider: EmbeddingProvider,
+    vectorStore: VectorStore,
+    config: RAGEngineConfig = {},
+  ) {
+    this.chunker = new DocumentChunker(config.chunking);
+    this.retriever = new Retriever(embeddingProvider, vectorStore, config.retriever);
+  }
+
+  /**
+   * ドキュメントをインデックスに追加
+   * チャンキング、埋め込み生成、インデックス追加を一括実行
+   */
+  async ingest(document: Document): Promise<Chunk[]> {
+    const chunks = this.chunker.chunk(document);
+    await this.retriever.addDocument(document, chunks);
+    return chunks;
+  }
+
+  /**
+   * 複数ドキュメントをバッチでインデックスに追加
+   */
+  async ingestBatch(documents: Document[]): Promise<Chunk[]> {
+    const allChunks: Chunk[] = [];
+
+    for (const document of documents) {
+      const chunks = this.chunker.chunk(document);
+      allChunks.push(...chunks);
+      await this.retriever.addDocument(document, chunks);
+    }
+
+    return allChunks;
+  }
+
+  /**
+   * クエリで関連チャンクを検索
+   */
+  async query(query: string): Promise<SearchResult[]> {
+    return this.retriever.search(query);
+  }
+
+  /**
+   * 複数クエリで検索
+   */
+  async queryMultiple(queries: string[]): Promise<SearchResult[]> {
+    return this.retriever.searchMultiple(queries);
+  }
+
+  /**
+   * チャンクを削除
+   */
+  async deleteChunk(chunkId: string): Promise<boolean> {
+    return this.retriever.deleteChunk(chunkId);
+  }
+
+  /**
+   * ドキュメントの全チャンクを削除
+   */
+  async deleteDocument(documentId: string, chunkCount: number): Promise<number> {
+    let deleted = 0;
+
+    for (let i = 0; i < chunkCount; i++) {
+      const chunkId = `${documentId}_chunk_${i}`;
+      if (await this.retriever.deleteChunk(chunkId)) {
+        deleted++;
+      }
+    }
+
+    return deleted;
+  }
+
+  /**
+   * チャンキング設定を更新して新しいチャンカーを作成
+   */
+  updateChunkingConfig(config: ChunkingConfig): void {
+    this.chunker = new DocumentChunker(config);
+  }
+
+  /**
+   * 検索設定を更新
+   */
+  updateRetrieverConfig(config: Partial<RetrieverConfig>): void {
+    this.retriever.updateConfig(config);
+  }
+
+  /**
+   * ドキュメントをチャンキングのみ実行（インデックス追加なし）
+   */
+  chunk(document: Document): Chunk[] {
+    return this.chunker.chunk(document);
+  }
+
+  /**
+   * 複数ドキュメントをチャンキングのみ実行
+   */
+  chunkBatch(documents: Document[]): Chunk[] {
+    return this.chunker.chunkBatch(documents);
+  }
+}