@sconedev/ai_toolkit
Version:
Simplify AI integration in web apps with local and offline model support
52 lines (51 loc) • 1.96 kB
JavaScript
// 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;