armor-editor
Version:
Advanced rich text editor with premium armor-grade security, real-time collaboration, spell checking, track changes, and framework-agnostic design for React, Vue, Angular, Next.js, Nuxt.js
40 lines (39 loc) • 1.31 kB
TypeScript
export interface LocalAIModel {
id: string;
name: string;
type: 'text-generation' | 'text-completion' | 'embedding' | 'classification';
modelUrl: string;
tokenizerUrl?: string;
maxTokens: number;
isLoaded: boolean;
size: string;
}
export declare class CompleteLocalAI {
private models;
private loadedModels;
private worker;
private tf;
private isInitialized;
constructor();
private initializeTensorFlow;
private loadTensorFlow;
private setupFallbackAI;
private setupWorker;
private handleWorkerMessage;
private loadDefaultModels;
addModel(model: LocalAIModel): void;
loadModel(modelId: string): Promise<boolean>;
generateText(modelId: string, prompt: string, options?: any): Promise<string>;
private fallbackTextGeneration;
private generateContinuation;
classifyText(modelId: string, text: string): Promise<any>;
private fallbackClassification;
embedText(modelId: string, text: string): Promise<number[]>;
private fallbackEmbedding;
getAvailableModels(): LocalAIModel[];
getLoadedModels(): LocalAIModel[];
unloadModel(modelId: string): Promise<boolean>;
getModelInfo(modelId: string): LocalAIModel | null;
benchmarkModel(modelId: string): Promise<any>;
destroy(): void;
}