gongzhiyong
1f21309597
Remove user-facing new-api naming; Docker/network/container names use heicode. Go imports updated; Dockerfiles and workflows ldflags fixed. Made-with: Cursor
41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
package common
|
|
|
|
import (
|
|
"github.com/heicode/manager/dto"
|
|
"github.com/heicode/manager/types"
|
|
)
|
|
|
|
func GuessRelayFormatFromRequest(req any) (types.RelayFormat, bool) {
|
|
switch req.(type) {
|
|
case *dto.GeneralOpenAIRequest, dto.GeneralOpenAIRequest:
|
|
return types.RelayFormatOpenAI, true
|
|
case *dto.OpenAIResponsesRequest, dto.OpenAIResponsesRequest:
|
|
return types.RelayFormatOpenAIResponses, true
|
|
case *dto.ClaudeRequest, dto.ClaudeRequest:
|
|
return types.RelayFormatClaude, true
|
|
case *dto.GeminiChatRequest, dto.GeminiChatRequest:
|
|
return types.RelayFormatGemini, true
|
|
case *dto.EmbeddingRequest, dto.EmbeddingRequest:
|
|
return types.RelayFormatEmbedding, true
|
|
case *dto.RerankRequest, dto.RerankRequest:
|
|
return types.RelayFormatRerank, true
|
|
case *dto.ImageRequest, dto.ImageRequest:
|
|
return types.RelayFormatOpenAIImage, true
|
|
case *dto.AudioRequest, dto.AudioRequest:
|
|
return types.RelayFormatOpenAIAudio, true
|
|
default:
|
|
return "", false
|
|
}
|
|
}
|
|
|
|
func AppendRequestConversionFromRequest(info *RelayInfo, req any) {
|
|
if info == nil {
|
|
return
|
|
}
|
|
format, ok := GuessRelayFormatFromRequest(req)
|
|
if !ok {
|
|
return
|
|
}
|
|
info.AppendRequestConversion(format)
|
|
}
|