edgeflowjs 0.1.0 → 0.2.0

package/dist/pipelines/index.js

@@ -1,6 +1,8 @@
 /**
  * edgeFlow.js - Pipeline Exports
  */
+import { getPluginPipeline } from '../core/plugin.js';
+import { registerAllBackends } from '../backends/index.js';
 // Base
 export { BasePipeline, registerPipeline, getPipelineFactory, SENTIMENT_LABELS, EMOTION_LABELS, IMAGENET_LABELS, } from './base.js';
 // Text Classification
@@ -9,10 +11,28 @@ export { TextClassificationPipeline, SentimentAnalysisPipeline, createTextClassi
 export { FeatureExtractionPipeline, createFeatureExtractionPipeline, } from './feature-extraction.js';
 // Image Classification
 export { ImageClassificationPipeline, createImageClassificationPipeline, } from './image-classification.js';
+// Text Generation
+export { TextGenerationPipeline, createTextGenerationPipeline, } from './text-generation.js';
+// Object Detection
+export { ObjectDetectionPipeline, createObjectDetectionPipeline, COCO_LABELS, } from './object-detection.js';
+// Automatic Speech Recognition
+export { AutomaticSpeechRecognitionPipeline, createASRPipeline, } from './automatic-speech-recognition.js';
+// Zero-shot Classification
+export { ZeroShotClassificationPipeline, createZeroShotClassificationPipeline, } from './zero-shot-classification.js';
+// Question Answering
+export { QuestionAnsweringPipeline, createQuestionAnsweringPipeline, } from './question-answering.js';
+// Image Segmentation
+export { ImageSegmentationPipeline, createImageSegmentationPipeline, } from './image-segmentation.js';
 // Import pipeline classes
 import { TextClassificationPipeline, SentimentAnalysisPipeline } from './text-classification.js';
 import { FeatureExtractionPipeline } from './feature-extraction.js';
 import { ImageClassificationPipeline } from './image-classification.js';
+import { TextGenerationPipeline } from './text-generation.js';
+import { ObjectDetectionPipeline } from './object-detection.js';
+import { AutomaticSpeechRecognitionPipeline } from './automatic-speech-recognition.js';
+import { ZeroShotClassificationPipeline } from './zero-shot-classification.js';
+import { QuestionAnsweringPipeline } from './question-answering.js';
+import { ImageSegmentationPipeline } from './image-segmentation.js';
 /**
  * Create a pipeline for a specific task
  *
@@ -29,6 +49,9 @@ import { ImageClassificationPipeline } from './image-classification.js';
  * ```
  */
 export async function pipeline(task, options) {
+    // Guarantee backends are registered before any model loads.
+    // registerAllBackends() is synchronous and idempotent (safe to call repeatedly).
+    registerAllBackends();
     const config = {
         task: task,
         model: options?.model ?? 'default',
@@ -50,8 +73,34 @@ export async function pipeline(task, options) {
         case 'image-classification':
             pipelineInstance = new ImageClassificationPipeline(config, options?.labels);
             break;
-        default:
-            throw new Error(`Unknown pipeline task: ${task}`);
+        case 'text-generation':
+            pipelineInstance = new TextGenerationPipeline(config);
+            break;
+        case 'object-detection':
+            pipelineInstance = new ObjectDetectionPipeline(config, options?.labels);
+            break;
+        case 'automatic-speech-recognition':
+            pipelineInstance = new AutomaticSpeechRecognitionPipeline(config);
+            break;
+        case 'zero-shot-classification':
+            pipelineInstance = new ZeroShotClassificationPipeline(config);
+            break;
+        case 'question-answering':
+            pipelineInstance = new QuestionAnsweringPipeline(config);
+            break;
+        case 'image-segmentation':
+            pipelineInstance = new ImageSegmentationPipeline(config);
+            break;
+        default: {
+            // Check if a plugin provides this pipeline task
+            const pluginEntry = getPluginPipeline(task);
+            if (pluginEntry) {
+                pipelineInstance = pluginEntry.factory(config);
+                break;
+            }
+            throw new Error(`Unknown pipeline task: "${task}". ` +
+                `Register a plugin with registerPlugin() to add custom pipeline tasks.`);
+        }
     }
     // Initialize the pipeline
     await pipelineInstance.initialize();

package/dist/pipelines/index.js.map

@@ -1 +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"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/pipelines/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AASH,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAE3D,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;AAEnC,kBAAkB;AAClB,OAAO,EACL,sBAAsB,EACtB,4BAA4B,GAQ7B,MAAM,sBAAsB,CAAC;AAE9B,mBAAmB;AACnB,OAAO,EACL,uBAAuB,EACvB,6BAA6B,EAC7B,WAAW,GAIZ,MAAM,uBAAuB,CAAC;AAE/B,+BAA+B;AAC/B,OAAO,EACL,kCAAkC,EAClC,iBAAiB,GAKlB,MAAM,mCAAmC,CAAC;AAE3C,2BAA2B;AAC3B,OAAO,EACL,8BAA8B,EAC9B,oCAAoC,GAGrC,MAAM,+BAA+B,CAAC;AAEvC,qBAAqB;AACrB,OAAO,EACL,yBAAyB,EACzB,+BAA+B,GAIhC,MAAM,yBAAyB,CAAC;AAEjC,qBAAqB;AACrB,OAAO,EACL,yBAAyB,EACzB,+BAA+B,GAMhC,MAAM,yBAAyB,CAAC;AAsCjC,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;AACxE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,kCAAkC,EAAE,MAAM,mCAAmC,CAAC;AACvF,OAAO,EAAE,8BAA8B,EAAE,MAAM,+BAA+B,CAAC;AAC/E,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAEpE;;;;;;;;;;;;;;GAcG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,IAAO,EACP,OAAgC;IAEhC,4DAA4D;IAC5D,iFAAiF;IACjF,mBAAmB,EAAE,CAAC;IAEtB,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;IAIF,IAAI,gBAA8B,CAAC;IAEnC,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,KAAK,iBAAiB;YACpB,gBAAgB,GAAG,IAAI,sBAAsB,CAAC,MAAM,CAAC,CAAC;YACtD,MAAM;QACR,KAAK,kBAAkB;YACrB,gBAAgB,GAAG,IAAI,uBAAuB,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACxE,MAAM;QACR,KAAK,8BAA8B;YACjC,gBAAgB,GAAG,IAAI,kCAAkC,CAAC,MAAM,CAAC,CAAC;YAClE,MAAM;QACR,KAAK,0BAA0B;YAC7B,gBAAgB,GAAG,IAAI,8BAA8B,CAAC,MAAM,CAAC,CAAC;YAC9D,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,OAAO,CAAC,CAAC,CAAC;YACR,gDAAgD;YAChD,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC5C,IAAI,WAAW,EAAE,CAAC;gBAChB,gBAAgB,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM;YACR,CAAC;YACD,MAAM,IAAI,KAAK,CACb,2BAA2B,IAAI,KAAK;gBACpC,uEAAuE,CACxE,CAAC;QACJ,CAAC;IACH,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/object-detection.d.ts

@@ -0,0 +1,44 @@
+/**
+ * edgeFlow.js - Object Detection Pipeline
+ *
+ * Detect objects in images with bounding boxes and class labels.
+ */
+import { BasePipeline, ObjectDetectionResult } from './base.js';
+import { EdgeFlowTensor } from '../core/tensor.js';
+import { PipelineConfig, PipelineOptions } from '../core/types.js';
+import { type ImageInput } from '../utils/preprocessor.js';
+export interface ObjectDetectionOptions extends PipelineOptions {
+    threshold?: number;
+    topK?: number;
+    nms?: boolean;
+    iouThreshold?: number;
+}
+export interface BoundingBox {
+    x: number;
+    y: number;
+    width: number;
+    height: number;
+}
+export interface Detection extends ObjectDetectionResult {
+    classId: number;
+    boxNormalized: BoundingBox;
+}
+export declare const COCO_LABELS: string[];
+export declare class ObjectDetectionPipeline extends BasePipeline<ImageInput | ImageInput[], Detection[]> {
+    private preprocessor;
+    private onnxModel;
+    private labels;
+    private modelUrl;
+    constructor(config?: PipelineConfig, labels?: string[]);
+    initialize(): Promise<void>;
+    setLabels(labels: string[]): void;
+    run(input: ImageInput | ImageInput[], options?: ObjectDetectionOptions): Promise<Detection[]>;
+    protected preprocess(input: ImageInput | ImageInput[]): Promise<EdgeFlowTensor[]>;
+    private runModelInference;
+    protected postprocess(outputs: EdgeFlowTensor[], options?: PipelineOptions): Promise<Detection[]>;
+    private parseDetections;
+    private nonMaxSuppression;
+    private computeIoU;
+}
+export declare function createObjectDetectionPipeline(config?: PipelineConfig, labels?: string[]): ObjectDetectionPipeline;
+//# sourceMappingURL=object-detection.d.ts.map

package/dist/pipelines/object-detection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"object-detection.d.ts","sourceRoot":"","sources":["../../src/pipelines/object-detection.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAE,qBAAqB,EAAoB,MAAM,WAAW,CAAC;AAClF,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,eAAe,EAAe,MAAM,kBAAkB,CAAC;AAChF,OAAO,EAAqB,KAAK,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAQ9E,MAAM,WAAW,sBAAuB,SAAQ,eAAe;IAC7D,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,OAAO,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,WAAW;IAC1B,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;IACV,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,SAAU,SAAQ,qBAAqB;IACtD,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,WAAW,CAAC;CAC5B;AAcD,eAAO,MAAM,WAAW,UAavB,CAAC;AAMF,qBAAa,uBAAwB,SAAQ,YAAY,CAAC,UAAU,GAAG,UAAU,EAAE,EAAE,SAAS,EAAE,CAAC;IAC/F,OAAO,CAAC,YAAY,CAAoB;IACxC,OAAO,CAAC,SAAS,CAA4B;IAC7C,OAAO,CAAC,MAAM,CAAW;IACzB,OAAO,CAAC,QAAQ,CAAS;gBAEb,MAAM,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,EAAE,MAAM,EAAE;IAiBvC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAS1C,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI;IAIlB,GAAG,CAChB,KAAK,EAAE,UAAU,GAAG,UAAU,EAAE,EAChC,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,SAAS,EAAE,CAAC;cAOP,UAAU,CAAC,KAAK,EAAE,UAAU,GAAG,UAAU,EAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;YAezE,iBAAiB;cAKf,WAAW,CACzB,OAAO,EAAE,cAAc,EAAE,EACzB,OAAO,CAAC,EAAE,eAAe,GACxB,OAAO,CAAC,SAAS,EAAE,CAAC;IAwBvB,OAAO,CAAC,eAAe;IAuFvB,OAAO,CAAC,iBAAiB;IAgCzB,OAAO,CAAC,UAAU;CAenB;AAMD,wBAAgB,6BAA6B,CAC3C,MAAM,CAAC,EAAE,cAAc,EACvB,MAAM,CAAC,EAAE,MAAM,EAAE,GAChB,uBAAuB,CAEzB"}

package/dist/pipelines/object-detection.js

@@ -0,0 +1,218 @@
+/**
+ * edgeFlow.js - Object Detection Pipeline
+ *
+ * Detect objects in images with bounding boxes and class labels.
+ */
+import { BasePipeline, registerPipeline } from './base.js';
+import { EdgeFlowTensor } from '../core/tensor.js';
+import { ImagePreprocessor } from '../utils/preprocessor.js';
+import { loadModelData } from '../utils/model-loader.js';
+import { loadModelFromBuffer, runInference } from '../core/runtime.js';
+// ============================================================================
+// Default Model (YOLOS-tiny, quantized)
+// ============================================================================
+const DEFAULT_MODELS = {
+    model: 'https://huggingface.co/Xenova/yolos-tiny/resolve/main/onnx/model_quantized.onnx',
+};
+// ============================================================================
+// COCO Labels
+// ============================================================================
+export const COCO_LABELS = [
+    'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck',
+    'boat', 'traffic light', 'fire hydrant', 'stop sign', 'parking meter', 'bench',
+    'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra',
+    'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee',
+    'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove',
+    'skateboard', 'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup',
+    'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange',
+    'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch',
+    'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse',
+    'remote', 'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink',
+    'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier',
+    'toothbrush'
+];
+// ============================================================================
+// Object Detection Pipeline
+// ============================================================================
+export class ObjectDetectionPipeline extends BasePipeline {
+    preprocessor;
+    onnxModel = null;
+    labels;
+    modelUrl;
+    constructor(config, labels) {
+        super(config ?? {
+            task: 'object-detection',
+            model: 'default',
+        });
+        this.labels = labels ?? COCO_LABELS;
+        this.modelUrl = (config?.model && config.model !== 'default') ? config.model : DEFAULT_MODELS.model;
+        this.preprocessor = new ImagePreprocessor({
+            width: 640,
+            height: 640,
+            mean: [0.485, 0.456, 0.406],
+            std: [0.229, 0.224, 0.225],
+            channelFormat: 'CHW',
+        });
+    }
+    async initialize() {
+        await super.initialize();
+        if (!this.onnxModel) {
+            const modelData = await loadModelData(this.modelUrl, { cache: this.config.cache ?? true });
+            this.onnxModel = await loadModelFromBuffer(modelData);
+        }
+    }
+    setLabels(labels) {
+        this.labels = labels;
+    }
+    async run(input, options) {
+        await this.initialize();
+        const tensorInputs = await this.preprocess(input);
+        const outputs = await this.runModelInference(tensorInputs);
+        return this.postprocess(outputs, options);
+    }
+    async preprocess(input) {
+        const inputs = Array.isArray(input) ? input : [input];
+        if (inputs.length === 1) {
+            const tensor = await this.preprocessor.process(inputs[0]);
+            return [new EdgeFlowTensor(tensor.toFloat32Array(), [1, ...tensor.shape], 'float32')];
+        }
+        return [await this.preprocessor.processBatch(inputs)];
+    }
+    async runModelInference(inputs) {
+        const outputs = await runInference(this.onnxModel, inputs);
+        return outputs;
+    }
+    async postprocess(outputs, options) {
+        const opts = options ?? {};
+        const threshold = opts.threshold ?? 0.5;
+        const topK = opts.topK ?? 100;
+        const nms = opts.nms ?? true;
+        const iouThreshold = opts.iouThreshold ?? 0.5;
+        if (!outputs[0]) {
+            return [];
+        }
+        const outputData = outputs[0].toFloat32Array();
+        const shape = [...outputs[0].shape];
+        const detections = this.parseDetections(outputData, shape, threshold);
+        let filtered = nms ? this.nonMaxSuppression(detections, iouThreshold) : detections;
+        filtered.sort((a, b) => b.score - a.score);
+        filtered = filtered.slice(0, topK);
+        return filtered;
+    }
+    parseDetections(data, shape, threshold) {
+        const detections = [];
+        const numBoxes = shape[1] ?? 0;
+        const boxSize = shape[2] ?? 0;
+        if (boxSize >= 5) {
+            const numClasses = boxSize - 5;
+            for (let i = 0; i < numBoxes; i++) {
+                const offset = i * boxSize;
+                const objectness = data[offset + 4] ?? 0;
+                if (objectness < threshold)
+                    continue;
+                let maxClassScore = 0;
+                let maxClassIdx = 0;
+                for (let c = 0; c < numClasses; c++) {
+                    const score = data[offset + 5 + c] ?? 0;
+                    if (score > maxClassScore) {
+                        maxClassScore = score;
+                        maxClassIdx = c;
+                    }
+                }
+                const confidence = objectness * maxClassScore;
+                if (confidence < threshold)
+                    continue;
+                const x = data[offset] ?? 0;
+                const y = data[offset + 1] ?? 0;
+                const w = data[offset + 2] ?? 0;
+                const h = data[offset + 3] ?? 0;
+                detections.push({
+                    label: this.labels[maxClassIdx] ?? `class_${maxClassIdx}`,
+                    score: confidence,
+                    classId: maxClassIdx,
+                    box: {
+                        x: Math.max(0, x - w / 2),
+                        y: Math.max(0, y - h / 2),
+                        width: w,
+                        height: h,
+                    },
+                    boxNormalized: {
+                        x: Math.max(0, x - w / 2),
+                        y: Math.max(0, y - h / 2),
+                        width: w,
+                        height: h,
+                    },
+                });
+            }
+        }
+        else if (boxSize === 4) {
+            for (let i = 0; i < numBoxes; i++) {
+                const offset = i * boxSize;
+                const x1 = data[offset] ?? 0;
+                const y1 = data[offset + 1] ?? 0;
+                const x2 = data[offset + 2] ?? 0;
+                const y2 = data[offset + 3] ?? 0;
+                detections.push({
+                    label: this.labels[0] ?? 'object',
+                    score: 1.0,
+                    classId: 0,
+                    box: {
+                        x: x1,
+                        y: y1,
+                        width: x2 - x1,
+                        height: y2 - y1,
+                    },
+                    boxNormalized: {
+                        x: x1,
+                        y: y1,
+                        width: x2 - x1,
+                        height: y2 - y1,
+                    },
+                });
+            }
+        }
+        return detections;
+    }
+    nonMaxSuppression(detections, iouThreshold) {
+        if (detections.length === 0)
+            return [];
+        const sorted = [...detections].sort((a, b) => b.score - a.score);
+        const selected = [];
+        const active = new Array(sorted.length).fill(true);
+        for (let i = 0; i < sorted.length; i++) {
+            if (!active[i])
+                continue;
+            const current = sorted[i];
+            selected.push(current);
+            for (let j = i + 1; j < sorted.length; j++) {
+                if (!active[j])
+                    continue;
+                const other = sorted[j];
+                if (current.classId !== other.classId)
+                    continue;
+                const iou = this.computeIoU(current.box, other.box);
+                if (iou > iouThreshold) {
+                    active[j] = false;
+                }
+            }
+        }
+        return selected;
+    }
+    computeIoU(a, b) {
+        const xOverlap = Math.max(0, Math.min(a.x + a.width, b.x + b.width) - Math.max(a.x, b.x));
+        const yOverlap = Math.max(0, Math.min(a.y + a.height, b.y + b.height) - Math.max(a.y, b.y));
+        const intersection = xOverlap * yOverlap;
+        const aArea = a.width * a.height;
+        const bArea = b.width * b.height;
+        const union = aArea + bArea - intersection;
+        return union > 0 ? intersection / union : 0;
+    }
+}
+// ============================================================================
+// Factory
+// ============================================================================
+export function createObjectDetectionPipeline(config, labels) {
+    return new ObjectDetectionPipeline(config, labels);
+}
+registerPipeline('object-detection', (config) => new ObjectDetectionPipeline(config));
+//# sourceMappingURL=object-detection.js.map

package/dist/pipelines/object-detection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"object-detection.js","sourceRoot":"","sources":["../../src/pipelines/object-detection.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAyB,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAClF,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,OAAO,EAAE,iBAAiB,EAAmB,MAAM,0BAA0B,CAAC;AAC9E,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAyBvE,+EAA+E;AAC/E,wCAAwC;AACxC,+EAA+E;AAE/E,MAAM,cAAc,GAAG;IACrB,KAAK,EAAE,iFAAiF;CACzF,CAAC;AAEF,+EAA+E;AAC/E,cAAc;AACd,+EAA+E;AAE/E,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO;IAC7E,MAAM,EAAE,eAAe,EAAE,cAAc,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO;IAC9E,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO;IAC1E,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,SAAS;IAC1E,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,gBAAgB;IAC5E,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK;IACzE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ;IACzE,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO;IAC3E,cAAc,EAAE,KAAK,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO;IACxE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;IAC1E,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,YAAY;IAC/E,YAAY;CACb,CAAC;AAEF,+EAA+E;AAC/E,4BAA4B;AAC5B,+EAA+E;AAE/E,MAAM,OAAO,uBAAwB,SAAQ,YAAoD;IACvF,YAAY,CAAoB;IAChC,SAAS,GAAuB,IAAI,CAAC;IACrC,MAAM,CAAW;IACjB,QAAQ,CAAS;IAEzB,YAAY,MAAuB,EAAE,MAAiB;QACpD,KAAK,CAAC,MAAM,IAAI;YACd,IAAI,EAAE,kBAAkB;YACxB,KAAK,EAAE,SAAS;SACjB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,WAAW,CAAC;QACpC,IAAI,CAAC,QAAQ,GAAG,CAAC,MAAM,EAAE,KAAK,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC;QACpG,IAAI,CAAC,YAAY,GAAG,IAAI,iBAAiB,CAAC;YACxC,KAAK,EAAE,GAAG;YACV,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC;YAC3B,GAAG,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC;YAC1B,aAAa,EAAE,KAAK;SACrB,CAAC,CAAC;IACL,CAAC;IAEQ,KAAK,CAAC,UAAU;QACvB,MAAM,KAAK,CAAC,UAAU,EAAE,CAAC;QAEzB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,SAAS,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,IAAI,EAAE,CAAC,CAAC;YAC3F,IAAI,CAAC,SAAS,GAAG,MAAM,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED,SAAS,CAAC,MAAgB;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEQ,KAAK,CAAC,GAAG,CAChB,KAAgC,EAChC,OAAgC;QAEhC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAC3D,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAES,KAAK,CAAC,UAAU,CAAC,KAAgC;QACzD,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAEtD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,CAAC;YAC3D,OAAO,CAAC,IAAI,cAAc,CACxB,MAAM,CAAC,cAAc,EAAE,EACvB,CAAC,CAAC,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,EACpB,SAAS,CACV,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;IACxD,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,MAAwB;QACtD,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,SAAU,EAAE,MAAM,CAAC,CAAC;QAC5D,OAAO,OAA2B,CAAC;IACrC,CAAC;IAES,KAAK,CAAC,WAAW,CACzB,OAAyB,EACzB,OAAyB;QAEzB,MAAM,IAAI,GAAG,OAAiC,IAAI,EAAE,CAAC;QACrD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,GAAG,CAAC;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC;QAC7B,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,GAAG,CAAC;QAE9C,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;YAChB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,UAAU,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;QAC/C,MAAM,KAAK,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAa,CAAC;QAEhD,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;QAEtE,IAAI,QAAQ,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QAEnF,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAC3C,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QAEnC,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,eAAe,CACrB,IAAkB,EAClB,KAAe,EACf,SAAiB;QAEjB,MAAM,UAAU,GAAgB,EAAE,CAAC;QAEnC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAE9B,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;YACjB,MAAM,UAAU,GAAG,OAAO,GAAG,CAAC,CAAC;YAE/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC;gBAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBAEzC,IAAI,UAAU,GAAG,SAAS;oBAAE,SAAS;gBAErC,IAAI,aAAa,GAAG,CAAC,CAAC;gBACtB,IAAI,WAAW,GAAG,CAAC,CAAC;gBAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;oBACpC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;oBACxC,IAAI,KAAK,GAAG,aAAa,EAAE,CAAC;wBAC1B,aAAa,GAAG,KAAK,CAAC;wBACtB,WAAW,GAAG,CAAC,CAAC;oBAClB,CAAC;gBACH,CAAC;gBAED,MAAM,UAAU,GAAG,UAAU,GAAG,aAAa,CAAC;gBAC9C,IAAI,UAAU,GAAG,SAAS;oBAAE,SAAS;gBAErC,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC5B,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBAChC,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBAChC,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBAEhC,UAAU,CAAC,IAAI,CAAC;oBACd,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,SAAS,WAAW,EAAE;oBACzD,KAAK,EAAE,UAAU;oBACjB,OAAO,EAAE,WAAW;oBACpB,GAAG,EAAE;wBACH,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wBACzB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wBACzB,KAAK,EAAE,CAAC;wBACR,MAAM,EAAE,CAAC;qBACV;oBACD,aAAa,EAAE;wBACb,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wBACzB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wBACzB,KAAK,EAAE,CAAC;wBACR,MAAM,EAAE,CAAC;qBACV;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;aAAM,IAAI,OAAO,KAAK,CAAC,EAAE,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC;gBAC3B,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC7B,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;gBAEjC,UAAU,CAAC,IAAI,CAAC;oBACd,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,QAAQ;oBACjC,KAAK,EAAE,GAAG;oBACV,OAAO,EAAE,CAAC;oBACV,GAAG,EAAE;wBACH,CAAC,EAAE,EAAE;wBACL,CAAC,EAAE,EAAE;wBACL,KAAK,EAAE,EAAE,GAAG,EAAE;wBACd,MAAM,EAAE,EAAE,GAAG,EAAE;qBAChB;oBACD,aAAa,EAAE;wBACb,CAAC,EAAE,EAAE;wBACL,CAAC,EAAE,EAAE;wBACL,KAAK,EAAE,EAAE,GAAG,EAAE;wBACd,MAAM,EAAE,EAAE,GAAG,EAAE;qBAChB;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,iBAAiB,CACvB,UAAuB,EACvB,YAAoB;QAEpB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEvC,MAAM,MAAM,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QACjE,MAAM,QAAQ,GAAgB,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;gBAAE,SAAS;YAEzB,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAE,CAAC;YAC3B,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEvB,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3C,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;oBAAE,SAAS;gBAEzB,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAE,CAAC;gBACzB,IAAI,OAAO,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO;oBAAE,SAAS;gBAEhD,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;gBACpD,IAAI,GAAG,GAAG,YAAY,EAAE,CAAC;oBACvB,MAAM,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;gBACpB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,UAAU,CAAC,CAAc,EAAE,CAAc;QAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EACzB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAC5D,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EACzB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAC9D,CAAC;QAEF,MAAM,YAAY,GAAG,QAAQ,GAAG,QAAQ,CAAC;QACzC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,CAAC;QACjC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,CAAC;QACjC,MAAM,KAAK,GAAG,KAAK,GAAG,KAAK,GAAG,YAAY,CAAC;QAE3C,OAAO,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC;CACF;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,MAAM,UAAU,6BAA6B,CAC3C,MAAuB,EACvB,MAAiB;IAEjB,OAAO,IAAI,uBAAuB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AACrD,CAAC;AAED,gBAAgB,CAAC,kBAAkB,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,uBAAuB,CAAC,MAAM,CAAC,CAAC,CAAC"}

package/dist/pipelines/question-answering.d.ts

@@ -0,0 +1,41 @@
+/**
+ * edgeFlow.js - Question Answering Pipeline
+ *
+ * Extract answers from context given a question using real ONNX QA models.
+ */
+import { BasePipeline, PipelineResult } from './base.js';
+import { EdgeFlowTensor } from '../core/tensor.js';
+import { PipelineConfig, PipelineOptions } from '../core/types.js';
+import { Tokenizer } from '../utils/tokenizer.js';
+export interface QAInput {
+    question: string;
+    context: string;
+}
+export interface QuestionAnsweringOptions extends PipelineOptions {
+    maxAnswerLength?: number;
+    maxQuestionLength?: number;
+    topK?: number;
+    threshold?: number;
+    handleImpossible?: boolean;
+}
+export interface QuestionAnsweringResult extends PipelineResult {
+    answer: string;
+    score: number;
+    start: number;
+    end: number;
+}
+export declare class QuestionAnsweringPipeline extends BasePipeline<QAInput | QAInput[], QuestionAnsweringResult | QuestionAnsweringResult[]> {
+    private tokenizer;
+    private onnxModel;
+    private modelUrl;
+    private tokenizerUrl;
+    constructor(config?: PipelineConfig);
+    initialize(): Promise<void>;
+    setTokenizer(tokenizer: Tokenizer): void;
+    run(input: QAInput | QAInput[], options?: QuestionAnsweringOptions): Promise<QuestionAnsweringResult | QuestionAnsweringResult[]>;
+    private answerQuestion;
+    protected preprocess(input: QAInput | QAInput[]): Promise<EdgeFlowTensor[]>;
+    protected postprocess(outputs: EdgeFlowTensor[], _options?: PipelineOptions): Promise<QuestionAnsweringResult | QuestionAnsweringResult[]>;
+}
+export declare function createQuestionAnsweringPipeline(config?: PipelineConfig): QuestionAnsweringPipeline;
+//# sourceMappingURL=question-answering.d.ts.map

package/dist/pipelines/question-answering.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"question-answering.d.ts","sourceRoot":"","sources":["../../src/pipelines/question-answering.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAE,cAAc,EAAoB,MAAM,WAAW,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAW,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,eAAe,EAAe,MAAM,kBAAkB,CAAC;AAChF,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAiBlD,MAAM,WAAW,OAAO;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,wBAAyB,SAAQ,eAAe;IAC/D,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,uBAAwB,SAAQ,cAAc;IAC7D,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;CACb;AAMD,qBAAa,yBAA0B,SAAQ,YAAY,CACzD,OAAO,GAAG,OAAO,EAAE,EACnB,uBAAuB,GAAG,uBAAuB,EAAE,CACpD;IACC,OAAO,CAAC,SAAS,CAA0B;IAC3C,OAAO,CAAC,SAAS,CAA4B;IAC7C,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,YAAY,CAAS;gBAEjB,MAAM,CAAC,EAAE,cAAc;IASpB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAa1C,YAAY,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI;IAIzB,GAAG,CAChB,KAAK,EAAE,OAAO,GAAG,OAAO,EAAE,EAC1B,OAAO,CAAC,EAAE,wBAAwB,GACjC,OAAO,CAAC,uBAAuB,GAAG,uBAAuB,EAAE,CAAC;YAWjD,cAAc;cAqFZ,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,EAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;cAyBjE,WAAW,CACzB,OAAO,EAAE,cAAc,EAAE,EACzB,QAAQ,CAAC,EAAE,eAAe,GACzB,OAAO,CAAC,uBAAuB,GAAG,uBAAuB,EAAE,CAAC;CAkChE;AAMD,wBAAgB,+BAA+B,CAC7C,MAAM,CAAC,EAAE,cAAc,GACtB,yBAAyB,CAE3B"}

package/dist/pipelines/question-answering.js

@@ -0,0 +1,164 @@
+/**
+ * edgeFlow.js - Question Answering Pipeline
+ *
+ * Extract answers from context given a question using real ONNX QA models.
+ */
+import { BasePipeline, registerPipeline } from './base.js';
+import { EdgeFlowTensor, softmax } from '../core/tensor.js';
+import { Tokenizer } from '../utils/tokenizer.js';
+import { loadModelData } from '../utils/model-loader.js';
+import { loadModelFromBuffer, runInferenceNamed } from '../core/runtime.js';
+// ============================================================================
+// Default Model (DistilBERT fine-tuned on SQuAD)
+// ============================================================================
+const DEFAULT_MODELS = {
+    model: 'https://huggingface.co/Xenova/distilbert-base-cased-distilled-squad/resolve/main/onnx/model_quantized.onnx',
+    tokenizer: 'https://huggingface.co/Xenova/distilbert-base-cased-distilled-squad/resolve/main/tokenizer.json',
+};
+// ============================================================================
+// Question Answering Pipeline
+// ============================================================================
+export class QuestionAnsweringPipeline extends BasePipeline {
+    tokenizer = null;
+    onnxModel = null;
+    modelUrl;
+    tokenizerUrl;
+    constructor(config) {
+        super(config ?? {
+            task: 'question-answering',
+            model: 'default',
+        });
+        this.modelUrl = (config?.model && config.model !== 'default') ? config.model : DEFAULT_MODELS.model;
+        this.tokenizerUrl = DEFAULT_MODELS.tokenizer;
+    }
+    async initialize() {
+        await super.initialize();
+        if (!this.tokenizer) {
+            this.tokenizer = await Tokenizer.fromUrl(this.tokenizerUrl);
+        }
+        if (!this.onnxModel) {
+            const modelData = await loadModelData(this.modelUrl, { cache: this.config.cache ?? true });
+            this.onnxModel = await loadModelFromBuffer(modelData);
+        }
+    }
+    setTokenizer(tokenizer) {
+        this.tokenizer = tokenizer;
+    }
+    async run(input, options) {
+        await this.initialize();
+        const inputs = Array.isArray(input) ? input : [input];
+        const results = await Promise.all(inputs.map(i => this.answerQuestion(i, options ?? {})));
+        return Array.isArray(input) ? results : results[0];
+    }
+    async answerQuestion(input, options) {
+        const startTime = performance.now();
+        const { question, context } = input;
+        const maxAnswerLength = options.maxAnswerLength ?? 30;
+        // No padding — QA runs one example at a time and padding wastes compute
+        const encoded = this.tokenizer.encode(question, {
+            textPair: context,
+            addSpecialTokens: true,
+            maxLength: 512,
+            truncation: true,
+            padding: 'do_not_pad',
+            returnAttentionMask: true,
+            returnTokenTypeIds: true,
+        });
+        const seqLen = encoded.inputIds.length;
+        const inputIds = new EdgeFlowTensor(BigInt64Array.from(encoded.inputIds.map(id => BigInt(id))), [1, seqLen], 'int64');
+        const attentionMask = new EdgeFlowTensor(BigInt64Array.from(encoded.attentionMask.map(m => BigInt(m))), [1, seqLen], 'int64');
+        const namedInputs = new Map();
+        namedInputs.set('input_ids', inputIds);
+        namedInputs.set('attention_mask', attentionMask);
+        const outputs = await runInferenceNamed(this.onnxModel, namedInputs);
+        if (outputs.length < 2) {
+            return { answer: '', score: 0, start: 0, end: 0, processingTime: performance.now() - startTime };
+        }
+        const startLogits = outputs[0].toFloat32Array();
+        const endLogits = outputs[1].toFloat32Array();
+        const startProbs = softmax(new EdgeFlowTensor(new Float32Array(startLogits), [seqLen], 'float32')).toFloat32Array();
+        const endProbs = softmax(new EdgeFlowTensor(new Float32Array(endLogits), [seqLen], 'float32')).toFloat32Array();
+        // Constrain answer span to the context portion only (tokenTypeIds === 1).
+        // tokenTypeIds: 0 = question tokens ([CLS], question, [SEP]), 1 = context tokens.
+        const typeIds = encoded.tokenTypeIds ?? new Array(seqLen).fill(1);
+        // Find where context starts (first index with typeId === 1)
+        const contextStart = typeIds.findIndex(t => t === 1);
+        const spanStart = contextStart >= 0 ? contextStart : 0;
+        const spanEnd = seqLen - 1; // last non-padding position
+        let bestStartIdx = spanStart;
+        let bestEndIdx = spanStart;
+        let bestScore = -Infinity;
+        for (let s = spanStart; s <= spanEnd; s++) {
+            for (let e = s; e < Math.min(s + maxAnswerLength, spanEnd + 1); e++) {
+                const score = (startProbs[s] ?? 0) * (endProbs[e] ?? 0);
+                if (score > bestScore) {
+                    bestScore = score;
+                    bestStartIdx = s;
+                    bestEndIdx = e;
+                }
+            }
+        }
+        // Decode the answer span directly from token IDs in the context portion
+        const answerTokenIds = encoded.inputIds.slice(bestStartIdx, bestEndIdx + 1);
+        const answer = this.tokenizer.decode(answerTokenIds, true);
+        return {
+            answer: answer || '',
+            score: Math.max(0, bestScore),
+            start: bestStartIdx,
+            end: bestEndIdx,
+            processingTime: performance.now() - startTime,
+        };
+    }
+    async preprocess(input) {
+        const qaInput = Array.isArray(input) ? input[0] : input;
+        const encoded = this.tokenizer.encode(qaInput.question, {
+            textPair: qaInput.context,
+            addSpecialTokens: true,
+            maxLength: 512,
+            truncation: true,
+            returnAttentionMask: true,
+            returnTokenTypeIds: true,
+        });
+        return [
+            new EdgeFlowTensor(BigInt64Array.from(encoded.inputIds.map(id => BigInt(id))), [1, encoded.inputIds.length], 'int64'),
+            new EdgeFlowTensor(BigInt64Array.from(encoded.attentionMask.map(m => BigInt(m))), [1, encoded.attentionMask.length], 'int64'),
+        ];
+    }
+    async postprocess(outputs, _options) {
+        if (outputs.length < 2) {
+            return { answer: '', score: 0, start: 0, end: 0 };
+        }
+        const startLogits = outputs[0].toFloat32Array();
+        const endLogits = outputs[1].toFloat32Array();
+        const seqLen = startLogits.length;
+        const startProbs = softmax(new EdgeFlowTensor(startLogits, [seqLen], 'float32')).toFloat32Array();
+        const endProbs = softmax(new EdgeFlowTensor(endLogits, [seqLen], 'float32')).toFloat32Array();
+        let bestStart = 0;
+        let bestEnd = 0;
+        let bestScore = 0;
+        for (let start = 0; start < seqLen; start++) {
+            for (let end = start; end < Math.min(start + 30, seqLen); end++) {
+                const score = (startProbs[start] ?? 0) * (endProbs[end] ?? 0);
+                if (score > bestScore) {
+                    bestScore = score;
+                    bestStart = start;
+                    bestEnd = end;
+                }
+            }
+        }
+        return {
+            answer: '',
+            score: bestScore,
+            start: bestStart,
+            end: bestEnd,
+        };
+    }
+}
+// ============================================================================
+// Factory
+// ============================================================================
+export function createQuestionAnsweringPipeline(config) {
+    return new QuestionAnsweringPipeline(config);
+}
+registerPipeline('question-answering', (config) => new QuestionAnsweringPipeline(config));
+//# sourceMappingURL=question-answering.js.map

package/dist/pipelines/question-answering.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"question-answering.js","sourceRoot":"","sources":["../../src/pipelines/question-answering.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAkB,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAE5D,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAE5E,+EAA+E;AAC/E,iDAAiD;AACjD,+EAA+E;AAE/E,MAAM,cAAc,GAAG;IACrB,KAAK,EAAE,4GAA4G;IACnH,SAAS,EAAE,iGAAiG;CAC7G,CAAC;AA0BF,+EAA+E;AAC/E,8BAA8B;AAC9B,+EAA+E;AAE/E,MAAM,OAAO,yBAA0B,SAAQ,YAG9C;IACS,SAAS,GAAqB,IAAI,CAAC;IACnC,SAAS,GAAuB,IAAI,CAAC;IACrC,QAAQ,CAAS;IACjB,YAAY,CAAS;IAE7B,YAAY,MAAuB;QACjC,KAAK,CAAC,MAAM,IAAI;YACd,IAAI,EAAE,oBAAoB;YAC1B,KAAK,EAAE,SAAS;SACjB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,GAAG,CAAC,MAAM,EAAE,KAAK,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC;QACpG,IAAI,CAAC,YAAY,GAAG,cAAc,CAAC,SAAS,CAAC;IAC/C,CAAC;IAEQ,KAAK,CAAC,UAAU;QACvB,MAAM,KAAK,CAAC,UAAU,EAAE,CAAC;QAEzB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,SAAS,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,IAAI,EAAE,CAAC,CAAC;YAC3F,IAAI,CAAC,SAAS,GAAG,MAAM,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED,YAAY,CAAC,SAAoB;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAEQ,KAAK,CAAC,GAAG,CAChB,KAA0B,EAC1B,OAAkC;QAElC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,CACvD,CAAC;QAEF,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC;IACtD,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,KAAc,EACd,OAAiC;QAEjC,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;QACpC,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC;QAEtD,wEAAwE;QACxE,MAAM,OAAO,GAAG,IAAI,CAAC,SAAU,CAAC,MAAM,CAAC,QAAQ,EAAE;YAC/C,QAAQ,EAAE,OAAO;YACjB,gBAAgB,EAAE,IAAI;YACtB,SAAS,EAAE,GAAG;YACd,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,YAAY;YACrB,mBAAmB,EAAE,IAAI;YACzB,kBAAkB,EAAE,IAAI;SACzB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;QAEvC,MAAM,QAAQ,GAAG,IAAI,cAAc,CACjC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAC1D,CAAC,CAAC,EAAE,MAAM,CAAC,EACX,OAAO,CACR,CAAC;QACF,MAAM,aAAa,GAAG,IAAI,cAAc,CACtC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EAC7D,CAAC,CAAC,EAAE,MAAM,CAAC,EACX,OAAO,CACR,CAAC;QAEF,MAAM,WAAW,GAAG,IAAI,GAAG,EAA0B,CAAC;QACtD,WAAW,CAAC,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QACvC,WAAW,CAAC,GAAG,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,SAAU,EAAE,WAAW,CAAC,CAAC;QAEtE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,cAAc,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC;QACnG,CAAC;QAED,MAAM,WAAW,GAAI,OAAO,CAAC,CAAC,CAAoB,CAAC,cAAc,EAAE,CAAC;QACpE,MAAM,SAAS,GAAI,OAAO,CAAC,CAAC,CAAoB,CAAC,cAAc,EAAE,CAAC;QAElE,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,cAAc,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;QACpH,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,cAAc,CAAC,IAAI,YAAY,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;QAEhH,0EAA0E;QAC1E,kFAAkF;QAClF,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClE,4DAA4D;QAC5D,MAAM,YAAY,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QACrD,MAAM,SAAS,GAAG,YAAY,IAAI,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC,4BAA4B;QAExD,IAAI,YAAY,GAAG,SAAS,CAAC;QAC7B,IAAI,UAAU,GAAG,SAAS,CAAC;QAC3B,IAAI,SAAS,GAAG,CAAC,QAAQ,CAAC;QAE1B,KAAK,IAAI,CAAC,GAAG,SAAS,EAAE,CAAC,IAAI,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,eAAe,EAAE,OAAO,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpE,MAAM,KAAK,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBACxD,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;oBACtB,SAAS,GAAG,KAAK,CAAC;oBAClB,YAAY,GAAG,CAAC,CAAC;oBACjB,UAAU,GAAG,CAAC,CAAC;gBACjB,CAAC;YACH,CAAC;QACH,CAAC;QAED,wEAAwE;QACxE,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,IAAI,CAAC,SAAU,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QAE5D,OAAO;YACL,MAAM,EAAE,MAAM,IAAI,EAAE;YACpB,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC;YAC7B,KAAK,EAAE,YAAY;YACnB,GAAG,EAAE,UAAU;YACf,cAAc,EAAE,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS;SAC9C,CAAC;IACJ,CAAC;IAGS,KAAK,CAAC,UAAU,CAAC,KAA0B;QACnD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,KAAK,CAAC;QACzD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAU,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;YACvD,QAAQ,EAAE,OAAO,CAAC,OAAO;YACzB,gBAAgB,EAAE,IAAI;YACtB,SAAS,EAAE,GAAG;YACd,UAAU,EAAE,IAAI;YAChB,mBAAmB,EAAE,IAAI;YACzB,kBAAkB,EAAE,IAAI;SACzB,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,cAAc,CAChB,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAC1D,CAAC,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC5B,OAAO,CACR;YACD,IAAI,cAAc,CAChB,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EAC7D,CAAC,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,EACjC,OAAO,CACR;SACF,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,WAAW,CACzB,OAAyB,EACzB,QAA0B;QAE1B,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;QACpD,CAAC;QAED,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC,cAAc,EAAE,CAAC;QACjD,MAAM,SAAS,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC,cAAc,EAAE,CAAC;QAC/C,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;QAElC,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,cAAc,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;QAClG,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,cAAc,CAAC,SAAS,EAAE,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;QAE9F,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;YAC5C,KAAK,IAAI,GAAG,GAAG,KAAK,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,EAAE,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;gBAChE,MAAM,KAAK,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC9D,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;oBACtB,SAAS,GAAG,KAAK,CAAC;oBAClB,SAAS,GAAG,KAAK,CAAC;oBAClB,OAAO,GAAG,GAAG,CAAC;gBAChB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO;YACL,MAAM,EAAE,EAAE;YACV,KAAK,EAAE,SAAS;YAChB,KAAK,EAAE,SAAS;YAChB,GAAG,EAAE,OAAO;SACb,CAAC;IACJ,CAAC;CACF;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,MAAM,UAAU,+BAA+B,CAC7C,MAAuB;IAEvB,OAAO,IAAI,yBAAyB,CAAC,MAAM,CAAC,CAAC;AAC/C,CAAC;AAED,gBAAgB,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC"}