@nahisaho/katashiro-rag 2.0.0

package/dist/embedding/OllamaEmbeddingProvider.d.ts

@@ -0,0 +1,69 @@
+/**
+ * Ollama Embedding Provider
+ *
+ * Local embedding provider using Ollama
+ *
+ * @requirement REQ-RAG-001
+ * @design DES-KATASHIRO-003-RAG
+ */
+import { BaseEmbeddingProvider } from './BaseEmbeddingProvider.js';
+import type { EmbeddingConfig } from '../types.js';
+/**
+ * Ollama設定
+ */
+export interface OllamaEmbeddingConfig extends EmbeddingConfig {
+    /** ベースURL（デフォルト: http://localhost:11434） */
+    baseUrl?: string;
+    /** モデル名（デフォルト: nomic-embed-text） */
+    model?: string;
+    /** Keep Alive（モデルをメモリに保持する時間） */
+    keepAlive?: string;
+}
+/**
+ * Ollama Embeddingプロバイダー
+ *
+ * ローカルでOllamaを使用した埋め込み生成
+ *
+ * @example
+ * ```typescript
+ * const provider = new OllamaEmbeddingProvider({
+ *   baseUrl: 'http://192.168.224.1:11434',
+ *   model: 'nomic-embed-text',
+ * });
+ *
+ * const embedding = await provider.embed('Hello, world!');
+ * ```
+ */
+export declare class OllamaEmbeddingProvider extends BaseEmbeddingProvider {
+    readonly name = "ollama";
+    private readonly baseUrl;
+    private readonly model;
+    private readonly keepAlive;
+    private _dimensions;
+    get dimensions(): number;
+    constructor(config?: OllamaEmbeddingConfig);
+    /**
+     * モデルごとのデフォルト次元数
+     */
+    private getDefaultDimensions;
+    /**
+     * 単一テキストの埋め込み生成
+     */
+    embed(text: string): Promise<number[]>;
+    /**
+     * バッチ埋め込み生成（Ollama固有実装）
+     *
+     * Ollamaはネイティブのバッチ処理をサポートしていないため、
+     * 並列リクエストで処理
+     */
+    protected embedBatchInternal(texts: string[]): Promise<number[][]>;
+    /**
+     * 利用可能なモデル一覧取得
+     */
+    listModels(): Promise<string[]>;
+    /**
+     * モデルの存在確認
+     */
+    isModelAvailable(): Promise<boolean>;
+}
+//# sourceMappingURL=OllamaEmbeddingProvider.d.ts.map

package/dist/embedding/OllamaEmbeddingProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OllamaEmbeddingProvider.d.ts","sourceRoot":"","sources":["../../src/embedding/OllamaEmbeddingProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEnD;;GAEG;AACH,MAAM,WAAW,qBAAsB,SAAQ,eAAe;IAC5D,4CAA4C;IAC5C,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,oCAAoC;IACpC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,iCAAiC;IACjC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAaD;;;;;;;;;;;;;;GAcG;AACH,qBAAa,uBAAwB,SAAQ,qBAAqB;IAChE,QAAQ,CAAC,IAAI,YAAY;IAEzB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,WAAW,CAAS;IAE5B,IAAI,UAAU,IAAI,MAAM,CAEvB;gBAEW,MAAM,GAAE,qBAA0B;IAa9C;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAmB5B;;OAEG;IACG,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA+B5C;;;;;OAKG;cACsB,kBAAkB,CACzC,KAAK,EAAE,MAAM,EAAE,GACd,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAmBtB;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAwBrC;;OAEG;IACG,gBAAgB,IAAI,OAAO,CAAC,OAAO,CAAC;CAQ3C"}

package/dist/embedding/OllamaEmbeddingProvider.js

@@ -0,0 +1,136 @@
+/**
+ * Ollama Embedding Provider
+ *
+ * Local embedding provider using Ollama
+ *
+ * @requirement REQ-RAG-001
+ * @design DES-KATASHIRO-003-RAG
+ */
+import { BaseEmbeddingProvider } from './BaseEmbeddingProvider.js';
+/**
+ * Ollama Embeddingプロバイダー
+ *
+ * ローカルでOllamaを使用した埋め込み生成
+ *
+ * @example
+ * ```typescript
+ * const provider = new OllamaEmbeddingProvider({
+ *   baseUrl: 'http://192.168.224.1:11434',
+ *   model: 'nomic-embed-text',
+ * });
+ *
+ * const embedding = await provider.embed('Hello, world!');
+ * ```
+ */
+export class OllamaEmbeddingProvider extends BaseEmbeddingProvider {
+    name = 'ollama';
+    baseUrl;
+    model;
+    keepAlive;
+    _dimensions;
+    get dimensions() {
+        return this._dimensions;
+    }
+    constructor(config = {}) {
+        super(config);
+        this.baseUrl = config.baseUrl ?? 'http://localhost:11434';
+        this.model = config.model ?? 'nomic-embed-text';
+        this.keepAlive = config.keepAlive ?? '5m';
+        // nomic-embed-text: 768次元
+        // mxbai-embed-large: 1024次元
+        // all-minilm: 384次元
+        this._dimensions = config.dimensions ?? this.getDefaultDimensions(this.model);
+    }
+    /**
+     * モデルごとのデフォルト次元数
+     */
+    getDefaultDimensions(model) {
+        const dimensionMap = {
+            'nomic-embed-text': 768,
+            'mxbai-embed-large': 1024,
+            'all-minilm': 384,
+            'snowflake-arctic-embed': 1024,
+            'bge-m3': 1024,
+        };
+        // モデル名の先頭マッチで検索
+        for (const [key, dims] of Object.entries(dimensionMap)) {
+            if (model.includes(key)) {
+                return dims;
+            }
+        }
+        return 768; // デフォルト
+    }
+    /**
+     * 単一テキストの埋め込み生成
+     */
+    async embed(text) {
+        return this.withRetry(async () => {
+            const url = `${this.baseUrl}/api/embeddings`;
+            const response = await this.fetchWithTimeout(url, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({
+                    model: this.model,
+                    prompt: text,
+                    keep_alive: this.keepAlive,
+                }),
+            }, this.config.timeout);
+            const data = (await response.json());
+            // 次元数を実際の値で更新（初回のみ）
+            if (this._dimensions === 768 && data.embedding.length !== 768) {
+                this._dimensions = data.embedding.length;
+            }
+            return data.embedding;
+        });
+    }
+    /**
+     * バッチ埋め込み生成（Ollama固有実装）
+     *
+     * Ollamaはネイティブのバッチ処理をサポートしていないため、
+     * 並列リクエストで処理
+     */
+    async embedBatchInternal(texts) {
+        // 並列度を制限（5並列）
+        const concurrency = 5;
+        const results = new Array(texts.length);
+        for (let i = 0; i < texts.length; i += concurrency) {
+            const batch = texts.slice(i, i + concurrency);
+            const embeddings = await Promise.all(batch.map((text) => this.embed(text)));
+            embeddings.forEach((embedding, j) => {
+                results[i + j] = embedding;
+            });
+        }
+        return results;
+    }
+    /**
+     * 利用可能なモデル一覧取得
+     */
+    async listModels() {
+        const url = `${this.baseUrl}/api/tags`;
+        const response = await this.fetchWithTimeout(url, {
+            method: 'GET',
+        });
+        const data = (await response.json());
+        // embeddingモデルをフィルタ（名前に embed を含むもの）
+        return data.models
+            .filter((m) => m.name.includes('embed') ||
+            m.name.includes('nomic') ||
+            m.name.includes('bge'))
+            .map((m) => m.name);
+    }
+    /**
+     * モデルの存在確認
+     */
+    async isModelAvailable() {
+        try {
+            const models = await this.listModels();
+            return models.some((m) => m.includes(this.model));
+        }
+        catch {
+            return false;
+        }
+    }
+}
+//# sourceMappingURL=OllamaEmbeddingProvider.js.map

package/dist/embedding/OllamaEmbeddingProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OllamaEmbeddingProvider.js","sourceRoot":"","sources":["../../src/embedding/OllamaEmbeddingProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AA0BnE;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,uBAAwB,SAAQ,qBAAqB;IACvD,IAAI,GAAG,QAAQ,CAAC;IAER,OAAO,CAAS;IAChB,KAAK,CAAS;IACd,SAAS,CAAS;IAC3B,WAAW,CAAS;IAE5B,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,YAAY,SAAgC,EAAE;QAC5C,KAAK,CAAC,MAAM,CAAC,CAAC;QAEd,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,wBAAwB,CAAC;QAC1D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,kBAAkB,CAAC;QAChD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC;QAE1C,0BAA0B;QAC1B,4BAA4B;QAC5B,oBAAoB;QACpB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChF,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,KAAa;QACxC,MAAM,YAAY,GAA2B;YAC3C,kBAAkB,EAAE,GAAG;YACvB,mBAAmB,EAAE,IAAI;YACzB,YAAY,EAAE,GAAG;YACjB,wBAAwB,EAAE,IAAI;YAC9B,QAAQ,EAAE,IAAI;SACf,CAAC;QAEF,gBAAgB;QAChB,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YACvD,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxB,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAED,OAAO,GAAG,CAAC,CAAC,QAAQ;IACtB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,iBAAiB,CAAC;YAE7C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC1C,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,IAAI;oBACZ,UAAU,EAAE,IAAI,CAAC,SAAS;iBAC3B,CAAC;aACH,EACD,IAAI,CAAC,MAAM,CAAC,OAAO,CACpB,CAAC;YAEF,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA4B,CAAC;YAEhE,oBAAoB;YACpB,IAAI,IAAI,CAAC,WAAW,KAAK,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC9D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;YAC3C,CAAC;YAED,OAAO,IAAI,CAAC,SAAS,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACgB,KAAK,CAAC,kBAAkB,CACzC,KAAe;QAEf,cAAc;QACd,MAAM,WAAW,GAAG,CAAC,CAAC;QACtB,MAAM,OAAO,GAAe,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAEpD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC;YACnD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,CAAC;YAC9C,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAClC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CACtC,CAAC;YAEF,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE;gBAClC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC;YAC7B,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,WAAW,CAAC;QAEvC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE;YAChD,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;QAMH,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAuB,CAAC;QAE3D,qCAAqC;QACrC,OAAO,IAAI,CAAC,MAAM;aACf,MAAM,CACL,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YACxB,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YACxB,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CACzB;aACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB;QACpB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;YACvC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACpD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}

package/dist/embedding/OpenAIEmbeddingProvider.d.ts

@@ -0,0 +1,83 @@
+/**
+ * OpenAI Embedding Provider
+ *
+ * OpenAI API embedding provider
+ *
+ * @requirement REQ-RAG-001
+ * @design DES-KATASHIRO-003-RAG
+ */
+import { BaseEmbeddingProvider } from './BaseEmbeddingProvider.js';
+import type { EmbeddingConfig } from '../types.js';
+/**
+ * OpenAI Embedding設定
+ */
+export interface OpenAIEmbeddingConfig extends EmbeddingConfig {
+    /** APIキー */
+    apiKey?: string;
+    /** ベースURL（カスタムエンドポイント用） */
+    baseUrl?: string;
+    /** モデル名 */
+    model?: string;
+    /** 組織ID */
+    organization?: string;
+    /** エンコーディングフォーマット */
+    encodingFormat?: 'float' | 'base64';
+}
+/**
+ * OpenAI Embeddingプロバイダー
+ *
+ * OpenAI API（またはOpenAI互換API）を使用した埋め込み生成
+ *
+ * @example
+ * ```typescript
+ * // OpenAI API
+ * const provider = new OpenAIEmbeddingProvider({
+ *   apiKey: process.env.OPENAI_API_KEY,
+ *   model: 'text-embedding-3-small',
+ * });
+ *
+ * // OpenAI互換API（vLLM, LocalAI等）
+ * const localProvider = new OpenAIEmbeddingProvider({
+ *   baseUrl: 'http://localhost:8000/v1',
+ *   model: 'local-embedding-model',
+ * });
+ *
+ * const embedding = await provider.embed('Hello, world!');
+ * ```
+ */
+export declare class OpenAIEmbeddingProvider extends BaseEmbeddingProvider {
+    readonly name = "openai";
+    private readonly apiKey;
+    private readonly baseUrl;
+    private readonly model;
+    private readonly organization?;
+    private readonly encodingFormat;
+    private _dimensions;
+    get dimensions(): number;
+    constructor(config?: OpenAIEmbeddingConfig);
+    /**
+     * モデルごとのデフォルト次元数
+     */
+    private getDefaultDimensions;
+    /**
+     * 単一テキストの埋め込み生成
+     */
+    embed(text: string): Promise<number[]>;
+    /**
+     * バッチ埋め込み生成（OpenAI固有実装）
+     *
+     * OpenAI APIはネイティブでバッチ処理をサポート
+     */
+    protected embedBatchInternal(texts: string[]): Promise<number[][]>;
+    /**
+     * 利用可能なモデル一覧取得
+     */
+    listModels(): Promise<string[]>;
+}
+/**
+ * OpenAI互換プロバイダー（ファクトリ関数）
+ *
+ * vLLM, LocalAI, LM Studio等のOpenAI互換エンドポイント用
+ */
+export declare function createOpenAICompatibleEmbeddingProvider(baseUrl: string, model: string, config?: Partial<OpenAIEmbeddingConfig>): OpenAIEmbeddingProvider;
+//# sourceMappingURL=OpenAIEmbeddingProvider.d.ts.map

package/dist/embedding/OpenAIEmbeddingProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenAIEmbeddingProvider.d.ts","sourceRoot":"","sources":["../../src/embedding/OpenAIEmbeddingProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEnD;;GAEG;AACH,MAAM,WAAW,qBAAsB,SAAQ,eAAe;IAC5D,YAAY;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,2BAA2B;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW;IACX,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW;IACX,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,qBAAqB;IACrB,cAAc,CAAC,EAAE,OAAO,GAAG,QAAQ,CAAC;CACrC;AAmBD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,uBAAwB,SAAQ,qBAAqB;IAChE,QAAQ,CAAC,IAAI,YAAY;IAEzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAqB;IACpD,OAAO,CAAC,WAAW,CAAS;IAE5B,IAAI,UAAU,IAAI,MAAM,CAEvB;gBAEW,MAAM,GAAE,qBAA0B;IAa9C;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAU5B;;OAEG;IACG,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAS5C;;;;OAIG;cACsB,kBAAkB,CACzC,KAAK,EAAE,MAAM,EAAE,GACd,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAqDtB;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;CAyBtC;AAED;;;;GAIG;AACH,wBAAgB,uCAAuC,CACrD,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,EACb,MAAM,GAAE,OAAO,CAAC,qBAAqB,CAAM,GAC1C,uBAAuB,CAOzB"}

package/dist/embedding/OpenAIEmbeddingProvider.js

@@ -0,0 +1,150 @@
+/**
+ * OpenAI Embedding Provider
+ *
+ * OpenAI API embedding provider
+ *
+ * @requirement REQ-RAG-001
+ * @design DES-KATASHIRO-003-RAG
+ */
+import { BaseEmbeddingProvider } from './BaseEmbeddingProvider.js';
+/**
+ * OpenAI Embeddingプロバイダー
+ *
+ * OpenAI API（またはOpenAI互換API）を使用した埋め込み生成
+ *
+ * @example
+ * ```typescript
+ * // OpenAI API
+ * const provider = new OpenAIEmbeddingProvider({
+ *   apiKey: process.env.OPENAI_API_KEY,
+ *   model: 'text-embedding-3-small',
+ * });
+ *
+ * // OpenAI互換API（vLLM, LocalAI等）
+ * const localProvider = new OpenAIEmbeddingProvider({
+ *   baseUrl: 'http://localhost:8000/v1',
+ *   model: 'local-embedding-model',
+ * });
+ *
+ * const embedding = await provider.embed('Hello, world!');
+ * ```
+ */
+export class OpenAIEmbeddingProvider extends BaseEmbeddingProvider {
+    name = 'openai';
+    apiKey;
+    baseUrl;
+    model;
+    organization;
+    encodingFormat;
+    _dimensions;
+    get dimensions() {
+        return this._dimensions;
+    }
+    constructor(config = {}) {
+        super(config);
+        this.apiKey = config.apiKey ?? process.env.OPENAI_API_KEY ?? '';
+        this.baseUrl = config.baseUrl ?? 'https://api.openai.com/v1';
+        this.model = config.model ?? 'text-embedding-3-small';
+        this.organization = config.organization;
+        this.encodingFormat = config.encodingFormat ?? 'float';
+        // モデルごとのデフォルト次元数
+        this._dimensions = config.dimensions ?? this.getDefaultDimensions(this.model);
+    }
+    /**
+     * モデルごとのデフォルト次元数
+     */
+    getDefaultDimensions(model) {
+        const dimensionMap = {
+            'text-embedding-3-small': 1536,
+            'text-embedding-3-large': 3072,
+            'text-embedding-ada-002': 1536,
+        };
+        return dimensionMap[model] ?? 1536;
+    }
+    /**
+     * 単一テキストの埋め込み生成
+     */
+    async embed(text) {
+        const embeddings = await this.embedBatchInternal([text]);
+        const result = embeddings[0];
+        if (!result) {
+            throw new Error('Failed to generate embedding');
+        }
+        return result;
+    }
+    /**
+     * バッチ埋め込み生成（OpenAI固有実装）
+     *
+     * OpenAI APIはネイティブでバッチ処理をサポート
+     */
+    async embedBatchInternal(texts) {
+        return this.withRetry(async () => {
+            const url = `${this.baseUrl}/embeddings`;
+            const headers = {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${this.apiKey}`,
+            };
+            if (this.organization) {
+                headers['OpenAI-Organization'] = this.organization;
+            }
+            const body = {
+                model: this.model,
+                input: texts,
+                encoding_format: this.encodingFormat,
+            };
+            // 次元数指定（text-embedding-3系のみサポート）
+            if (this._dimensions &&
+                (this.model.includes('text-embedding-3') ||
+                    this.baseUrl !== 'https://api.openai.com/v1')) {
+                body.dimensions = this._dimensions;
+            }
+            const response = await this.fetchWithTimeout(url, {
+                method: 'POST',
+                headers,
+                body: JSON.stringify(body),
+            }, this.config.timeout);
+            const data = (await response.json());
+            // インデックス順にソート
+            const sortedData = [...data.data].sort((a, b) => a.index - b.index);
+            // 実際の次元数を更新
+            const firstItem = sortedData[0];
+            if (firstItem) {
+                this._dimensions = firstItem.embedding.length;
+            }
+            return sortedData.map((d) => d.embedding);
+        });
+    }
+    /**
+     * 利用可能なモデル一覧取得
+     */
+    async listModels() {
+        const url = `${this.baseUrl}/models`;
+        const headers = {
+            Authorization: `Bearer ${this.apiKey}`,
+        };
+        if (this.organization) {
+            headers['OpenAI-Organization'] = this.organization;
+        }
+        const response = await this.fetchWithTimeout(url, {
+            method: 'GET',
+            headers,
+        });
+        const data = (await response.json());
+        // embeddingモデルをフィルタ
+        return data.data.filter((m) => m.id.includes('embedding')).map((m) => m.id);
+    }
+}
+/**
+ * OpenAI互換プロバイダー（ファクトリ関数）
+ *
+ * vLLM, LocalAI, LM Studio等のOpenAI互換エンドポイント用
+ */
+export function createOpenAICompatibleEmbeddingProvider(baseUrl, model, config = {}) {
+    return new OpenAIEmbeddingProvider({
+        ...config,
+        baseUrl,
+        model,
+        apiKey: config.apiKey ?? 'not-required', // 一部のローカルAPIはキー不要
+    });
+}
+//# sourceMappingURL=OpenAIEmbeddingProvider.js.map

package/dist/embedding/OpenAIEmbeddingProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenAIEmbeddingProvider.js","sourceRoot":"","sources":["../../src/embedding/OpenAIEmbeddingProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAoCnE;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,uBAAwB,SAAQ,qBAAqB;IACvD,IAAI,GAAG,QAAQ,CAAC;IAER,MAAM,CAAS;IACf,OAAO,CAAS;IAChB,KAAK,CAAS;IACd,YAAY,CAAU;IACtB,cAAc,CAAqB;IAC5C,WAAW,CAAS;IAE5B,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,YAAY,SAAgC,EAAE;QAC5C,KAAK,CAAC,MAAM,CAAC,CAAC;QAEd,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC;QAChE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,2BAA2B,CAAC;QAC7D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,wBAAwB,CAAC;QACtD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,OAAO,CAAC;QAEvD,iBAAiB;QACjB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChF,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,KAAa;QACxC,MAAM,YAAY,GAA2B;YAC3C,wBAAwB,EAAE,IAAI;YAC9B,wBAAwB,EAAE,IAAI;YAC9B,wBAAwB,EAAE,IAAI;SAC/B,CAAC;QAEF,OAAO,YAAY,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACgB,KAAK,CAAC,kBAAkB,CACzC,KAAe;QAEf,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,aAAa,CAAC;YAEzC,MAAM,OAAO,GAA2B;gBACtC,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;aACvC,CAAC;YAEF,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,OAAO,CAAC,qBAAqB,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC;YACrD,CAAC;YAED,MAAM,IAAI,GAA4B;gBACpC,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,KAAK;gBACZ,eAAe,EAAE,IAAI,CAAC,cAAc;aACrC,CAAC;YAEF,iCAAiC;YACjC,IACE,IAAI,CAAC,WAAW;gBAChB,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC;oBACtC,IAAI,CAAC,OAAO,KAAK,2BAA2B,CAAC,EAC/C,CAAC;gBACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC;YACrC,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC1C,GAAG,EACH;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO;gBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,IAAI,CAAC,MAAM,CAAC,OAAO,CACpB,CAAC;YAEF,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA4B,CAAC;YAEhE,cAAc;YACd,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;YAEpE,YAAY;YACZ,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAChC,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC;YAChD,CAAC;YAED,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,SAAS,CAAC;QAErC,MAAM,OAAO,GAA2B;YACtC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;SACvC,CAAC;QAEF,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,CAAC,qBAAqB,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC;QACrD,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE;YAChD,MAAM,EAAE,KAAK;YACb,OAAO;SACR,CAAC,CAAC;QAMH,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAE7D,oBAAoB;QACpB,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC9E,CAAC;CACF;AAED;;;;GAIG;AACH,MAAM,UAAU,uCAAuC,CACrD,OAAe,EACf,KAAa,EACb,SAAyC,EAAE;IAE3C,OAAO,IAAI,uBAAuB,CAAC;QACjC,GAAG,MAAM;QACT,OAAO;QACP,KAAK;QACL,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,cAAc,EAAE,kBAAkB;KAC5D,CAAC,CAAC;AACL,CAAC"}

package/dist/embedding/index.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Embedding module exports
+ */
+export { EmbeddingManager } from './EmbeddingManager.js';
+export { BaseEmbeddingProvider } from './BaseEmbeddingProvider.js';
+export { MockEmbeddingProvider } from './MockEmbeddingProvider.js';
+export type { MockEmbeddingProviderConfig } from './MockEmbeddingProvider.js';
+export { OllamaEmbeddingProvider } from './OllamaEmbeddingProvider.js';
+export type { OllamaEmbeddingConfig } from './OllamaEmbeddingProvider.js';
+export { OpenAIEmbeddingProvider, createOpenAICompatibleEmbeddingProvider } from './OpenAIEmbeddingProvider.js';
+export type { OpenAIEmbeddingConfig } from './OpenAIEmbeddingProvider.js';
+export { AzureOpenAIEmbeddingProvider } from './AzureOpenAIEmbeddingProvider.js';
+export type { AzureOpenAIEmbeddingConfig } from './AzureOpenAIEmbeddingProvider.js';
+export { EmbeddingFactory, createEmbeddingProvider, getDefaultEmbeddingProvider, } from './EmbeddingFactory.js';
+export type { EmbeddingProviderType, EmbeddingProviderConfigMap, } from './EmbeddingFactory.js';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/embedding/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAGzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAGnE,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,YAAY,EAAE,2BAA2B,EAAE,MAAM,4BAA4B,CAAC;AAE9E,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AACvE,YAAY,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAE1E,OAAO,EAAE,uBAAuB,EAAE,uCAAuC,EAAE,MAAM,8BAA8B,CAAC;AAChH,YAAY,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAE1E,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,YAAY,EAAE,0BAA0B,EAAE,MAAM,mCAAmC,CAAC;AAGpF,OAAO,EACL,gBAAgB,EAChB,uBAAuB,EACvB,2BAA2B,GAC5B,MAAM,uBAAuB,CAAC;AAC/B,YAAY,EACV,qBAAqB,EACrB,0BAA0B,GAC3B,MAAM,uBAAuB,CAAC"}

package/dist/embedding/index.js

@@ -0,0 +1,15 @@
+/**
+ * Embedding module exports
+ */
+// Manager
+export { EmbeddingManager } from './EmbeddingManager.js';
+// Base
+export { BaseEmbeddingProvider } from './BaseEmbeddingProvider.js';
+// Providers
+export { MockEmbeddingProvider } from './MockEmbeddingProvider.js';
+export { OllamaEmbeddingProvider } from './OllamaEmbeddingProvider.js';
+export { OpenAIEmbeddingProvider, createOpenAICompatibleEmbeddingProvider } from './OpenAIEmbeddingProvider.js';
+export { AzureOpenAIEmbeddingProvider } from './AzureOpenAIEmbeddingProvider.js';
+// Factory
+export { EmbeddingFactory, createEmbeddingProvider, getDefaultEmbeddingProvider, } from './EmbeddingFactory.js';
+//# sourceMappingURL=index.js.map

package/dist/embedding/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/embedding/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,UAAU;AACV,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,OAAO;AACP,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,YAAY;AACZ,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAGnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAGvE,OAAO,EAAE,uBAAuB,EAAE,uCAAuC,EAAE,MAAM,8BAA8B,CAAC;AAGhH,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAGjF,UAAU;AACV,OAAO,EACL,gBAAgB,EAChB,uBAAuB,EACvB,2BAA2B,GAC5B,MAAM,uBAAuB,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,20 @@
+/**
+ * RAG Package - Main Entry Point
+ *
+ * @packageDocumentation
+ * @module @nahisaho/katashiro-rag
+ */
+export type { Document, Chunk, SearchResult, RetrievalResult, Vector, EmbeddingProvider, VectorStore, ChunkingConfig, RetrieverConfig, RAGEngineConfig, } from './types.js';
+export { EmbeddingManager } from './embedding/index.js';
+export { MockEmbeddingProvider } from './embedding/index.js';
+export type { MockEmbeddingProviderConfig } from './embedding/index.js';
+export { InMemoryVectorStore } from './vectordb/index.js';
+export type { InMemoryVectorStoreConfig } from './vectordb/index.js';
+export { DocumentChunker } from './chunking/index.js';
+export { Retriever } from './Retriever.js';
+export { RAGEngine } from './RAGEngine.js';
+export { RAGPipeline, DEFAULT_RAG_SYSTEM_PROMPT, DEFAULT_CONTEXT_TEMPLATE } from './RAGPipeline.js';
+export type { RAGPipelineConfig, RAGPipelineResult, RAGQueryOptions } from './RAGPipeline.js';
+export { LLMReranker, rerankResults, DEFAULT_RERANKER_CONFIG, DEFAULT_RERANKER_PROMPT, } from './reranking/index.js';
+export type { LLMRerankerConfig, RerankedResult, RerankerResponse, } from './reranking/index.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,YAAY,EACV,QAAQ,EACR,KAAK,EACL,YAAY,EACZ,eAAe,EACf,MAAM,EACN,iBAAiB,EACjB,WAAW,EACX,cAAc,EACd,eAAe,EACf,eAAe,GAChB,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,YAAY,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AAGxE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,YAAY,EAAE,yBAAyB,EAAE,MAAM,qBAAqB,CAAC;AAGrE,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAGtD,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAG3C,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAG3C,OAAO,EAAE,WAAW,EAAE,yBAAyB,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AACpG,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAG9F,OAAO,EACL,WAAW,EACX,aAAa,EACb,uBAAuB,EACvB,uBAAuB,GACxB,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EACV,iBAAiB,EACjB,cAAc,EACd,gBAAgB,GACjB,MAAM,sBAAsB,CAAC"}

package/dist/index.js

@@ -0,0 +1,22 @@
+/**
+ * RAG Package - Main Entry Point
+ *
+ * @packageDocumentation
+ * @module @nahisaho/katashiro-rag
+ */
+// Embedding
+export { EmbeddingManager } from './embedding/index.js';
+export { MockEmbeddingProvider } from './embedding/index.js';
+// Vector Store
+export { InMemoryVectorStore } from './vectordb/index.js';
+// Chunking
+export { DocumentChunker } from './chunking/index.js';
+// Retriever
+export { Retriever } from './Retriever.js';
+// RAG Engine
+export { RAGEngine } from './RAGEngine.js';
+// RAG Pipeline (REQ-RAG-101)
+export { RAGPipeline, DEFAULT_RAG_SYSTEM_PROMPT, DEFAULT_CONTEXT_TEMPLATE } from './RAGPipeline.js';
+// Reranking (REQ-RAG-103)
+export { LLMReranker, rerankResults, DEFAULT_RERANKER_CONFIG, DEFAULT_RERANKER_PROMPT, } from './reranking/index.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAgBH,YAAY;AACZ,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAG7D,eAAe;AACf,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAG1D,WAAW;AACX,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,YAAY;AACZ,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAE3C,aAAa;AACb,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAE3C,6BAA6B;AAC7B,OAAO,EAAE,WAAW,EAAE,yBAAyB,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AAGpG,0BAA0B;AAC1B,OAAO,EACL,WAAW,EACX,aAAa,EACb,uBAAuB,EACvB,uBAAuB,GACxB,MAAM,sBAAsB,CAAC"}