Codex
bb7968b612
Publish Docker image (Multi-arch) / Build & push (amd64) (push) Failing after 9m20s
Release (Linux, macOS, Windows) / Linux Release (push) Failing after 55m7s
Publish Docker image (Multi-arch) / Build & push (arm64) (push) Canceled after 0s
Release (Linux, macOS, Windows) / macOS Release (push) Canceled after 0s
Release (Linux, macOS, Windows) / Windows Release (push) Canceled after 0s
Publish Docker image (Multi-arch) / Create multi-arch manifests (push) Canceled after 0s
73 lines
2.5 KiB
Go
73 lines
2.5 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) {
|
|
result, err := relayconvert.ConvertRequest(c, info, target, request)
|
|
if result != nil {
|
|
info.RecordConversionDiagnostics(c, result.Diagnostics)
|
|
}
|
|
return result, err
|
|
}
|
|
|
|
func ConvertRequestByID(c *gin.Context, info *relaycommon.RelayInfo, converter string, request any) (*relayconvert.RequestResult, error) {
|
|
result, err := relayconvert.ConvertRequestByID(c, info, converter, request)
|
|
if result != nil {
|
|
info.RecordConversionDiagnostics(c, result.Diagnostics)
|
|
}
|
|
return result, err
|
|
}
|
|
|
|
func ConvertRequestVia(c *gin.Context, info *relaycommon.RelayInfo, request any, path ...types.RelayFormat) (*relayconvert.RequestResult, error) {
|
|
result, err := relayconvert.ConvertRequestVia(c, info, request, path...)
|
|
if result != nil {
|
|
info.RecordConversionDiagnostics(c, result.Diagnostics)
|
|
}
|
|
return result, err
|
|
}
|
|
|
|
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
|
|
}
|