update close ws-client conn

devinyf · Jun 6, 2024 · 216adb4 · 216adb4
1 parent c27267d
commit 216adb4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/example/paraformer/realtime/speech2text.go b/example/paraformer/realtime/speech2text.go
@@ -63,7 +63,7 @@ func main() {
 
 	// 等待语音识别结果输出
 	time.Sleep(5 * time.Second)
-	// 手动关闭语音识别
+	// 关闭语音识别连接
 	if err := cli.CloseSpeechToTextGeneration(); err != nil {
 		panic(err)
 	}

diff --git a/tongyiclient.go b/tongyiclient.go
@@ -252,7 +252,13 @@ func (q *TongyiClient) CloseSpeechToTextGeneration() error {
 		return errors.New("wsCli is nil")
 	}
 
-	return paraformer.CloseRecognitionClient(q.wsCli)
+	if err := paraformer.CloseRecognitionClient(q.wsCli); err != nil {
+		return err
+	}
+
+	q.wsCli = nil
+
+	return nil
 }
 
 func (q *TongyiClient) CreateEmbedding(ctx context.Context, r *embedding.Request) ([][]float64, int, error) {