knowpro: querying

ts/examples/chat/src/memory/knowproMemory.ts

@@ -165,6 +165,7 @@ export async function createKnowproCommands(
             description: "Search current knowPro conversation by terms",
             options: {
                 maxToDisplay: argNum("Maximum matches to display", 25),
+                type: arg("Knowledge type"),
             },
         };
     }
@@ -188,9 +189,10 @@ export async function createKnowproCommands(
                 `Searching ${conversation.nameTag}...`,
             );
 
-            const matches = await kp.searchTermsInConversation(
+            const matches = await kp.searchConversation(
                 conversation,
                 terms,
+                namedArgs.type,
             );
             if (matches === undefined || matches.size === 0) {
                 context.printer.writeLine("No matches");

ts/packages/knowPro/src/accumulators.ts

@@ -0,0 +1,379 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { collections, createTopNList } from "typeagent";
+import {
+    IMessage,
+    KnowledgeType,
+    MessageIndex,
+    ScoredSemanticRef,
+    SemanticRef,
+    SemanticRefIndex,
+    Term,
+    TextRange,
+} from "./dataFormat.js";
+import { isInTextRange } from "./query.js";
+
+export interface Match<T = any> {
+    value: T;
+    score: number;
+    hitCount: number;
+}
+
+/**
+ * Sort in place
+ * @param matches
+ */
+export function sortMatchesByRelevance(matches: Match[]) {
+    matches.sort((x, y) => y.score - x.score);
+}
+
+export class MatchAccumulator<T = any> {
+    private matches: Map<T, Match<T>>;
+    private maxHitCount: number;
+
+    constructor() {
+        this.matches = new Map<T, Match<T>>();
+        this.maxHitCount = 0;
+    }
+
+    public get numMatches(): number {
+        return this.matches.size;
+    }
+
+    public get maxHits(): number {
+        return this.maxHitCount;
+    }
+
+    public has(value: T): boolean {
+        return this.matches.has(value);
+    }
+
+    public getMatch(value: T): Match<T> | undefined {
+        return this.matches.get(value);
+    }
+
+    public setMatch(match: Match<T>): void {
+        this.matches.set(match.value, match);
+        if (match.hitCount > this.maxHitCount) {
+            this.maxHitCount = match.hitCount;
+        }
+    }
+
+    public setMatches(matches: Match<T>[] | IterableIterator<Match<T>>): void {
+        for (const match of matches) {
+            this.setMatch(match);
+        }
+    }
+
+    public add(value: T, score: number): void {
+        let match = this.matches.get(value);
+        if (match !== undefined) {
+            match.hitCount += 1;
+            match.score += score;
+        } else {
+            match = {
+                value,
+                score,
+                hitCount: 1,
+            };
+            this.matches.set(value, match);
+        }
+        if (match.hitCount > this.maxHitCount) {
+            this.maxHitCount = match.hitCount;
+        }
+    }
+
+    public getSortedByScore(minHitCount?: number): Match<T>[] {
+        if (this.matches.size === 0) {
+            return [];
+        }
+        const matches = [...this.matchesWithMinHitCount(minHitCount)];
+        matches.sort((x, y) => y.score - x.score);
+        return matches;
+    }
+
+    /**
+     * Return the top N scoring matches
+     * @param maxMatches
+     * @returns
+     */
+    public getTopNScoring(
+        maxMatches?: number,
+        minHitCount?: number,
+    ): Match<T>[] {
+        if (this.matches.size === 0) {
+            return [];
+        }
+        if (maxMatches && maxMatches > 0) {
+            const topList = createTopNList<T>(maxMatches);
+            for (const match of this.matchesWithMinHitCount(minHitCount)) {
+                topList.push(match.value, match.score);
+            }
+            const ranked = topList.byRank();
+            return ranked.map((m) => this.matches.get(m.item)!);
+        } else {
+            return this.getSortedByScore(minHitCount);
+        }
+    }
+
+    public getMatches(): IterableIterator<Match<T>> {
+        return this.matches.values();
+    }
+
+    public *getMatchesWhere(
+        predicate: (match: Match<T>) => boolean,
+    ): IterableIterator<Match<T>> {
+        for (const match of this.matches.values()) {
+            if (predicate(match)) {
+                yield match;
+            }
+        }
+    }
+
+    public clearMatches(): void {
+        this.matches.clear();
+        this.maxHitCount = 0;
+    }
+
+    public reduceTopNScoring(
+        maxMatches?: number,
+        minHitCount?: number,
+    ): number {
+        const topN = this.getTopNScoring(maxMatches, minHitCount);
+        this.clearMatches();
+        if (topN.length > 0) {
+            this.setMatches(topN);
+        }
+        return topN.length;
+    }
+
+    public union(other: MatchAccumulator<T>): void {
+        for (const matchFrom of other.matches.values()) {
+            const matchTo = this.matches.get(matchFrom.value);
+            if (matchTo !== undefined) {
+                // Existing
+                matchTo.hitCount += matchFrom.hitCount;
+                matchTo.score += matchFrom.score;
+            } else {
+                this.matches.set(matchFrom.value, matchFrom);
+            }
+        }
+    }
+
+    private matchesWithMinHitCount(
+        minHitCount: number | undefined,
+    ): IterableIterator<Match<T>> {
+        return minHitCount !== undefined && minHitCount > 0
+            ? this.getMatchesWhere((m) => m.hitCount >= minHitCount)
+            : this.matches.values();
+    }
+}
+
+export class SemanticRefAccumulator extends MatchAccumulator<SemanticRefIndex> {
+    constructor(public queryTermMatches = new QueryTermAccumulator()) {
+        super();
+    }
+
+    public addTermMatch(
+        term: Term,
+        semanticRefs: ScoredSemanticRef[] | undefined,
+        scoreBoost?: number,
+    ) {
+        if (semanticRefs) {
+            scoreBoost ??= term.score ?? 0;
+            for (const match of semanticRefs) {
+                this.add(match.semanticRefIndex, match.score + scoreBoost);
+            }
+            this.queryTermMatches.add(term);
+        }
+    }
+
+    public addRelatedTermMatch(
+        primaryTerm: Term,
+        relatedTerm: Term,
+        semanticRefs: ScoredSemanticRef[] | undefined,
+        scoreBoost?: number,
+    ) {
+        if (semanticRefs) {
+            // Related term matches count as matches for the queryTerm...
+            // BUT are scored with the score of the related term
+            scoreBoost ??= relatedTerm.score ?? 0;
+            for (const semanticRef of semanticRefs) {
+                let score = semanticRef.score + scoreBoost;
+                let match = this.getMatch(semanticRef.semanticRefIndex);
+                if (match !== undefined) {
+                    if (match.score < score) {
+                        match.score = score;
+                    }
+                } else {
+                    match = {
+                        value: semanticRef.semanticRefIndex,
+                        score,
+                        hitCount: 1,
+                    };
+                    this.setMatch(match);
+                }
+            }
+            this.queryTermMatches.add(primaryTerm, relatedTerm);
+        }
+    }
+
+    public override getSortedByScore(
+        minHitCount?: number,
+    ): Match<SemanticRefIndex>[] {
+        return super.getSortedByScore(this.getMinHitCount(minHitCount));
+    }
+
+    public override getTopNScoring(
+        maxMatches?: number,
+        minHitCount?: number,
+    ): Match<SemanticRefIndex>[] {
+        return super.getTopNScoring(
+            maxMatches,
+            this.getMinHitCount(minHitCount),
+        );
+    }
+
+    public groupMatchesByKnowledgeType(
+        semanticRefs: SemanticRef[],
+    ): Map<KnowledgeType, SemanticRefAccumulator> {
+        const groups = new Map<KnowledgeType, SemanticRefAccumulator>();
+        for (const match of this.getMatches()) {
+            const semanticRef = semanticRefs[match.value];
+            let group = groups.get(semanticRef.knowledgeType);
+            if (group === undefined) {
+                group = new SemanticRefAccumulator();
+                group.queryTermMatches = this.queryTermMatches;
+                groups.set(semanticRef.knowledgeType, group);
+            }
+            group.setMatch(match);
+        }
+        return groups;
+    }
+
+    public toScoredSemanticRefs(): ScoredSemanticRef[] {
+        return this.getSortedByScore(0).map((m) => {
+            return {
+                semanticRefIndex: m.value,
+                score: m.score,
+            };
+        }, 0);
+    }
+
+    private getMinHitCount(minHitCount?: number): number {
+        return minHitCount !== undefined
+            ? minHitCount
+            : //: this.queryTermMatches.termMatches.size;
+              this.maxHits;
+    }
+}
+
+export class QueryTermAccumulator {
+    constructor(
+        public termMatches: Set<string> = new Set<string>(),
+        public relatedTermToTerms: Map<string, Set<string>> = new Map<
+            string,
+            Set<string>
+        >(),
+    ) {}
+
+    public add(term: Term, relatedTerm?: Term) {
+        this.termMatches.add(term.text);
+        if (relatedTerm !== undefined) {
+            let relatedTermToTerms = this.relatedTermToTerms.get(
+                relatedTerm.text,
+            );
+            if (relatedTermToTerms === undefined) {
+                relatedTermToTerms = new Set<string>();
+                this.relatedTermToTerms.set(
+                    relatedTerm.text,
+                    relatedTermToTerms,
+                );
+            }
+            relatedTermToTerms.add(term.text);
+        }
+    }
+
+    public matched(testText: string | string[], expectedText: string): boolean {
+        if (Array.isArray(testText)) {
+            if (testText.length > 0) {
+                for (const text of testText) {
+                    if (this.matched(text, expectedText)) {
+                        return true;
+                    }
+                }
+            }
+            return false;
+        }
+
+        if (
+            this.termMatches.has(testText) &&
+            collections.stringEquals(testText, expectedText, false)
+        ) {
+            return true;
+        }
+
+        // Maybe the test text matched a related term.
+        // If so, the matching related term should have matched *on behalf* of
+        // of expectedTerm
+        const relatedTermToTerms = this.relatedTermToTerms.get(testText);
+        return relatedTermToTerms !== undefined
+            ? relatedTermToTerms.has(expectedText)
+            : false;
+    }
+
+    public didValueMatch(
+        obj: Record<string, any>,
+        key: string,
+        expectedValue: string,
+    ): boolean {
+        const value = obj[key];
+        if (value === undefined) {
+            return false;
+        }
+        if (Array.isArray(value)) {
+            for (const item of value) {
+                if (this.didValueMatch(item, key, expectedValue)) {
+                    return true;
+                }
+            }
+            return false;
+        } else {
+            const stringValue = value.toString().toLowerCase();
+            return this.matched(stringValue, expectedValue);
+        }
+    }
+}
+
+export class TextRangeAccumulator {
+    constructor(
+        public rangesForMessage: Map<MessageIndex, TextRange[]> = new Map<
+            MessageIndex,
+            TextRange[]
+        >(),
+    ) {}
+
+    public addTextRange(textRange: TextRange) {
+        const messageIndex = textRange.start.messageIndex;
+        let textRanges = this.rangesForMessage.get(messageIndex);
+        if (textRanges === undefined) {
+            textRanges = [textRange];
+        }
+        textRanges.push(textRange);
+    }
+
+    public isInRange(textRange: TextRange): boolean {
+        const textRanges = this.rangesForMessage.get(
+            textRange.start.messageIndex,
+        );
+        if (textRanges === undefined) {
+            return false;
+        }
+        return textRanges.some((outerRange) =>
+            isInTextRange(outerRange, textRange),
+        );
+    }
+}
+
+export class MessageAccumulator extends MatchAccumulator<IMessage> {}