xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-19 18:16:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #30 from qistchan/fix-2109

Browse Source 
Fix 2109
This commit is contained in:
Laisky.Cai 2025-02-26 09:02:24 +08:00 committed by GitHub
commit 07d9a8e144
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
relay/adaptor/doubao/main.go
View File

@ -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

2
relay/adaptor/openai/main.go
View File

@ -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
}

8
relay/adaptor/openai/util.go
View File

@ -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
}