UNPKG

@sconedev/ai_toolkit

Version:

Simplify AI integration in web apps with local and offline model support

45 lines (44 loc) 2.16 kB
import { summarizeText, summarizePDF } from './core/pdfSummarize.browser'; import { chat, LocalChatMessage } from './core/chat'; import { generateImage } from './core/generateImage'; import { configure, getConfig } from './core/config'; import { OnnxModelError, OnnxRuntimeError, runModelFromPath, unloadOnnxModel, isWasmSupported, downloadModelForOfflineUse, isModelInCache } from './core/onnxRuntime'; import { registerLocalModel, getLocalModel, listLocalModels, downloadAndRegisterModel } from './models/modelRegistry'; import { runModelFromFile, loadModelFromUrl, isWebGPUSupported } from './core/onnxRuntime.browser'; import { extractTextFromURL } from './utils/web'; import { processText } from './utils/processText'; export { configure, getConfig, chat, summarizeText, summarizePDF, generateImage, extractTextFromURL, processText, OnnxModelError, OnnxRuntimeError, }; export type { LocalChatMessage }; export declare const aiToolkit: { configure: typeof configure; getConfig: typeof getConfig; chat: typeof chat; summarizeText: typeof summarizeText; summarizePDF: typeof summarizePDF; generateImage: typeof generateImage; utils: { extractTextFromURL: typeof extractTextFromURL; processText: typeof processText; }; onnx: { runModel: typeof runModelFromFile; unloadModel: typeof unloadOnnxModel; runLocalModel: typeof runModelFromPath; getModels: () => string[]; getModelConfig: (name: string) => import("./core/onnxRuntime").OnnxModelConfig; isWasmSupported: typeof isWasmSupported; isWebGPUSupported: typeof isWebGPUSupported; downloadModel: typeof downloadModelForOfflineUse; loadFromUrl: typeof loadModelFromUrl; isModelCached: typeof isModelInCache; registerLocalModel: typeof registerLocalModel; getLocalModel: typeof getLocalModel; listLocalModels: typeof listLocalModels; downloadAndRegister: typeof downloadAndRegisterModel; }; errors: { OnnxModelError: typeof OnnxModelError; OnnxRuntimeError: typeof OnnxRuntimeError; }; }; export default aiToolkit;