@memberjunction/query-gen 0.0.1 → 2.126.1

package/dist/vectors/SimilaritySearch.js

@@ -0,0 +1,85 @@
+"use strict";
+/**
+ * SimilaritySearch - Finds similar golden queries using weighted cosine similarity
+ *
+ * Implements weighted similarity search across multiple fields (name, userQuestion,
+ * description, technicalDescription) to find the most relevant few-shot examples.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SimilaritySearch = void 0;
+const ai_vectors_memory_1 = require("@memberjunction/ai-vectors-memory");
+/**
+ * SimilaritySearch class
+ * Finds most similar golden queries using weighted cosine similarity across multiple fields
+ */
+class SimilaritySearch extends ai_vectors_memory_1.SimpleVectorService {
+    /**
+     * Weights for each field in similarity calculation
+     * Total weights sum to 1.0
+     *
+     * Weight distribution prioritizes technical specifications:
+     * - userQuestion: 0.20 (less important - natural language is variable)
+     * - description: 0.40 (high-level business logic matching)
+     * - technicalDescription: 0.40 (technical implementation details)
+     */
+    weights = {
+        userQuestion: 0.20,
+        description: 0.40,
+        technicalDescription: 0.40
+    };
+    /**
+     * Find similar golden queries using weighted cosine similarity
+     *
+     * Calculates similarity for each field separately, then combines with weights.
+     * ALWAYS returns topK results, even if below threshold (threshold is informational).
+     *
+     * @param queryEmbeddings - Embeddings for the user's query (one per field)
+     * @param goldenEmbeddings - Array of golden queries with their embeddings
+     * @param topK - Number of most similar queries to return (default: 5)
+     * @returns Array of top-K most similar golden queries with scores
+     */
+    async findSimilarQueries(queryEmbeddings, goldenEmbeddings, topK = 5) {
+        // Calculate weighted similarity for each golden query
+        const similarities = goldenEmbeddings.map(golden => {
+            const fieldScores = this.calculateFieldSimilarities(queryEmbeddings, golden.embeddings);
+            const weightedScore = this.calculateWeightedScore(fieldScores);
+            return {
+                query: golden.query,
+                similarity: weightedScore,
+                fieldScores
+            };
+        });
+        // Sort by similarity (highest first) and return top K
+        return this.selectTopK(similarities, topK);
+    }
+    /**
+     * Calculate cosine similarity for each field separately
+     * Note: name field excluded as queries don't have names during generation
+     */
+    calculateFieldSimilarities(queryEmbeddings, goldenEmbeddings) {
+        return {
+            userQuestionSim: this.CosineSimilarity(queryEmbeddings.userQuestion, goldenEmbeddings.userQuestion),
+            descSim: this.CosineSimilarity(queryEmbeddings.description, goldenEmbeddings.description),
+            techDescSim: this.CosineSimilarity(queryEmbeddings.technicalDescription, goldenEmbeddings.technicalDescription)
+        };
+    }
+    /**
+     * Calculate weighted sum of field similarities
+     * Note: name field excluded, weights adjusted accordingly
+     */
+    calculateWeightedScore(fieldScores) {
+        return (fieldScores.userQuestionSim * this.weights.userQuestion +
+            fieldScores.descSim * this.weights.description +
+            fieldScores.techDescSim * this.weights.technicalDescription);
+    }
+    /**
+     * Select top K results sorted by similarity
+     */
+    selectTopK(similarities, topK) {
+        return similarities
+            .sort((a, b) => b.similarity - a.similarity)
+            .slice(0, topK);
+    }
+}
+exports.SimilaritySearch = SimilaritySearch;
+//# sourceMappingURL=SimilaritySearch.js.map

package/dist/vectors/SimilaritySearch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SimilaritySearch.js","sourceRoot":"","sources":["../../src/vectors/SimilaritySearch.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAGH,yEAAwE;AAYxE;;;GAGG;AACH,MAAa,gBAAiB,SAAQ,uCAAmB;IACvD;;;;;;;;OAQG;IACc,OAAO,GAAiB;QACvC,YAAY,EAAE,IAAI;QAClB,WAAW,EAAE,IAAI;QACjB,oBAAoB,EAAE,IAAI;KAC3B,CAAC;IAEF;;;;;;;;;;OAUG;IACH,KAAK,CAAC,kBAAkB,CACtB,eAAgC,EAChC,gBAAuC,EACvC,OAAe,CAAC;QAEhB,sDAAsD;QACtD,MAAM,YAAY,GAAG,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;YACjD,MAAM,WAAW,GAAG,IAAI,CAAC,0BAA0B,CAAC,eAAe,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;YACxF,MAAM,aAAa,GAAG,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;YAE/D,OAAO;gBACL,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,UAAU,EAAE,aAAa;gBACzB,WAAW;aACZ,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,sDAAsD;QACtD,OAAO,IAAI,CAAC,UAAU,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAC7C,CAAC;IAED;;;OAGG;IACK,0BAA0B,CAChC,eAAgC,EAChC,gBAAiC;QAEjC,OAAO;YACL,eAAe,EAAE,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,YAAY,EAAE,gBAAgB,CAAC,YAAY,CAAC;YACnG,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,WAAW,EAAE,gBAAgB,CAAC,WAAW,CAAC;YACzF,WAAW,EAAE,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,oBAAoB,EAAE,gBAAgB,CAAC,oBAAoB,CAAC;SAChH,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,sBAAsB,CAAC,WAAwC;QACrE,OAAO,CACL,WAAW,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY;YACvD,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW;YAC9C,WAAW,CAAC,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAC5D,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,YAA4B,EAAE,IAAY;QAC3D,OAAO,YAAY;aAChB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC;aAC3C,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACpB,CAAC;CACF;AAnFD,4CAmFC"}