69 lines
2.3 KiB
Go
69 lines
2.3 KiB
Go
package provider
|
|
|
|
import "github.com/leonardotrapani/hyprvoice/internal/language"
|
|
|
|
// ModelType represents the type of a model
|
|
type ModelType int
|
|
|
|
const (
|
|
Transcription ModelType = iota
|
|
LLM
|
|
)
|
|
|
|
// Model represents a model with full metadata
|
|
type Model struct {
|
|
ID string // unique identifier (e.g., "whisper-1", "gpt-4o-mini")
|
|
Name string // display name (e.g., "Whisper 1", "GPT-4o Mini")
|
|
Description string // short description
|
|
Type ModelType // transcription or LLM
|
|
Streaming bool // supports streaming
|
|
Local bool // runs locally (no API call)
|
|
AdapterType string // which adapter to use (e.g., "openai", "elevenlabs", "whisper-cpp")
|
|
SupportedLanguages []string // explicit list of supported language codes
|
|
Endpoint *EndpointConfig // nil for local models
|
|
LocalInfo *LocalModelInfo // nil for cloud models
|
|
}
|
|
|
|
// EndpointConfig holds HTTP/WebSocket endpoint configuration
|
|
type EndpointConfig struct {
|
|
BaseURL string // e.g., "https://api.openai.com" or "wss://api.deepgram.com"
|
|
Path string // e.g., "/v1/audio/transcriptions"
|
|
}
|
|
|
|
// LocalModelInfo holds metadata for downloadable local models
|
|
type LocalModelInfo struct {
|
|
Filename string // e.g., "ggml-base.en.bin"
|
|
Size string // human readable size (e.g., "142MB")
|
|
DownloadURL string // full URL to download from
|
|
}
|
|
|
|
// NeedsDownload returns true if this is a local model that requires downloading
|
|
func (m *Model) NeedsDownload() bool {
|
|
return m.LocalInfo != nil
|
|
}
|
|
|
|
// IsStreaming returns true if this model supports streaming
|
|
func (m *Model) IsStreaming() bool {
|
|
return m.Streaming
|
|
}
|
|
|
|
// SupportsLanguage returns true if the model supports the given language code.
|
|
// Auto-detect (empty string) is always supported.
|
|
func (m *Model) SupportsLanguage(code string) bool {
|
|
if code == "" {
|
|
return true // auto always supported
|
|
}
|
|
for _, supported := range m.SupportedLanguages {
|
|
if supported == code {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// SupportsAllLanguages returns true if the model supports all 57 languages
|
|
func (m *Model) SupportsAllLanguages() bool {
|
|
allCodes := language.AllLanguageCodes()
|
|
return len(m.SupportedLanguages) == len(allCodes)
|
|
}
|