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

clean

Browse Source
This commit is contained in:
Seefs
2025-10-14 00:16:08 +08:00
parent 10aa680890
commit 31f565b87f
1 changed files with 15 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
relay/common/relay_info.go
View File

@@ -426,46 +426,38 @@ func genBaseRelayInfo(c *gin.Context, request dto.Request) *RelayInfo {
} }
func GenRelayInfo(c *gin.Context, relayFormat types.RelayFormat, request dto.Request, ws *websocket.Conn) (*RelayInfo, error) { func GenRelayInfo(c *gin.Context, relayFormat types.RelayFormat, request dto.Request, ws *websocket.Conn) (*RelayInfo, error) {
var info *RelayInfo
switch relayFormat { switch relayFormat {
case types.RelayFormatOpenAI: case types.RelayFormatOpenAI:
info = GenRelayInfoOpenAI(c, request) return GenRelayInfoOpenAI(c, request), nil
case types.RelayFormatOpenAIAudio: case types.RelayFormatOpenAIAudio:
info = GenRelayInfoOpenAIAudio(c, request) return GenRelayInfoOpenAIAudio(c, request), nil
case types.RelayFormatOpenAIImage: case types.RelayFormatOpenAIImage:
info = GenRelayInfoImage(c, request) return GenRelayInfoImage(c, request), nil
case types.RelayFormatOpenAIRealtime: case types.RelayFormatOpenAIRealtime:
info = GenRelayInfoWs(c, ws) return GenRelayInfoWs(c, ws), nil
case types.RelayFormatClaude: case types.RelayFormatClaude:
info = GenRelayInfoClaude(c, request) return GenRelayInfoClaude(c, request), nil
case types.RelayFormatRerank: case types.RelayFormatRerank:
rerankReq, ok := request.(*dto.RerankRequest) if request, ok := request.(*dto.RerankRequest); ok {
if !ok { return GenRelayInfoRerank(c, request), nil
}
return nil, errors.New("request is not a RerankRequest") return nil, errors.New("request is not a RerankRequest")
}
info = GenRelayInfoRerank(c, rerankReq)
case types.RelayFormatGemini: case types.RelayFormatGemini:
info = GenRelayInfoGemini(c, request) return GenRelayInfoGemini(c, request), nil
case types.RelayFormatEmbedding: case types.RelayFormatEmbedding:
info = GenRelayInfoEmbedding(c, request) return GenRelayInfoEmbedding(c, request), nil
case types.RelayFormatOpenAIResponses: case types.RelayFormatOpenAIResponses:
responsesReq, ok := request.(*dto.OpenAIResponsesRequest) if request, ok := request.(*dto.OpenAIResponsesRequest); ok {
if !ok { return GenRelayInfoResponses(c, request), nil
return nil, errors.New("request is not a OpenAIResponsesRequest")
} }
info = GenRelayInfoResponses(c, responsesReq) return nil, errors.New("request is not a OpenAIResponsesRequest")
case types.RelayFormatTask: case types.RelayFormatTask:
info = genBaseRelayInfo(c, nil) return genBaseRelayInfo(c, nil), nil
info.RelayFormat = types.RelayFormatTask
case types.RelayFormatMjProxy: case types.RelayFormatMjProxy:
info = genBaseRelayInfo(c, nil) return genBaseRelayInfo(c, nil), nil
info.RelayFormat = types.RelayFormatMjProxy
default: default:
return nil, errors.New("invalid relay format") return nil, errors.New("invalid relay format")
} }
return info, nil
} }
func (info *RelayInfo) SetPromptTokens(promptTokens int) { func (info *RelayInfo) SetPromptTokens(promptTokens int) {