8e603bd5db
* fix: mismatch between model json and path * chore: revert preserve model settings * test: add tests
37 lines
1.1 KiB
TypeScript
37 lines
1.1 KiB
TypeScript
import { BaseExtension, ExtensionTypeEnum } from '../extension'
|
|
import {
|
|
GpuSetting,
|
|
HuggingFaceRepoData,
|
|
ImportingModel,
|
|
Model,
|
|
ModelFile,
|
|
ModelInterface,
|
|
OptionType,
|
|
} from '../../types'
|
|
|
|
/**
|
|
* Model extension for managing models.
|
|
*/
|
|
export abstract class ModelExtension extends BaseExtension implements ModelInterface {
|
|
/**
|
|
* Model extension type.
|
|
*/
|
|
type(): ExtensionTypeEnum | undefined {
|
|
return ExtensionTypeEnum.Model
|
|
}
|
|
|
|
abstract downloadModel(
|
|
model: Model,
|
|
gpuSettings?: GpuSetting,
|
|
network?: { proxy: string; ignoreSSL?: boolean }
|
|
): Promise<void>
|
|
abstract cancelModelDownload(modelId: string): Promise<void>
|
|
abstract deleteModel(model: ModelFile): Promise<void>
|
|
abstract getDownloadedModels(): Promise<ModelFile[]>
|
|
abstract getConfiguredModels(): Promise<ModelFile[]>
|
|
abstract importModels(models: ImportingModel[], optionType: OptionType): Promise<void>
|
|
abstract updateModelInfo(modelInfo: Partial<ModelFile>): Promise<ModelFile>
|
|
abstract fetchHuggingFaceRepoData(repoId: string): Promise<HuggingFaceRepoData>
|
|
abstract getDefaultModel(): Promise<Model>
|
|
}
|