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