huangzhenpc/newapi-yx-diy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: support gemini-3-pro-image-preview via images/generations

Browse Source
This commit is contained in:
feitianbubu
2025-11-26 00:29:13 +08:00
parent a6a20a2069
commit bff04514a8
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
relay/channel/gemini/adaptor.go
View File

@@ -56,8 +56,15 @@ func (a *Adaptor) ConvertAudioRequest(c *gin.Context, info *relaycommon.RelayInf
} }
func (a *Adaptor) ConvertImageRequest(c *gin.Context, info *relaycommon.RelayInfo, request dto.ImageRequest) (any, error) { func (a *Adaptor) ConvertImageRequest(c *gin.Context, info *relaycommon.RelayInfo, request dto.ImageRequest) (any, error) {
if !strings.HasPrefix(info.UpstreamModelName, "imagen") { if strings.HasPrefix(info.UpstreamModelName, "gemini-3-pro-image") {
return nil, errors.New("not supported model for image generation") chatRequest := dto.GeneralOpenAIRequest{
Model: request.Model,
Messages: []dto.Message{
{Role: "user", Content: request.Prompt},
},
N: int(request.N),
}
return a.ConvertOpenAIRequest(c, info, &chatRequest)
} }
// convert size to aspect ratio but allow user to specify aspect ratio // convert size to aspect ratio but allow user to specify aspect ratio

4
setting/model_setting/gemini.go
View File

@@ -26,6 +26,10 @@ var defaultGeminiSettings = GeminiSettings{
SupportedImagineModels: []string{ SupportedImagineModels: []string{
"gemini-2.0-flash-exp-image-generation", "gemini-2.0-flash-exp-image-generation",
"gemini-2.0-flash-exp", "gemini-2.0-flash-exp",
"gemini-3-pro-image-preview",
"gemini-2.5-flash-image",
"nano-banana",
"nano-banana-pro",
}, },
ThinkingAdapterEnabled: false, ThinkingAdapterEnabled: false,
ThinkingAdapterBudgetTokensPercentage: 0.6, ThinkingAdapterBudgetTokensPercentage: 0.6,