Просмотр исходного кода

fix: Fix the issue of 'unknown relay mode' when making an embedding request (close #93)

1808837298@qq.com 2 лет назад
Родитель
Сommit
fb39b6b30e
1 измененных файлов с 2 добавлено и 0 удалено
  1. 2 0
      relay/relay-text.go

+ 2 - 0
relay/relay-text.go

@@ -169,6 +169,8 @@ func getPromptTokens(textRequest *dto.GeneralOpenAIRequest, info *relaycommon.Re
 		promptTokens, err = service.CountTokenInput(textRequest.Prompt, textRequest.Model), nil
 		promptTokens, err = service.CountTokenInput(textRequest.Prompt, textRequest.Model), nil
 	case relayconstant.RelayModeModerations:
 	case relayconstant.RelayModeModerations:
 		promptTokens, err = service.CountTokenInput(textRequest.Input, textRequest.Model), nil
 		promptTokens, err = service.CountTokenInput(textRequest.Input, textRequest.Model), nil
+	case relayconstant.RelayModeEmbeddings:
+		promptTokens, err = service.CountTokenInput(textRequest.Input, textRequest.Model), nil
 	default:
 	default:
 		err = errors.New("unknown relay mode")
 		err = errors.New("unknown relay mode")
 		promptTokens = 0
 		promptTokens = 0