diff --git a/relay/adaptor/doubao/main.go b/relay/adaptor/doubao/main.go index dd43d06c..669a6751 100644 --- a/relay/adaptor/doubao/main.go +++ b/relay/adaptor/doubao/main.go @@ -9,6 +9,9 @@ import ( func GetRequestURL(meta *meta.Meta) (string, error) { switch meta.Mode { case relaymode.ChatCompletions: + if strings.HasPrefix(meta.ActualModelName, "bot") { + return fmt.Sprintf("%s/api/v3/bots/chat/completions", meta.BaseURL), nil + } return fmt.Sprintf("%s/api/v3/chat/completions", meta.BaseURL), nil case relaymode.Embeddings: return fmt.Sprintf("%s/api/v3/embeddings", meta.BaseURL), nil diff --git a/relay/adaptor/openai/main.go b/relay/adaptor/openai/main.go index f9f564be..8b62619a 100644 --- a/relay/adaptor/openai/main.go +++ b/relay/adaptor/openai/main.go @@ -38,7 +38,7 @@ func StreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*model.E doneRendered := false for scanner.Scan() { - data := scanner.Text() + data := NormalizeDataLine(scanner.Text()) if len(data) < dataPrefixLength { // ignore blank line or wrong format continue } diff --git a/relay/adaptor/openai/util.go b/relay/adaptor/openai/util.go index 83beadba..0ee3896b 100644 --- a/relay/adaptor/openai/util.go +++ b/relay/adaptor/openai/util.go @@ -21,3 +21,11 @@ func ErrorWrapper(err error, code string, statusCode int) *model.ErrorWithStatus StatusCode: statusCode, } } + +func NormalizeDataLine(data string) string { + if strings.HasPrefix(data, "data:") { + content := strings.TrimLeft(data[len("data:"):], " ") + return "data: " + content + } + return data +}