feat(yocsef): add error if nowOutput is empty

JingYiJun · Jun 2, 2024 · 47102a3 · 47102a3
1 parent 2956a9a
commit 47102a3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/service/yocsef.go b/service/yocsef.go
@@ -65,7 +65,9 @@ func InferYocsef(
 	if err != nil {
 		return
 	}
-	defer res.Body.Close()
+	defer func(Body io.ReadCloser) {
+		_ = Body.Close()
+	}(res.Body)
 
 	if res.StatusCode != http.StatusOK {
 		return nil, errors.New("yocsef 推理模型暂不可用")
@@ -135,6 +137,10 @@ func InferYocsef(
 		})
 	}
 
+	if nowOutput == "" {
+		return nil, errors.New("yocsef 推理模型暂不可用")
+	}
+
 	if ctx.Err() != nil {
 		return nil, ctx.Err()
 	}