61 lines
2.2 KiB
Go
61 lines
2.2 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
relaycommon "github.com/QuantumNous/new-api/relay/common"
|
||
|
|
"github.com/QuantumNous/new-api/relaykit/dto"
|
||
|
|
"github.com/QuantumNous/new-api/relaykit/relayconvert"
|
||
|
|
"github.com/QuantumNous/new-api/relaykit/types"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
relayconvert.SetMediaResolver(relayconvert.MediaResolver{
|
||
|
|
// relayconvert is gin-free; recover the gin context when the caller
|
||
|
|
// passed one so file caching/cleanup keeps working.
|
||
|
|
GetBase64Data: func(ctx context.Context, source types.FileSource, reason ...string) (string, string, error) {
|
||
|
|
ginCtx, _ := ctx.(*gin.Context)
|
||
|
|
return GetBase64Data(ginCtx, source, reason...)
|
||
|
|
},
|
||
|
|
DecodeBase64FileData: DecodeBase64FileData,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func ConvertRequest(c *gin.Context, info *relaycommon.RelayInfo, target types.RelayFormat, request any) (*relayconvert.RequestResult, error) {
|
||
|
|
return relayconvert.ConvertRequest(c, info, target, request)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ConvertRequestByID(c *gin.Context, info *relaycommon.RelayInfo, converter string, request any) (*relayconvert.RequestResult, error) {
|
||
|
|
return relayconvert.ConvertRequestByID(c, info, converter, request)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ConvertRequestVia(c *gin.Context, info *relaycommon.RelayInfo, request any, path ...types.RelayFormat) (*relayconvert.RequestResult, error) {
|
||
|
|
return relayconvert.ConvertRequestVia(c, info, request, path...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ClaudeToOpenAIRequest(claudeRequest dto.ClaudeRequest, info *relaycommon.RelayInfo) (*dto.GeneralOpenAIRequest, error) {
|
||
|
|
result, err := ConvertRequest(nil, info, types.RelayFormatOpenAI, &claudeRequest)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
openAIRequest, ok := result.Value.(*dto.GeneralOpenAIRequest)
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf("expected OpenAI chat completions request, got %T", result.Value)
|
||
|
|
}
|
||
|
|
return openAIRequest, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func GeminiToOpenAIRequest(geminiRequest *dto.GeminiChatRequest, info *relaycommon.RelayInfo) (*dto.GeneralOpenAIRequest, error) {
|
||
|
|
result, err := ConvertRequest(nil, info, types.RelayFormatOpenAI, geminiRequest)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
openAIRequest, ok := result.Value.(*dto.GeneralOpenAIRequest)
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf("expected OpenAI chat completions request, got %T", result.Value)
|
||
|
|
}
|
||
|
|
return openAIRequest, nil
|
||
|
|
}
|