diff --git a/relay/adaptor/doubao/main.go b/relay/adaptor/doubao/main.go index dd43d06c53..669a675155 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 970807384f..e8bdb8cf8d 100644 --- a/relay/adaptor/openai/main.go +++ b/relay/adaptor/openai/main.go @@ -34,7 +34,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 83beadbafc..0ee3896bad 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 +}