@reftrixmcp/mcp-server 0.1.7

package/dist/services/part/part-search.service.d.ts

@@ -0,0 +1,180 @@
+/**
+ * EmbeddingServiceインターフェース
+ * EmbeddingService interface
+ */
+export interface PartSearchEmbeddingService {
+    generateEmbedding(text: string, type: "query" | "passage"): Promise<number[]>;
+}
+/**
+ * PrismaClientインターフェース（部分的）
+ * PrismaClient interface (partial)
+ */
+export interface PartSearchPrismaClient {
+    $queryRawUnsafe: <T>(query: string, ...values: unknown[]) => Promise<T>;
+}
+/**
+ * パーツ検索オプション
+ * Part search options
+ */
+export interface PartSearchOptions {
+    /** 返却件数 / Result limit (default 10) */
+    limit: number;
+    /** オフセット / Offset (default 0) */
+    offset: number;
+    /** 最小類似度閾値 / Minimum similarity threshold (default 0.3) */
+    minSimilarity: number;
+    /** パーツタイプフィルタ / Part type filter */
+    partType?: string;
+    /** セクションタイプフィルタ（section_patterns結合） / Section type filter (via section_patterns join) */
+    sectionType?: string;
+    /** CSSフレームワークフィルタ（section_patterns.css_framework） / CSS framework filter */
+    cssFramework?: string;
+    /** 検索モード / Search mode (default 'hybrid') */
+    searchMode: "visual" | "text" | "hybrid";
+}
+/**
+ * 検索結果アイテム
+ * Search result item
+ */
+export interface PartSearchResultItem {
+    id: string;
+    partType: string;
+    partSubtype: string | null;
+    sectionType: string;
+    webPageUrl: string;
+    similarity: number;
+    visualSimilarity?: number;
+    textSimilarity?: number;
+    boundingBox: Record<string, unknown>;
+    computedStyles?: Record<string, string>;
+    htmlSnippet?: string;
+}
+/**
+ * 検索結果
+ * Search result
+ */
+export interface PartSearchResult {
+    results: PartSearchResultItem[];
+    total: number;
+    query: {
+        text?: string;
+        referencePartId?: string;
+    };
+}
+/**
+ * PartSearchServiceInterface インターフェース
+ * PartSearchServiceInterface interface
+ */
+export interface PartSearchServiceInterface {
+    generateQueryEmbedding(query: string): Promise<number[] | null>;
+    searchParts(embedding: number[], options: PartSearchOptions): Promise<PartSearchResult>;
+    searchPartsHybrid(queryText: string, embedding: number[], options: PartSearchOptions): Promise<PartSearchResult>;
+    searchPartsByVisual(referencePartId: string, options: PartSearchOptions): Promise<PartSearchResult>;
+}
+/**
+ * EmbeddingServiceファクトリを設定
+ * Set EmbeddingService factory
+ */
+export declare function setPartSearchEmbeddingServiceFactory(factory: () => PartSearchEmbeddingService): void;
+/**
+ * EmbeddingServiceファクトリをリセット
+ * Reset EmbeddingService factory
+ */
+export declare function resetPartSearchEmbeddingServiceFactory(): void;
+/**
+ * PrismaClientファクトリを設定
+ * Set PrismaClient factory
+ */
+export declare function setPartSearchPrismaClientFactory(factory: () => PartSearchPrismaClient): void;
+/**
+ * PrismaClientファクトリをリセット
+ * Reset PrismaClient factory
+ */
+export declare function resetPartSearchPrismaClientFactory(): void;
+/**
+ * WHERE句構築結果
+ * WHERE clause build result
+ */
+interface BuildWhereResult {
+    clause: string;
+    params: unknown[];
+    nextIndex: number;
+}
+/**
+ * フィルター条件をWHERE句に変換
+ * Convert filter conditions to WHERE clause
+ */
+export declare function buildPartSearchWhereClause(options: Pick<PartSearchOptions, "partType" | "sectionType" | "cssFramework">, startIndex?: number): BuildWhereResult;
+/**
+ * PartSearchServiceクラス
+ * PartSearchService class
+ *
+ * コンポーネントパーツのベクトル検索、全文検索、ハイブリッド検索を提供。
+ * Provides vector, full-text, and hybrid search for component parts.
+ */
+export declare class PartSearchService implements PartSearchServiceInterface {
+    private embeddingService;
+    private prismaClient;
+    /**
+     * EmbeddingServiceを取得
+     * Get EmbeddingService
+     */
+    private getEmbeddingService;
+    /**
+     * PrismaClientを取得
+     * Get PrismaClient
+     */
+    private getPrismaClient;
+    /**
+     * クエリテキストからEmbeddingを生成
+     * Generate embedding from query text
+     *
+     * EmbeddingServiceが利用できない場合はnullを返す。
+     * Returns null if EmbeddingService is not available.
+     */
+    generateQueryEmbedding(query: string): Promise<number[] | null>;
+    /**
+     * テキストベクトル検索（e5-base text_embedding）
+     * Text vector search (e5-base text_embedding)
+     */
+    searchParts(embedding: number[], options: PartSearchOptions): Promise<PartSearchResult>;
+    /**
+     * ハイブリッド検索: テキストベクトル + 全文検索をRRFで統合
+     * Hybrid search: text vector + full-text search merged with RRF
+     *
+     * 両検索を並列実行し、Reciprocal Rank Fusion (60% vector + 40% fulltext) で
+     * 結果をマージする。全文検索が失敗した場合はベクトル検索のみで結果を返す。
+     *
+     * Both searches run in parallel, merged with RRF (60% vector + 40% fulltext).
+     * Falls back to vector-only if full-text search fails.
+     */
+    searchPartsHybrid(queryText: string, embedding: number[], options: PartSearchOptions): Promise<PartSearchResult>;
+    /**
+     * 参照パーツIDによるビジュアル類似検索
+     * Visual similarity search by reference part ID
+     *
+     * 参照パーツのvisual_embeddingをDBから取得し、
+     * それをクエリベクトルとしてDINOv2 HNSWビジュアル検索を実行する。
+     *
+     * Fetches the reference part's visual_embedding from DB and
+     * uses it as query vector for DINOv2 HNSW visual search.
+     */
+    searchPartsByVisual(referencePartId: string, options: PartSearchOptions): Promise<PartSearchResult>;
+}
+/**
+ * PartSearchServiceインスタンスを取得
+ * Get PartSearchService instance
+ */
+export declare function getPartSearchService(): PartSearchService;
+/**
+ * PartSearchServiceインスタンスをリセット
+ * Reset PartSearchService instance
+ */
+export declare function resetPartSearchService(): void;
+/**
+ * PartSearchServiceファクトリを作成
+ * Create PartSearchService factory
+ */
+export declare function createPartSearchServiceFactory(): () => PartSearchServiceInterface;
+export default PartSearchService;
+//# sourceMappingURL=part-search.service.d.ts.map

package/dist/services/part/part-search.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"part-search.service.d.ts","sourceRoot":"","sources":["../../../src/services/part/part-search.service.ts"],"names":[],"mappings":"AAwCA;;;GAGG;AACH,MAAM,WAAW,0BAA0B;IACzC,iBAAiB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CAC/E;AAED;;;GAGG;AACH,MAAM,WAAW,sBAAsB;IACrC,eAAe,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;CACzE;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,uCAAuC;IACvC,KAAK,EAAE,MAAM,CAAC;IACd,iCAAiC;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,2DAA2D;IAC3D,aAAa,EAAE,MAAM,CAAC;IACtB,oCAAoC;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,yFAAyF;IACzF,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,4EAA4E;IAC5E,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,6CAA6C;IAC7C,UAAU,EAAE,QAAQ,GAAG,MAAM,GAAG,QAAQ,CAAC;CAC1C;AAED;;;GAGG;AACH,MAAM,WAAW,oBAAoB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACxC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;;GAGG;AACH,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,oBAAoB,EAAE,CAAC;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,eAAe,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CACpD;AAED;;;GAGG;AACH,MAAM,WAAW,0BAA0B;IACzC,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC;IAChE,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACxF,iBAAiB,CACf,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EAAE,EACnB,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC7B,mBAAmB,CACjB,eAAe,EAAE,MAAM,EACvB,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,gBAAgB,CAAC,CAAC;CAC9B;AA6BD;;;GAGG;AACH,wBAAgB,oCAAoC,CAClD,OAAO,EAAE,MAAM,0BAA0B,GACxC,IAAI,CAEN;AAED;;;GAGG;AACH,wBAAgB,sCAAsC,IAAI,IAAI,CAE7D;AAED;;;GAGG;AACH,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,MAAM,sBAAsB,GAAG,IAAI,CAE5F;AAED;;;GAGG;AACH,wBAAgB,kCAAkC,IAAI,IAAI,CAEzD;AAMD;;;GAGG;AACH,UAAU,gBAAgB;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,OAAO,EAAE,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE,UAAU,GAAG,aAAa,GAAG,cAAc,CAAC,EAC7E,UAAU,GAAE,MAAU,GACrB,gBAAgB,CA4BlB;AAsDD;;;;;;GAMG;AACH,qBAAa,iBAAkB,YAAW,0BAA0B;IAClE,OAAO,CAAC,gBAAgB,CAA2C;IACnE,OAAO,CAAC,YAAY,CAAuC;IAE3D;;;OAGG;IACH,OAAO,CAAC,mBAAmB;IAa3B;;;OAGG;IACH,OAAO,CAAC,eAAe;IAavB;;;;;;OAMG;IACG,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;IAyBrE;;;OAGG;IACG,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAqG7F;;;;;;;;;OASG;IACG,iBAAiB,CACrB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EAAE,EACnB,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,gBAAgB,CAAC;IAyJ5B;;;;;;;;;OASG;IACG,mBAAmB,CACvB,eAAe,EAAE,MAAM,EACvB,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,gBAAgB,CAAC;CAyH7B;AAQD;;;GAGG;AACH,wBAAgB,oBAAoB,IAAI,iBAAiB,CAKxD;AAED;;;GAGG;AACH,wBAAgB,sBAAsB,IAAI,IAAI,CAE7C;AAED;;;GAGG;AACH,wBAAgB,8BAA8B,IAAI,MAAM,0BAA0B,CAEjF;AAED,eAAe,iBAAiB,CAAC"}