feat: err if cannot connect to yocsef inference

JingYiJun · May 24, 2024 · 0027af1 · 0027af1
1 parent d191aca
commit 0027af1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/apis/record/yocsef.go b/apis/record/yocsef.go
@@ -46,7 +46,7 @@ func InferYocsefAsyncAPI(c *websocket.Conn) {
 		}
 
 		if body.Request == "" {
-			return BadRequest("request is empty")
+			return BadRequest("内容不能为空")
 		}
 
 		ctx, cancel := context.WithCancelCause(context.Background())

diff --git a/service/yocsef.go b/service/yocsef.go
@@ -67,6 +67,10 @@ func InferYocsef(
 	}
 	defer res.Body.Close()
 
+	if res.StatusCode != http.StatusOK {
+		return nil, errors.New("yocsef 推理模型暂不可用")
+	}
+
 	var reader = bufio.NewReader(res.Body)
 	var resultBuilder strings.Builder
 	var nowOutput string
@@ -143,7 +147,7 @@ func InferYocsef(
 		})
 	}
 
-	err = w.WriteJSON(InferResponseModel{
+	_ = w.WriteJSON(InferResponseModel{
 		Status: 0,
 		Output: nowOutput,
 		Stage:  "MOSS",