UNPKG

openvino-genai-node

Version:

OpenVINO™ GenAI pipelines for using from Node.js environment

68 lines 3.4 kB
// Copyright (C) 2025-2026 Intel Corporation // SPDX-License-Identifier: Apache-2.0 import { LLMPipeline as LLM } from "./pipelines/llmPipeline.js"; import { VLMPipeline as VLM } from "./pipelines/vlmPipeline.js"; import { TextEmbeddingPipeline as Embedding } from "./pipelines/textEmbeddingPipeline.js"; import { TextRerankPipeline as TextRerank, } from "./pipelines/textRerankPipeline.js"; import { WhisperPipeline as Whisper } from "./pipelines/whisperPipeline.js"; import { Text2ImagePipeline as Text2Image } from "./pipelines/text2ImagePipeline.js"; import { Image2ImagePipeline as Image2Image } from "./pipelines/image2ImagePipeline.js"; import { InpaintingPipeline as Inpainting } from "./pipelines/inpaintingPipeline.js"; import { Text2SpeechPipeline as Text2Speech } from "./pipelines/text2SpeechPipeline.js"; class PipelineFactory { static async LLMPipeline(modelPath, device, properties = {}) { if (device === undefined) device = "CPU"; if (typeof device !== "string") { throw new Error("The second argument must be a device string. If you want to pass LLMPipelineProperties, please use the third argument."); } const pipeline = new LLM(modelPath, device, properties); await pipeline.init(); return pipeline; } static async VLMPipeline(modelPath, device = "CPU", properties = {}) { const pipeline = new VLM(modelPath, device, properties); await pipeline.init(); return pipeline; } static async TextEmbeddingPipeline(modelPath, device = "CPU", config = {}) { const pipeline = new Embedding(modelPath, device, config); await pipeline.init(); return pipeline; } static async TextRerankPipeline(modelPath, options = {}) { const pipeline = new TextRerank(modelPath, options); await pipeline.init(); return pipeline; } static async WhisperPipeline(modelPath, device = "CPU", properties = {}) { const pipeline = new Whisper(modelPath, device, properties); await pipeline.init(); return pipeline; } static async Text2ImagePipeline(modelPath, device = "CPU", properties = {}) { const pipeline = new Text2Image(modelPath, device, properties); await pipeline.init(); return pipeline; } static async Image2ImagePipeline(modelPath, device = "CPU", properties = {}) { const pipeline = new Image2Image(modelPath, device, properties); await pipeline.init(); return pipeline; } static async InpaintingPipeline(modelPath, device = "CPU", properties = {}) { const pipeline = new Inpainting(modelPath, device, properties); await pipeline.init(); return pipeline; } static async Text2SpeechPipeline(modelPath, device = "CPU", properties = {}) { const pipeline = new Text2Speech(modelPath, device, properties); await pipeline.init(); return pipeline; } } export const { LLMPipeline, VLMPipeline, TextEmbeddingPipeline, TextRerankPipeline, WhisperPipeline, Text2ImagePipeline, Image2ImagePipeline, InpaintingPipeline, Text2SpeechPipeline, } = PipelineFactory; export { DecodedResults, VLMDecodedResults, WhisperDecodedResults, Text2SpeechDecodedResults, } from "./decodedResults.js"; export * from "./utils.js"; export * from "./addon.js"; //# sourceMappingURL=index.js.map