wanx: download option

README.md

@@ -85,6 +85,7 @@ func main() {
 		Params: wanx.ImageSynthesisParams{
 			N: 1,
 		},
+		Download: true // 从 URL 下载图片
 	}
 	ctx := context.TODO()
 
@@ -94,6 +95,8 @@ func main() {
 	}
 
 	for _, blob := range imgBlobs {
+		// blob.Data 会在 request 中设置了 Download: true 时下载
+		// 否则使用 blob.ImgURL
 		saveImg2Desktop(blob.ImgType, blob.Data)
 	}
 }

dtypes.go

@@ -0,0 +1,5 @@
+package dashscopego
+
+const (
+	DashscopeTokenEnvName = "DASHSCOPE_API_KEY" //nolint:gosec
+)

tongyiclient_test.go

@@ -193,6 +193,7 @@ func TestImageGeneration(t *testing.T) {
 		Input: wanx.ImageSynthesisInput{
 			Prompt: "A beautiful sunset",
 		},
+		Download: true,
 	}
 
 	imgBlobs, err := cli.CreateImageGeneration(ctx, req)
@@ -201,6 +202,7 @@ func TestImageGeneration(t *testing.T) {
 
 	for _, blob := range imgBlobs {
 		assert.NotEmpty(t, blob.Data)
+		assert.NotEmpty(t, blob.ImgURL)
 		assert.Equal(t, "image/png", blob.ImgType)
 	}
 }

wanx/wanx_dtypes.go

@@ -42,9 +42,10 @@ type ImageSynthesisInput struct {
 }
 
 type ImageSynthesisRequest struct {
-	Model  string               `json:"model"`
-	Input  ImageSynthesisInput  `json:"input"`
-	Params ImageSynthesisParams `json:"parameters"`
+	Model    string               `json:"model"`
+	Input    ImageSynthesisInput  `json:"input"`
+	Params   ImageSynthesisParams `json:"parameters"`
+	Download bool                 `json:"-"`
 }
 
 type Output struct {
@@ -76,6 +77,7 @@ type ImageResponse struct {
 type ImgBlob struct {
 	//	types include: "image/png".
 	ImgType string
+	ImgURL  string
 	// Raw bytes for media formats.
 	Data []byte
 }

wanx/wanxcli.go

@@ -31,12 +31,21 @@ func CreateImageGeneration(ctx context.Context, payload *ImageSynthesisRequest,
 
 	blobList := make([]*ImgBlob, 0, len(resp.Results))
 	for _, img := range resp.Results {
-		imgByte, err := httpcli.GetImage(ctx, img.URL, tokenOpt)
-		if err != nil {
-			return nil, err
+		tmpImgBlob := &ImgBlob{
+			ImgType: "image/png",
+			ImgURL:  img.URL,
+		}
+
+		if payload.Download {
+			imgByte, err := httpcli.GetImage(ctx, img.URL, tokenOpt)
+			if err != nil {
+				return nil, err
+			}
+
+			tmpImgBlob.Data = imgByte
 		}
 
-		blobList = append(blobList, &ImgBlob{Data: imgByte, ImgType: "image/png"})
+		blobList = append(blobList, tmpImgBlob)
 	}
 
 	return blobList, nil