2024-06-08 05:26:26 +00:00
|
|
|
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) {
|
2024-07-22 14:38:50 +00:00
|
|
|
switch meta.Mode {
|
|
|
|
case relaymode.ChatCompletions:
|
2024-06-08 05:26:26 +00:00
|
|
|
return fmt.Sprintf("%s/api/v3/chat/completions", meta.BaseURL), nil
|
2024-07-22 14:38:50 +00:00
|
|
|
case relaymode.Embeddings:
|
|
|
|
return fmt.Sprintf("%s/api/v3/embeddings", meta.BaseURL), nil
|
|
|
|
default:
|
2024-06-08 05:26:26 +00:00
|
|
|
}
|
|
|
|
return "", fmt.Errorf("unsupported relay mode %d for doubao", meta.Mode)
|
|
|
|
}
|