edgeflowjs 0.1.0

package/dist/pipelines/feature-extraction.js

@@ -0,0 +1,197 @@
+/**
+ * edgeFlow.js - Feature Extraction Pipeline
+ *
+ * Extract embeddings/features from text, images, or other data.
+ */
+import { EdgeFlowTensor } from '../core/tensor.js';
+import { createBasicTokenizer } from '../utils/tokenizer.js';
+import { BasePipeline, registerPipeline, } from './base.js';
+/**
+ * FeatureExtractionPipeline - Extract embeddings from text
+ */
+export class FeatureExtractionPipeline extends BasePipeline {
+    tokenizer = null;
+    embeddingDim;
+    constructor(config, embeddingDim = 768) {
+        super(config);
+        this.embeddingDim = embeddingDim;
+    }
+    /**
+     * Initialize pipeline
+     */
+    async initialize() {
+        await super.initialize();
+        if (!this.tokenizer) {
+            this.tokenizer = createBasicTokenizer();
+        }
+    }
+    /**
+     * Run feature extraction
+     */
+    async run(input, options) {
+        const isBatch = Array.isArray(input);
+        const inputs = isBatch ? input : [input];
+        await this.initialize();
+        const startTime = performance.now();
+        const results = [];
+        for (const text of inputs) {
+            // Preprocess
+            const tensorInputs = await this.preprocess(text);
+            // Run inference
+            const outputs = await this.runInference(tensorInputs);
+            // Postprocess
+            const result = await this.postprocess(outputs, options);
+            results.push(result);
+        }
+        const processingTime = performance.now() - startTime;
+        for (const result of results) {
+            result.processingTime = processingTime / results.length;
+        }
+        return isBatch ? results : results[0];
+    }
+    /**
+     * Preprocess text input
+     */
+    async preprocess(input) {
+        const text = Array.isArray(input) ? input[0] : input;
+        const encoded = this.tokenizer.encode(text, {
+            maxLength: 128,
+            padding: 'max_length',
+            truncation: true,
+        });
+        const inputIds = new EdgeFlowTensor(new Float32Array(encoded.inputIds), [1, encoded.inputIds.length], 'float32');
+        const attentionMask = new EdgeFlowTensor(new Float32Array(encoded.attentionMask), [1, encoded.attentionMask.length], 'float32');
+        return [inputIds, attentionMask];
+    }
+    /**
+     * Run model inference
+     */
+    async runInference(inputs) {
+        // Generate mock embeddings for demo
+        // In production, this would call the actual model
+        const seqLen = inputs[0]?.shape[1] ?? 128;
+        const embeddings = new Float32Array(seqLen * this.embeddingDim);
+        // Generate deterministic pseudo-embeddings based on input
+        const inputData = inputs[0]?.toFloat32Array() ?? new Float32Array(0);
+        for (let i = 0; i < seqLen; i++) {
+            for (let j = 0; j < this.embeddingDim; j++) {
+                const inputVal = inputData[i] ?? 0;
+                embeddings[i * this.embeddingDim + j] =
+                    Math.sin(inputVal * (j + 1) * 0.01) * 0.1;
+            }
+        }
+        return [new EdgeFlowTensor(embeddings, [1, seqLen, this.embeddingDim], 'float32')];
+    }
+    /**
+     * Postprocess model outputs
+     */
+    async postprocess(outputs, options) {
+        const hiddenStates = outputs[0];
+        if (!hiddenStates) {
+            return { embeddings: [] };
+        }
+        const pooling = options?.pooling ?? 'mean';
+        const normalize = options?.normalize ?? true;
+        let embeddings;
+        switch (pooling) {
+            case 'cls':
+                // Use first token (CLS) embedding
+                embeddings = this.extractCLSEmbedding(hiddenStates);
+                break;
+            case 'max':
+                // Max pooling
+                embeddings = this.maxPooling(hiddenStates);
+                break;
+            case 'none':
+                // Return all token embeddings (flattened)
+                embeddings = hiddenStates.toArray();
+                break;
+            case 'mean':
+            default:
+                // Mean pooling
+                embeddings = this.meanPooling(hiddenStates);
+                break;
+        }
+        // Normalize if requested
+        if (normalize) {
+            embeddings = this.normalizeVector(embeddings);
+        }
+        // Dimension reduction if requested
+        if (options?.outputDim && options.outputDim < embeddings.length) {
+            embeddings = embeddings.slice(0, options.outputDim);
+        }
+        return { embeddings };
+    }
+    /**
+     * Extract CLS token embedding
+     */
+    extractCLSEmbedding(hiddenStates) {
+        const data = hiddenStates.toFloat32Array();
+        const embeddingDim = hiddenStates.shape[2] ?? this.embeddingDim;
+        return Array.from(data.slice(0, embeddingDim));
+    }
+    /**
+     * Mean pooling over sequence
+     */
+    meanPooling(hiddenStates) {
+        const data = hiddenStates.toFloat32Array();
+        const seqLen = hiddenStates.shape[1] ?? 1;
+        const embeddingDim = hiddenStates.shape[2] ?? this.embeddingDim;
+        const result = new Float32Array(embeddingDim);
+        for (let i = 0; i < seqLen; i++) {
+            for (let j = 0; j < embeddingDim; j++) {
+                result[j] = (result[j] ?? 0) + (data[i * embeddingDim + j] ?? 0) / seqLen;
+            }
+        }
+        return Array.from(result);
+    }
+    /**
+     * Max pooling over sequence
+     */
+    maxPooling(hiddenStates) {
+        const data = hiddenStates.toFloat32Array();
+        const seqLen = hiddenStates.shape[1] ?? 1;
+        const embeddingDim = hiddenStates.shape[2] ?? this.embeddingDim;
+        const result = new Array(embeddingDim).fill(-Infinity);
+        for (let i = 0; i < seqLen; i++) {
+            for (let j = 0; j < embeddingDim; j++) {
+                const val = data[i * embeddingDim + j] ?? 0;
+                if (val > (result[j] ?? -Infinity)) {
+                    result[j] = val;
+                }
+            }
+        }
+        return result;
+    }
+    /**
+     * L2 normalize vector
+     */
+    normalizeVector(vec) {
+        let norm = 0;
+        for (const v of vec) {
+            norm += v * v;
+        }
+        norm = Math.sqrt(norm);
+        if (norm === 0)
+            return vec;
+        return vec.map(v => v / norm);
+    }
+}
+// ============================================================================
+// Factory Function
+// ============================================================================
+/**
+ * Create feature extraction pipeline
+ */
+export function createFeatureExtractionPipeline(config = {}) {
+    return new FeatureExtractionPipeline({
+        task: 'feature-extraction',
+        model: config.model ?? 'default',
+        runtime: config.runtime,
+        cache: config.cache ?? true,
+        quantization: config.quantization,
+    });
+}
+// Register pipeline
+registerPipeline('feature-extraction', (config) => new FeatureExtractionPipeline(config));
+//# sourceMappingURL=feature-extraction.js.map

package/dist/pipelines/feature-extraction.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"feature-extraction.js","sourceRoot":"","sources":["../../src/pipelines/feature-extraction.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAa,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AACxE,OAAO,EACL,YAAY,EAEZ,gBAAgB,GACjB,MAAM,WAAW,CAAC;AAkBnB;;GAEG;AACH,MAAM,OAAO,yBAA0B,SAAQ,YAG9C;IACS,SAAS,GAAqB,IAAI,CAAC;IACnC,YAAY,CAAS;IAE7B,YAAY,MAAsB,EAAE,eAAuB,GAAG;QAC5D,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAED;;OAEG;IACM,KAAK,CAAC,UAAU;QACvB,MAAM,KAAK,CAAC,UAAU,EAAE,CAAC;QAEzB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,oBAAoB,EAAE,CAAC;QAC1C,CAAC;IACH,CAAC;IAED;;OAEG;IACM,KAAK,CAAC,GAAG,CAChB,KAAwB,EACxB,OAAkC;QAElC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAEzC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,OAAO,GAA8B,EAAE,CAAC;QAE9C,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAC1B,aAAa;YACb,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAEjD,gBAAgB;YAChB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;YAEtD,cAAc;YACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACxD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAErD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,CAAC,cAAc,GAAG,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;QAC1D,CAAC;QAED,OAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACgB,KAAK,CAAC,UAAU,CAAC,KAAwB;QAC1D,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,KAAK,CAAC;QAEtD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAU,CAAC,MAAM,CAAC,IAAI,EAAE;YAC3C,SAAS,EAAE,GAAG;YACd,OAAO,EAAE,YAAY;YACrB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,CACjC,IAAI,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,EAClC,CAAC,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC5B,SAAS,CACV,CAAC;QAEF,MAAM,aAAa,GAAG,IAAI,cAAc,CACtC,IAAI,YAAY,CAAC,OAAO,CAAC,aAAa,CAAC,EACvC,CAAC,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,EACjC,SAAS,CACV,CAAC;QAEF,OAAO,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY,CAAC,MAAwB;QACjD,oCAAoC;QACpC,kDAAkD;QAClD,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC;QAEhE,0DAA0D;QAC1D,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QAErE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3C,MAAM,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACnC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;oBACnC,IAAI,CAAC,GAAG,CAAC,QAAQ,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,GAAG,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,OAAO,CAAC,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC;IACrF,CAAC;IAED;;OAEG;IACgB,KAAK,CAAC,WAAW,CAClC,OAAyB,EACzB,OAAkC;QAElC,MAAM,YAAY,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAChC,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAC5B,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,MAAM,CAAC;QAC3C,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC;QAE7C,IAAI,UAAoB,CAAC;QAEzB,QAAQ,OAAO,EAAE,CAAC;YAChB,KAAK,KAAK;gBACR,kCAAkC;gBAClC,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;gBACpD,MAAM;YACR,KAAK,KAAK;gBACR,cAAc;gBACd,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;gBAC3C,MAAM;YACR,KAAK,MAAM;gBACT,0CAA0C;gBAC1C,UAAU,GAAG,YAAY,CAAC,OAAO,EAAE,CAAC;gBACpC,MAAM;YACR,KAAK,MAAM,CAAC;YACZ;gBACE,eAAe;gBACf,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;gBAC5C,MAAM;QACV,CAAC;QAED,yBAAyB;QACzB,IAAI,SAAS,EAAE,CAAC;YACd,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QAChD,CAAC;QAED,mCAAmC;QACnC,IAAI,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,SAAS,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC;YAChE,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QACtD,CAAC;QAED,OAAO,EAAE,UAAU,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,YAA4B;QACtD,MAAM,IAAI,GAAG,YAAY,CAAC,cAAc,EAAE,CAAC;QAC3C,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC;QAChE,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,YAA4B;QAC9C,MAAM,IAAI,GAAG,YAAY,CAAC,cAAc,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC;QAEhE,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;QAE9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,YAAY,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC;YAC5E,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,YAA4B;QAC7C,MAAM,IAAI,GAAG,YAAY,CAAC,cAAc,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC;QAEhE,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAa,CAAC;QAEnE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,GAAG,YAAY,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC5C,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACnC,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;gBAClB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,GAAa;QACnC,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,KAAK,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;QAChB,CAAC;QACD,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEvB,IAAI,IAAI,KAAK,CAAC;YAAE,OAAO,GAAG,CAAC;QAE3B,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAChC,CAAC;CACF;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,+BAA+B,CAC7C,SAAkC,EAAE;IAEpC,OAAO,IAAI,yBAAyB,CAAC;QACnC,IAAI,EAAE,oBAAoB;QAC1B,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,SAAS;QAChC,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI;QAC3B,YAAY,EAAE,MAAM,CAAC,YAAY;KAClC,CAAC,CAAC;AACL,CAAC;AAED,oBAAoB;AACpB,gBAAgB,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC"}

package/dist/pipelines/image-classification.d.ts

@@ -0,0 +1,61 @@
+/**
+ * edgeFlow.js - Image Classification Pipeline
+ *
+ * Classify images into categories using vision models.
+ */
+import { PipelineConfig, PipelineOptions } from '../core/types.js';
+import { EdgeFlowTensor } from '../core/tensor.js';
+import { BasePipeline, ImageClassificationResult } from './base.js';
+/**
+ * Image classification options
+ */
+export interface ImageClassificationOptions extends PipelineOptions {
+    /** Return all labels with scores */
+    returnAllScores?: boolean;
+    /** Custom labels */
+    labels?: string[];
+    /** Number of top predictions to return */
+    topK?: number;
+}
+/**
+ * Image classification input types
+ */
+export type ImageInput = HTMLImageElement | HTMLCanvasElement | ImageBitmap | ImageData | string;
+/**
+ * ImageClassificationPipeline - Classify images
+ */
+export declare class ImageClassificationPipeline extends BasePipeline<ImageInput | ImageInput[], ImageClassificationResult | ImageClassificationResult[]> {
+    private preprocessor;
+    private labels;
+    private numClasses;
+    constructor(config: PipelineConfig, labels?: string[], numClasses?: number);
+    /**
+     * Initialize pipeline
+     */
+    initialize(): Promise<void>;
+    /**
+     * Set custom labels
+     */
+    setLabels(labels: string[]): void;
+    /**
+     * Run classification
+     */
+    run(input: ImageInput | ImageInput[], options?: ImageClassificationOptions): Promise<ImageClassificationResult | ImageClassificationResult[]>;
+    /**
+     * Preprocess image input
+     */
+    protected preprocess(input: ImageInput | ImageInput[]): Promise<EdgeFlowTensor[]>;
+    /**
+     * Run model inference
+     */
+    private runInference;
+    /**
+     * Postprocess model outputs
+     */
+    protected postprocess(outputs: EdgeFlowTensor[], options?: ImageClassificationOptions): Promise<ImageClassificationResult>;
+}
+/**
+ * Create image classification pipeline
+ */
+export declare function createImageClassificationPipeline(config?: Partial<PipelineConfig>, labels?: string[]): ImageClassificationPipeline;
+//# sourceMappingURL=image-classification.d.ts.map

package/dist/pipelines/image-classification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-classification.d.ts","sourceRoot":"","sources":["../../src/pipelines/image-classification.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,cAAc,EAAW,MAAM,mBAAmB,CAAC;AAE5D,OAAO,EACL,YAAY,EACZ,yBAAyB,EAG1B,MAAM,WAAW,CAAC;AAMnB;;GAEG;AACH,MAAM,WAAW,0BAA2B,SAAQ,eAAe;IACjE,oCAAoC;IACpC,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,oBAAoB;IACpB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,0CAA0C;IAC1C,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,MAAM,MAAM,UAAU,GAClB,gBAAgB,GAChB,iBAAiB,GACjB,WAAW,GACX,SAAS,GACT,MAAM,CAAC;AAEX;;GAEG;AACH,qBAAa,2BAA4B,SAAQ,YAAY,CAC3D,UAAU,GAAG,UAAU,EAAE,EACzB,yBAAyB,GAAG,yBAAyB,EAAE,CACxD;IACC,OAAO,CAAC,YAAY,CAAkC;IACtD,OAAO,CAAC,MAAM,CAAW;IACzB,OAAO,CAAC,UAAU,CAAS;gBAGzB,MAAM,EAAE,cAAc,EACtB,MAAM,CAAC,EAAE,MAAM,EAAE,EACjB,UAAU,GAAE,MAAa;IAO3B;;OAEG;IACY,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ1C;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI;IAKjC;;OAEG;IACY,GAAG,CAChB,KAAK,EAAE,UAAU,GAAG,UAAU,EAAE,EAChC,OAAO,CAAC,EAAE,0BAA0B,GACnC,OAAO,CAAC,yBAAyB,GAAG,yBAAyB,EAAE,CAAC;IA8BnE;;OAEG;cACsB,UAAU,CAAC,KAAK,EAAE,UAAU,GAAG,UAAU,EAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAchG;;OAEG;YACW,YAAY;IAmB1B;;OAEG;cACsB,WAAW,CAClC,OAAO,EAAE,cAAc,EAAE,EACzB,OAAO,CAAC,EAAE,0BAA0B,GACnC,OAAO,CAAC,yBAAyB,CAAC;CAkCtC;AAMD;;GAEG;AACH,wBAAgB,iCAAiC,CAC/C,MAAM,GAAE,OAAO,CAAC,cAAc,CAAM,EACpC,MAAM,CAAC,EAAE,MAAM,EAAE,GAChB,2BAA2B,CAW7B"}

package/dist/pipelines/image-classification.js

@@ -0,0 +1,140 @@
+/**
+ * edgeFlow.js - Image Classification Pipeline
+ *
+ * Classify images into categories using vision models.
+ */
+import { EdgeFlowTensor, softmax } from '../core/tensor.js';
+import { createImagePreprocessor } from '../utils/preprocessor.js';
+import { BasePipeline, registerPipeline, IMAGENET_LABELS, } from './base.js';
+/**
+ * ImageClassificationPipeline - Classify images
+ */
+export class ImageClassificationPipeline extends BasePipeline {
+    preprocessor = null;
+    labels;
+    numClasses;
+    constructor(config, labels, numClasses = 1000) {
+        super(config);
+        this.labels = labels ?? IMAGENET_LABELS;
+        this.numClasses = numClasses;
+    }
+    /**
+     * Initialize pipeline
+     */
+    async initialize() {
+        await super.initialize();
+        if (!this.preprocessor) {
+            this.preprocessor = createImagePreprocessor('imagenet');
+        }
+    }
+    /**
+     * Set custom labels
+     */
+    setLabels(labels) {
+        this.labels = labels;
+        this.numClasses = labels.length;
+    }
+    /**
+     * Run classification
+     */
+    async run(input, options) {
+        const isBatch = Array.isArray(input);
+        const inputs = isBatch ? input : [input];
+        await this.initialize();
+        const startTime = performance.now();
+        const results = [];
+        for (const image of inputs) {
+            // Preprocess
+            const tensorInputs = await this.preprocess(image);
+            // Run inference
+            const outputs = await this.runInference(tensorInputs);
+            // Postprocess
+            const result = await this.postprocess(outputs, options);
+            results.push(result);
+        }
+        const processingTime = performance.now() - startTime;
+        for (const result of results) {
+            result.processingTime = processingTime / results.length;
+        }
+        return isBatch ? results : results[0];
+    }
+    /**
+     * Preprocess image input
+     */
+    async preprocess(input) {
+        const image = Array.isArray(input) ? input[0] : input;
+        // Process image
+        const tensor = await this.preprocessor.process(image);
+        // Add batch dimension if needed
+        if (tensor.shape.length === 3) {
+            return [tensor.reshape([1, ...tensor.shape])];
+        }
+        return [tensor];
+    }
+    /**
+     * Run model inference
+     */
+    async runInference(inputs) {
+        // Generate mock classification logits for demo
+        // In production, this would call the actual model
+        const logits = new Float32Array(this.numClasses);
+        // Generate deterministic pseudo-logits based on input
+        const inputData = inputs[0]?.toFloat32Array() ?? new Float32Array(0);
+        let sum = 0;
+        for (let i = 0; i < Math.min(1000, inputData.length); i++) {
+            sum += inputData[i] ?? 0;
+        }
+        for (let i = 0; i < this.numClasses; i++) {
+            logits[i] = Math.sin(sum * (i + 1) * 0.1) * 3;
+        }
+        return [new EdgeFlowTensor(logits, [1, this.numClasses], 'float32')];
+    }
+    /**
+     * Postprocess model outputs
+     */
+    async postprocess(outputs, options) {
+        const logits = outputs[0];
+        if (!logits) {
+            return { label: 'unknown', score: 0 };
+        }
+        // Apply softmax
+        const probs = softmax(logits, -1);
+        const probsArray = probs.toFloat32Array();
+        const topK = options?.topK ?? 1;
+        if (topK > 1 || options?.returnAllScores) {
+            // Return top-K results (simplified to top-1 here)
+        }
+        // Find argmax
+        let maxIdx = 0;
+        let maxScore = probsArray[0] ?? 0;
+        for (let i = 1; i < probsArray.length; i++) {
+            if ((probsArray[i] ?? 0) > maxScore) {
+                maxScore = probsArray[i] ?? 0;
+                maxIdx = i;
+            }
+        }
+        const label = options?.labels?.[maxIdx] ?? this.labels[maxIdx] ?? `class_${maxIdx}`;
+        return {
+            label,
+            score: maxScore,
+        };
+    }
+}
+// ============================================================================
+// Factory Function
+// ============================================================================
+/**
+ * Create image classification pipeline
+ */
+export function createImageClassificationPipeline(config = {}, labels) {
+    return new ImageClassificationPipeline({
+        task: 'image-classification',
+        model: config.model ?? 'default',
+        runtime: config.runtime,
+        cache: config.cache ?? true,
+        quantization: config.quantization,
+    }, labels);
+}
+// Register pipeline
+registerPipeline('image-classification', (config) => new ImageClassificationPipeline(config));
+//# sourceMappingURL=image-classification.js.map

package/dist/pipelines/image-classification.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-classification.js","sourceRoot":"","sources":["../../src/pipelines/image-classification.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EAAqB,uBAAuB,EAAE,MAAM,0BAA0B,CAAC;AACtF,OAAO,EACL,YAAY,EAEZ,gBAAgB,EAChB,eAAe,GAChB,MAAM,WAAW,CAAC;AA4BnB;;GAEG;AACH,MAAM,OAAO,2BAA4B,SAAQ,YAGhD;IACS,YAAY,GAA6B,IAAI,CAAC;IAC9C,MAAM,CAAW;IACjB,UAAU,CAAS;IAE3B,YACE,MAAsB,EACtB,MAAiB,EACjB,aAAqB,IAAI;QAEzB,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,eAAe,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED;;OAEG;IACM,KAAK,CAAC,UAAU;QACvB,MAAM,KAAK,CAAC,UAAU,EAAE,CAAC;QAEzB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,IAAI,CAAC,YAAY,GAAG,uBAAuB,CAAC,UAAU,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,MAAgB;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC;IAClC,CAAC;IAED;;OAEG;IACM,KAAK,CAAC,GAAG,CAChB,KAAgC,EAChC,OAAoC;QAEpC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAEzC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,OAAO,GAAgC,EAAE,CAAC;QAEhD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,aAAa;YACb,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAElD,gBAAgB;YAChB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;YAEtD,cAAc;YACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACxD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAErD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,CAAC,cAAc,GAAG,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;QAC1D,CAAC;QAED,OAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACgB,KAAK,CAAC,UAAU,CAAC,KAAgC;QAClE,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,KAAK,CAAC;QAEvD,gBAAgB;QAChB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAa,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAEvD,gCAAgC;QAChC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY,CAAC,MAAwB;QACjD,+CAA+C;QAC/C,kDAAkD;QAClD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAEjD,sDAAsD;QACtD,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACrE,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1D,GAAG,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,CAAC,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC;IACvE,CAAC;IAED;;OAEG;IACgB,KAAK,CAAC,WAAW,CAClC,OAAyB,EACzB,OAAoC;QAEpC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACxC,CAAC;QAED,gBAAgB;QAChB,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAmB,CAAC;QACpD,MAAM,UAAU,GAAG,KAAK,CAAC,cAAc,EAAE,CAAC;QAE1C,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,CAAC,CAAC;QAEhC,IAAI,IAAI,GAAG,CAAC,IAAI,OAAO,EAAE,eAAe,EAAE,CAAC;YACzC,kDAAkD;QACpD,CAAC;QAED,cAAc;QACd,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,QAAQ,EAAE,CAAC;gBACpC,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC9B,MAAM,GAAG,CAAC,CAAC;YACb,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,SAAS,MAAM,EAAE,CAAC;QAEpF,OAAO;YACL,KAAK;YACL,KAAK,EAAE,QAAQ;SAChB,CAAC;IACJ,CAAC;CACF;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,iCAAiC,CAC/C,SAAkC,EAAE,EACpC,MAAiB;IAEjB,OAAO,IAAI,2BAA2B,CACpC;QACE,IAAI,EAAE,sBAAsB;QAC5B,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,SAAS;QAChC,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI;QAC3B,YAAY,EAAE,MAAM,CAAC,YAAY;KAClC,EACD,MAAM,CACP,CAAC;AACJ,CAAC;AAED,oBAAoB;AACpB,gBAAgB,CAAC,sBAAsB,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,2BAA2B,CAAC,MAAM,CAAC,CAAC,CAAC"}

package/dist/pipelines/index.d.ts

@@ -0,0 +1,58 @@
+/**
+ * edgeFlow.js - Pipeline Exports
+ */
+import { RuntimeType, QuantizationType } from '../core/types.js';
+export { BasePipeline, registerPipeline, getPipelineFactory, SENTIMENT_LABELS, EMOTION_LABELS, IMAGENET_LABELS, type PipelineResult, type TextClassificationResult, type FeatureExtractionResult, type ImageClassificationResult, type ObjectDetectionResult, } from './base.js';
+export { TextClassificationPipeline, SentimentAnalysisPipeline, createTextClassificationPipeline, createSentimentAnalysisPipeline, type TextClassificationOptions, } from './text-classification.js';
+export { FeatureExtractionPipeline, createFeatureExtractionPipeline, type FeatureExtractionOptions, } from './feature-extraction.js';
+export { ImageClassificationPipeline, createImageClassificationPipeline, type ImageClassificationOptions, type ImageInput, } from './image-classification.js';
+/**
+ * Pipeline options for the factory function
+ */
+export interface PipelineFactoryOptions {
+    /** Model ID or URL */
+    model?: string;
+    /** Runtime to use */
+    runtime?: RuntimeType;
+    /** Enable caching */
+    cache?: boolean;
+    /** Quantization type */
+    quantization?: QuantizationType;
+    /** Custom labels for classification */
+    labels?: string[];
+}
+/**
+ * Supported pipeline task mapping
+ */
+type PipelineTaskMap = {
+    'text-classification': TextClassificationPipeline;
+    'sentiment-analysis': SentimentAnalysisPipeline;
+    'feature-extraction': FeatureExtractionPipeline;
+    'image-classification': ImageClassificationPipeline;
+};
+import { TextClassificationPipeline, SentimentAnalysisPipeline } from './text-classification.js';
+import { FeatureExtractionPipeline } from './feature-extraction.js';
+import { ImageClassificationPipeline } from './image-classification.js';
+/**
+ * Create a pipeline for a specific task
+ *
+ * @example
+ * ```typescript
+ * // Create a sentiment analysis pipeline
+ * const sentiment = await pipeline('sentiment-analysis');
+ * const result = await sentiment.run('I love this product!');
+ *
+ * // Create an image classifier with custom model
+ * const classifier = await pipeline('image-classification', {
+ *   model: 'https://example.com/model.bin',
+ * });
+ * ```
+ */
+export declare function pipeline<T extends keyof PipelineTaskMap>(task: T, options?: PipelineFactoryOptions): Promise<PipelineTaskMap[T]>;
+/**
+ * Create multiple pipelines at once
+ */
+export declare function createPipelines<T extends (keyof PipelineTaskMap)[]>(tasks: T, options?: PipelineFactoryOptions): Promise<{
+    [K in T[number]]: PipelineTaskMap[K];
+}>;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/pipelines/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAGL,WAAW,EACX,gBAAgB,EACjB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,cAAc,EACd,eAAe,EACf,KAAK,cAAc,EACnB,KAAK,wBAAwB,EAC7B,KAAK,uBAAuB,EAC5B,KAAK,yBAAyB,EAC9B,KAAK,qBAAqB,GAC3B,MAAM,WAAW,CAAC;AAGnB,OAAO,EACL,0BAA0B,EAC1B,yBAAyB,EACzB,gCAAgC,EAChC,+BAA+B,EAC/B,KAAK,yBAAyB,GAC/B,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,KAAK,wBAAwB,GAC9B,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EACL,2BAA2B,EAC3B,iCAAiC,EACjC,KAAK,0BAA0B,EAC/B,KAAK,UAAU,GAChB,MAAM,2BAA2B,CAAC;AAMnC;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,sBAAsB;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,qBAAqB;IACrB,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,qBAAqB;IACrB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,wBAAwB;IACxB,YAAY,CAAC,EAAE,gBAAgB,CAAC;IAChC,uCAAuC;IACvC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CACnB;AAED;;GAEG;AACH,KAAK,eAAe,GAAG;IACrB,qBAAqB,EAAE,0BAA0B,CAAC;IAClD,oBAAoB,EAAE,yBAAyB,CAAC;IAChD,oBAAoB,EAAE,yBAAyB,CAAC;IAChD,sBAAsB,EAAE,2BAA2B,CAAC;CACrD,CAAC;AAGF,OAAO,EAAE,0BAA0B,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACjG,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AAExE;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,QAAQ,CAAC,CAAC,SAAS,MAAM,eAAe,EAC5D,IAAI,EAAE,CAAC,EACP,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAgC7B;AAED;;GAEG;AACH,wBAAsB,eAAe,CAAC,CAAC,SAAS,CAAC,MAAM,eAAe,CAAC,EAAE,EACvE,KAAK,EAAE,CAAC,EACR,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC;KAAG,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,GAAG,eAAe,CAAC,CAAC,CAAC;CAAE,CAAC,CAanD"}

package/dist/pipelines/index.js

@@ -0,0 +1,72 @@
+/**
+ * edgeFlow.js - Pipeline Exports
+ */
+// Base
+export { BasePipeline, registerPipeline, getPipelineFactory, SENTIMENT_LABELS, EMOTION_LABELS, IMAGENET_LABELS, } from './base.js';
+// Text Classification
+export { TextClassificationPipeline, SentimentAnalysisPipeline, createTextClassificationPipeline, createSentimentAnalysisPipeline, } from './text-classification.js';
+// Feature Extraction
+export { FeatureExtractionPipeline, createFeatureExtractionPipeline, } from './feature-extraction.js';
+// Image Classification
+export { ImageClassificationPipeline, createImageClassificationPipeline, } from './image-classification.js';
+// Import pipeline classes
+import { TextClassificationPipeline, SentimentAnalysisPipeline } from './text-classification.js';
+import { FeatureExtractionPipeline } from './feature-extraction.js';
+import { ImageClassificationPipeline } from './image-classification.js';
+/**
+ * Create a pipeline for a specific task
+ *
+ * @example
+ * ```typescript
+ * // Create a sentiment analysis pipeline
+ * const sentiment = await pipeline('sentiment-analysis');
+ * const result = await sentiment.run('I love this product!');
+ *
+ * // Create an image classifier with custom model
+ * const classifier = await pipeline('image-classification', {
+ *   model: 'https://example.com/model.bin',
+ * });
+ * ```
+ */
+export async function pipeline(task, options) {
+    const config = {
+        task: task,
+        model: options?.model ?? 'default',
+        runtime: options?.runtime,
+        cache: options?.cache ?? true,
+        quantization: options?.quantization,
+    };
+    let pipelineInstance;
+    switch (task) {
+        case 'text-classification':
+            pipelineInstance = new TextClassificationPipeline(config, options?.labels);
+            break;
+        case 'sentiment-analysis':
+            pipelineInstance = new SentimentAnalysisPipeline(config);
+            break;
+        case 'feature-extraction':
+            pipelineInstance = new FeatureExtractionPipeline(config);
+            break;
+        case 'image-classification':
+            pipelineInstance = new ImageClassificationPipeline(config, options?.labels);
+            break;
+        default:
+            throw new Error(`Unknown pipeline task: ${task}`);
+    }
+    // Initialize the pipeline
+    await pipelineInstance.initialize();
+    return pipelineInstance;
+}
+/**
+ * Create multiple pipelines at once
+ */
+export async function createPipelines(tasks, options) {
+    const pipelines = await Promise.all(tasks.map(task => pipeline(task, options)));
+    const result = {};
+    for (let i = 0; i < tasks.length; i++) {
+        const task = tasks[i];
+        result[task] = pipelines[i];
+    }
+    return result;
+}
+//# sourceMappingURL=index.js.map

package/dist/pipelines/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/pipelines/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AASH,OAAO;AACP,OAAO,EACL,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,cAAc,EACd,eAAe,GAMhB,MAAM,WAAW,CAAC;AAEnB,sBAAsB;AACtB,OAAO,EACL,0BAA0B,EAC1B,yBAAyB,EACzB,gCAAgC,EAChC,+BAA+B,GAEhC,MAAM,0BAA0B,CAAC;AAElC,qBAAqB;AACrB,OAAO,EACL,yBAAyB,EACzB,+BAA+B,GAEhC,MAAM,yBAAyB,CAAC;AAEjC,uBAAuB;AACvB,OAAO,EACL,2BAA2B,EAC3B,iCAAiC,GAGlC,MAAM,2BAA2B,CAAC;AAgCnC,0BAA0B;AAC1B,OAAO,EAAE,0BAA0B,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACjG,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AAExE;;;;;;;;;;;;;;GAcG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,IAAO,EACP,OAAgC;IAEhC,MAAM,MAAM,GAAmB;QAC7B,IAAI,EAAE,IAAoB;QAC1B,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,SAAS;QAClC,OAAO,EAAE,OAAO,EAAE,OAAO;QACzB,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,IAAI;QAC7B,YAAY,EAAE,OAAO,EAAE,YAAY;KACpC,CAAC;IAEF,IAAI,gBAAkI,CAAC;IAEvI,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,qBAAqB;YACxB,gBAAgB,GAAG,IAAI,0BAA0B,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YAC3E,MAAM;QACR,KAAK,oBAAoB;YACvB,gBAAgB,GAAG,IAAI,yBAAyB,CAAC,MAAM,CAAC,CAAC;YACzD,MAAM;QACR,KAAK,oBAAoB;YACvB,gBAAgB,GAAG,IAAI,yBAAyB,CAAC,MAAM,CAAC,CAAC;YACzD,MAAM;QACR,KAAK,sBAAsB;YACzB,gBAAgB,GAAG,IAAI,2BAA2B,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YAC5E,MAAM;QACR;YACE,MAAM,IAAI,KAAK,CAAC,0BAA0B,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,0BAA0B;IAC1B,MAAM,gBAAgB,CAAC,UAAU,EAAE,CAAC;IAEpC,OAAO,gBAAsC,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,KAAQ,EACR,OAAgC;IAEhC,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,GAAG,CACjC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAC3C,CAAC;IAEF,MAAM,MAAM,GAAsD,EAAE,CAAC;IAErE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;QACvB,MAAM,CAAC,IAAiB,CAAC,GAAG,SAAS,CAAC,CAAC,CAA+B,CAAC;IACzE,CAAC;IAED,OAAO,MAAkD,CAAC;AAC5D,CAAC"}

package/dist/pipelines/text-classification.d.ts

@@ -0,0 +1,71 @@
+/**
+ * edgeFlow.js - Text Classification Pipeline
+ *
+ * High-level API for text classification tasks including
+ * sentiment analysis, topic classification, etc.
+ */
+import { PipelineConfig, PipelineOptions } from '../core/types.js';
+import { EdgeFlowTensor } from '../core/tensor.js';
+import { BasePipeline, TextClassificationResult } from './base.js';
+/**
+ * Text classification options
+ */
+export interface TextClassificationOptions extends PipelineOptions {
+    /** Return all labels with scores */
+    returnAllScores?: boolean;
+    /** Custom labels */
+    labels?: string[];
+    /** Number of labels to return */
+    topK?: number;
+}
+/**
+ * TextClassificationPipeline - Classify text into categories
+ */
+export declare class TextClassificationPipeline extends BasePipeline<string | string[], TextClassificationResult | TextClassificationResult[]> {
+    private tokenizer;
+    private labels;
+    constructor(config: PipelineConfig, labels?: string[]);
+    /**
+     * Initialize pipeline
+     */
+    initialize(): Promise<void>;
+    /**
+     * Set custom labels
+     */
+    setLabels(labels: string[]): void;
+    /**
+     * Run classification
+     */
+    run(input: string | string[], options?: TextClassificationOptions): Promise<TextClassificationResult | TextClassificationResult[]>;
+    /**
+     * Preprocess text input
+     */
+    protected preprocess(input: string | string[]): Promise<EdgeFlowTensor[]>;
+    /**
+     * Run model inference
+     */
+    private runInference;
+    /**
+     * Postprocess model outputs
+     */
+    protected postprocess(outputs: EdgeFlowTensor[], options?: TextClassificationOptions): Promise<TextClassificationResult>;
+}
+/**
+ * SentimentAnalysisPipeline - Specialized for sentiment analysis
+ */
+export declare class SentimentAnalysisPipeline extends TextClassificationPipeline {
+    constructor(config: PipelineConfig);
+    /**
+     * Analyze sentiment
+     */
+    analyze(text: string | string[], options?: TextClassificationOptions): Promise<TextClassificationResult | TextClassificationResult[]>;
+}
+/**
+ * Create text classification pipeline
+ */
+export declare function createTextClassificationPipeline(config?: Partial<PipelineConfig>): TextClassificationPipeline;
+/**
+ * Create sentiment analysis pipeline
+ */
+export declare function createSentimentAnalysisPipeline(config?: Partial<PipelineConfig>): SentimentAnalysisPipeline;
+//# sourceMappingURL=text-classification.d.ts.map

package/dist/pipelines/text-classification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"text-classification.d.ts","sourceRoot":"","sources":["../../src/pipelines/text-classification.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,cAAc,EAAW,MAAM,mBAAmB,CAAC;AAE5D,OAAO,EACL,YAAY,EACZ,wBAAwB,EAGzB,MAAM,WAAW,CAAC;AAMnB;;GAEG;AACH,MAAM,WAAW,yBAA0B,SAAQ,eAAe;IAChE,oCAAoC;IACpC,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,oBAAoB;IACpB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,iCAAiC;IACjC,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,qBAAa,0BAA2B,SAAQ,YAAY,CAC1D,MAAM,GAAG,MAAM,EAAE,EACjB,wBAAwB,GAAG,wBAAwB,EAAE,CACtD;IACC,OAAO,CAAC,SAAS,CAA0B;IAC3C,OAAO,CAAC,MAAM,CAAW;gBAEb,MAAM,EAAE,cAAc,EAAE,MAAM,CAAC,EAAE,MAAM,EAAE;IAKrD;;OAEG;IACY,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAS1C;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI;IAIjC;;OAEG;IACY,GAAG,CAChB,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,EACxB,OAAO,CAAC,EAAE,yBAAyB,GAClC,OAAO,CAAC,wBAAwB,GAAG,wBAAwB,EAAE,CAAC;IA+BjE;;OAEG;cACsB,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IA0BxF;;OAEG;YACW,YAAY;IAkB1B;;OAEG;cACsB,WAAW,CAClC,OAAO,EAAE,cAAc,EAAE,EACzB,OAAO,CAAC,EAAE,yBAAyB,GAClC,OAAO,CAAC,wBAAwB,CAAC;CAqCrC;AAMD;;GAEG;AACH,qBAAa,yBAA0B,SAAQ,0BAA0B;gBAC3D,MAAM,EAAE,cAAc;IAIlC;;OAEG;IACG,OAAO,CACX,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,EACvB,OAAO,CAAC,EAAE,yBAAyB,GAClC,OAAO,CAAC,wBAAwB,GAAG,wBAAwB,EAAE,CAAC;CAGlE;AAMD;;GAEG;AACH,wBAAgB,gCAAgC,CAC9C,MAAM,GAAE,OAAO,CAAC,cAAc,CAAM,GACnC,0BAA0B,CAQ5B;AAED;;GAEG;AACH,wBAAgB,+BAA+B,CAC7C,MAAM,GAAE,OAAO,CAAC,cAAc,CAAM,GACnC,yBAAyB,CAQ3B"}