qing-client 0.0.4 → 0.0.6

package/lib/srvice/AigcService.d.ts

@@ -1,9 +1,20 @@
 import { BaseClient } from "../client/BaseClient";
 import { ClientConfig, RequestOptions } from "../types";
-import { TextGenerationRequest, TextGenerationResponse, ImageGenerationRequest, ImageGenerationResponse, ChatCompletionRequest, ChatCompletionResponse } from "../types/aigc";
+import { TextGenerationRequest, TextGenerationResponse, ImageGenerationRequest, ImageGenerationResponse, ChatCompletionRequest, ChatCompletionResponse, GaoqingTaskRequest, GaoqingTaskResponse, TwoDToThreeDTaskRequest, TwoDToThreeDTaskResponse, ImageRewriteTaskRequest, ImageRewriteTaskResponse, IndustrialDesignTaskRequest, IndustrialDesignTaskResponse, InpaintingTaskRequest, InpaintingTaskResponse, InteriorDesignTaskRequest, InteriorDesignTaskResponse, OldPhotoRestoreTaskRequest, OldPhotoRestoreTaskResponse, ProductBackgroundTaskRequest, ProductBackgroundTaskResponse, QingpingguoTaskRequest, QingpingguoTaskResponse, StyleTransferTaskRequest, StyleTransferTaskResponse, TaskStatusRequest, TaskStatusResponse } from "../types/aigc";
 export declare class AigcService extends BaseClient {
     constructor(config: ClientConfig);
     generateText(request: TextGenerationRequest, options?: RequestOptions): Promise<TextGenerationResponse>;
     generateImage(request: ImageGenerationRequest, options?: RequestOptions): Promise<ImageGenerationResponse>;
     chatCompletion(request: ChatCompletionRequest, options?: RequestOptions): Promise<ChatCompletionResponse>;
+    submitGaoqingTask(request: GaoqingTaskRequest, options?: RequestOptions): Promise<GaoqingTaskResponse>;
+    submit2DToThreeDTask(request: TwoDToThreeDTaskRequest, options?: RequestOptions): Promise<TwoDToThreeDTaskResponse>;
+    submitImageRewriteTask(request: ImageRewriteTaskRequest, options?: RequestOptions): Promise<ImageRewriteTaskResponse>;
+    submitIndustrialDesignTask(request: IndustrialDesignTaskRequest, options?: RequestOptions): Promise<IndustrialDesignTaskResponse>;
+    submitInpaintingTask(request: InpaintingTaskRequest, options?: RequestOptions): Promise<InpaintingTaskResponse>;
+    submitInteriorDesignTask(request: InteriorDesignTaskRequest, options?: RequestOptions): Promise<InteriorDesignTaskResponse>;
+    submitOldPhotoRestoreTask(request: OldPhotoRestoreTaskRequest, options?: RequestOptions): Promise<OldPhotoRestoreTaskResponse>;
+    submitProductBackgroundTask(request: ProductBackgroundTaskRequest, options?: RequestOptions): Promise<ProductBackgroundTaskResponse>;
+    submitQingpingguoTask(request: QingpingguoTaskRequest, options?: RequestOptions): Promise<QingpingguoTaskResponse>;
+    submitStyleTransferTask(request: StyleTransferTaskRequest, options?: RequestOptions): Promise<StyleTransferTaskResponse>;
+    getTaskStatus(request: TaskStatusRequest, options?: RequestOptions): Promise<TaskStatusResponse>;
 }

package/lib/srvice/AigcService.js

@@ -6,7 +6,6 @@ class AigcService extends BaseClient_1.BaseClient {
     constructor(config) {
         super(config, 'aigc');
     }
-    // 文本生成
     async generateText(request, options) {
         return this.request('/text/generation', {
             ...options,
@@ -14,7 +13,6 @@ class AigcService extends BaseClient_1.BaseClient {
             body: request
         });
     }
-    // 图片生成
     async generateImage(request, options) {
         return this.request('/image/generation', {
             ...options,
@@ -22,7 +20,6 @@ class AigcService extends BaseClient_1.BaseClient {
             body: request
         });
     }
-    // 对话补全
     async chatCompletion(request, options) {
         return this.request('/chat/completion', {
             ...options,
@@ -30,5 +27,92 @@ class AigcService extends BaseClient_1.BaseClient {
             body: request
         });
     }
+    // 高清面部修复任务
+    async submitGaoqingTask(request, options) {
+        return this.request('/lib/gaoqing/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 2D转3D任务
+    async submit2DToThreeDTask(request, options) {
+        return this.request('/lib/2d23d/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 图生图重写任务
+    async submitImageRewriteTask(request, options) {
+        return this.request('/lib/rewrite/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 工业设计任务
+    async submitIndustrialDesignTask(request, options) {
+        return this.request('/lib/industrial-design/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 智能涂抹消除任务
+    async submitInpaintingTask(request, options) {
+        return this.request('/lib/inpainting/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 智能装修任务
+    async submitInteriorDesignTask(request, options) {
+        return this.request('/lib/interior-design/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 老照片修复任务
+    async submitOldPhotoRestoreTask(request, options) {
+        return this.request('/old-photo/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 产品换背景任务
+    async submitProductBackgroundTask(request, options) {
+        return this.request('/product-background/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 青苹果风格洗稿任务
+    async submitQingpingguoTask(request, options) {
+        return this.request('/qingpingguo/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 风格转绘任务
+    async submitStyleTransferTask(request, options) {
+        return this.request('/style-transfer/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+    // 获取任务状态
+    async getTaskStatus(request, options) {
+        return this.request(`/comfy/tasks/${request.taskId}`, {
+            ...options,
+            method: 'GET'
+        });
+    }
 }
 exports.AigcService = AigcService;

package/lib/types/aigc.d.ts

@@ -50,3 +50,113 @@ export interface ChatCompletionResponse {
         total_tokens: number;
     };
 }
+export interface GaoqingTaskRequest {
+    faceImageUrl: string;
+    posterImageUrl: string;
+}
+export interface GaoqingTaskResponse {
+    taskId: string;
+    result: any;
+    statusUrl: string;
+}
+export interface TwoDToThreeDTaskRequest {
+    imageUrl: string;
+}
+export interface TwoDToThreeDTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    result: any;
+}
+export interface ImageRewriteTaskRequest {
+    imageUrl: string;
+    denoise?: number;
+    seed?: number;
+    batchSize?: number;
+    loraStrength?: number;
+}
+export interface ImageRewriteTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    result: any;
+}
+export interface IndustrialDesignTaskRequest {
+    styleImageUrl: string;
+    productImageUrl: string;
+}
+export interface IndustrialDesignTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    result: any;
+}
+export interface InpaintingTaskRequest {
+    imageUrl: string;
+    batchSize?: number;
+    resolution?: number;
+}
+export interface InpaintingTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    author: string;
+    features: string[];
+    batchSize: number;
+}
+export interface InteriorDesignTaskRequest {
+    originalImage: string;
+    designImage: string;
+}
+export interface InteriorDesignTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    author: string;
+    features: string[];
+}
+export interface OldPhotoRestoreTaskRequest {
+    imageUrl: string;
+    description?: string;
+}
+export interface OldPhotoRestoreTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    author: string;
+    features: string[];
+}
+export interface ProductBackgroundTaskRequest {
+    imageUrl: string;
+    promptText: string;
+}
+export interface ProductBackgroundTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    result: any;
+}
+export interface QingpingguoTaskRequest {
+    imageUrl: string;
+    denoise?: number;
+    seed?: number;
+    batchSize?: number;
+    loraStrength?: number;
+}
+export interface QingpingguoTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    authorStyle: string;
+    resolution: number;
+}
+export interface StyleTransferTaskRequest {
+    imageUrl: string;
+    styleCode?: number;
+}
+export interface StyleTransferTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    selectedStyle: string;
+    result: any;
+}
+export interface TaskStatusRequest {
+    taskId: string;
+}
+export interface TaskStatusResponse {
+    status: any;
+    progress: number;
+    images: any[];
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "qing-client",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",
   "scripts": {

package/src/index.ts

@@ -14,5 +14,4 @@ export * from './types/msg';
 export * from './types/token';
 export * from './types/user';
 
-
 export { BaseClient } from './client/BaseClient';

package/src/srvice/AigcService.ts

@@ -6,7 +6,29 @@ import {
     ImageGenerationRequest,
     ImageGenerationResponse,
     ChatCompletionRequest,
-    ChatCompletionResponse
+    ChatCompletionResponse,
+    GaoqingTaskRequest,
+    GaoqingTaskResponse,
+    TwoDToThreeDTaskRequest,
+    TwoDToThreeDTaskResponse,
+    ImageRewriteTaskRequest,
+    ImageRewriteTaskResponse,
+    IndustrialDesignTaskRequest,
+    IndustrialDesignTaskResponse,
+    InpaintingTaskRequest,
+    InpaintingTaskResponse,
+    InteriorDesignTaskRequest,
+    InteriorDesignTaskResponse,
+    OldPhotoRestoreTaskRequest,
+    OldPhotoRestoreTaskResponse,
+    ProductBackgroundTaskRequest,
+    ProductBackgroundTaskResponse,
+    QingpingguoTaskRequest,
+    QingpingguoTaskResponse,
+    StyleTransferTaskRequest,
+    StyleTransferTaskResponse,
+    TaskStatusRequest,
+    TaskStatusResponse
 } from "../types/aigc";
 
 export class AigcService extends BaseClient {
@@ -14,7 +36,6 @@ export class AigcService extends BaseClient {
         super(config, 'aigc');
     }
 
-    // 文本生成
     async generateText(request: TextGenerationRequest, options?: RequestOptions): Promise<TextGenerationResponse> {
         return this.request<TextGenerationResponse>('/text/generation', {
             ...options,
@@ -23,7 +44,6 @@ export class AigcService extends BaseClient {
         });
     }
 
-    // 图片生成
     async generateImage(request: ImageGenerationRequest, options?: RequestOptions): Promise<ImageGenerationResponse> {
         return this.request<ImageGenerationResponse>('/image/generation', {
             ...options,
@@ -32,7 +52,6 @@ export class AigcService extends BaseClient {
         });
     }
 
-    // 对话补全
     async chatCompletion(request: ChatCompletionRequest, options?: RequestOptions): Promise<ChatCompletionResponse> {
         return this.request<ChatCompletionResponse>('/chat/completion', {
             ...options,
@@ -40,4 +59,102 @@ export class AigcService extends BaseClient {
             body: request
         });
     }
+
+    // 高清面部修复任务
+    async submitGaoqingTask(request: GaoqingTaskRequest, options?: RequestOptions): Promise<GaoqingTaskResponse> {
+        return this.request<GaoqingTaskResponse>('/lib/gaoqing/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 2D转3D任务
+    async submit2DToThreeDTask(request: TwoDToThreeDTaskRequest, options?: RequestOptions): Promise<TwoDToThreeDTaskResponse> {
+        return this.request<TwoDToThreeDTaskResponse>('/lib/2d23d/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 图生图重写任务
+    async submitImageRewriteTask(request: ImageRewriteTaskRequest, options?: RequestOptions): Promise<ImageRewriteTaskResponse> {
+        return this.request<ImageRewriteTaskResponse>('/lib/rewrite/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 工业设计任务
+    async submitIndustrialDesignTask(request: IndustrialDesignTaskRequest, options?: RequestOptions): Promise<IndustrialDesignTaskResponse> {
+        return this.request<IndustrialDesignTaskResponse>('/lib/industrial-design/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 智能涂抹消除任务
+    async submitInpaintingTask(request: InpaintingTaskRequest, options?: RequestOptions): Promise<InpaintingTaskResponse> {
+        return this.request<InpaintingTaskResponse>('/lib/inpainting/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 智能装修任务
+    async submitInteriorDesignTask(request: InteriorDesignTaskRequest, options?: RequestOptions): Promise<InteriorDesignTaskResponse> {
+        return this.request<InteriorDesignTaskResponse>('/lib/interior-design/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 老照片修复任务
+    async submitOldPhotoRestoreTask(request: OldPhotoRestoreTaskRequest, options?: RequestOptions): Promise<OldPhotoRestoreTaskResponse> {
+        return this.request<OldPhotoRestoreTaskResponse>('/old-photo/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 产品换背景任务
+    async submitProductBackgroundTask(request: ProductBackgroundTaskRequest, options?: RequestOptions): Promise<ProductBackgroundTaskResponse> {
+        return this.request<ProductBackgroundTaskResponse>('/product-background/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 青苹果风格洗稿任务
+    async submitQingpingguoTask(request: QingpingguoTaskRequest, options?: RequestOptions): Promise<QingpingguoTaskResponse> {
+        return this.request<QingpingguoTaskResponse>('/qingpingguo/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 风格转绘任务
+    async submitStyleTransferTask(request: StyleTransferTaskRequest, options?: RequestOptions): Promise<StyleTransferTaskResponse> {
+        return this.request<StyleTransferTaskResponse>('/style-transfer/tasks', {
+            ...options,
+            method: 'POST',
+            body: request
+        });
+    }
+
+    // 获取任务状态
+    async getTaskStatus(request: TaskStatusRequest, options?: RequestOptions): Promise<TaskStatusResponse> {
+        return this.request<TaskStatusResponse>(`/comfy/tasks/${request.taskId}`, {
+            ...options,
+            method: 'GET'
+        });
+    }
 }

package/src/types/aigc.ts

@@ -54,4 +54,136 @@ export interface ChatCompletionResponse {
         completion_tokens: number;
         total_tokens: number;
     };
+}
+
+export interface GaoqingTaskRequest {
+    faceImageUrl: string;
+    posterImageUrl: string;
+}
+
+export interface GaoqingTaskResponse {
+    taskId: string;
+    result: any;
+    statusUrl: string;
+}
+
+export interface TwoDToThreeDTaskRequest {
+    imageUrl: string;
+}
+
+export interface TwoDToThreeDTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    result: any;
+}
+
+export interface ImageRewriteTaskRequest {
+    imageUrl: string;
+    denoise?: number;
+    seed?: number;
+    batchSize?: number;
+    loraStrength?: number;
+}
+
+export interface ImageRewriteTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    result: any;
+}
+
+export interface IndustrialDesignTaskRequest {
+    styleImageUrl: string;
+    productImageUrl: string;
+}
+
+export interface IndustrialDesignTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    result: any;
+}
+
+export interface InpaintingTaskRequest {
+    imageUrl: string;
+    batchSize?: number;
+    resolution?: number;
+}
+
+export interface InpaintingTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    author: string;
+    features: string[];
+    batchSize: number;
+}
+
+export interface InteriorDesignTaskRequest {
+    originalImage: string;
+    designImage: string;
+}
+
+export interface InteriorDesignTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    author: string;
+    features: string[];
+}
+
+export interface OldPhotoRestoreTaskRequest {
+    imageUrl: string;
+    description?: string;
+}
+
+export interface OldPhotoRestoreTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    author: string;
+    features: string[];
+}
+
+export interface ProductBackgroundTaskRequest {
+    imageUrl: string;
+    promptText: string;
+}
+
+export interface ProductBackgroundTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    result: any;
+}
+
+export interface QingpingguoTaskRequest {
+    imageUrl: string;
+    denoise?: number;
+    seed?: number;
+    batchSize?: number;
+    loraStrength?: number;
+}
+
+export interface QingpingguoTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    authorStyle: string;
+    resolution: number;
+}
+
+export interface StyleTransferTaskRequest {
+    imageUrl: string;
+    styleCode?: number;
+}
+
+export interface StyleTransferTaskResponse {
+    taskId: string;
+    statusUrl: string;
+    selectedStyle: string;
+    result: any;
+}
+
+export interface TaskStatusRequest {
+    taskId: string;
+}
+
+export interface TaskStatusResponse {
+    status: any;
+    progress: number;
+    images: any[];
 }