From 4ea58bc6f31777c50852f6dd946140a2fc1fd0ee Mon Sep 17 00:00:00 2001
From: wizardchen <wizardchen@tencent.com>
Date: Thu, 2 Apr 2026 21:06:21 +0800
Subject: [PATCH] feat(assets): add ASR test audio file and embed it in the
 application

- Introduced a new audio file, asr_test.wav, for ASR testing purposes.
- Created embed.go to embed the ASR test audio file as a byte slice.
- Updated initialization.go to utilize the embedded ASR test audio for model transcription checks.
---
 internal/{handler/testdata => assets}/asr_test.wav | Bin
 internal/assets/embed.go                           |   6 ++++++
 internal/handler/initialization.go                 |   7 ++-----
 3 files changed, 8 insertions(+), 5 deletions(-)
 rename internal/{handler/testdata => assets}/asr_test.wav (100%)
 create mode 100644 internal/assets/embed.go

diff --git a/internal/handler/testdata/asr_test.wav b/internal/assets/asr_test.wav
similarity index 100%
rename from internal/handler/testdata/asr_test.wav
rename to internal/assets/asr_test.wav
diff --git a/internal/assets/embed.go b/internal/assets/embed.go
new file mode 100644
index 00000000..ad33689c
--- /dev/null
+++ b/internal/assets/embed.go
@@ -0,0 +1,6 @@
+package assets
+
+import _ "embed"
+
+//go:embed asr_test.wav
+var ASRTestWAV []byte
diff --git a/internal/handler/initialization.go b/internal/handler/initialization.go
index 9575a9a0..5bb7f26f 100644
--- a/internal/handler/initialization.go
+++ b/internal/handler/initialization.go
@@ -2,7 +2,6 @@ package handler
 
 import (
 	"context"
-	_ "embed"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -15,6 +14,7 @@ import (
 	"time"
 
 	chatpipeline "github.com/Tencent/WeKnora/internal/application/service/chat_pipeline"
+	"github.com/Tencent/WeKnora/internal/assets"
 	"github.com/Tencent/WeKnora/internal/config"
 	"github.com/Tencent/WeKnora/internal/errors"
 	"github.com/Tencent/WeKnora/internal/logger"
@@ -31,9 +31,6 @@ import (
 	"github.com/ollama/ollama/api"
 )
 
-//go:embed testdata/asr_test.wav
-var asrTestWAV []byte
-
 // DownloadTask 下载任务信息
 type DownloadTask struct {
 	ID        string     `json:"id"`
@@ -1848,7 +1845,7 @@ func (h *InitializationHandler) CheckASRModel(c *gin.Context) {
 		return
 	}
 
-	text, err := asrInstance.Transcribe(ctx, asrTestWAV, "asr_test.wav")
+	text, err := asrInstance.Transcribe(ctx, assets.ASRTestWAV, "asr_test.wav")
 	available := true
 	message := "ASR连接成功"