xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-11-06 06:13:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
1808837298@qq.com
2024-03-05 23:04:57 +08:00
parent fe7f42fc2e
commit fb39b6b30e
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
relay/relay-text.go
View File

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