oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(adaptor): update relay mode handling #1419

Browse Source
This commit is contained in:
CaIon
2025-07-23 19:28:58 +08:00
parent 6f74e7b738
commit 13bdb80958
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
relay/channel/siliconflow/adaptor.go
View File

@@ -46,7 +46,7 @@ func (a *Adaptor) GetRequestURL(info *relaycommon.RelayInfo) (string, error) {
} else if info.RelayMode == constant.RelayModeCompletions { } else if info.RelayMode == constant.RelayModeCompletions {
return fmt.Sprintf("%s/v1/completions", info.BaseUrl), nil return fmt.Sprintf("%s/v1/completions", info.BaseUrl), nil
} }
return "", errors.New("invalid relay mode") return fmt.Sprintf("%s/v1/chat/completions", info.BaseUrl), nil
} }
func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Header, info *relaycommon.RelayInfo) error { func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Header, info *relaycommon.RelayInfo) error {
@@ -80,16 +80,19 @@ func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, info *relaycom
switch info.RelayMode { switch info.RelayMode {
case constant.RelayModeRerank: case constant.RelayModeRerank:
usage, err = siliconflowRerankHandler(c, info, resp) usage, err = siliconflowRerankHandler(c, info, resp)
case constant.RelayModeEmbeddings:
usage, err = openai.OpenaiHandler(c, info, resp)
case constant.RelayModeCompletions: case constant.RelayModeCompletions:
fallthrough fallthrough
case constant.RelayModeChatCompletions: case constant.RelayModeChatCompletions:
fallthrough
default:
if info.IsStream { if info.IsStream {
usage, err = openai.OaiStreamHandler(c, info, resp) usage, err = openai.OaiStreamHandler(c, info, resp)
} else { } else {
usage, err = openai.OpenaiHandler(c, info, resp) usage, err = openai.OpenaiHandler(c, info, resp)
} }
case constant.RelayModeEmbeddings:
usage, err = openai.OpenaiHandler(c, info, resp)
} }
return return
} }