@okrapdf/cli 0.1.0 → 0.1.2

package/dist/lib/processors.test.js

@@ -0,0 +1,83 @@
+/**
+ * Tests for processor plugin system
+ */
+import { describe, it, expect } from 'vitest';
+import { resolveProcessor, getProcessor, listProcessors, BUILTIN_PROCESSORS, PROCESSOR_ALIASES, } from './processors.js';
+describe('processors', () => {
+    describe('BUILTIN_PROCESSORS', () => {
+        it('should have expected processors', () => {
+            expect(BUILTIN_PROCESSORS).toHaveProperty('docai');
+            expect(BUILTIN_PROCESSORS).toHaveProperty('qwen');
+            expect(BUILTIN_PROCESSORS).toHaveProperty('llamaparse');
+            expect(BUILTIN_PROCESSORS).toHaveProperty('gemini');
+            expect(BUILTIN_PROCESSORS).toHaveProperty('pdfplumber');
+            expect(BUILTIN_PROCESSORS).toHaveProperty('default');
+        });
+        it('should have required fields for each processor', () => {
+            for (const [id, proc] of Object.entries(BUILTIN_PROCESSORS)) {
+                expect(proc.id).toBe(id);
+                expect(proc.name).toBeTruthy();
+                expect(proc.description).toBeTruthy();
+                expect(proc.apiEndpoint).toBeTruthy();
+            }
+        });
+    });
+    describe('resolveProcessor', () => {
+        it('should return processor ID as-is if not an alias', () => {
+            expect(resolveProcessor('docai')).toBe('docai');
+            expect(resolveProcessor('gemini')).toBe('gemini');
+        });
+        it('should resolve aliases to processor IDs', () => {
+            expect(resolveProcessor('google')).toBe('docai');
+            expect(resolveProcessor('gcp')).toBe('docai');
+            expect(resolveProcessor('llama')).toBe('llamaparse');
+            expect(resolveProcessor('lp')).toBe('llamaparse');
+            expect(resolveProcessor('auto')).toBe('default');
+        });
+        it('should be case-insensitive', () => {
+            expect(resolveProcessor('DOCAI')).toBe('docai');
+            expect(resolveProcessor('Google')).toBe('docai');
+        });
+        it('should return unknown processor as-is', () => {
+            expect(resolveProcessor('unknown')).toBe('unknown');
+        });
+    });
+    describe('getProcessor', () => {
+        it('should return processor info for valid ID', () => {
+            const proc = getProcessor('docai');
+            expect(proc).toBeDefined();
+            expect(proc?.id).toBe('docai');
+            expect(proc?.name).toBe('Google Document AI');
+        });
+        it('should resolve aliases', () => {
+            const proc = getProcessor('google');
+            expect(proc).toBeDefined();
+            expect(proc?.id).toBe('docai');
+        });
+        it('should return undefined for unknown processor', () => {
+            expect(getProcessor('nonexistent')).toBeUndefined();
+        });
+    });
+    describe('listProcessors', () => {
+        it('should return all processors', () => {
+            const processors = listProcessors();
+            expect(processors.length).toBe(Object.keys(BUILTIN_PROCESSORS).length);
+        });
+        it('should return processors with required fields', () => {
+            const processors = listProcessors();
+            for (const proc of processors) {
+                expect(proc.id).toBeTruthy();
+                expect(proc.name).toBeTruthy();
+                expect(proc.description).toBeTruthy();
+            }
+        });
+    });
+    describe('PROCESSOR_ALIASES', () => {
+        it('should map to valid processors', () => {
+            for (const [alias, target] of Object.entries(PROCESSOR_ALIASES)) {
+                expect(BUILTIN_PROCESSORS[target]).toBeDefined();
+            }
+        });
+    });
+});
+//# sourceMappingURL=processors.test.js.map

package/dist/lib/processors.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"processors.test.js","sourceRoot":"","sources":["../../src/lib/processors.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,gBAAgB,EAChB,YAAY,EACZ,cAAc,EACd,kBAAkB,EAClB,iBAAiB,GAClB,MAAM,iBAAiB,CAAC;AAEzB,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,CAAC,kBAAkB,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACnD,MAAM,CAAC,kBAAkB,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAClD,MAAM,CAAC,kBAAkB,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YACxD,MAAM,CAAC,kBAAkB,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YACpD,MAAM,CAAC,kBAAkB,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YACxD,MAAM,CAAC,kBAAkB,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAAE,CAAC;gBAC5D,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACzB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;gBAC/B,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;gBACtC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;YACxC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAChD,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACjD,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACrD,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAChD,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC/B,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;YACpC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,MAAM,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;YACpC,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,UAAU,EAAE,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;gBAC/B,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;YACxC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBAChE,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACnD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/lib/progress.d.ts

@@ -0,0 +1,44 @@
+/**
+ * Progress indicators and spinners
+ */
+import { type Ora } from 'ora';
+/**
+ * Create a spinner
+ */
+export declare function createSpinner(text: string): Ora;
+/**
+ * Run an async operation with a spinner
+ */
+export declare function withSpinner<T>(text: string, operation: () => Promise<T>, successText?: string): Promise<T>;
+/**
+ * Simple progress bar
+ */
+export declare class ProgressBar {
+    private current;
+    private total;
+    private width;
+    private label;
+    constructor(total: number, label?: string, width?: number);
+    update(current: number): void;
+    increment(amount?: number): void;
+    private render;
+    complete(): void;
+}
+/**
+ * Poll a job until completion with progress
+ */
+export declare function pollWithProgress<T>(pollFn: () => Promise<{
+    done: boolean;
+    progress?: number;
+    total?: number;
+    data: T;
+}>, options?: {
+    interval?: number;
+    timeout?: number;
+    label?: string;
+}): Promise<T>;
+/**
+ * Sleep helper
+ */
+export declare function sleep(ms: number): Promise<void>;
+//# sourceMappingURL=progress.d.ts.map

package/dist/lib/progress.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"progress.d.ts","sourceRoot":"","sources":["../../src/lib/progress.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAY,EAAE,KAAK,GAAG,EAAE,MAAM,KAAK,CAAC;AAGpC;;GAEG;AACH,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,GAAG,CAK/C;AAED;;GAEG;AACH,wBAAsB,WAAW,CAAC,CAAC,EACjC,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAC3B,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,CAAC,CAAC,CAYZ;AAED;;GAEG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,OAAO,CAAK;IACpB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,KAAK,CAAS;gBAEV,KAAK,EAAE,MAAM,EAAE,KAAK,SAAa,EAAE,KAAK,SAAK;IAMzD,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAK7B,SAAS,CAAC,MAAM,SAAI,GAAG,IAAI;IAK3B,OAAO,CAAC,MAAM;IAWd,QAAQ,IAAI,IAAI;CAKjB;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CAAC,CAAC,EACtC,MAAM,EAAE,MAAM,OAAO,CAAC;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,CAAC,CAAA;CAAE,CAAC,EACpF,OAAO,GAAE;IACP,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CACX,GACL,OAAO,CAAC,CAAC,CAAC,CAiCZ;AAED;;GAEG;AACH,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE/C"}

package/dist/lib/progress.js

@@ -0,0 +1,102 @@
+/**
+ * Progress indicators and spinners
+ */
+import ora from 'ora';
+import chalk from 'chalk';
+/**
+ * Create a spinner
+ */
+export function createSpinner(text) {
+    return ora({
+        text,
+        spinner: 'dots',
+    });
+}
+/**
+ * Run an async operation with a spinner
+ */
+export async function withSpinner(text, operation, successText) {
+    const spinner = createSpinner(text);
+    spinner.start();
+    try {
+        const result = await operation();
+        spinner.succeed(successText || text);
+        return result;
+    }
+    catch (error) {
+        spinner.fail();
+        throw error;
+    }
+}
+/**
+ * Simple progress bar
+ */
+export class ProgressBar {
+    current = 0;
+    total;
+    width;
+    label;
+    constructor(total, label = 'Progress', width = 30) {
+        this.total = total;
+        this.width = width;
+        this.label = label;
+    }
+    update(current) {
+        this.current = current;
+        this.render();
+    }
+    increment(amount = 1) {
+        this.current = Math.min(this.current + amount, this.total);
+        this.render();
+    }
+    render() {
+        const percent = this.total > 0 ? this.current / this.total : 0;
+        const filled = Math.round(this.width * percent);
+        const empty = this.width - filled;
+        const bar = chalk.green('█'.repeat(filled)) + chalk.gray('░'.repeat(empty));
+        const percentStr = (percent * 100).toFixed(0).padStart(3);
+        process.stdout.write(`\r${this.label}: ${bar} ${percentStr}% (${this.current}/${this.total})`);
+    }
+    complete() {
+        this.current = this.total;
+        this.render();
+        console.log(); // New line after completion
+    }
+}
+/**
+ * Poll a job until completion with progress
+ */
+export async function pollWithProgress(pollFn, options = {}) {
+    const { interval = 2000, timeout = 300000, label = 'Processing' } = options;
+    const startTime = Date.now();
+    const spinner = createSpinner(label);
+    spinner.start();
+    let lastProgress = 0;
+    let lastTotal = 0;
+    while (true) {
+        const elapsed = Date.now() - startTime;
+        if (elapsed > timeout) {
+            spinner.fail('Timeout waiting for completion');
+            throw new Error('Operation timed out');
+        }
+        const result = await pollFn();
+        if (result.progress !== undefined && result.total !== undefined) {
+            lastProgress = result.progress;
+            lastTotal = result.total;
+            const percent = lastTotal > 0 ? Math.round((lastProgress / lastTotal) * 100) : 0;
+            spinner.text = `${label}: ${percent}% (${lastProgress}/${lastTotal} pages)`;
+        }
+        if (result.done) {
+            spinner.succeed(`${label}: Complete`);
+            return result.data;
+        }
+        await sleep(interval);
+    }
+}
+/**
+ * Sleep helper
+ */
+export function sleep(ms) {
+    return new Promise(resolve => setTimeout(resolve, ms));
+}
+//# sourceMappingURL=progress.js.map

package/dist/lib/progress.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"progress.js","sourceRoot":"","sources":["../../src/lib/progress.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,GAAiB,MAAM,KAAK,CAAC;AACpC,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,IAAY;IACxC,OAAO,GAAG,CAAC;QACT,IAAI;QACJ,OAAO,EAAE,MAAM;KAChB,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAAY,EACZ,SAA2B,EAC3B,WAAoB;IAEpB,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IACpC,OAAO,CAAC,KAAK,EAAE,CAAC;IAEhB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,EAAE,CAAC;QACjC,OAAO,CAAC,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC;QACrC,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,WAAW;IACd,OAAO,GAAG,CAAC,CAAC;IACZ,KAAK,CAAS;IACd,KAAK,CAAS;IACd,KAAK,CAAS;IAEtB,YAAY,KAAa,EAAE,KAAK,GAAG,UAAU,EAAE,KAAK,GAAG,EAAE;QACvD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,MAAM,CAAC,OAAe;QACpB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAED,SAAS,CAAC,MAAM,GAAG,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3D,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAEO,MAAM;QACZ,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;QAElC,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAC5E,MAAM,UAAU,GAAG,CAAC,OAAO,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAE1D,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,KAAK,KAAK,GAAG,IAAI,UAAU,MAAM,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;IACjG,CAAC;IAED,QAAQ;QACN,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;QACd,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,4BAA4B;IAC7C,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,MAAoF,EACpF,UAII,EAAE;IAEN,MAAM,EAAE,QAAQ,GAAG,IAAI,EAAE,OAAO,GAAG,MAAM,EAAE,KAAK,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC;IAE5E,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;IACrC,OAAO,CAAC,KAAK,EAAE,CAAC;IAEhB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QACvC,IAAI,OAAO,GAAG,OAAO,EAAE,CAAC;YACtB,OAAO,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;YAC/C,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,MAAM,EAAE,CAAC;QAE9B,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAChE,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC;YAC/B,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;YACzB,MAAM,OAAO,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjF,OAAO,CAAC,IAAI,GAAG,GAAG,KAAK,KAAK,OAAO,MAAM,YAAY,IAAI,SAAS,SAAS,CAAC;QAC9E,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,OAAO,CAAC,OAAO,CAAC,GAAG,KAAK,YAAY,CAAC,CAAC;YACtC,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC;QAED,MAAM,KAAK,CAAC,QAAQ,CAAC,CAAC;IACxB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,KAAK,CAAC,EAAU;IAC9B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AACzD,CAAC"}

package/dist/lib/templates.d.ts

@@ -0,0 +1,56 @@
+/**
+ * Extraction Templates
+ *
+ * Reusable patterns for common document types.
+ * Users can create custom templates stored in ~/.okrapdf/templates/
+ */
+export interface ExtractField {
+    name: string;
+    description: string;
+    type: 'string' | 'number' | 'date' | 'currency' | 'array' | 'table';
+    required?: boolean;
+    format?: string;
+}
+export interface Template {
+    id: string;
+    name: string;
+    description: string;
+    /** Document type this template is for */
+    documentType: string;
+    /** Preferred processor for this template */
+    processor?: string;
+    /** Fields to extract */
+    fields: ExtractField[];
+    /** System prompt for LLM-based extraction */
+    systemPrompt?: string;
+    /** Example output format */
+    exampleOutput?: Record<string, unknown>;
+    /** Whether this is a built-in template */
+    builtin?: boolean;
+}
+export declare const BUILTIN_TEMPLATES: Record<string, Template>;
+/**
+ * Get the templates directory path
+ */
+export declare function getTemplatesDir(): string;
+/**
+ * Load a template by ID (checks user templates first, then builtins)
+ */
+export declare function loadTemplate(id: string): Template | undefined;
+/**
+ * Save a custom template
+ */
+export declare function saveTemplate(template: Template): void;
+/**
+ * List all available templates (user + builtin)
+ */
+export declare function listTemplates(): Template[];
+/**
+ * Delete a user template
+ */
+export declare function deleteTemplate(id: string): boolean;
+/**
+ * Format template for display
+ */
+export declare function formatTemplateInfo(template: Template): string;
+//# sourceMappingURL=templates.d.ts.map

package/dist/lib/templates.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"templates.d.ts","sourceRoot":"","sources":["../../src/lib/templates.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,QAAQ,GAAG,QAAQ,GAAG,MAAM,GAAG,UAAU,GAAG,OAAO,GAAG,OAAO,CAAC;IACpE,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,yCAAyC;IACzC,YAAY,EAAE,MAAM,CAAC;IACrB,4CAA4C;IAC5C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,wBAAwB;IACxB,MAAM,EAAE,YAAY,EAAE,CAAC;IACvB,6CAA6C;IAC7C,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,4BAA4B;IAC5B,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACxC,0CAA0C;IAC1C,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAGD,eAAO,MAAM,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAmGtD,CAAC;AAEF;;GAEG;AACH,wBAAgB,eAAe,IAAI,MAAM,CAMxC;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAc7D;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAGrD;AAED;;GAEG;AACH,wBAAgB,aAAa,IAAI,QAAQ,EAAE,CA4B1C;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAQlD;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,GAAG,MAAM,CAe7D"}

package/dist/lib/templates.js

@@ -0,0 +1,204 @@
+/**
+ * Extraction Templates
+ *
+ * Reusable patterns for common document types.
+ * Users can create custom templates stored in ~/.okrapdf/templates/
+ */
+import { readFileSync, writeFileSync, existsSync, mkdirSync, readdirSync } from 'fs';
+import { join } from 'path';
+import { homedir } from 'os';
+// Built-in templates for common document types
+export const BUILTIN_TEMPLATES = {
+    'invoice': {
+        id: 'invoice',
+        name: 'Invoice',
+        description: 'Extract data from invoices and bills',
+        documentType: 'invoice',
+        processor: 'docai',
+        fields: [
+            { name: 'vendor_name', description: 'Vendor/supplier name', type: 'string', required: true },
+            { name: 'invoice_number', description: 'Invoice number', type: 'string', required: true },
+            { name: 'invoice_date', description: 'Invoice date', type: 'date', format: 'YYYY-MM-DD' },
+            { name: 'due_date', description: 'Payment due date', type: 'date', format: 'YYYY-MM-DD' },
+            { name: 'total_amount', description: 'Total amount due', type: 'currency', required: true },
+            { name: 'tax_amount', description: 'Tax amount', type: 'currency' },
+            { name: 'line_items', description: 'Line items', type: 'table' },
+        ],
+        systemPrompt: 'Extract invoice data. Return structured JSON with vendor_name, invoice_number, invoice_date, due_date, total_amount, tax_amount, and line_items array.',
+        builtin: true,
+    },
+    'receipt': {
+        id: 'receipt',
+        name: 'Receipt',
+        description: 'Extract data from receipts',
+        documentType: 'receipt',
+        processor: 'gemini',
+        fields: [
+            { name: 'merchant_name', description: 'Store/merchant name', type: 'string', required: true },
+            { name: 'date', description: 'Transaction date', type: 'date' },
+            { name: 'total', description: 'Total amount', type: 'currency', required: true },
+            { name: 'tax', description: 'Tax amount', type: 'currency' },
+            { name: 'items', description: 'Purchased items', type: 'array' },
+            { name: 'payment_method', description: 'Payment method', type: 'string' },
+        ],
+        builtin: true,
+    },
+    'financial-statement': {
+        id: 'financial-statement',
+        name: 'Financial Statement',
+        description: 'Extract data from financial statements (balance sheet, income statement)',
+        documentType: 'financial',
+        processor: 'llamaparse',
+        fields: [
+            { name: 'company_name', description: 'Company name', type: 'string', required: true },
+            { name: 'period', description: 'Reporting period', type: 'string', required: true },
+            { name: 'statement_type', description: 'Type of statement', type: 'string' },
+            { name: 'total_assets', description: 'Total assets', type: 'currency' },
+            { name: 'total_liabilities', description: 'Total liabilities', type: 'currency' },
+            { name: 'total_equity', description: 'Total equity', type: 'currency' },
+            { name: 'revenue', description: 'Total revenue', type: 'currency' },
+            { name: 'net_income', description: 'Net income', type: 'currency' },
+            { name: 'tables', description: 'Financial tables', type: 'table' },
+        ],
+        builtin: true,
+    },
+    'contract': {
+        id: 'contract',
+        name: 'Contract',
+        description: 'Extract key terms from contracts and agreements',
+        documentType: 'contract',
+        processor: 'qwen',
+        fields: [
+            { name: 'parties', description: 'Contracting parties', type: 'array', required: true },
+            { name: 'effective_date', description: 'Effective date', type: 'date' },
+            { name: 'termination_date', description: 'Termination/expiry date', type: 'date' },
+            { name: 'contract_value', description: 'Contract value', type: 'currency' },
+            { name: 'key_terms', description: 'Key terms and conditions', type: 'array' },
+            { name: 'obligations', description: 'Key obligations', type: 'array' },
+        ],
+        builtin: true,
+    },
+    'resume': {
+        id: 'resume',
+        name: 'Resume/CV',
+        description: 'Extract structured data from resumes',
+        documentType: 'resume',
+        processor: 'gemini',
+        fields: [
+            { name: 'name', description: 'Full name', type: 'string', required: true },
+            { name: 'email', description: 'Email address', type: 'string' },
+            { name: 'phone', description: 'Phone number', type: 'string' },
+            { name: 'location', description: 'Location', type: 'string' },
+            { name: 'summary', description: 'Professional summary', type: 'string' },
+            { name: 'experience', description: 'Work experience', type: 'array' },
+            { name: 'education', description: 'Education', type: 'array' },
+            { name: 'skills', description: 'Skills', type: 'array' },
+        ],
+        builtin: true,
+    },
+    'table': {
+        id: 'table',
+        name: 'Table Extraction',
+        description: 'Extract all tables from a document',
+        documentType: 'any',
+        processor: 'default',
+        fields: [
+            { name: 'tables', description: 'All tables in the document', type: 'table', required: true },
+        ],
+        builtin: true,
+    },
+};
+/**
+ * Get the templates directory path
+ */
+export function getTemplatesDir() {
+    const dir = join(homedir(), '.okrapdf', 'templates');
+    if (!existsSync(dir)) {
+        mkdirSync(dir, { recursive: true });
+    }
+    return dir;
+}
+/**
+ * Load a template by ID (checks user templates first, then builtins)
+ */
+export function loadTemplate(id) {
+    // Check user templates first
+    const userPath = join(getTemplatesDir(), `${id}.json`);
+    if (existsSync(userPath)) {
+        try {
+            const content = readFileSync(userPath, 'utf-8');
+            return JSON.parse(content);
+        }
+        catch {
+            // Fall through to builtin
+        }
+    }
+    // Check builtins
+    return BUILTIN_TEMPLATES[id];
+}
+/**
+ * Save a custom template
+ */
+export function saveTemplate(template) {
+    const path = join(getTemplatesDir(), `${template.id}.json`);
+    writeFileSync(path, JSON.stringify(template, null, 2), 'utf-8');
+}
+/**
+ * List all available templates (user + builtin)
+ */
+export function listTemplates() {
+    const templates = [];
+    // Add user templates
+    const userDir = getTemplatesDir();
+    if (existsSync(userDir)) {
+        const files = readdirSync(userDir).filter(f => f.endsWith('.json'));
+        for (const file of files) {
+            try {
+                const content = readFileSync(join(userDir, file), 'utf-8');
+                const template = JSON.parse(content);
+                template.builtin = false;
+                templates.push(template);
+            }
+            catch {
+                // Skip invalid templates
+            }
+        }
+    }
+    // Add builtins (that aren't overridden)
+    const userIds = new Set(templates.map(t => t.id));
+    for (const builtin of Object.values(BUILTIN_TEMPLATES)) {
+        if (!userIds.has(builtin.id)) {
+            templates.push(builtin);
+        }
+    }
+    return templates.sort((a, b) => a.id.localeCompare(b.id));
+}
+/**
+ * Delete a user template
+ */
+export function deleteTemplate(id) {
+    const path = join(getTemplatesDir(), `${id}.json`);
+    if (existsSync(path)) {
+        const { unlinkSync } = require('fs');
+        unlinkSync(path);
+        return true;
+    }
+    return false;
+}
+/**
+ * Format template for display
+ */
+export function formatTemplateInfo(template) {
+    const lines = [
+        `ID: ${template.id}${template.builtin ? ' (builtin)' : ''}`,
+        `Name: ${template.name}`,
+        `Description: ${template.description}`,
+        `Document Type: ${template.documentType}`,
+        template.processor ? `Processor: ${template.processor}` : null,
+        '',
+        'Fields:',
+        ...template.fields.map(f => `  - ${f.name} (${f.type})${f.required ? ' *required' : ''}: ${f.description}`),
+    ].filter(Boolean);
+    return lines.join('\n');
+}
+//# sourceMappingURL=templates.js.map

package/dist/lib/templates.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"templates.js","sourceRoot":"","sources":["../../src/lib/templates.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,IAAI,CAAC;AACrF,OAAO,EAAE,IAAI,EAAY,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AA4B7B,+CAA+C;AAC/C,MAAM,CAAC,MAAM,iBAAiB,GAA6B;IACzD,SAAS,EAAE;QACT,EAAE,EAAE,SAAS;QACb,IAAI,EAAE,SAAS;QACf,WAAW,EAAE,sCAAsC;QACnD,YAAY,EAAE,SAAS;QACvB,SAAS,EAAE,OAAO;QAClB,MAAM,EAAE;YACN,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,sBAAsB,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;YAC5F,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,gBAAgB,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;YACzF,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE;YACzF,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,kBAAkB,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE;YACzF,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,kBAAkB,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE;YAC3F,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE;YACnE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE;SACjE;QACD,YAAY,EAAE,wJAAwJ;QACtK,OAAO,EAAE,IAAI;KACd;IACD,SAAS,EAAE;QACT,EAAE,EAAE,SAAS;QACb,IAAI,EAAE,SAAS;QACf,WAAW,EAAE,4BAA4B;QACzC,YAAY,EAAE,SAAS;QACvB,SAAS,EAAE,QAAQ;QACnB,MAAM,EAAE;YACN,EAAE,IAAI,EAAE,eAAe,EAAE,WAAW,EAAE,qBAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;YAC7F,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,kBAAkB,EAAE,IAAI,EAAE,MAAM,EAAE;YAC/D,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE;YAChF,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE;YAC5D,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,OAAO,EAAE;YAChE,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,gBAAgB,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC1E;QACD,OAAO,EAAE,IAAI;KACd;IACD,qBAAqB,EAAE;QACrB,EAAE,EAAE,qBAAqB;QACzB,IAAI,EAAE,qBAAqB;QAC3B,WAAW,EAAE,0EAA0E;QACvF,YAAY,EAAE,WAAW;QACzB,SAAS,EAAE,YAAY;QACvB,MAAM,EAAE;YACN,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;YACrF,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,kBAAkB,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;YACnF,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,mBAAmB,EAAE,IAAI,EAAE,QAAQ,EAAE;YAC5E,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,UAAU,EAAE;YACvE,EAAE,IAAI,EAAE,mBAAmB,EAAE,WAAW,EAAE,mBAAmB,EAAE,IAAI,EAAE,UAAU,EAAE;YACjF,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,UAAU,EAAE;YACvE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,eAAe,EAAE,IAAI,EAAE,UAAU,EAAE;YACnE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE;YACnE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,kBAAkB,EAAE,IAAI,EAAE,OAAO,EAAE;SACnE;QACD,OAAO,EAAE,IAAI;KACd;IACD,UAAU,EAAE;QACV,EAAE,EAAE,UAAU;QACd,IAAI,EAAE,UAAU;QAChB,WAAW,EAAE,iDAAiD;QAC9D,YAAY,EAAE,UAAU;QACxB,SAAS,EAAE,MAAM;QACjB,MAAM,EAAE;YACN,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,qBAAqB,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE;YACtF,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,EAAE;YACvE,EAAE,IAAI,EAAE,kBAAkB,EAAE,WAAW,EAAE,yBAAyB,EAAE,IAAI,EAAE,MAAM,EAAE;YAClF,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,gBAAgB,EAAE,IAAI,EAAE,UAAU,EAAE;YAC3E,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,0BAA0B,EAAE,IAAI,EAAE,OAAO,EAAE;YAC7E,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,OAAO,EAAE;SACvE;QACD,OAAO,EAAE,IAAI;KACd;IACD,QAAQ,EAAE;QACR,EAAE,EAAE,QAAQ;QACZ,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,sCAAsC;QACnD,YAAY,EAAE,QAAQ;QACtB,SAAS,EAAE,QAAQ;QACnB,MAAM,EAAE;YACN,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;YAC1E,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,IAAI,EAAE,QAAQ,EAAE;YAC/D,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,QAAQ,EAAE;YAC9D,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE;YAC7D,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,sBAAsB,EAAE,IAAI,EAAE,QAAQ,EAAE;YACxE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,OAAO,EAAE;YACrE,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE;YAC9D,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE;SACzD;QACD,OAAO,EAAE,IAAI;KACd;IACD,OAAO,EAAE;QACP,EAAE,EAAE,OAAO;QACX,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE,oCAAoC;QACjD,YAAY,EAAE,KAAK;QACnB,SAAS,EAAE,SAAS;QACpB,MAAM,EAAE;YACN,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,4BAA4B,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE;SAC7F;QACD,OAAO,EAAE,IAAI;KACd;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,eAAe;IAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;IACrD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACrB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,EAAU;IACrC,6BAA6B;IAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IACvD,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAa,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,0BAA0B;QAC5B,CAAC;IACH,CAAC;IAED,iBAAiB;IACjB,OAAO,iBAAiB,CAAC,EAAE,CAAC,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,QAAkB;IAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,EAAE,EAAE,GAAG,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAC;IAC5D,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;AAClE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa;IAC3B,MAAM,SAAS,GAAe,EAAE,CAAC;IAEjC,qBAAqB;IACrB,MAAM,OAAO,GAAG,eAAe,EAAE,CAAC;IAClC,IAAI,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QACxB,MAAM,KAAK,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QACpE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;gBAC3D,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAa,CAAC;gBACjD,QAAQ,CAAC,OAAO,GAAG,KAAK,CAAC;gBACzB,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC3B,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,wCAAwC;IACxC,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAClD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACvD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;YAC7B,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,EAAU;IACvC,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IACnD,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACrB,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QACrC,UAAU,CAAC,IAAI,CAAC,CAAC;QACjB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAkB;IACnD,MAAM,KAAK,GAAG;QACZ,OAAO,QAAQ,CAAC,EAAE,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,EAAE;QAC3D,SAAS,QAAQ,CAAC,IAAI,EAAE;QACxB,gBAAgB,QAAQ,CAAC,WAAW,EAAE;QACtC,kBAAkB,QAAQ,CAAC,YAAY,EAAE;QACzC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI;QAC9D,EAAE;QACF,SAAS;QACT,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CACzB,OAAO,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,WAAW,EAAE,CAC/E;KACF,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAElB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/lib/templates.test.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Tests for extraction templates
+ */
+export {};
+//# sourceMappingURL=templates.test.d.ts.map

package/dist/lib/templates.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"templates.test.d.ts","sourceRoot":"","sources":["../../src/lib/templates.test.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/lib/templates.test.js

@@ -0,0 +1,101 @@
+/**
+ * Tests for extraction templates
+ */
+import { describe, it, expect } from 'vitest';
+import { BUILTIN_TEMPLATES, loadTemplate, listTemplates, formatTemplateInfo, } from './templates.js';
+describe('templates', () => {
+    describe('BUILTIN_TEMPLATES', () => {
+        it('should have expected templates', () => {
+            expect(BUILTIN_TEMPLATES).toHaveProperty('invoice');
+            expect(BUILTIN_TEMPLATES).toHaveProperty('receipt');
+            expect(BUILTIN_TEMPLATES).toHaveProperty('financial-statement');
+            expect(BUILTIN_TEMPLATES).toHaveProperty('contract');
+            expect(BUILTIN_TEMPLATES).toHaveProperty('resume');
+            expect(BUILTIN_TEMPLATES).toHaveProperty('table');
+        });
+        it('should have required fields for each template', () => {
+            for (const [id, tpl] of Object.entries(BUILTIN_TEMPLATES)) {
+                expect(tpl.id).toBe(id);
+                expect(tpl.name).toBeTruthy();
+                expect(tpl.description).toBeTruthy();
+                expect(tpl.documentType).toBeTruthy();
+                expect(Array.isArray(tpl.fields)).toBe(true);
+                expect(tpl.builtin).toBe(true);
+            }
+        });
+        it('invoice template should have expected fields', () => {
+            const invoice = BUILTIN_TEMPLATES.invoice;
+            const fieldNames = invoice.fields.map(f => f.name);
+            expect(fieldNames).toContain('vendor_name');
+            expect(fieldNames).toContain('invoice_number');
+            expect(fieldNames).toContain('total_amount');
+        });
+        it('receipt template should have expected fields', () => {
+            const receipt = BUILTIN_TEMPLATES.receipt;
+            const fieldNames = receipt.fields.map(f => f.name);
+            expect(fieldNames).toContain('merchant_name');
+            expect(fieldNames).toContain('total');
+            expect(fieldNames).toContain('items');
+        });
+    });
+    describe('loadTemplate', () => {
+        it('should load builtin templates', () => {
+            const invoice = loadTemplate('invoice');
+            expect(invoice).toBeDefined();
+            expect(invoice?.id).toBe('invoice');
+            expect(invoice?.builtin).toBe(true);
+        });
+        it('should return undefined for unknown templates', () => {
+            expect(loadTemplate('nonexistent')).toBeUndefined();
+        });
+    });
+    describe('listTemplates', () => {
+        it('should return all builtin templates', () => {
+            const templates = listTemplates();
+            const builtinCount = Object.keys(BUILTIN_TEMPLATES).length;
+            // Should have at least all builtins
+            expect(templates.length).toBeGreaterThanOrEqual(builtinCount);
+            // All builtins should be present
+            for (const id of Object.keys(BUILTIN_TEMPLATES)) {
+                expect(templates.find(t => t.id === id)).toBeDefined();
+            }
+        });
+        it('should return sorted templates', () => {
+            const templates = listTemplates();
+            const ids = templates.map(t => t.id);
+            const sortedIds = [...ids].sort();
+            expect(ids).toEqual(sortedIds);
+        });
+    });
+    describe('formatTemplateInfo', () => {
+        it('should format template info as string', () => {
+            const invoice = BUILTIN_TEMPLATES.invoice;
+            const formatted = formatTemplateInfo(invoice);
+            expect(formatted).toContain('invoice');
+            expect(formatted).toContain('Invoice');
+            expect(formatted).toContain('Fields:');
+            expect(formatted).toContain('vendor_name');
+        });
+        it('should indicate builtin status', () => {
+            const invoice = BUILTIN_TEMPLATES.invoice;
+            const formatted = formatTemplateInfo(invoice);
+            expect(formatted).toContain('builtin');
+        });
+    });
+    describe('template fields', () => {
+        it('should have valid field types', () => {
+            const validTypes = ['string', 'number', 'date', 'currency', 'array', 'table'];
+            for (const tpl of Object.values(BUILTIN_TEMPLATES)) {
+                for (const field of tpl.fields) {
+                    expect(validTypes).toContain(field.type);
+                }
+            }
+        });
+        it('required fields should be marked', () => {
+            const invoice = BUILTIN_TEMPLATES.invoice;
+            const vendorField = invoice.fields.find(f => f.name === 'vendor_name');
+            expect(vendorField?.required).toBe(true);
+        });
+    });
+});
+//# sourceMappingURL=templates.test.js.map

package/dist/lib/templates.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"templates.test.js","sourceRoot":"","sources":["../../src/lib/templates.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAyB,MAAM,QAAQ,CAAC;AAIrE,OAAO,EACL,iBAAiB,EACjB,YAAY,EAEZ,aAAa,EAEb,kBAAkB,GAEnB,MAAM,gBAAgB,CAAC;AAExB,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,iBAAiB,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YACpD,MAAM,CAAC,iBAAiB,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YACpD,MAAM,CAAC,iBAAiB,CAAC,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;YAChE,MAAM,CAAC,iBAAiB,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YACrD,MAAM,CAAC,iBAAiB,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,CAAC,iBAAiB,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,KAAK,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBAC1D,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACxB,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;gBAC9B,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;gBACrC,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC;gBACtC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC7C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC;YAC1C,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAEnD,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YAC5C,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC/C,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC;YAC1C,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAEnD,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,OAAO,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YAC9B,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACpC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;YAClC,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,MAAM,CAAC;YAE3D,oCAAoC;YACpC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC;YAE9D,iCAAiC;YACjC,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBAChD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACzD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;YAClC,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,SAAS,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YAElC,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC;YAC1C,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAE9C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC;YAC1C,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAE9C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,UAAU,GAAG,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAE9E,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACnD,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;oBAC/B,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3C,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC;YAC1C,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;YAEvE,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}