fix: support embedding models for doubao (#1662)

Fixes #1594
This commit is contained in:
igophper
2024-07-22 22:38:50 +08:00
committed by GitHub
parent 2a892c1937
commit 39383e5532
2 changed files with 33 additions and 27 deletions

View File

@@ -7,8 +7,12 @@ import (
)
func GetRequestURL(meta *meta.Meta) (string, error) {
if meta.Mode == relaymode.ChatCompletions {
switch meta.Mode {
case relaymode.ChatCompletions:
return fmt.Sprintf("%s/api/v3/chat/completions", meta.BaseURL), nil
case relaymode.Embeddings:
return fmt.Sprintf("%s/api/v3/embeddings", meta.BaseURL), nil
default:
}
return "", fmt.Errorf("unsupported relay mode %d for doubao", meta.Mode)
}