update
This commit is contained in:
		
							
								
								
									
										40
									
								
								mock/common.ts
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										40
									
								
								mock/common.ts
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,40 @@
 | 
			
		||||
import dotenv from 'dotenv';
 | 
			
		||||
import util from 'node:util';
 | 
			
		||||
 | 
			
		||||
dotenv.config({ path: '.env' });
 | 
			
		||||
 | 
			
		||||
import { QwenText2Image } from '../src/dashscope/dashscope.ts';
 | 
			
		||||
 | 
			
		||||
const main = async () => {
 | 
			
		||||
  // 1. create image
 | 
			
		||||
  const qwen = new QwenText2Image(process.env.BAILIAN_API_KEY || '');
 | 
			
		||||
  const res = await qwen.createImage({
 | 
			
		||||
    model: 'wan2.2-t2i-flash',
 | 
			
		||||
    input: {
 | 
			
		||||
      prompt: '一个可爱的小的熊猫,卡通风格,抱着竹子。',
 | 
			
		||||
    },
 | 
			
		||||
  });
 | 
			
		||||
  console.log('create image response:', res);
 | 
			
		||||
  // 2. for-of loop get status
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
const id = '857ac37e-d470-4008-a98b-afe8ad054f81';
 | 
			
		||||
const loopGetStatus = async (taskId: string) => {
 | 
			
		||||
  const qwen = new QwenText2Image(process.env.BAILIAN_API_KEY || '');
 | 
			
		||||
  let status: string;
 | 
			
		||||
  do {
 | 
			
		||||
    await qwen.getTaskStatus(taskId);
 | 
			
		||||
    const taskResponse = qwen.taskResponse;
 | 
			
		||||
    console.log('task status:', taskResponse);
 | 
			
		||||
    status = taskResponse?.output?.task_status;
 | 
			
		||||
    if (status === 'SUCCEEDED') {
 | 
			
		||||
      break;
 | 
			
		||||
    }
 | 
			
		||||
    await new Promise((resolve) => setTimeout(resolve, 1000));
 | 
			
		||||
  } while (status !== 'SUCCEEDED' && status !== 'FAILED');
 | 
			
		||||
 | 
			
		||||
  console.log(util.inspect(qwen.taskResponse, { depth: null, colors: true }));
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// main()
 | 
			
		||||
loopGetStatus(id);
 | 
			
		||||
@@ -13,5 +13,8 @@
 | 
			
		||||
  "type": "module",
 | 
			
		||||
  "devDependencies": {
 | 
			
		||||
    "@types/node": "^24.4.0"
 | 
			
		||||
  },
 | 
			
		||||
  "dependencies": {
 | 
			
		||||
    "dotenv": "^17.2.2"
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										10
									
								
								pnpm-lock.yaml
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										10
									
								
								pnpm-lock.yaml
									
									
									
										generated
									
									
									
								
							@@ -7,6 +7,10 @@ settings:
 | 
			
		||||
importers:
 | 
			
		||||
 | 
			
		||||
  .:
 | 
			
		||||
    dependencies:
 | 
			
		||||
      dotenv:
 | 
			
		||||
        specifier: ^17.2.2
 | 
			
		||||
        version: 17.2.2
 | 
			
		||||
    devDependencies:
 | 
			
		||||
      '@types/node':
 | 
			
		||||
        specifier: ^24.4.0
 | 
			
		||||
@@ -17,6 +21,10 @@ packages:
 | 
			
		||||
  '@types/node@24.4.0':
 | 
			
		||||
    resolution: {integrity: sha512-gUuVEAK4/u6F9wRLznPUU4WGUacSEBDPoC2TrBkw3GAnOLHBL45QdfHOXp1kJ4ypBGLxTOB+t7NJLpKoC3gznQ==}
 | 
			
		||||
 | 
			
		||||
  dotenv@17.2.2:
 | 
			
		||||
    resolution: {integrity: sha512-Sf2LSQP+bOlhKWWyhFsn0UsfdK/kCWRv1iuA2gXAwt3dyNabr6QSj00I2V10pidqz69soatm9ZwZvpQMTIOd5Q==}
 | 
			
		||||
    engines: {node: '>=12'}
 | 
			
		||||
 | 
			
		||||
  undici-types@7.11.0:
 | 
			
		||||
    resolution: {integrity: sha512-kt1ZriHTi7MU+Z/r9DOdAI3ONdaR3M3csEaRc6ewa4f4dTvX4cQCbJ4NkEn0ohE4hHtq85+PhPSTY+pO/1PwgA==}
 | 
			
		||||
 | 
			
		||||
@@ -26,4 +34,6 @@ snapshots:
 | 
			
		||||
    dependencies:
 | 
			
		||||
      undici-types: 7.11.0
 | 
			
		||||
 | 
			
		||||
  dotenv@17.2.2: {}
 | 
			
		||||
 | 
			
		||||
  undici-types@7.11.0: {}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										114
									
								
								src/dashscope/dashscope.ts
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										114
									
								
								src/dashscope/dashscope.ts
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,114 @@
 | 
			
		||||
export class QwenText2Image {
 | 
			
		||||
  text2imageApi = 'https://dashscope.aliyuncs.com/api/v1/services/aigc/text2image/image-synthesis';
 | 
			
		||||
  tasksApi = 'https://dashscope.aliyuncs.com/api/v1/tasks/{task_id}';
 | 
			
		||||
  apiKey: string;
 | 
			
		||||
  taskResponse: TaskResponse;
 | 
			
		||||
  constructor(apiKey: string) {
 | 
			
		||||
    this.apiKey = apiKey;
 | 
			
		||||
  }
 | 
			
		||||
  async createImage(data: CreateImageRequest): Promise<CreateImageResponse> {
 | 
			
		||||
    const res = await fetch(this.text2imageApi, {
 | 
			
		||||
      method: 'POST',
 | 
			
		||||
      headers: {
 | 
			
		||||
        'Content-Type': 'application/json',
 | 
			
		||||
        'X-DashScope-Async': 'enable',
 | 
			
		||||
        Authorization: `Bearer ${this.apiKey}`,
 | 
			
		||||
      },
 | 
			
		||||
      body: JSON.stringify(data),
 | 
			
		||||
    });
 | 
			
		||||
    return await res.json();
 | 
			
		||||
  }
 | 
			
		||||
  /**
 | 
			
		||||
   *  // PENDING:任务排队中
 | 
			
		||||
      // RUNNING:任务处理中
 | 
			
		||||
      // SUCCEEDED:任务执行成功
 | 
			
		||||
      // FAILED:任务执行失败
 | 
			
		||||
      // CANCELED:任务取消成功
 | 
			
		||||
      // UNKNOWN:任务不存在或状态未知
 | 
			
		||||
   * @param taskId 
 | 
			
		||||
   * @returns 
 | 
			
		||||
   */
 | 
			
		||||
  async getTaskStatus(taskId: string): Promise<TaskResponse> {
 | 
			
		||||
    const url = this.tasksApi.replace('{task_id}', taskId);
 | 
			
		||||
    const res = await fetch(url, {
 | 
			
		||||
      method: 'GET',
 | 
			
		||||
      headers: {
 | 
			
		||||
        'Content-Type': 'application/json',
 | 
			
		||||
        Authorization: `Bearer ${this.apiKey}`,
 | 
			
		||||
      },
 | 
			
		||||
    });
 | 
			
		||||
    this.taskResponse = await res.json();
 | 
			
		||||
    return this.taskResponse;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type TaskStatus = 'PENDING' | 'SUCCESS' | 'FAILURE' | 'CANCELED' | 'UNKNOWN';
 | 
			
		||||
type CreateImageResponse = {
 | 
			
		||||
  output: {
 | 
			
		||||
    task_status: TaskStatus;
 | 
			
		||||
    task_id: string;
 | 
			
		||||
  };
 | 
			
		||||
  request_id: string;
 | 
			
		||||
  code?: string;
 | 
			
		||||
  message?: string;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
type CreateImageRequest = {
 | 
			
		||||
  model: ImageModel;
 | 
			
		||||
  input: {
 | 
			
		||||
    prompt: string;
 | 
			
		||||
    negative_prompt?: string;
 | 
			
		||||
  };
 | 
			
		||||
  parameters?: {
 | 
			
		||||
    size?: string; // 图片尺寸,支持 "256x256", "512x512", "1024x1024"
 | 
			
		||||
    n: number; // 生成图片数量,默认为4,1-4之间
 | 
			
		||||
    seed?: number; // 随机种子,seed参数取值范围是[0, 2147483647]。
 | 
			
		||||
    prompt_extend?: boolean; // 是否启用提示词增强,默认为true 是否开启prompt智能改写。开启后会使用大模型对输入prompt进行智能改写,仅对正向提示词有效。对于较短的输入prompt生成效果提升明显,但会增加
 | 
			
		||||
    watermark?: boolean; // 是否添加水印,默认为false 是否在生成的图片上添加水印, "AI 生成" 字样
 | 
			
		||||
  };
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
type TaskResponse = {
 | 
			
		||||
  output: {
 | 
			
		||||
    task_status: TaskStatus;
 | 
			
		||||
    task_id: string;
 | 
			
		||||
    submit_time: string;
 | 
			
		||||
    scheduled_time: string;
 | 
			
		||||
    end_time: string;
 | 
			
		||||
    results: {
 | 
			
		||||
      orig_prompt: string;
 | 
			
		||||
      actual_prompt: string;
 | 
			
		||||
      url: string;
 | 
			
		||||
      code: string;
 | 
			
		||||
      message: string;
 | 
			
		||||
      [key: string]: any;
 | 
			
		||||
    }[];
 | 
			
		||||
    task_metrics: {
 | 
			
		||||
      TOTAL: number;
 | 
			
		||||
      SUCCEEDED: number;
 | 
			
		||||
      FAILED: number;
 | 
			
		||||
    };
 | 
			
		||||
    [key: string]: any;
 | 
			
		||||
  };
 | 
			
		||||
  request_id: string;
 | 
			
		||||
  usage: {
 | 
			
		||||
    image_count: number;
 | 
			
		||||
    [key: string]: any;
 | 
			
		||||
  };
 | 
			
		||||
  code?: string;
 | 
			
		||||
  message?: string;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
export const models = ['wan2.2-t2i-flash', 'wan2.2-t2i-plus', 'wanx2.1-t2i-turbo', 'wanx2.1-t2i-plus', 'wanx2.0-t2i-turbo'] as const;
 | 
			
		||||
 | 
			
		||||
export type ImageModel = (typeof models)[number];
 | 
			
		||||
 | 
			
		||||
export const sizes = [
 | 
			
		||||
  '1664x928', // 16:9
 | 
			
		||||
  '1472x1140', // 4:3
 | 
			
		||||
  '1328x1328', // 1:1
 | 
			
		||||
  '1140x1472', // 3:4
 | 
			
		||||
  '928x1664', // 9:16
 | 
			
		||||
] as const;
 | 
			
		||||
 | 
			
		||||
export type ImageSize = (typeof sizes)[number];
 | 
			
		||||
							
								
								
									
										1
									
								
								src/dashscope/index.ts
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								src/dashscope/index.ts
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1 @@
 | 
			
		||||
export { QwenText2Image } from './dashscope.ts';
 | 
			
		||||
							
								
								
									
										17
									
								
								src/dashscope/readme.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								src/dashscope/readme.md
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
			
		||||
https://help.aliyun.com/zh/model-studio/text-to-image-v2-api-reference?spm=a2c4g.11186623.help-menu-2400256.d_2_2_3.3b3256e5iOwcyt#f281932a43b03
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
链接仅在 24 小时内有效。
 | 
			
		||||
```
 | 
			
		||||
# OSS域名列表
 | 
			
		||||
dashscope-result-bj.oss-cn-beijing.aliyuncs.com
 | 
			
		||||
dashscope-result-hz.oss-cn-hangzhou.aliyuncs.com
 | 
			
		||||
dashscope-result-sh.oss-cn-shanghai.aliyuncs.com
 | 
			
		||||
dashscope-result-wlcb.oss-cn-wulanchabu.aliyuncs.com
 | 
			
		||||
dashscope-result-zjk.oss-cn-zhangjiakou.aliyuncs.com
 | 
			
		||||
dashscope-result-sz.oss-cn-shenzhen.aliyuncs.com
 | 
			
		||||
dashscope-result-hy.oss-cn-heyuan.aliyuncs.com
 | 
			
		||||
dashscope-result-cd.oss-cn-chengdu.aliyuncs.com
 | 
			
		||||
dashscope-result-gz.oss-cn-guangzhou.aliyuncs.com
 | 
			
		||||
dashscope-result-wlcb-acdr-1.oss-cn-wulanchabu-acdr-1.aliyuncs.com
 | 
			
		||||
```
 | 
			
		||||
		Reference in New Issue
	
	Block a user