[email protected] 1 год назад
Родитель
Сommit
f9ba10f180
2 измененных файлов с 6 добавлено и 3 удалено
  1. 2 1
      controller/relay.go
  2. 4 2
      web/src/components/Playground.js

+ 2 - 1
controller/relay.go

@@ -75,7 +75,8 @@ func Playground(c *gin.Context) {
 	}
 	channel, err := model.CacheGetRandomSatisfiedChannel(group, playgroundRequest.Model, 0)
 	if err != nil {
-		openaiErr = service.OpenAIErrorWrapperLocal(err, "get_playground_channel_failed", http.StatusInternalServerError)
+		message := fmt.Sprintf("当前分组 %s 下对于模型 %s 无可用渠道", group, playgroundRequest.Model)
+		openaiErr = service.OpenAIErrorWrapperLocal(errors.New(message), "get_playground_channel_failed", http.StatusInternalServerError)
 		return
 	}
 	middleware.SetupContextForSelectedChannel(c, channel, playgroundRequest.Model)

+ 4 - 2
web/src/components/Playground.js

@@ -133,7 +133,9 @@ const Playground = () => {
           completeMessage();
         } else {
           let text = payload.choices[0].delta.content;
-          generateMockResponse(text);
+          if (text) {
+            generateMockResponse(text);
+          }
         }
       } else {
         completeMessage();
@@ -186,7 +188,7 @@ const Playground = () => {
           stream: true,
           model: inputs.model,
           group: inputs.group,
-          max_tokens: inputs.max_tokens,
+          max_tokens: parseInt(inputs.max_tokens),
           temperature: inputs.temperature,
         };
       };