ai-output-assert 0.3.0

package/dist/setup.js

@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setupAIAssertions = setupAIAssertions;
+exports.getGlobalOptions = getGlobalOptions;
+const format_1 = require("./matchers/format");
+const content_1 = require("./matchers/content");
+const tone_1 = require("./matchers/tone");
+const structural_1 = require("./matchers/structural");
+const safety_1 = require("./matchers/safety");
+const quality_1 = require("./matchers/quality");
+const semantic_1 = require("./matchers/semantic");
+let globalOptions = {};
+function setupAIAssertions(options) {
+    globalOptions = { ...globalOptions, ...options };
+    const expectFn = typeof globalThis.expect !== 'undefined'
+        ? globalThis.expect
+        : null;
+    if (expectFn && typeof expectFn.extend === 'function') {
+        expectFn.extend(buildMatchers(globalOptions));
+    }
+}
+function getGlobalOptions() {
+    return globalOptions;
+}
+function wrapSync(fn) {
+    return function (received, ...args) {
+        const result = fn(received, ...args);
+        return {
+            pass: result.pass,
+            message: result.message,
+        };
+    };
+}
+function wrapAsync(fn) {
+    return async function (received, ...args) {
+        const result = await fn(received, ...args);
+        return {
+            pass: result.pass,
+            message: result.message,
+        };
+    };
+}
+function buildMatchers(opts) {
+    return {
+        // format
+        toStartWith: wrapSync((r, prefix) => (0, format_1.toStartWith)(r, prefix)),
+        toEndWith: wrapSync((r, suffix) => (0, format_1.toEndWith)(r, suffix)),
+        toBeFormattedAs: wrapSync((r, fmt) => (0, format_1.toBeFormattedAs)(r, fmt)),
+        toHaveListItems: wrapSync((r, items) => (0, format_1.toHaveListItems)(r, items)),
+        // content
+        toContainAllOf: wrapSync((r, phrases) => (0, content_1.toContainAllOf)(r, phrases)),
+        toContainAnyOf: wrapSync((r, phrases) => (0, content_1.toContainAnyOf)(r, phrases)),
+        toNotContain: wrapSync((r, phrase) => (0, content_1.toNotContain)(r, phrase)),
+        toMentionEntity: wrapSync((r, entity, aliases) => (0, content_1.toMentionEntity)(r, entity, aliases)),
+        // tone
+        toHaveSentiment: wrapSync((r, sentiment) => (0, tone_1.toHaveSentiment)(r, sentiment)),
+        toHaveTone: wrapSync((r, tone) => (0, tone_1.toHaveTone)(r, tone)),
+        toBeConcise: wrapSync((r, maxWords) => (0, tone_1.toBeConcise)(r, maxWords)),
+        toNotBeVerbose: wrapSync((r, options) => (0, tone_1.toNotBeVerbose)(r, options)),
+        // structural
+        toBeValidJSON: wrapSync((r) => (0, structural_1.toBeValidJSON)(r)),
+        toMatchSchema: wrapSync((r, schema) => (0, structural_1.toMatchSchema)(r, schema)),
+        toHaveJSONFields: wrapSync((r, fields) => (0, structural_1.toHaveJSONFields)(r, fields)),
+        toBeValidMarkdown: wrapSync((r) => (0, structural_1.toBeValidMarkdown)(r)),
+        toContainCodeBlock: wrapSync((r, lang) => (0, structural_1.toContainCodeBlock)(r, lang)),
+        // safety
+        toNotContainPII: wrapSync((r, patterns) => (0, safety_1.toNotContainPII)(r, patterns)),
+        toNotContainToxicContent: wrapSync((r, words) => (0, safety_1.toNotContainToxicContent)(r, words)),
+        toNotLeakSystemPrompt: wrapSync((r, patterns) => (0, safety_1.toNotLeakSystemPrompt)(r, patterns)),
+        toNotBeRefusal: wrapSync((r, phrases) => (0, safety_1.toNotBeRefusal)(r, phrases)),
+        // quality
+        toNotBeTruncated: wrapSync((r) => (0, quality_1.toNotBeTruncated)(r)),
+        toNotBeHedged: wrapSync((r, phrases, threshold) => (0, quality_1.toNotBeHedged)(r, phrases, threshold)),
+        toBeCompleteJSON: wrapSync((r) => (0, quality_1.toBeCompleteJSON)(r)),
+        toNotRepeat: wrapSync((r, options) => (0, quality_1.toNotRepeat)(r, options)),
+        // semantic (async)
+        toBeSemanticallySimilarTo: wrapAsync((r, expected, options) => (0, semantic_1.toBeSemanticallySimilarTo)(r, expected, {
+            ...(opts.embedFn ? { embedFn: opts.embedFn } : {}),
+            ...(opts.semanticThreshold ? { threshold: opts.semanticThreshold } : {}),
+            ...(options ?? {}),
+        })),
+        toAnswerQuestion: wrapAsync((r, question, options) => (0, semantic_1.toAnswerQuestion)(r, question, {
+            ...(opts.embedFn ? { embedFn: opts.embedFn } : {}),
+            ...(options ?? {}),
+        })),
+        toBeFactuallyConsistentWith: wrapAsync((r, reference, options) => (0, semantic_1.toBeFactuallyConsistentWith)(r, reference, {
+            ...(opts.embedFn ? { embedFn: opts.embedFn } : {}),
+            ...(options ?? {}),
+        })),
+    };
+}
+//# sourceMappingURL=setup.js.map

package/dist/setup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.js","sourceRoot":"","sources":["../src/setup.ts"],"names":[],"mappings":";;AA+BA,8CAaC;AAED,4CAEC;AA/CD,8CAA6F;AAC7F,gDAAmG;AACnG,0CAA2F;AAC3F,sDAM+B;AAC/B,8CAK2B;AAC3B,gDAK4B;AAC5B,kDAI6B;AAE7B,IAAI,aAAa,GAAuB,EAAE,CAAC;AAE3C,SAAgB,iBAAiB,CAAC,OAA4B;IAC5D,aAAa,GAAG,EAAE,GAAG,aAAa,EAAE,GAAG,OAAO,EAAE,CAAC;IAEjD,MAAM,QAAQ,GACZ,OAAQ,UAAsC,CAAC,MAAM,KAAK,WAAW;QACnE,CAAC,CAAE,UAAsC,CAAC,MAAM;QAChD,CAAC,CAAC,IAAI,CAAC;IAEX,IAAI,QAAQ,IAAI,OAAQ,QAAiC,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;QAC/E,QAAoE,CAAC,MAAM,CAC1E,aAAa,CAAC,aAAa,CAAC,CAC7B,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAgB,gBAAgB;IAC9B,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,SAAS,QAAQ,CAAC,EAAyC;IACzD,OAAO,UAAoC,QAAgB,EAAE,GAAG,IAAe;QAC7E,MAAM,MAAM,GAAG,EAAE,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,CAAC;QACrC,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,EAAkD;IACnE,OAAO,KAAK,WAAqC,QAAgB,EAAE,GAAG,IAAe;QACnF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,CAAC;QAC3C,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,IAAwB;IAC7C,OAAO;QACL,SAAS;QACT,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,IAAA,oBAAW,EAAC,CAAW,EAAE,MAAgB,CAAC,CAAC;QAChF,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,IAAA,kBAAS,EAAC,CAAW,EAAE,MAAgB,CAAC,CAAC;QAC5E,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CACnC,IAAA,wBAAe,EAAC,CAAW,EAAE,GAA4C,CAAC,CAC3E;QACD,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,IAAA,wBAAe,EAAC,CAAW,EAAE,KAAiB,CAAC,CAAC;QAExF,UAAU;QACV,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC,IAAA,wBAAc,EAAC,CAAW,EAAE,OAAmB,CAAC,CAAC;QAC1F,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC,IAAA,wBAAc,EAAC,CAAW,EAAE,OAAmB,CAAC,CAAC;QAC1F,YAAY,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,IAAA,sBAAY,EAAC,CAAW,EAAE,MAAgB,CAAC,CAAC;QAClF,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,CAC/C,IAAA,yBAAe,EAAC,CAAW,EAAE,MAAgB,EAAE,OAA+B,CAAC,CAChF;QAED,OAAO;QACP,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,CACzC,IAAA,sBAAe,EAAC,CAAW,EAAE,SAAkD,CAAC,CACjF;QACD,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CAC/B,IAAA,iBAAU,EAAC,CAAW,EAAE,IAAwC,CAAC,CAClE;QACD,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,CACpC,IAAA,kBAAW,EAAC,CAAW,EAAE,QAA8B,CAAC,CACzD;QACD,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CACtC,IAAA,qBAAc,EAAC,CAAW,EAAE,OAA+C,CAAC,CAC7E;QAED,aAAa;QACb,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,0BAAa,EAAC,CAAW,CAAC,CAAC;QAC1D,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,IAAA,0BAAa,EAAC,CAAW,EAAE,MAAgB,CAAC,CAAC;QACpF,gBAAgB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,IAAA,6BAAgB,EAAC,CAAW,EAAE,MAAkB,CAAC,CAAC;QAC5F,iBAAiB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,8BAAiB,EAAC,CAAW,CAAC,CAAC;QAClE,kBAAkB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CACvC,IAAA,+BAAkB,EAAC,CAAW,EAAE,IAA0B,CAAC,CAC5D;QAED,SAAS;QACT,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,CACxC,IAAA,wBAAe,EACb,CAAW,EACX,QAAiD,CAClD,CACF;QACD,wBAAwB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAC9C,IAAA,iCAAwB,EACtB,CAAW,EACX,KAAuD,CACxD,CACF;QACD,qBAAqB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,CAC9C,IAAA,8BAAqB,EAAC,CAAW,EAAE,QAAgC,CAAC,CACrE;QACD,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CACtC,IAAA,uBAAc,EAAC,CAAW,EAAE,OAA+B,CAAC,CAC7D;QAED,UAAU;QACV,gBAAgB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,0BAAgB,EAAC,CAAW,CAAC,CAAC;QAChE,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,CAChD,IAAA,uBAAa,EAAC,CAAW,EAAE,OAA+B,EAAE,SAA+B,CAAC,CAC7F;QACD,gBAAgB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,0BAAgB,EAAC,CAAW,CAAC,CAAC;QAChE,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CACnC,IAAA,qBAAW,EAAC,CAAW,EAAE,OAA4C,CAAC,CACvE;QAED,mBAAmB;QACnB,yBAAyB,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,CAC5D,IAAA,oCAAyB,EACvB,CAAW,EACX,QAAkB,EAClB;YACE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClD,GAAG,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxE,GAAG,CAAC,OAAiB,IAAI,EAAE,CAAC;SAC7B,CACF,CACF;QACD,gBAAgB,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,CACnD,IAAA,2BAAgB,EAAC,CAAW,EAAE,QAAkB,EAAE;YAChD,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClD,GAAG,CAAC,OAAiB,IAAI,EAAE,CAAC;SAC7B,CAAC,CACH;QACD,2BAA2B,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,CAC/D,IAAA,sCAA2B,EAAC,CAAW,EAAE,SAAmB,EAAE;YAC5D,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClD,GAAG,CAAC,OAAiB,IAAI,EAAE,CAAC;SAC7B,CAAC,CACH;KACF,CAAC;AACJ,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,43 @@
+export interface MatcherResult {
+    pass: boolean;
+    message: () => string;
+    details: Record<string, unknown>;
+}
+export type EmbedFn = (text: string) => Promise<number[]>;
+export type Tone = 'formal' | 'casual' | 'technical' | 'friendly';
+export type ToneScores = Record<Tone, number>;
+export type Sentiment = 'positive' | 'negative' | 'neutral';
+export type OutputFormat = 'json' | 'markdown' | 'list' | 'csv' | 'xml' | 'yaml' | 'table';
+export interface PIIPattern {
+    type: string;
+    pattern: RegExp;
+    validate?: (match: string) => boolean;
+    label: string;
+}
+export interface ToxicWord {
+    word: string;
+    severity: 'critical' | 'warning' | 'info';
+}
+export type PIIMatch = {
+    type: string;
+    value: string;
+    position: [number, number];
+};
+export interface AIAssertionOptions {
+    embedFn?: EmbedFn;
+    semanticThreshold?: number;
+    answerThreshold?: number;
+    consistencyThreshold?: number;
+    conciseMaxTokens?: number;
+    verboseMaxSentences?: number;
+    hedgingMaxRatio?: number;
+    repeatMaxRepetitions?: number;
+    systemPromptLeakThreshold?: number;
+    sentimentNeutralRange?: [number, number];
+    customPIIPatterns?: PIIPattern[];
+    customToxicWords?: ToxicWord[];
+    customEntityAliases?: Record<string, string[]>;
+    customHedgingPhrases?: string[];
+    customRefusalPhrases?: string[];
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,OAAO,CAAC;IACd,OAAO,EAAE,MAAM,MAAM,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAClC;AAED,MAAM,MAAM,OAAO,GAAG,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;AAE1D,MAAM,MAAM,IAAI,GAAG,QAAQ,GAAG,QAAQ,GAAG,WAAW,GAAG,UAAU,CAAC;AAClE,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AAC9C,MAAM,MAAM,SAAS,GAAG,UAAU,GAAG,UAAU,GAAG,SAAS,CAAC;AAC5D,MAAM,MAAM,YAAY,GAAG,MAAM,GAAG,UAAU,GAAG,MAAM,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,GAAG,OAAO,CAAC;AAE3F,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC;IACtC,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,UAAU,GAAG,SAAS,GAAG,MAAM,CAAC;CAC3C;AAED,MAAM,MAAM,QAAQ,GAAG;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC5B,CAAC;AAEF,MAAM,WAAW,kBAAkB;IACjC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,qBAAqB,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACzC,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC;IACjC,gBAAgB,CAAC,EAAE,SAAS,EAAE,CAAC;IAC/B,mBAAmB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAC/C,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;IAChC,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;CACjC"}

package/dist/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/dist/utils/cosine-similarity.d.ts

@@ -0,0 +1,2 @@
+export declare function cosineSimilarity(a: number[], b: number[]): number;
+//# sourceMappingURL=cosine-similarity.d.ts.map

package/dist/utils/cosine-similarity.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cosine-similarity.d.ts","sourceRoot":"","sources":["../../src/utils/cosine-similarity.ts"],"names":[],"mappings":"AAAA,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAYjE"}

package/dist/utils/cosine-similarity.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cosineSimilarity = cosineSimilarity;
+function cosineSimilarity(a, b) {
+    if (a.length !== b.length) {
+        throw new Error(`Dimension mismatch: ${a.length} vs ${b.length}`);
+    }
+    let dot = 0, normA = 0, normB = 0;
+    for (let i = 0; i < a.length; i++) {
+        dot += a[i] * b[i];
+        normA += a[i] * a[i];
+        normB += b[i] * b[i];
+    }
+    if (normA === 0 || normB === 0)
+        return 0;
+    return dot / (Math.sqrt(normA) * Math.sqrt(normB));
+}
+//# sourceMappingURL=cosine-similarity.js.map

package/dist/utils/cosine-similarity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cosine-similarity.js","sourceRoot":"","sources":["../../src/utils/cosine-similarity.ts"],"names":[],"mappings":";;AAAA,4CAYC;AAZD,SAAgB,gBAAgB,CAAC,CAAW,EAAE,CAAW;IACvD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,MAAM,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,GAAG,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,IAAI,KAAK,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IACzC,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;AACrD,CAAC"}

package/dist/utils/embedding-cache.d.ts

@@ -0,0 +1,3 @@
+import type { EmbedFn } from '../types';
+export declare function createCachedEmbedFn(embedFn: EmbedFn): EmbedFn;
+//# sourceMappingURL=embedding-cache.d.ts.map

package/dist/utils/embedding-cache.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding-cache.d.ts","sourceRoot":"","sources":["../../src/utils/embedding-cache.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,UAAU,CAAC;AAExC,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAQ7D"}

package/dist/utils/embedding-cache.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createCachedEmbedFn = createCachedEmbedFn;
+function createCachedEmbedFn(embedFn) {
+    const cache = new Map();
+    return async (text) => {
+        if (cache.has(text))
+            return cache.get(text);
+        const embedding = await embedFn(text);
+        cache.set(text, embedding);
+        return embedding;
+    };
+}
+//# sourceMappingURL=embedding-cache.js.map

package/dist/utils/embedding-cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding-cache.js","sourceRoot":"","sources":["../../src/utils/embedding-cache.ts"],"names":[],"mappings":";;AAEA,kDAQC;AARD,SAAgB,mBAAmB,CAAC,OAAgB;IAClD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC1C,OAAO,KAAK,EAAE,IAAY,EAAE,EAAE;QAC5B,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAE,CAAC;QAC7C,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;QACtC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC3B,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;AACJ,CAAC"}

package/dist/utils/json-extract.d.ts

@@ -0,0 +1,2 @@
+export declare function extractJSONFromCodeFence(text: string): string | null;
+//# sourceMappingURL=json-extract.d.ts.map

package/dist/utils/json-extract.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"json-extract.d.ts","sourceRoot":"","sources":["../../src/utils/json-extract.ts"],"names":[],"mappings":"AAAA,wBAAgB,wBAAwB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAGpE"}

package/dist/utils/json-extract.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.extractJSONFromCodeFence = extractJSONFromCodeFence;
+function extractJSONFromCodeFence(text) {
+    const match = text.match(/```(?:json)?\s*\n([\s\S]*?)\n```/i);
+    return match ? match[1].trim() : null;
+}
+//# sourceMappingURL=json-extract.js.map

package/dist/utils/json-extract.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"json-extract.js","sourceRoot":"","sources":["../../src/utils/json-extract.ts"],"names":[],"mappings":";;AAAA,4DAGC;AAHD,SAAgB,wBAAwB,CAAC,IAAY;IACnD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;IAC9D,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACxC,CAAC"}

package/dist/utils/luhn.d.ts

@@ -0,0 +1,2 @@
+export declare function luhnCheck(num: string): boolean;
+//# sourceMappingURL=luhn.d.ts.map

package/dist/utils/luhn.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"luhn.d.ts","sourceRoot":"","sources":["../../src/utils/luhn.ts"],"names":[],"mappings":"AAAA,wBAAgB,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAe9C"}

package/dist/utils/luhn.js

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.luhnCheck = luhnCheck;
+function luhnCheck(num) {
+    const digits = num.replace(/\D/g, '');
+    if (digits.length === 0)
+        return false;
+    let sum = 0;
+    let isEven = false;
+    for (let i = digits.length - 1; i >= 0; i--) {
+        let d = parseInt(digits[i], 10);
+        if (isEven) {
+            d *= 2;
+            if (d > 9)
+                d -= 9;
+        }
+        sum += d;
+        isEven = !isEven;
+    }
+    return sum % 10 === 0;
+}
+//# sourceMappingURL=luhn.js.map

package/dist/utils/luhn.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"luhn.js","sourceRoot":"","sources":["../../src/utils/luhn.ts"],"names":[],"mappings":";;AAAA,8BAeC;AAfD,SAAgB,SAAS,CAAC,GAAW;IACnC,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACtC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACtC,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAChC,IAAI,MAAM,EAAE,CAAC;YACX,CAAC,IAAI,CAAC,CAAC;YACP,IAAI,CAAC,GAAG,CAAC;gBAAE,CAAC,IAAI,CAAC,CAAC;QACpB,CAAC;QACD,GAAG,IAAI,CAAC,CAAC;QACT,MAAM,GAAG,CAAC,MAAM,CAAC;IACnB,CAAC;IACD,OAAO,GAAG,GAAG,EAAE,KAAK,CAAC,CAAC;AACxB,CAAC"}

package/dist/utils/ngrams.d.ts

@@ -0,0 +1,2 @@
+export declare function extractNgrams(tokens: string[], n: number): string[];
+//# sourceMappingURL=ngrams.d.ts.map

package/dist/utils/ngrams.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ngrams.d.ts","sourceRoot":"","sources":["../../src/utils/ngrams.ts"],"names":[],"mappings":"AAAA,wBAAgB,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAOnE"}

package/dist/utils/ngrams.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.extractNgrams = extractNgrams;
+function extractNgrams(tokens, n) {
+    if (n <= 0 || tokens.length < n)
+        return [];
+    const result = [];
+    for (let i = 0; i <= tokens.length - n; i++) {
+        result.push(tokens.slice(i, i + n).join(' '));
+    }
+    return result;
+}
+//# sourceMappingURL=ngrams.js.map

package/dist/utils/ngrams.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ngrams.js","sourceRoot":"","sources":["../../src/utils/ngrams.ts"],"names":[],"mappings":";;AAAA,sCAOC;AAPD,SAAgB,aAAa,CAAC,MAAgB,EAAE,CAAS;IACvD,IAAI,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IAC3C,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/utils/regex-escape.d.ts

@@ -0,0 +1,2 @@
+export declare function escapeRegex(str: string): string;
+//# sourceMappingURL=regex-escape.d.ts.map

package/dist/utils/regex-escape.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"regex-escape.d.ts","sourceRoot":"","sources":["../../src/utils/regex-escape.ts"],"names":[],"mappings":"AAAA,wBAAgB,WAAW,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAE/C"}

package/dist/utils/regex-escape.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.escapeRegex = escapeRegex;
+function escapeRegex(str) {
+    return str.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+}
+//# sourceMappingURL=regex-escape.js.map

package/dist/utils/regex-escape.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"regex-escape.js","sourceRoot":"","sources":["../../src/utils/regex-escape.ts"],"names":[],"mappings":";;AAAA,kCAEC;AAFD,SAAgB,WAAW,CAAC,GAAW;IACrC,OAAO,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;AACpD,CAAC"}

package/dist/utils/sentences.d.ts

@@ -0,0 +1,2 @@
+export declare function splitSentences(text: string): string[];
+//# sourceMappingURL=sentences.d.ts.map

package/dist/utils/sentences.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sentences.d.ts","sourceRoot":"","sources":["../../src/utils/sentences.ts"],"names":[],"mappings":"AAEA,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAgCrD"}

package/dist/utils/sentences.js

@@ -0,0 +1,36 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.splitSentences = splitSentences;
+const ABBREV_PATTERN = /\b(?:Mr|Mrs|Ms|Dr|Prof|Sr|Jr|Inc|Ltd|Corp|vs|etc|e\.g|i\.e|U\.S|U\.K|Jan|Feb|Mar|Apr|Jun|Jul|Aug|Sep|Oct|Nov|Dec)\.$/i;
+function splitSentences(text) {
+    if (!text || text.trim().length === 0)
+        return [];
+    const result = [];
+    // Split on sentence-ending punctuation followed by whitespace + uppercase letter
+    // We use a manual approach to skip abbreviations
+    const raw = text.split(/(?<=[.!?]['")]*)\s+(?=[A-Z"])/);
+    for (const part of raw) {
+        const trimmed = part.trim();
+        if (trimmed.length === 0)
+            continue;
+        // Check if the last "sentence" ends with a known abbreviation — if so, it got
+        // split incorrectly and we should merge it back with the next part.
+        // Because we already split, we instead check each segment itself.
+        // If a segment ends with an abbreviation dot (e.g. "Dr."), it should have been
+        // attached to the next segment. We handle this by checking if the TRIMMED segment
+        // (without trailing whitespace) ends with an abbreviation.
+        if (result.length > 0) {
+            const prev = result[result.length - 1];
+            // If prev ends with a known abbreviation, merge
+            if (ABBREV_PATTERN.test(prev)) {
+                result[result.length - 1] = prev + ' ' + trimmed;
+                continue;
+            }
+        }
+        result.push(trimmed);
+    }
+    // Final pass: if the last element ends with an abbreviation (no merge needed),
+    // it's still a valid sentence — leave it as is.
+    return result;
+}
+//# sourceMappingURL=sentences.js.map

package/dist/utils/sentences.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sentences.js","sourceRoot":"","sources":["../../src/utils/sentences.ts"],"names":[],"mappings":";;AAEA,wCAgCC;AAlCD,MAAM,cAAc,GAAG,uHAAuH,CAAC;AAE/I,SAAgB,cAAc,CAAC,IAAY;IACzC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAEjD,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,iFAAiF;IACjF,iDAAiD;IACjD,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;IAExD,KAAK,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAEnC,8EAA8E;QAC9E,oEAAoE;QACpE,kEAAkE;QAClE,+EAA+E;QAC/E,kFAAkF;QAClF,2DAA2D;QAC3D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACvC,gDAAgD;YAChD,IAAI,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9B,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,GAAG,GAAG,OAAO,CAAC;gBACjD,SAAS;YACX,CAAC;QACH,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAED,+EAA+E;IAC/E,gDAAgD;IAChD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/utils/tokenizer.d.ts

@@ -0,0 +1,2 @@
+export declare function tokenize(text: string): string[];
+//# sourceMappingURL=tokenizer.d.ts.map

package/dist/utils/tokenizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenizer.d.ts","sourceRoot":"","sources":["../../src/utils/tokenizer.ts"],"names":[],"mappings":"AAAA,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAE/C"}

package/dist/utils/tokenizer.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tokenize = tokenize;
+function tokenize(text) {
+    return text.split(/\s+/).filter(Boolean);
+}
+//# sourceMappingURL=tokenizer.js.map

package/dist/utils/tokenizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenizer.js","sourceRoot":"","sources":["../../src/utils/tokenizer.ts"],"names":[],"mappings":";;AAAA,4BAEC;AAFD,SAAgB,QAAQ,CAAC,IAAY;IACnC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAC3C,CAAC"}

package/package.json

@@ -0,0 +1,36 @@
+{
+  "name": "ai-output-assert",
+  "version": "0.3.0",
+  "description": "Rich assertion library for LLM outputs as Jest/Vitest matchers",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "test": "vitest run",
+    "lint": "eslint src/",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "MIT",
+  "engines": {
+    "node": ">=18"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "devDependencies": {
+    "@types/node": "^25.5.0",
+    "@typescript-eslint/eslint-plugin": "^8.57.1",
+    "@typescript-eslint/parser": "^8.57.1",
+    "eslint": "^10.1.0",
+    "typescript": "^5.9.3",
+    "vitest": "^4.1.0"
+  },
+  "dependencies": {
+    "ajv": "^8.18.0"
+  }
+}