openvino-genai-node
Version:
OpenVINO™ GenAI pipelines for using from Node.js environment
16 lines (15 loc) • 709 B
TypeScript
import { TextEmbeddingPipelineWrapper, EmbeddingResult, EmbeddingResults, TextEmbeddingConfig } from "../addon.js";
export declare class TextEmbeddingPipeline {
modelPath: string;
device: string;
config: TextEmbeddingConfig;
ovProperties: object;
pipeline: TextEmbeddingPipelineWrapper | null;
isInitialized: boolean;
constructor(modelPath: string, device: string, config: TextEmbeddingConfig, ovProperties?: object);
init(): Promise<void>;
embedDocumentsSync(texts: string[]): EmbeddingResults;
embedQuerySync(text: string): EmbeddingResult;
embedDocuments(documents: string[]): Promise<EmbeddingResults>;
embedQuery(text: string): Promise<EmbeddingResult>;
}