Kaynağa Gözat

fix: use gpt-3.5's encoder if not found (close #110)

JustSong 2 yıl önce
ebeveyn
işleme
fa79e8b7a3
1 değiştirilmiş dosya ile 5 ekleme ve 1 silme
  1. 5 1
      controller/relay-utils.go

+ 5 - 1
controller/relay-utils.go

@@ -15,7 +15,11 @@ func getTokenEncoder(model string) *tiktoken.Tiktoken {
 	}
 	tokenEncoder, err := tiktoken.EncodingForModel(model)
 	if err != nil {
-		common.FatalLog(fmt.Sprintf("failed to get token encoder for model %s: %s", model, err.Error()))
+		common.SysError(fmt.Sprintf("failed to get token encoder for model %s: %s, using encoder for gpt-3.5-turbo", model, err.Error()))
+		tokenEncoder, err = tiktoken.EncodingForModel("gpt-3.5-turbo")
+		if err != nil {
+			common.FatalLog(fmt.Sprintf("failed to get token encoder for model gpt-3.5-turbo: %s", err.Error()))
+		}
 	}
 	tokenEncoderMap[model] = tokenEncoder
 	return tokenEncoder