5191ea6f55
* Modify the test module, add the file upload test, and add the image editing api * fix golangci-lint * fix golangci-lint * Static file deletion, file directory name modification * fix * test-server-related logic encapsulated in a single tidy struct --------- Co-authored-by: julian_huang <julian.huang@yuansuan.com>
124 lines
3.1 KiB
Go
124 lines
3.1 KiB
Go
package gogpt
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
// Image sizes defined by the OpenAI API.
|
|
const (
|
|
CreateImageSize256x256 = "256x256"
|
|
CreateImageSize512x512 = "512x512"
|
|
CreateImageSize1024x1024 = "1024x1024"
|
|
)
|
|
|
|
const (
|
|
CreateImageResponseFormatURL = "url"
|
|
CreateImageResponseFormatB64JSON = "b64_json"
|
|
)
|
|
|
|
// ImageRequest represents the request structure for the image API.
|
|
type ImageRequest struct {
|
|
Prompt string `json:"prompt,omitempty"`
|
|
N int `json:"n,omitempty"`
|
|
Size string `json:"size,omitempty"`
|
|
ResponseFormat string `json:"response_format,omitempty"`
|
|
User string `json:"user,omitempty"`
|
|
}
|
|
|
|
// ImageResponse represents a response structure for image API.
|
|
type ImageResponse struct {
|
|
Created int64 `json:"created,omitempty"`
|
|
Data []ImageResponseDataInner `json:"data,omitempty"`
|
|
}
|
|
|
|
// ImageResponseData represents a response data structure for image API.
|
|
type ImageResponseDataInner struct {
|
|
URL string `json:"url,omitempty"`
|
|
B64JSON string `json:"b64_json,omitempty"`
|
|
}
|
|
|
|
// CreateImage - API call to create an image. This is the main endpoint of the DALL-E API.
|
|
func (c *Client) CreateImage(ctx context.Context, request ImageRequest) (response ImageResponse, err error) {
|
|
var reqBytes []byte
|
|
reqBytes, err = json.Marshal(request)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
urlSuffix := "/images/generations"
|
|
req, err := http.NewRequest(http.MethodPost, c.fullURL(urlSuffix), bytes.NewBuffer(reqBytes))
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
req = req.WithContext(ctx)
|
|
err = c.sendRequest(req, &response)
|
|
return
|
|
}
|
|
|
|
// ImageEditRequest represents the request structure for the image API.
|
|
type ImageEditRequest struct {
|
|
Image *os.File `json:"image,omitempty"`
|
|
Mask *os.File `json:"mask,omitempty"`
|
|
Prompt string `json:"prompt,omitempty"`
|
|
N int `json:"n,omitempty"`
|
|
Size string `json:"size,omitempty"`
|
|
}
|
|
|
|
// CreateEditImage - API call to create an image. This is the main endpoint of the DALL-E API.
|
|
func (c *Client) CreateEditImage(ctx context.Context, request ImageEditRequest) (response ImageResponse, err error) {
|
|
body := &bytes.Buffer{}
|
|
writer := multipart.NewWriter(body)
|
|
|
|
// image
|
|
image, err := writer.CreateFormFile("image", request.Image.Name())
|
|
if err != nil {
|
|
return
|
|
}
|
|
_, err = io.Copy(image, request.Image)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// mask
|
|
mask, err := writer.CreateFormFile("mask", request.Mask.Name())
|
|
if err != nil {
|
|
return
|
|
}
|
|
_, err = io.Copy(mask, request.Mask)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = writer.WriteField("prompt", request.Prompt)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = writer.WriteField("n", strconv.Itoa(request.N))
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = writer.WriteField("size", request.Size)
|
|
if err != nil {
|
|
return
|
|
}
|
|
writer.Close()
|
|
urlSuffix := "/images/edits"
|
|
req, err := http.NewRequest(http.MethodPost, c.fullURL(urlSuffix), body)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
req = req.WithContext(ctx)
|
|
req.Header.Set("Content-Type", writer.FormDataContentType())
|
|
err = c.sendRequest(req, &response)
|
|
return
|
|
}
|