From d435cd4b9754de4a567261b1a877add3af977d9a Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Sun, 07 Jun 2026 01:41:26 +0800
Subject: [PATCH] Add X-ASR experimental backend

---
 privatevoice.src/internal/model/model_test.go |   83 +++++++++++++++++++++++++++++++++++++++--
 1 files changed, 79 insertions(+), 4 deletions(-)

diff --git a/privatevoice.src/internal/model/model_test.go b/privatevoice.src/internal/model/model_test.go
index c334d19..1e01c14 100644
--- a/privatevoice.src/internal/model/model_test.go
+++ b/privatevoice.src/internal/model/model_test.go
@@ -80,6 +80,25 @@
 	}
 }
 
+func TestRegistryReturnsXASRZhEn960(t *testing.T) {
+	profile, err := GetModelProfile(XASRZhEn960ModelID)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if profile.BackendKind != BackendXASRStreaming {
+		t.Fatalf("backend kind = %q, want %q", profile.BackendKind, BackendXASRStreaming)
+	}
+	if profile.InstallDirName != "x-asr-zh-en-960ms" {
+		t.Fatalf("install dir = %q, want x-asr-zh-en-960ms", profile.InstallDirName)
+	}
+	if profile.Tier != "experimental" {
+		t.Fatalf("tier = %q, want experimental", profile.Tier)
+	}
+	if len(profile.DownloadURLs) != 0 {
+		t.Fatal("expected x-asr experiment to use manual/scripted install, not in-app tar download")
+	}
+}
+
 func TestChineseLanguageOffersQwen3ASRUpgrade(t *testing.T) {
 	profile, err := GetLanguageProfile(DefaultLanguageID)
 	if err != nil {
@@ -88,8 +107,8 @@
 	if profile.DefaultModelID != DefaultModelID {
 		t.Fatalf("Chinese default model = %q, want %q", profile.DefaultModelID, DefaultModelID)
 	}
-	if len(profile.UpgradeModelIDs) != 1 || profile.UpgradeModelIDs[0] != Qwen3ASRModelID {
-		t.Fatalf("Chinese upgrade models = %v, want [%s]", profile.UpgradeModelIDs, Qwen3ASRModelID)
+	if !stringSliceContains(profile.UpgradeModelIDs, Qwen3ASRModelID) || !stringSliceContains(profile.UpgradeModelIDs, XASRZhEn960ModelID) {
+		t.Fatalf("Chinese upgrade models = %v, want qwen3 and x-asr", profile.UpgradeModelIDs)
 	}
 }
 
@@ -101,8 +120,8 @@
 	if profile.DefaultModelID != MoonshineModelID {
 		t.Fatalf("English default model = %q, want %q", profile.DefaultModelID, MoonshineModelID)
 	}
-	if len(profile.UpgradeModelIDs) != 1 || profile.UpgradeModelIDs[0] != ParakeetModelID {
-		t.Fatalf("English upgrade models = %v, want [%s]", profile.UpgradeModelIDs, ParakeetModelID)
+	if !stringSliceContains(profile.UpgradeModelIDs, ParakeetModelID) || !stringSliceContains(profile.UpgradeModelIDs, XASRZhEn960ModelID) {
+		t.Fatalf("English upgrade models = %v, want parakeet and x-asr", profile.UpgradeModelIDs)
 	}
 }
 
@@ -282,6 +301,42 @@
 	result := ValidateModelDir(profile, dir)
 	if result.Valid {
 		t.Fatal("expected tokenizer file to fail directory validation")
+	}
+}
+
+func TestValidateXASRZhEn960RequiredFiles(t *testing.T) {
+	profile, err := GetModelProfile(XASRZhEn960ModelID)
+	if err != nil {
+		t.Fatal(err)
+	}
+	dir := t.TempDir()
+	createValidXASRZhEn960(t, dir)
+
+	result := ValidateModelDir(profile, dir)
+	if !result.Valid {
+		t.Fatalf("expected valid x-asr dir, missing=%v problems=%v", result.Missing, result.Problems)
+	}
+	for _, role := range []string{"encoder", "decoder", "joiner", "tokens"} {
+		if result.Files[role] == "" {
+			t.Fatalf("missing resolved file role %q", role)
+		}
+	}
+}
+
+func TestValidateXASRZhEn960RejectsMissingJoiner(t *testing.T) {
+	profile, err := GetModelProfile(XASRZhEn960ModelID)
+	if err != nil {
+		t.Fatal(err)
+	}
+	dir := t.TempDir()
+	createValidXASRZhEn960(t, dir)
+	if err := os.Remove(filepath.Join(dir, "joiner-960ms.onnx")); err != nil {
+		t.Fatal(err)
+	}
+
+	result := ValidateModelDir(profile, dir)
+	if result.Valid {
+		t.Fatal("expected missing joiner to fail validation")
 	}
 }
 
@@ -681,6 +736,26 @@
 	writeTestFile(t, filepath.Join(dir, "tokenizer", "vocab.json"), "vocab")
 }
 
+func createValidXASRZhEn960(t *testing.T, dir string) {
+	t.Helper()
+	if err := os.MkdirAll(dir, 0755); err != nil {
+		t.Fatal(err)
+	}
+	writeTestFile(t, filepath.Join(dir, "encoder-960ms.onnx"), "encoder")
+	writeTestFile(t, filepath.Join(dir, "decoder-960ms.onnx"), "decoder")
+	writeTestFile(t, filepath.Join(dir, "joiner-960ms.onnx"), "joiner")
+	writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens")
+}
+
+func stringSliceContains(values []string, target string) bool {
+	for _, value := range values {
+		if value == target {
+			return true
+		}
+	}
+	return false
+}
+
 func tarArchive(t *testing.T, files map[string]string) []byte {
 	t.Helper()
 	var buf bytes.Buffer

--
Gitblit v1.9.3