mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-17 09:16:36 +08:00
23 lines
562 B
Go
23 lines
562 B
Go
package doubao
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/songquanpeng/one-api/relay/meta"
|
|
"github.com/songquanpeng/one-api/relay/relaymode"
|
|
)
|
|
|
|
func GetRequestURL(meta *meta.Meta) (string, error) {
|
|
var context = ""
|
|
if meta.Cache {
|
|
context = "context/"
|
|
}
|
|
switch meta.Mode {
|
|
case relaymode.ChatCompletions:
|
|
return fmt.Sprintf("%s/api/v3/%schat/completions", meta.BaseURL, context), 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)
|
|
}
|