@workglow/test 0.0.92 → 0.0.93

package/dist/node.js

@@ -32,566 +32,6 @@ class InMemoryTaskOutputRepository extends TaskOutputTabularRepository {
     });
   }
 }
-// src/samples/index.ts
-import {
-  HuggingFaceTransformersProvider,
-  TensorFlowMediaPipeProvider
-} from "@workglow/ai-provider";
-import { HFT_TASKS } from "@workglow/ai-provider/hf-transformers";
-import { TFMP_TASKS } from "@workglow/ai-provider/tf-mediapipe";
-
-// src/samples/MediaPipeModelSamples.ts
-import { getGlobalModelRepository } from "@workglow/ai";
-import { TENSORFLOW_MEDIAPIPE } from "@workglow/ai-provider";
-async function registerMediaPipeTfJsLocalModels() {
-  const models = [
-    {
-      model_id: "media-pipe:Universal Sentence Encoder",
-      title: "Universal Sentence Encoder",
-      description: "Universal Sentence Encoder",
-      tasks: ["TextEmbeddingTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "text",
-        pipeline: "text-embedder",
-        model_path: "https://storage.googleapis.com/mediapipe-tasks/text_embedder/universal_sentence_encoder.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:BERT Text Classifier",
-      title: "BERT Text Classifier",
-      description: "BERT-based text classification model",
-      tasks: ["TextClassificationTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "text",
-        pipeline: "text-classifier",
-        model_path: "https://storage.googleapis.com/mediapipe-models/text_classifier/bert_classifier/float32/1/bert_classifier.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:Language Detector",
-      title: "Language Detector",
-      description: "Language detection model",
-      tasks: ["TextLanguageDetectionTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "text",
-        pipeline: "text-language-detector",
-        model_path: "https://storage.googleapis.com/mediapipe-models/language_detector/language_detector/float32/1/language_detector.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:EfficientNet Lite0 Image Classifier",
-      title: "EfficientNet Lite0",
-      description: "Lightweight image classification model",
-      tasks: ["ImageClassificationTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-image-classifier",
-        model_path: "https://storage.googleapis.com/mediapipe-models/image_classifier/efficientnet_lite0/float32/1/efficientnet_lite0.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:MobileNet V3 Image Embedder",
-      title: "MobileNet V3 Small",
-      description: "Lightweight image embedding model",
-      tasks: ["ImageEmbeddingTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-image-embedder",
-        model_path: "https://storage.googleapis.com/mediapipe-models/image_embedder/mobilenet_v3_small/float32/1/mobilenet_v3_small.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:EfficientDet Lite0 Object Detector",
-      title: "EfficientDet Lite0",
-      description: "Lightweight object detection model",
-      tasks: ["ObjectDetectionTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-object-detector",
-        model_path: "https://storage.googleapis.com/mediapipe-models/object_detector/efficientdet_lite0/float32/1/efficientdet_lite0.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:DeepLab V3 Image Segmenter",
-      title: "DeepLab V3",
-      description: "Image segmentation model",
-      tasks: ["ImageSegmentationTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-image-segmenter",
-        model_path: "https://storage.googleapis.com/mediapipe-models/image_segmenter/deeplab_v3/float32/1/deeplab_v3.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:YAMNet Audio Classifier",
-      title: "YAMNet",
-      description: "Audio event classification model",
-      tasks: ["AudioClassificationTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "audio",
-        pipeline: "audio-classifier",
-        model_path: "https://storage.googleapis.com/mediapipe-models/audio_classifier/yamnet/float32/1/yamnet.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:Gesture Recognizer",
-      title: "Gesture Recognizer",
-      description: "Recognizes hand gestures (thumbs up, victory, etc.)",
-      tasks: ["GestureRecognizerTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-gesture-recognizer",
-        model_path: "https://storage.googleapis.com/mediapipe-models/gesture_recognizer/gesture_recognizer/float16/1/gesture_recognizer.task"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:Hand Landmarker",
-      title: "Hand Landmarker",
-      description: "Detects 21 hand landmarks",
-      tasks: ["HandLandmarkerTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-hand-landmarker",
-        model_path: "https://storage.googleapis.com/mediapipe-models/hand_landmarker/hand_landmarker/float16/1/hand_landmarker.task"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:Face Detector",
-      title: "Face Detector",
-      description: "Detects faces with bounding boxes and keypoints",
-      tasks: ["FaceDetectorTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-face-detector",
-        model_path: "https://storage.googleapis.com/mediapipe-models/face_detector/blaze_face_short_range/float16/1/blaze_face_short_range.tflite"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:Face Landmarker",
-      title: "Face Landmarker",
-      description: "Detects 478 facial landmarks with blendshapes",
-      tasks: ["FaceLandmarkerTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-face-landmarker",
-        model_path: "https://storage.googleapis.com/mediapipe-models/face_landmarker/face_landmarker/float16/1/face_landmarker.task"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "media-pipe:Pose Landmarker",
-      title: "Pose Landmarker",
-      description: "Detects 33 body pose landmarks",
-      tasks: ["PoseLandmarkerTask"],
-      provider: TENSORFLOW_MEDIAPIPE,
-      provider_config: {
-        task_engine: "vision",
-        pipeline: "vision-pose-landmarker",
-        model_path: "https://storage.googleapis.com/mediapipe-models/pose_landmarker/pose_landmarker_lite/float16/1/pose_landmarker_lite.task"
-      },
-      metadata: {}
-    }
-  ];
-  for (const model of models) {
-    await getGlobalModelRepository().addModel(model);
-  }
-}
-// src/samples/ONNXModelSamples.ts
-import { getGlobalModelRepository as getGlobalModelRepository2 } from "@workglow/ai";
-import { HF_TRANSFORMERS_ONNX } from "@workglow/ai-provider";
-async function registerHuggingfaceLocalModels() {
-  const onnxModels = [
-    {
-      model_id: "onnx:Xenova/all-MiniLM-L6-v2:q8",
-      title: "All MiniLM L6 V2 384D",
-      description: "Xenova/all-MiniLM-L6-v2",
-      tasks: ["TextEmbeddingTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "feature-extraction",
-        model_path: "Xenova/all-MiniLM-L6-v2",
-        native_dimensions: 384
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/bge-base-en-v1.5:q8",
-      title: "BGE Base English V1.5 768D",
-      description: "Xenova/bge-base-en-v1.5",
-      tasks: ["TextEmbeddingTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "feature-extraction",
-        model_path: "Xenova/bge-base-en-v1.5",
-        native_dimensions: 768
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/gte-small:q8",
-      title: "GTE Small 384D",
-      description: "Xenova/gte-small",
-      tasks: ["TextEmbeddingTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "feature-extraction",
-        model_path: "Xenova/gte-small",
-        native_dimensions: 384
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:onnx-community/bert_uncased_L-2_H-128_A-2-ONNX:q8",
-      title: "BERT Uncased 128D",
-      description: "onnx-community/bert_uncased_L-2_H-128_A-2-ONNX",
-      tasks: ["TextEmbeddingTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "feature-extraction",
-        model_path: "onnx-community/bert_uncased_L-2_H-128_A-2-ONNX",
-        native_dimensions: 128
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/paraphrase-albert-base-v2:q8",
-      title: "Paraphrase ALBERT Base V2 768D",
-      description: "Xenova/paraphrase-albert-base-v2",
-      tasks: ["TextEmbeddingTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "feature-extraction",
-        model_path: "Xenova/paraphrase-albert-base-v2",
-        native_dimensions: 768
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Qwen3-Embedding-0.6B:auto",
-      title: "Qwen3 Embedding 0.6B",
-      description: "onnx-community/Qwen3-Embedding-0.6B-ONNX",
-      tasks: ["TextEmbeddingTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "feature-extraction",
-        model_path: "onnx-community/Qwen3-Embedding-0.6B-ONNX",
-        native_dimensions: 1024,
-        mrl: true
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:onnx-community/NeuroBERT-NER-ONNX:q8",
-      title: "NeuroBERT NER",
-      description: "onnx-community/NeuroBERT-NER-ONNX",
-      tasks: ["TextNamedEntityRecognitionTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "token-classification",
-        model_path: "onnx-community/NeuroBERT-NER-ONNX"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/distilbert-base-uncased-distilled-squad:q8",
-      title: "distilbert-base-uncased-distilled-squad",
-      description: "Xenova/distilbert-base-uncased-distilled-squad quantized to 8bit",
-      tasks: ["TextQuestionAnswerTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "question-answering",
-        model_path: "Xenova/distilbert-base-uncased-distilled-squad"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:onnx-community/ModernBERT-finetuned-squad-ONNX",
-      title: "ModernBERT-finetuned-squad-ONNX",
-      description: "onnx-community/ModernBERT-finetuned-squad-ONNX quantized to int8",
-      tasks: ["TextQuestionAnswerTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "question-answering",
-        model_path: "onnx-community/ModernBERT-finetuned-squad-ONNX",
-        dtype: "int8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/gpt2:q8",
-      title: "gpt2",
-      description: "Xenova/gpt2 quantized to 8bit",
-      tasks: ["TextGenerationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "text-generation",
-        model_path: "Xenova/gpt2",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/Phi-3-mini-4k-instruct:q4f16",
-      title: "Phi-3-mini-4k-instruct:q4f16",
-      description: "Xenova/Phi-3-mini-4k-instruct quantized to q4f16",
-      tasks: ["TextGenerationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "text-generation",
-        model_path: "Xenova/Phi-3-mini-4k-instruct",
-        dtype: "q4f16",
-        use_external_data_format: true
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/distilgpt2:q8",
-      title: "distilgpt2",
-      description: "Xenova/distilgpt2 quantized to 8bit",
-      tasks: ["TextGenerationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "text-generation",
-        model_path: "Xenova/distilgpt2",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/LaMini-Flan-T5-783M:q8",
-      title: "LaMini-Flan-T5-783M",
-      description: "Xenova/LaMini-Flan-T5-783M quantized to 8bit",
-      tasks: ["TextGenerationTask", "TextRewriterTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "text2text-generation",
-        model_path: "Xenova/LaMini-Flan-T5-783M",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Falconsai/text_summarization:fp32",
-      title: "text_summarization",
-      description: "Falconsai/text_summarization",
-      tasks: ["TextSummaryTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "summarization",
-        model_path: "Falconsai/text_summarization",
-        dtype: "fp32"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/nllb-200-distilled-600M:q8",
-      title: "nllb-200-distilled-600M",
-      description: "Xenova/nllb-200-distilled-600M quantized to 8bit",
-      tasks: ["TextTranslationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "translation",
-        model_path: "Xenova/nllb-200-distilled-600M",
-        language_style: "FLORES-200",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/m2m100_418M:q8",
-      title: "m2m100_418M",
-      description: "Xenova/m2m100_418M quantized to 8bit",
-      tasks: ["TextTranslationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "translation",
-        model_path: "Xenova/m2m100_418M",
-        language_style: "ISO-639",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/m2m100_418M:q8",
-      title: "m2m100_418M",
-      description: "Xenova/m2m100_418M quantized to 8bit",
-      tasks: ["TextTranslationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "translation",
-        model_path: "Xenova/m2m100_418M",
-        language_style: "ISO-639",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/mbart-large-50-many-to-many-mmt:q8",
-      title: "mbart-large-50-many-to-many-mmt",
-      description: "Xenova/mbart-large-50-many-to-many-mmt quantized to 8bit",
-      tasks: ["TextTranslationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "translation",
-        model_path: "Xenova/mbart-large-50-many-to-many-mmt",
-        language_style: "ISO-639_ISO-3166-1-alpha-2",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/vit-base-patch16-224:q8",
-      title: "ViT Base Patch16 224",
-      description: "Vision Transformer for image classification",
-      tasks: ["ImageClassificationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "image-classification",
-        model_path: "Xenova/vit-base-patch16-224",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/clip-vit-base-patch32:q8",
-      title: "CLIP ViT Base Patch32",
-      description: "CLIP model for zero-shot image classification and embeddings",
-      tasks: ["ImageClassificationTask", "ImageEmbeddingTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "zero-shot-image-classification",
-        model_path: "Xenova/clip-vit-base-patch32",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/detr-resnet-50:q8",
-      title: "DETR ResNet-50",
-      description: "Object detection model",
-      tasks: ["ObjectDetectionTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "object-detection",
-        model_path: "Xenova/detr-resnet-50",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/owlvit-base-patch32:q8",
-      title: "OWL-ViT Base Patch32",
-      description: "Zero-shot object detection model",
-      tasks: ["ObjectDetectionTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "zero-shot-object-detection",
-        model_path: "Xenova/owlvit-base-patch32",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/segformer-b0-finetuned-ade-512-512:q8",
-      title: "Segformer B0 ADE",
-      description: "Image segmentation model trained on ADE20K dataset",
-      tasks: ["ImageSegmentationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "image-segmentation",
-        model_path: "Xenova/segformer-b0-finetuned-ade-512-512",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/vit-gpt2-image-captioning:q8",
-      title: "ViT GPT2 Image Captioning",
-      description: "Image to text captioning model",
-      tasks: ["ImageToTextTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "image-to-text",
-        model_path: "Xenova/vit-gpt2-image-captioning",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/modnet:q8",
-      title: "MODNet Background Removal",
-      description: "Background removal model",
-      tasks: ["BackgroundRemovalTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "background-removal",
-        model_path: "Xenova/modnet",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/mobilebert-uncased-mnli:q8",
-      title: "MobileBERT MNLI",
-      description: "Zero-shot text classification model",
-      tasks: ["TextClassificationTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "zero-shot-classification",
-        model_path: "Xenova/mobilebert-uncased-mnli",
-        dtype: "q8"
-      },
-      metadata: {}
-    },
-    {
-      model_id: "onnx:Xenova/bge-reranker-base:q8",
-      title: "BGE Reranker Base",
-      description: "Cross-encoder reranker model for relevance scoring",
-      tasks: ["TextClassificationTask", "RerankerTask"],
-      provider: HF_TRANSFORMERS_ONNX,
-      provider_config: {
-        pipeline: "text-classification",
-        model_path: "Xenova/bge-reranker-base",
-        dtype: "q8"
-      },
-      metadata: {}
-    }
-  ];
-  for (const model of onnxModels) {
-    await getGlobalModelRepository2().addModel(model);
-  }
-}
-
-// src/samples/index.ts
-async function register_HFT_InMemoryQueue() {
-  await new HuggingFaceTransformersProvider(HFT_TASKS).register({ mode: "inline" });
-}
-async function register_TFMP_InMemoryQueue() {
-  await new TensorFlowMediaPipeProvider(TFMP_TASKS).register({ mode: "inline" });
-}
 // src/binding/FsFolderTaskGraphRepository.ts
 import { FsFolderTabularStorage } from "@workglow/storage";
 import {
@@ -729,10 +169,6 @@ class SqliteTaskOutputRepository extends TaskOutputTabularRepository5 {
   }
 }
 export {
-  register_TFMP_InMemoryQueue,
-  register_HFT_InMemoryQueue,
-  registerMediaPipeTfJsLocalModels,
-  registerHuggingfaceLocalModels,
   SqliteTaskOutputRepository,
   SqliteTaskGraphRepository,
   SQLITE_TASK_OUTPUT_REPOSITORY,
@@ -755,4 +191,4 @@ export {
   FS_FOLDER_TASK_GRAPH_REPOSITORY
 };
 
-//# debugId=A492DE3C008ECA7464756E2164756E21
+//# debugId=E54A8018F73E78C164756E2164756E21