package model
|
|
const (
|
DefaultModelID = "sensevoice-zh"
|
MoonshineModelID = "moonshine-en"
|
DefaultLanguageID = "zh-CN"
|
EnglishLanguageID = "en"
|
)
|
|
type ModelProfile struct {
|
ID string
|
DisplayName string
|
BackendKind string
|
Tier string
|
SupportedLanguageIDs []string
|
RecommendedFor []string
|
Description string
|
ApproxSize string
|
DownloadURLs []string
|
InstallDirName string
|
LegacyDirNames []string
|
RequiredFiles []RequiredFileRule
|
ProviderOrder []string
|
NumThreads int
|
}
|
|
type RequiredFileRule struct {
|
Role string
|
AllOf []string
|
AnyOf []string
|
Required bool
|
}
|
|
type LanguageProfile struct {
|
ID string
|
DisplayName string
|
NativeName string
|
UILocale string
|
SystemMatchers []string
|
DefaultModelID string
|
UpgradeModelIDs []string
|
}
|
|
type ModelInstallStatus string
|
|
const (
|
ModelNotInstalled ModelInstallStatus = "not_installed"
|
ModelInstalledValid ModelInstallStatus = "installed_valid"
|
ModelInstalledIncomplete ModelInstallStatus = "installed_incomplete"
|
ModelStateInvalidFallbackFound ModelInstallStatus = "state_invalid_fallback_found"
|
ModelStateInvalidNoFallback ModelInstallStatus = "state_invalid_no_fallback"
|
)
|
|
type ResolvedModel struct {
|
ModelID string
|
Profile ModelProfile
|
RootDir string
|
SourceDirKind string
|
Files map[string]string
|
BackendKind string
|
ProviderOrder []string
|
Status ModelInstallStatus
|
Missing []string
|
Problems []string
|
}
|
|
func (r ResolvedModel) IsUsable() bool {
|
return r.Status == ModelInstalledValid || r.Status == ModelStateInvalidFallbackFound
|
}
|
