feat: support /v1/rerank router

This commit is contained in:
nongqiqin
2024-04-30 12:29:41 +08:00
parent 2720e1a358
commit 893c9adeb9
8 changed files with 133 additions and 0 deletions

View File

@@ -11,4 +11,5 @@ const (
AudioSpeech
AudioTranscription
AudioTranslation
Rerank
)

View File

@@ -24,6 +24,9 @@ func GetByPath(path string) int {
relayMode = AudioTranscription
} else if strings.HasPrefix(path, "/v1/audio/translations") {
relayMode = AudioTranslation
} else if strings.HasPrefix(path, "/v1/rerank") {
relayMode = Rerank
}
return relayMode
}