@nahisaho/musubix-neural-search 2.1.1 → 2.2.0

package/dist/scorer/ContextAwareScorer.js

@@ -0,0 +1,281 @@
+/**
+ * ContextAwareScorer - Project Context-Aware Branch Scoring
+ * @module @nahisaho/musubix-neural-search
+ * @see TSK-NS-103
+ * @see DES-NS-103
+ * @see REQ-NS-103
+ *
+ * プロジェクトコンテキストを30%以上の重みで反映するスコアラー
+ * - プロジェクトパターンの検出
+ * - コードベース埋め込みとの類似度
+ * - 詳細なスコア内訳の提供
+ */
+import { BranchScorer } from './BranchScorer.js';
+// =============================================================================
+// Implementation
+// =============================================================================
+/**
+ * Default ContextAwareScorer implementation
+ */
+export class DefaultContextAwareScorer {
+    config;
+    baseScorer;
+    projectContext = null;
+    feedbackHistory = new Map();
+    // Statistics
+    totalScored = 0;
+    scoreSum = 0;
+    confidenceSum = 0;
+    contextContributionSum = 0;
+    // Cache for breakdowns
+    breakdownCache = new WeakMap();
+    constructor(config = {}) {
+        // Ensure context weight is at least 30%
+        const contextWeight = Math.max(config.contextWeight ?? 0.3, 0.3);
+        this.config = {
+            contextWeight,
+            enableProjectContext: config.enableProjectContext ?? true,
+            baseWeights: config.baseWeights ?? {},
+            patternMatchBonus: config.patternMatchBonus ?? 0.1,
+        };
+        this.baseScorer = new BranchScorer(undefined, this.config.baseWeights);
+    }
+    // =========================================================================
+    // IBranchScorer Implementation
+    // =========================================================================
+    async score(branch, context) {
+        // Get base score
+        const baseResult = await this.baseScorer.score(branch, context);
+        const baseScore = baseResult.score;
+        // Compute context score
+        const contextScore = this.computeContextScore(branch, context);
+        // Combine scores
+        const effectiveContextWeight = this.config.enableProjectContext
+            ? this.config.contextWeight
+            : 0;
+        const finalScore = baseScore * (1 - effectiveContextWeight) +
+            contextScore * effectiveContextWeight;
+        // Update statistics
+        this.updateStatistics(finalScore, baseResult.confidence, contextScore * effectiveContextWeight);
+        // Create result
+        const result = {
+            branch,
+            score: finalScore,
+            confidence: baseResult.confidence,
+            components: baseResult.components,
+            _contextScore: contextScore,
+            _baseScore: baseScore,
+        };
+        // Cache breakdown
+        this.breakdownCache.set(result, {
+            baseScore,
+            contextScore,
+            contextWeight: effectiveContextWeight,
+            finalScore,
+            patternMatches: this.findPatternMatches(branch),
+        });
+        return result;
+    }
+    async scoreBatch(branches, context) {
+        return Promise.all(branches.map((branch) => this.score(branch, context)));
+    }
+    update(feedback) {
+        const history = this.feedbackHistory.get(feedback.branchId) ?? [];
+        history.push(feedback);
+        this.feedbackHistory.set(feedback.branchId, history);
+        // Also update base scorer
+        this.baseScorer.update(feedback);
+    }
+    // =========================================================================
+    // Project Context
+    // =========================================================================
+    setProjectContext(context) {
+        this.projectContext = context;
+    }
+    // =========================================================================
+    // Score Breakdown
+    // =========================================================================
+    getScoreBreakdown(score) {
+        // Check cache
+        const cached = this.breakdownCache.get(score);
+        if (cached) {
+            return cached;
+        }
+        // Reconstruct breakdown if not cached
+        const extendedScore = score;
+        const baseScore = extendedScore._baseScore ?? score.score;
+        const contextScore = extendedScore._contextScore ?? 0;
+        const contextWeight = this.config.enableProjectContext
+            ? this.config.contextWeight
+            : 0;
+        return {
+            baseScore,
+            contextScore,
+            contextWeight,
+            finalScore: score.score,
+        };
+    }
+    // =========================================================================
+    // Statistics
+    // =========================================================================
+    getStatistics() {
+        return {
+            totalScored: this.totalScored,
+            averageScore: this.totalScored > 0 ? this.scoreSum / this.totalScored : 0,
+            averageConfidence: this.totalScored > 0 ? this.confidenceSum / this.totalScored : 0,
+            averageContextContribution: this.totalScored > 0
+                ? this.contextContributionSum / this.totalScored
+                : 0,
+        };
+    }
+    resetStatistics() {
+        this.totalScored = 0;
+        this.scoreSum = 0;
+        this.confidenceSum = 0;
+        this.contextContributionSum = 0;
+    }
+    // =========================================================================
+    // Serialization
+    // =========================================================================
+    toJSON() {
+        return JSON.stringify({
+            config: {
+                contextWeight: this.config.contextWeight,
+                enableProjectContext: this.config.enableProjectContext,
+                patternMatchBonus: this.config.patternMatchBonus,
+            },
+            statistics: this.getStatistics(),
+            projectContext: this.projectContext
+                ? {
+                    projectName: this.projectContext.projectName,
+                    language: this.projectContext.language,
+                    patterns: this.projectContext.patterns,
+                }
+                : null,
+        });
+    }
+    fromJSON(json) {
+        const data = JSON.parse(json);
+        if (data.config) {
+            if (data.config.contextWeight !== undefined) {
+                this.config.contextWeight = Math.max(data.config.contextWeight, 0.3);
+            }
+            if (data.config.enableProjectContext !== undefined) {
+                this.config.enableProjectContext = data.config.enableProjectContext;
+            }
+            if (data.config.patternMatchBonus !== undefined) {
+                this.config.patternMatchBonus = data.config.patternMatchBonus;
+            }
+        }
+        if (data.statistics) {
+            this.totalScored = data.statistics.totalScored ?? 0;
+            this.scoreSum =
+                (data.statistics.averageScore ?? 0) * this.totalScored;
+            this.confidenceSum =
+                (data.statistics.averageConfidence ?? 0) * this.totalScored;
+            this.contextContributionSum =
+                (data.statistics.averageContextContribution ?? 0) * this.totalScored;
+        }
+    }
+    // =========================================================================
+    // Private Methods
+    // =========================================================================
+    computeContextScore(branch, context) {
+        if (!this.config.enableProjectContext || !this.projectContext) {
+            return 0.5; // Neutral score when no context
+        }
+        let score = 0.5; // Base
+        // Pattern matching bonus
+        const patternMatches = this.findPatternMatches(branch);
+        if (patternMatches.length > 0) {
+            score += this.config.patternMatchBonus * patternMatches.length;
+        }
+        // Codebase embedding similarity
+        if (this.projectContext.codebaseEmbedding) {
+            const similarity = this.cosineSimilarity(branch.features.codeEmbedding, this.projectContext.codebaseEmbedding);
+            score += similarity * 0.2;
+        }
+        // Language alignment
+        if (this.isLanguageAligned(branch, context)) {
+            score += 0.1;
+        }
+        // Domain keyword matching
+        if (this.projectContext.domainKeywords) {
+            const keywordScore = this.computeKeywordScore(branch, context);
+            score += keywordScore * 0.1;
+        }
+        // Clamp to [0, 1]
+        return Math.max(0, Math.min(1, score));
+    }
+    findPatternMatches(branch) {
+        if (!this.projectContext) {
+            return [];
+        }
+        const matches = [];
+        const action = branch.action.toLowerCase();
+        for (const pattern of this.projectContext.patterns) {
+            if (action.includes(pattern.toLowerCase()) ||
+                pattern.toLowerCase().includes(action)) {
+                matches.push(pattern);
+            }
+        }
+        return matches;
+    }
+    cosineSimilarity(a, b) {
+        if (a.length !== b.length || a.length === 0) {
+            return 0;
+        }
+        let dotProduct = 0;
+        let normA = 0;
+        let normB = 0;
+        for (let i = 0; i < a.length; i++) {
+            dotProduct += a[i] * b[i];
+            normA += a[i] * a[i];
+            normB += b[i] * b[i];
+        }
+        const denominator = Math.sqrt(normA) * Math.sqrt(normB);
+        if (denominator === 0) {
+            return 0;
+        }
+        return dotProduct / denominator;
+    }
+    isLanguageAligned(_branch, _context) {
+        if (!this.projectContext) {
+            return false;
+        }
+        // Simple heuristic: assume TypeScript/JavaScript projects
+        const lang = this.projectContext.language.toLowerCase();
+        return lang === 'typescript' || lang === 'javascript';
+    }
+    computeKeywordScore(branch, _context) {
+        if (!this.projectContext?.domainKeywords) {
+            return 0;
+        }
+        const code = branch.to.code.toLowerCase();
+        let matches = 0;
+        for (const keyword of this.projectContext.domainKeywords) {
+            if (code.includes(keyword.toLowerCase())) {
+                matches++;
+            }
+        }
+        return Math.min(1, matches / Math.max(1, this.projectContext.domainKeywords.length));
+    }
+    updateStatistics(score, confidence, contextContribution) {
+        this.totalScored++;
+        this.scoreSum += score;
+        this.confidenceSum += confidence;
+        this.contextContributionSum += contextContribution;
+    }
+}
+// =============================================================================
+// Factory Function
+// =============================================================================
+/**
+ * Create a ContextAwareScorer instance
+ * @param config - Optional configuration
+ * @returns ContextAwareScorer instance
+ */
+export function createContextAwareScorer(config = {}) {
+    return new DefaultContextAwareScorer(config);
+}
+//# sourceMappingURL=ContextAwareScorer.js.map

package/dist/scorer/ContextAwareScorer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContextAwareScorer.js","sourceRoot":"","sources":["../../src/scorer/ContextAwareScorer.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAUH,OAAO,EAAE,YAAY,EAAqB,MAAM,mBAAmB,CAAC;AA8FpE,gFAAgF;AAChF,iBAAiB;AACjB,gFAAgF;AAEhF;;GAEG;AACH,MAAM,OAAO,yBAAyB;IAC5B,MAAM,CAAqC;IAC3C,UAAU,CAAe;IACzB,cAAc,GAA0B,IAAI,CAAC;IAC7C,eAAe,GAAiC,IAAI,GAAG,EAAE,CAAC;IAElE,aAAa;IACL,WAAW,GAAG,CAAC,CAAC;IAChB,QAAQ,GAAG,CAAC,CAAC;IACb,aAAa,GAAG,CAAC,CAAC;IAClB,sBAAsB,GAAG,CAAC,CAAC;IAEnC,uBAAuB;IACf,cAAc,GAAyC,IAAI,OAAO,EAAE,CAAC;IAE7E,YAAY,SAAmC,EAAE;QAC/C,wCAAwC;QACxC,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,GAAG,EAAE,GAAG,CAAC,CAAC;QAEjE,IAAI,CAAC,MAAM,GAAG;YACZ,aAAa;YACb,oBAAoB,EAAE,MAAM,CAAC,oBAAoB,IAAI,IAAI;YACzD,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,EAAE;YACrC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,GAAG;SACnD,CAAC;QAEF,IAAI,CAAC,UAAU,GAAG,IAAI,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IACzE,CAAC;IAED,4EAA4E;IAC5E,+BAA+B;IAC/B,4EAA4E;IAE5E,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,OAAsB;QAChD,iBAAiB;QACjB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChE,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC;QAEnC,wBAAwB;QACxB,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAE/D,iBAAiB;QACjB,MAAM,sBAAsB,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB;YAC7D,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa;YAC3B,CAAC,CAAC,CAAC,CAAC;QAEN,MAAM,UAAU,GACd,SAAS,GAAG,CAAC,CAAC,GAAG,sBAAsB,CAAC;YACxC,YAAY,GAAG,sBAAsB,CAAC;QAExC,oBAAoB;QACpB,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,UAAU,CAAC,UAAU,EAAE,YAAY,GAAG,sBAAsB,CAAC,CAAC;QAEhG,gBAAgB;QAChB,MAAM,MAAM,GAAwB;YAClC,MAAM;YACN,KAAK,EAAE,UAAU;YACjB,UAAU,EAAE,UAAU,CAAC,UAAU;YACjC,UAAU,EAAE,UAAU,CAAC,UAAU;YACjC,aAAa,EAAE,YAAY;YAC3B,UAAU,EAAE,SAAS;SACtB,CAAC;QAEF,kBAAkB;QAClB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,EAAE;YAC9B,SAAS;YACT,YAAY;YACZ,aAAa,EAAE,sBAAsB;YACrC,UAAU;YACV,cAAc,EAAE,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC;SAChD,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,UAAU,CACd,QAAkB,EAClB,OAAsB;QAEtB,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAC5E,CAAC;IAED,MAAM,CAAC,QAAuB;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAClE,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAErD,0BAA0B;QAC1B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,4EAA4E;IAC5E,kBAAkB;IAClB,4EAA4E;IAE5E,iBAAiB,CAAC,OAAuB;QACvC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;IAChC,CAAC;IAED,4EAA4E;IAC5E,kBAAkB;IAClB,4EAA4E;IAE5E,iBAAiB,CAAC,KAAkB;QAClC,cAAc;QACd,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,sCAAsC;QACtC,MAAM,aAAa,GAAG,KAA4B,CAAC;QACnD,MAAM,SAAS,GAAG,aAAa,CAAC,UAAU,IAAI,KAAK,CAAC,KAAK,CAAC;QAC1D,MAAM,YAAY,GAAG,aAAa,CAAC,aAAa,IAAI,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB;YACpD,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa;YAC3B,CAAC,CAAC,CAAC,CAAC;QAEN,OAAO;YACL,SAAS;YACT,YAAY;YACZ,aAAa;YACb,UAAU,EAAE,KAAK,CAAC,KAAK;SACxB,CAAC;IACJ,CAAC;IAED,4EAA4E;IAC5E,aAAa;IACb,4EAA4E;IAE5E,aAAa;QACX,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,YAAY,EAAE,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACzE,iBAAiB,EACf,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YAClE,0BAA0B,EACxB,IAAI,CAAC,WAAW,GAAG,CAAC;gBAClB,CAAC,CAAC,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,WAAW;gBAChD,CAAC,CAAC,CAAC;SACR,CAAC;IACJ,CAAC;IAED,eAAe;QACb,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;QAClB,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;QACvB,IAAI,CAAC,sBAAsB,GAAG,CAAC,CAAC;IAClC,CAAC;IAED,4EAA4E;IAC5E,gBAAgB;IAChB,4EAA4E;IAE5E,MAAM;QACJ,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,MAAM,EAAE;gBACN,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;gBACxC,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB;gBACtD,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB;aACjD;YACD,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE;YAChC,cAAc,EAAE,IAAI,CAAC,cAAc;gBACjC,CAAC,CAAC;oBACE,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,WAAW;oBAC5C,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ;oBACtC,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ;iBACvC;gBACH,CAAC,CAAC,IAAI;SACT,CAAC,CAAC;IACL,CAAC;IAED,QAAQ,CAAC,IAAY;QACnB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAE9B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;gBAC5C,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;YACvE,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,KAAK,SAAS,EAAE,CAAC;gBACnD,IAAI,CAAC,MAAM,CAAC,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC;YACtE,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;gBAChD,IAAI,CAAC,MAAM,CAAC,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAChE,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,IAAI,CAAC,CAAC;YACpD,IAAI,CAAC,QAAQ;gBACX,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC;YACzD,IAAI,CAAC,aAAa;gBAChB,CAAC,IAAI,CAAC,UAAU,CAAC,iBAAiB,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC;YAC9D,IAAI,CAAC,sBAAsB;gBACzB,CAAC,IAAI,CAAC,UAAU,CAAC,0BAA0B,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC;QACzE,CAAC;IACH,CAAC;IAED,4EAA4E;IAC5E,kBAAkB;IAClB,4EAA4E;IAEpE,mBAAmB,CAAC,MAAc,EAAE,OAAsB;QAChE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAC9D,OAAO,GAAG,CAAC,CAAC,gCAAgC;QAC9C,CAAC;QAED,IAAI,KAAK,GAAG,GAAG,CAAC,CAAC,OAAO;QAExB,yBAAyB;QACzB,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QACvD,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,GAAG,cAAc,CAAC,MAAM,CAAC;QACjE,CAAC;QAED,gCAAgC;QAChC,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,EAAE,CAAC;YAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CACtC,MAAM,CAAC,QAAQ,CAAC,aAAa,EAC7B,IAAI,CAAC,cAAc,CAAC,iBAAiB,CACtC,CAAC;YACF,KAAK,IAAI,UAAU,GAAG,GAAG,CAAC;QAC5B,CAAC;QAED,qBAAqB;QACrB,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;YAC5C,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,0BAA0B;QAC1B,IAAI,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC;YACvC,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC/D,KAAK,IAAI,YAAY,GAAG,GAAG,CAAC;QAC9B,CAAC;QAED,kBAAkB;QAClB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;IACzC,CAAC;IAEO,kBAAkB,CAAC,MAAc;QACvC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QAE3C,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC;YACnD,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBACtC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3C,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,gBAAgB,CAAC,CAAY,EAAE,CAAY;QACjD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,OAAO,CAAC,CAAC;QACX,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,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1B,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,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,EAAE,CAAC;YACtB,OAAO,CAAC,CAAC;QACX,CAAC;QAED,OAAO,UAAU,GAAG,WAAW,CAAC;IAClC,CAAC;IAEO,iBAAiB,CAAC,OAAe,EAAE,QAAuB;QAChE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,0DAA0D;QAC1D,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;QACxD,OAAO,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,YAAY,CAAC;IACxD,CAAC;IAEO,mBAAmB,CAAC,MAAc,EAAE,QAAuB;QACjE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,cAAc,EAAE,CAAC;YACzC,OAAO,CAAC,CAAC;QACX,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QAC1C,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC;YACzD,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;gBACzC,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;IACvF,CAAC;IAEO,gBAAgB,CACtB,KAAa,EACb,UAAkB,EAClB,mBAA2B;QAE3B,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,IAAI,CAAC,QAAQ,IAAI,KAAK,CAAC;QACvB,IAAI,CAAC,aAAa,IAAI,UAAU,CAAC;QACjC,IAAI,CAAC,sBAAsB,IAAI,mBAAmB,CAAC;IACrD,CAAC;CACF;AAED,gFAAgF;AAChF,mBAAmB;AACnB,gFAAgF;AAEhF;;;;GAIG;AACH,MAAM,UAAU,wBAAwB,CACtC,SAAmC,EAAE;IAErC,OAAO,IAAI,yBAAyB,CAAC,MAAM,CAAC,CAAC;AAC/C,CAAC"}

package/dist/scorer/index.d.ts

@@ -6,4 +6,6 @@ export { EmbeddingModel } from './EmbeddingModel.js';
 export { BranchScorer } from './BranchScorer.js';
 export type { ScoreWeights } from './BranchScorer.js';
 export { ContextEncoder } from './ContextEncoder.js';
+export { createContextAwareScorer, DefaultContextAwareScorer, } from './ContextAwareScorer.js';
+export type { ContextAwareScorer, ContextAwareScorerConfig, ProjectContext, ScoreBreakdown, ScoringStatistics, } from './ContextAwareScorer.js';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/scorer/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,YAAY,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/scorer/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,YAAY,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAGrD,OAAO,EACL,wBAAwB,EACxB,yBAAyB,GAC1B,MAAM,yBAAyB,CAAC;AACjC,YAAY,EACV,kBAAkB,EAClB,wBAAwB,EACxB,cAAc,EACd,cAAc,EACd,iBAAiB,GAClB,MAAM,yBAAyB,CAAC"}

package/dist/scorer/index.js

@@ -5,4 +5,6 @@
 export { EmbeddingModel } from './EmbeddingModel.js';
 export { BranchScorer } from './BranchScorer.js';
 export { ContextEncoder } from './ContextEncoder.js';
+// v2.2.0 NEW!
+export { createContextAwareScorer, DefaultContextAwareScorer, } from './ContextAwareScorer.js';
 //# sourceMappingURL=index.js.map

package/dist/scorer/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/scorer/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/scorer/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,cAAc;AACd,OAAO,EACL,wBAAwB,EACxB,yBAAyB,GAC1B,MAAM,yBAAyB,CAAC"}

package/dist/search/AdaptiveBeamSearch.d.ts

@@ -0,0 +1,98 @@
+/**
+ * AdaptiveBeamSearch - Adaptive Beam Width Search
+ *
+ * Extends BeamSearch with dynamic beam width adjustment based on
+ * search progress. When stagnation is detected (no improvement for
+ * N iterations), beam width is increased up to a maximum.
+ *
+ * @module @nahisaho/musubix-neural-search
+ * @see TSK-NS-102
+ * @see DES-NS-102
+ * @see REQ-NS-102
+ *
+ * @example
+ * ```typescript
+ * import { AdaptiveBeamSearch } from './AdaptiveBeamSearch.js';
+ *
+ * const search = new AdaptiveBeamSearch({
+ *   initialBeamWidth: 5,
+ *   maxBeamWidth: 100,
+ *   stagnationThreshold: 10,
+ * });
+ * const results = await search.search(initial, expand, scorer, context, config);
+ * ```
+ */
+import type { Branch, IBranchScorer, IBeamSearch, SearchConfig, SearchContext, SearchResult, SearchState, SearchStatistics } from '../types.js';
+/**
+ * Configuration for adaptive beam search
+ */
+export interface AdaptiveBeamConfig {
+    /** Initial beam width (default: 5) */
+    initialBeamWidth: number;
+    /** Maximum beam width allowed (default: 100) */
+    maxBeamWidth: number;
+    /** Iterations without improvement before adaptation (default: 10) */
+    stagnationThreshold: number;
+    /** Fraction to increase beam width on stagnation (default: 0.5 = 50%) */
+    beamWidthIncrease: number;
+    /** Minimum score improvement to not count as stagnation */
+    improvementThreshold: number;
+}
+/**
+ * Adaptive search statistics
+ */
+export interface AdaptiveStatistics {
+    /** Current beam width */
+    currentBeamWidth: number;
+    /** Number of beam width adjustments made */
+    beamWidthAdjustments: number;
+    /** Current stagnation counter */
+    stagnationCount: number;
+    /** Best score seen */
+    bestScore: number;
+    /** Iterations since last improvement */
+    iterationsSinceImprovement: number;
+}
+/**
+ * Adaptive Beam Search implementation with dynamic beam width adjustment
+ */
+export declare class AdaptiveBeamSearch implements IBeamSearch {
+    private readonly adaptiveConfig;
+    private currentBeam;
+    private statistics;
+    private currentBeamWidth;
+    private beamWidthAdjustments;
+    private stagnationCount;
+    private bestScore;
+    private iterationsSinceImprovement;
+    constructor(config?: Partial<AdaptiveBeamConfig>);
+    /**
+     * Execute adaptive beam search
+     */
+    search(initial: SearchState, expand: (state: SearchState) => Promise<Branch[]>, scorer: IBranchScorer, context: SearchContext, config: SearchConfig): Promise<SearchResult[]>;
+    /**
+     * Get current beam
+     */
+    getCurrentBeam(): SearchState[];
+    /**
+     * Get search statistics
+     */
+    getStatistics(): SearchStatistics;
+    /**
+     * Get adaptive-specific statistics
+     */
+    getAdaptiveStatistics(): AdaptiveStatistics;
+    /**
+     * Reset adaptive state
+     */
+    reset(): void;
+    /**
+     * Adjust beam width when stagnation detected
+     */
+    private adjustBeamWidth;
+    /**
+     * Build path from state to root
+     */
+    private buildPath;
+}
+//# sourceMappingURL=AdaptiveBeamSearch.d.ts.map

package/dist/search/AdaptiveBeamSearch.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AdaptiveBeamSearch.d.ts","sourceRoot":"","sources":["../../src/search/AdaptiveBeamSearch.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AAEH,OAAO,KAAK,EACV,MAAM,EACN,aAAa,EACb,WAAW,EACX,YAAY,EACZ,aAAa,EACb,YAAY,EACZ,WAAW,EACX,gBAAgB,EACjB,MAAM,aAAa,CAAC;AAOrB;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,sCAAsC;IACtC,gBAAgB,EAAE,MAAM,CAAC;IACzB,gDAAgD;IAChD,YAAY,EAAE,MAAM,CAAC;IACrB,qEAAqE;IACrE,mBAAmB,EAAE,MAAM,CAAC;IAC5B,yEAAyE;IACzE,iBAAiB,EAAE,MAAM,CAAC;IAC1B,2DAA2D;IAC3D,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,yBAAyB;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,4CAA4C;IAC5C,oBAAoB,EAAE,MAAM,CAAC;IAC7B,iCAAiC;IACjC,eAAe,EAAE,MAAM,CAAC;IACxB,sBAAsB;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,wCAAwC;IACxC,0BAA0B,EAAE,MAAM,CAAC;CACpC;AAkBD;;GAEG;AACH,qBAAa,kBAAmB,YAAW,WAAW;IACpD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAqB;IACpD,OAAO,CAAC,WAAW,CAAgB;IACnC,OAAO,CAAC,UAAU,CAAmB;IAGrC,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,oBAAoB,CAAa;IACzC,OAAO,CAAC,eAAe,CAAa;IACpC,OAAO,CAAC,SAAS,CAAqB;IACtC,OAAO,CAAC,0BAA0B,CAAa;gBAEnC,MAAM,GAAE,OAAO,CAAC,kBAAkB,CAAM;IAapD;;OAEG;IACG,MAAM,CACV,OAAO,EAAE,WAAW,EACpB,MAAM,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,EACjD,MAAM,EAAE,aAAa,EACrB,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,YAAY,GACnB,OAAO,CAAC,YAAY,EAAE,CAAC;IAkI1B;;OAEG;IACH,cAAc,IAAI,WAAW,EAAE;IAI/B;;OAEG;IACH,aAAa,IAAI,gBAAgB;IAIjC;;OAEG;IACH,qBAAqB,IAAI,kBAAkB;IAU3C;;OAEG;IACH,KAAK,IAAI,IAAI;IAoBb;;OAEG;IACH,OAAO,CAAC,eAAe;IAevB;;OAEG;IACH,OAAO,CAAC,SAAS;CAYlB"}

package/dist/search/AdaptiveBeamSearch.js

@@ -0,0 +1,240 @@
+/**
+ * AdaptiveBeamSearch - Adaptive Beam Width Search
+ *
+ * Extends BeamSearch with dynamic beam width adjustment based on
+ * search progress. When stagnation is detected (no improvement for
+ * N iterations), beam width is increased up to a maximum.
+ *
+ * @module @nahisaho/musubix-neural-search
+ * @see TSK-NS-102
+ * @see DES-NS-102
+ * @see REQ-NS-102
+ *
+ * @example
+ * ```typescript
+ * import { AdaptiveBeamSearch } from './AdaptiveBeamSearch.js';
+ *
+ * const search = new AdaptiveBeamSearch({
+ *   initialBeamWidth: 5,
+ *   maxBeamWidth: 100,
+ *   stagnationThreshold: 10,
+ * });
+ * const results = await search.search(initial, expand, scorer, context, config);
+ * ```
+ */
+import { SearchTimeoutError } from '../errors.js';
+// =============================================================================
+// Default Configuration
+// =============================================================================
+const DEFAULT_CONFIG = {
+    initialBeamWidth: 5,
+    maxBeamWidth: 100,
+    stagnationThreshold: 10,
+    beamWidthIncrease: 0.5,
+    improvementThreshold: 0.001,
+};
+// =============================================================================
+// Implementation
+// =============================================================================
+/**
+ * Adaptive Beam Search implementation with dynamic beam width adjustment
+ */
+export class AdaptiveBeamSearch {
+    adaptiveConfig;
+    currentBeam;
+    statistics;
+    // Adaptive state
+    currentBeamWidth;
+    beamWidthAdjustments = 0;
+    stagnationCount = 0;
+    bestScore = -Infinity;
+    iterationsSinceImprovement = 0;
+    constructor(config = {}) {
+        this.adaptiveConfig = { ...DEFAULT_CONFIG, ...config };
+        this.currentBeamWidth = this.adaptiveConfig.initialBeamWidth;
+        this.currentBeam = [];
+        this.statistics = {
+            totalExpanded: 0,
+            totalPruned: 0,
+            maxDepthReached: 0,
+            averageScore: 0,
+            timeElapsed: 0,
+        };
+    }
+    /**
+     * Execute adaptive beam search
+     */
+    async search(initial, expand, scorer, context, config) {
+        const startTime = Date.now();
+        // Use config beam width or adaptive width
+        this.currentBeamWidth = config.beamWidth ?? this.adaptiveConfig.initialBeamWidth;
+        this.currentBeam = [initial];
+        const allResults = [];
+        let iteration = 0;
+        let totalPruned = 0;
+        let totalExpanded = 0;
+        let maxDepth = 0;
+        const scores = [];
+        // Reset adaptive state
+        this.bestScore = -Infinity;
+        this.iterationsSinceImprovement = 0;
+        while (iteration < config.maxIterations && this.currentBeam.length > 0) {
+            // Check timeout
+            if (Date.now() - startTime > config.timeout) {
+                throw new SearchTimeoutError(config.timeout);
+            }
+            // Check depth
+            const maxBeamDepth = Math.max(...this.currentBeam.map((s) => s.depth));
+            maxDepth = Math.max(maxDepth, maxBeamDepth);
+            if (maxBeamDepth >= config.maxDepth) {
+                break;
+            }
+            // Expand all states in beam
+            const candidates = [];
+            for (const state of this.currentBeam) {
+                const branches = await expand(state);
+                totalExpanded += branches.length;
+                // Score all branches
+                const scored = await scorer.scoreBatch(branches, context);
+                for (const branchScore of scored) {
+                    // Apply pruning threshold
+                    if (config.pruneThreshold !== undefined &&
+                        branchScore.score < config.pruneThreshold) {
+                        totalPruned++;
+                        continue;
+                    }
+                    const path = this.buildPath(branchScore.branch.to, context.history);
+                    candidates.push({
+                        state: branchScore.branch.to,
+                        score: branchScore.score,
+                        path,
+                    });
+                    scores.push(branchScore.score);
+                }
+            }
+            // Sort by score and keep top beamWidth
+            candidates.sort((a, b) => b.score - a.score);
+            const nextBeam = candidates.slice(0, this.currentBeamWidth);
+            // Track best score and check for improvement
+            const iterationBestScore = nextBeam.length > 0 ? nextBeam[0].score : -Infinity;
+            const improved = iterationBestScore > this.bestScore + this.adaptiveConfig.improvementThreshold;
+            if (improved) {
+                this.bestScore = iterationBestScore;
+                this.iterationsSinceImprovement = 0;
+            }
+            else {
+                this.iterationsSinceImprovement++;
+            }
+            // Check for stagnation and adapt
+            if (this.iterationsSinceImprovement >= this.adaptiveConfig.stagnationThreshold) {
+                this.adjustBeamWidth();
+                this.iterationsSinceImprovement = 0;
+                this.stagnationCount++;
+            }
+            // Add to results
+            for (const candidate of nextBeam) {
+                allResults.push({
+                    state: candidate.state,
+                    score: candidate.score,
+                    path: candidate.path,
+                    iterations: iteration + 1,
+                    pruned: totalPruned,
+                });
+            }
+            // Update beam
+            this.currentBeam = nextBeam.map((c) => c.state);
+            iteration++;
+        }
+        // Update statistics
+        this.statistics = {
+            totalExpanded,
+            totalPruned,
+            maxDepthReached: maxDepth,
+            averageScore: scores.length > 0
+                ? scores.reduce((a, b) => a + b, 0) / scores.length
+                : 0,
+            timeElapsed: Date.now() - startTime,
+        };
+        // Return best results (deduplicated)
+        const seen = new Set();
+        return allResults
+            .filter((r) => {
+            if (seen.has(r.state.id))
+                return false;
+            seen.add(r.state.id);
+            return true;
+        })
+            .sort((a, b) => b.score - a.score)
+            .slice(0, this.currentBeamWidth);
+    }
+    /**
+     * Get current beam
+     */
+    getCurrentBeam() {
+        return [...this.currentBeam];
+    }
+    /**
+     * Get search statistics
+     */
+    getStatistics() {
+        return { ...this.statistics };
+    }
+    /**
+     * Get adaptive-specific statistics
+     */
+    getAdaptiveStatistics() {
+        return {
+            currentBeamWidth: this.currentBeamWidth,
+            beamWidthAdjustments: this.beamWidthAdjustments,
+            stagnationCount: this.stagnationCount,
+            bestScore: this.bestScore,
+            iterationsSinceImprovement: this.iterationsSinceImprovement,
+        };
+    }
+    /**
+     * Reset adaptive state
+     */
+    reset() {
+        this.currentBeamWidth = this.adaptiveConfig.initialBeamWidth;
+        this.beamWidthAdjustments = 0;
+        this.stagnationCount = 0;
+        this.bestScore = -Infinity;
+        this.iterationsSinceImprovement = 0;
+        this.currentBeam = [];
+        this.statistics = {
+            totalExpanded: 0,
+            totalPruned: 0,
+            maxDepthReached: 0,
+            averageScore: 0,
+            timeElapsed: 0,
+        };
+    }
+    // ===========================================================================
+    // Private Methods
+    // ===========================================================================
+    /**
+     * Adjust beam width when stagnation detected
+     */
+    adjustBeamWidth() {
+        const increase = Math.ceil(this.currentBeamWidth * this.adaptiveConfig.beamWidthIncrease);
+        const newWidth = Math.min(this.currentBeamWidth + increase, this.adaptiveConfig.maxBeamWidth);
+        if (newWidth > this.currentBeamWidth) {
+            this.currentBeamWidth = newWidth;
+            this.beamWidthAdjustments++;
+        }
+    }
+    /**
+     * Build path from state to root
+     */
+    buildPath(state, history) {
+        const path = [state];
+        for (let i = history.length - 1; i >= 0; i--) {
+            if (history[i].id === state.parent) {
+                path.unshift(history[i]);
+                break;
+            }
+        }
+        return path;
+    }
+}
+//# sourceMappingURL=AdaptiveBeamSearch.js.map