feat: add whisper git config?

telepace · Dec 30, 2024 · 7c4b7f8 · 7c4b7f8
1 parent 1a27bb7
commit 7c4b7f8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/stt/whisper/whisper.go b/internal/stt/whisper/whisper.go
@@ -43,6 +43,11 @@ func NewWhisperSTT() *WhisperSTT {
 }
 
 func (w *WhisperSTT) Recognize(audioData []byte, audioURL string) (string, error) {
+	if w.model != "whisper-v3-turbo" {
+		logger.Warnf("检测到不正确的模型名称: %s，自动修正为: whisper-v3-turbo", w.model)
+		w.model = "whisper-v3-turbo"
+	}
+
 	body := &bytes.Buffer{}
 	writer := multipart.NewWriter(body)