thread/hyprvoice
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: models test and fixes

Browse Source
This commit is contained in:
leonardotrapani
2026-02-02 01:13:33 +01:00
parent 2729882b9d
commit 195f9f5115
34 changed files with 507 additions and 154 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/models/whisper/models.go
+3
View File
@@ -28,7 +28,10 @@ var models = []ModelInfo{
{ID: "base", Name: "Base", Filename: "ggml-base.bin", Size: "142MB", SizeBytes: 142_000_000, Multilingual: true},
{ID: "small", Name: "Small", Filename: "ggml-small.bin", Size: "466MB", SizeBytes: 466_000_000, Multilingual: true},
{ID: "medium", Name: "Medium", Filename: "ggml-medium.bin", Size: "1.5GB", SizeBytes: 1_500_000_000, Multilingual: true},
{ID: "large-v1", Name: "Large V1", Filename: "ggml-large-v1.bin", Size: "2.9GB", SizeBytes: 2_900_000_000, Multilingual: true},
{ID: "large-v2", Name: "Large V2", Filename: "ggml-large-v2.bin", Size: "2.9GB", SizeBytes: 2_900_000_000, Multilingual: true},
{ID: "large-v3", Name: "Large V3", Filename: "ggml-large-v3.bin", Size: "3GB", SizeBytes: 3_000_000_000, Multilingual: true},
{ID: "large-v3-turbo", Name: "Large V3 Turbo", Filename: "ggml-large-v3-turbo.bin", Size: "1.6GB", SizeBytes: 1_600_000_000, Multilingual: true},
}
// modelByID maps model ID to ModelInfo for quick lookup
internal/models/whisper/whisper_test.go
+5 -5
View File
@@ -108,8 +108,8 @@ func TestGetModel(t *testing.T) {
func TestListModels(t *testing.T) {
models := ListModels()
if len(models) != 9 {
t.Errorf("ListModels() returned %d models, want 9", len(models))
if len(models) != 12 {
t.Errorf("ListModels() returned %d models, want 12", len(models))
}
// verify known models exist
@@ -118,7 +118,7 @@ func TestListModels(t *testing.T) {
ids[m.ID] = true
}
expected := []string{"tiny.en", "base.en", "small.en", "medium.en", "tiny", "base", "small", "medium", "large-v3"}
expected := []string{"tiny.en", "base.en", "small.en", "medium.en", "tiny", "base", "small", "medium", "large-v1", "large-v2", "large-v3", "large-v3-turbo"}
for _, id := range expected {
if !ids[id] {
t.Errorf("ListModels() missing model %s", id)
@@ -128,8 +128,8 @@ func TestListModels(t *testing.T) {
func TestListMultilingualModels(t *testing.T) {
models := ListMultilingualModels()
if len(models) != 5 {
t.Errorf("ListMultilingualModels() returned %d models, want 5", len(models))
if len(models) != 8 {
t.Errorf("ListMultilingualModels() returned %d models, want 8", len(models))
}
for _, m := range models {