xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-11 10:53:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: implement image editing request handling and response conversion for Imagen API

Browse Source
This commit is contained in:
Laisky.Cai
2025-03-16 14:21:38 +00:00
parent fa794e7bd5
commit 580fec6359
9 changed files with 363 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
relay/model/general.go
View File

@@ -1,6 +1,10 @@
package model
import "github.com/songquanpeng/one-api/relay/adaptor/openrouter"
import (
"mime/multipart"
"github.com/songquanpeng/one-api/relay/adaptor/openrouter"
)
type ResponseFormat struct {
Type string `json:"type,omitempty"`
@@ -142,3 +146,19 @@ func (r GeneralOpenAIRequest) ParseInput() []string {
}
return input
}
// OpenaiImageEditRequest is the request body for the OpenAI image edit API.
type OpenaiImageEditRequest struct {
Image *multipart.FileHeader `json:"image" form:"image" binding:"required"`
Prompt string `json:"prompt" form:"prompt" binding:"required"`
Mask *multipart.FileHeader `json:"mask" form:"mask" binding:"required"`
Model string `json:"model" form:"model" binding:"required"`
N int `json:"n" form:"n" binding:"min=0,max=10"`
Size string `json:"size" form:"size"`
ResponseFormat string `json:"response_format" form:"response_format"`
// -------------------------------------
// Imagen-3
// -------------------------------------
EditMode *string `json:"edit_mode" form:"edit_mode"`
MaskMode *string `json:"mask_mode" form:"mask_mode"`
}