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连接成功"