UNPKG

@sconedev/ai_toolkit

Version:

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

52 lines (51 loc) 1.96 kB
// Re-export browser-compatible functions import { summarizeText, summarizePDF } from './core/pdfSummarize.browser'; import { chat } 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 { predefinedModels } from './models/onnxModels'; 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 the public API export { configure, getConfig, chat, summarizeText, summarizePDF, generateImage, extractTextFromURL, processText, // Removed duplicate and undefined export OnnxModelError, OnnxRuntimeError, }; // Export a unified API export const aiToolkit = { configure, getConfig, chat, summarizeText, summarizePDF, generateImage, utils: { extractTextFromURL, processText }, onnx: { runModel: runModelFromFile, unloadModel: unloadOnnxModel, runLocalModel: runModelFromPath, getModels: () => Object.keys(predefinedModels), getModelConfig: (name) => predefinedModels[name], isWasmSupported, isWebGPUSupported, downloadModel: downloadModelForOfflineUse, loadFromUrl: loadModelFromUrl, isModelCached: isModelInCache, // Local model registry functions registerLocalModel, getLocalModel, listLocalModels, downloadAndRegister: downloadAndRegisterModel, }, errors: { OnnxModelError, OnnxRuntimeError, } }; // Default export export default aiToolkit;