Support new fields for /v1/images/generation API (#530)

* add support for new image/generation api

* fix one lint

* add revised_prompt to response

* fix lints

* add CreateImageQualityStandard
This commit is contained in:
Mike Cutalo
2023-11-07 23:16:22 -08:00
committed by GitHub
parent a20eb08b79
commit a0159ad2b0
2 changed files with 32 additions and 3 deletions

View File

@@ -19,7 +19,14 @@ func TestImages(t *testing.T) {
defer teardown()
server.RegisterHandler("/v1/images/generations", handleImageEndpoint)
_, err := client.CreateImage(context.Background(), openai.ImageRequest{
Prompt: "Lorem ipsum",
Prompt: "Lorem ipsum",
Model: openai.CreateImageModelDallE3,
N: 1,
Quality: openai.CreateImageQualityHD,
Size: openai.CreateImageSize1024x1024,
Style: openai.CreateImageStyleVivid,
ResponseFormat: openai.CreateImageResponseFormatURL,
User: "user",
})
checks.NoError(t, err, "CreateImage error")
}