@vectorx/ai-sdk 0.1.3 → 0.4.0

package/lib/models/QwenDocTurbo/index.js

@@ -0,0 +1,178 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QwenDocTurbo = void 0;
+const model_type_1 = require("../../model-type");
+const stream_1 = require("../../stream");
+const utils_1 = require("../../utils");
+const Chat_1 = require("../Chat");
+class QwenDocTurbo extends Chat_1.SimpleChatModel {
+    constructor(req, baseUrl, modelName, tokenManager) {
+        super(req, baseUrl, QwenDocTurbo.SUB_URL, tokenManager);
+        this.modelName = modelName;
+    }
+    modelRequest(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            const fetchHeaders = yield this.createAuthHeaders(options === null || options === void 0 ? void 0 : options.headers);
+            const isStreaming = (_b = (_a = data.parameters) === null || _a === void 0 ? void 0 : _a.stream) !== null && _b !== void 0 ? _b : false;
+            if (isStreaming) {
+                fetchHeaders["X-DashScope-SSE"] = "enable";
+            }
+            console.log("=== QwenDocTurbo Request ===", {
+                method: "post",
+                headers: Object.assign({}, fetchHeaders),
+                body: JSON.stringify(data),
+                url: `${this.baseUrl}/${this.subUrl}`,
+                stream: isStreaming,
+            });
+            const { data: responseData, header } = (yield this.req.fetch({
+                method: "post",
+                headers: Object.assign({}, fetchHeaders),
+                body: JSON.stringify(data),
+                url: `${this.baseUrl}/${this.subUrl}`,
+                stream: isStreaming,
+            }));
+            return (0, utils_1.handleResponseData)(responseData, header);
+        });
+    }
+    normalizeResponse(response) {
+        var _a, _b, _c, _d, _e;
+        const output = response.output || {};
+        const choice = (_a = output.choices) === null || _a === void 0 ? void 0 : _a[0];
+        const requestId = response.request_id || "";
+        const created = Math.floor(Date.now() / 1000);
+        const content = ((_b = choice === null || choice === void 0 ? void 0 : choice.message) === null || _b === void 0 ? void 0 : _b.content) || output.text || "";
+        return {
+            id: requestId,
+            object: "chat.completion",
+            created: created,
+            model: this.modelName,
+            log_id: requestId,
+            error: "",
+            code: 0,
+            choices: [
+                {
+                    index: 0,
+                    message: {
+                        id: requestId,
+                        role: "assistant",
+                        type: "answer",
+                        content: content,
+                        reasoning_content: "",
+                    },
+                    finish_reason: (choice === null || choice === void 0 ? void 0 : choice.finish_reason) || "stop",
+                },
+            ],
+            usage: {
+                prompt_tokens: ((_c = response.usage) === null || _c === void 0 ? void 0 : _c.input_tokens) || 0,
+                completion_tokens: ((_d = response.usage) === null || _d === void 0 ? void 0 : _d.output_tokens) || 0,
+                knowledge_tokens: 0,
+                reasoning_tokens: 0,
+                total_tokens: ((_e = response.usage) === null || _e === void 0 ? void 0 : _e.total_tokens) || 0,
+            },
+        };
+    }
+    doGenerate(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const qwenDocTurboData = this.convertToQwenDocTurboRequestOptions(data, false);
+            const res = yield this.modelRequest(qwenDocTurboData, options);
+            return this.normalizeResponse(res);
+        });
+    }
+    doStream(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const qwenDocTurboData = this.convertToQwenDocTurboRequestOptions(data, true);
+            const _stream = (yield this.modelRequest(qwenDocTurboData, options));
+            const stream = (0, stream_1.toPolyfillReadable)(_stream);
+            const rawStream = (0, stream_1.intoStandardStream)(stream);
+            const standardStream = this.normalizeStreamChunks(rawStream);
+            return (0, stream_1.createAsyncIterable)(standardStream);
+        });
+    }
+    normalizeStreamChunks(stream) {
+        const modelName = this.modelName;
+        let previousContent = "";
+        return stream.pipeThrough(new stream_1.TransformStream({
+            transform(chunk, controller) {
+                var _a, _b, _c, _d, _e;
+                const output = chunk.output || {};
+                const choice = (_a = output.choices) === null || _a === void 0 ? void 0 : _a[0];
+                const requestId = chunk.request_id || "";
+                const created = Math.floor(Date.now() / 1000);
+                const fullContent = ((_b = choice === null || choice === void 0 ? void 0 : choice.message) === null || _b === void 0 ? void 0 : _b.content) || output.text || "";
+                const deltaContent = fullContent.slice(previousContent.length);
+                previousContent = fullContent;
+                if (!deltaContent && (choice === null || choice === void 0 ? void 0 : choice.finish_reason) !== "stop") {
+                    return;
+                }
+                const standardChunk = {
+                    id: requestId,
+                    object: "chat.completion.chunk",
+                    created: created,
+                    model: modelName,
+                    log_id: requestId,
+                    error: "",
+                    code: 0,
+                    choices: [
+                        {
+                            index: 0,
+                            message: {
+                                id: requestId,
+                                role: "assistant",
+                                type: "answer",
+                                content: deltaContent,
+                                reasoning_content: "",
+                            },
+                            finish_reason: (choice === null || choice === void 0 ? void 0 : choice.finish_reason) || null,
+                        },
+                    ],
+                    usage: {
+                        prompt_tokens: ((_c = chunk.usage) === null || _c === void 0 ? void 0 : _c.input_tokens) || 0,
+                        completion_tokens: ((_d = chunk.usage) === null || _d === void 0 ? void 0 : _d.output_tokens) || 0,
+                        knowledge_tokens: 0,
+                        reasoning_tokens: 0,
+                        total_tokens: ((_e = chunk.usage) === null || _e === void 0 ? void 0 : _e.total_tokens) || 0,
+                    },
+                };
+                controller.enqueue(standardChunk);
+            },
+        }));
+    }
+    convertToQwenDocTurboRequestOptions(data, stream) {
+        const clamp = (value, min, max, defaultValue) => value !== undefined ? Math.max(min, Math.min(max, value)) : defaultValue;
+        const messages = (data.messages || []).map((msg) => {
+            var _a;
+            const role = msg.role;
+            if (role === "system") {
+                return {
+                    role,
+                    content: Array.isArray(msg.content) ? ((_a = msg.content.find((c) => c.type === "text")) === null || _a === void 0 ? void 0 : _a.text) || "" : msg.content,
+                };
+            }
+            return {
+                role,
+                content: Array.isArray(msg.content) ? (0, model_type_1.filterContentByTypes)(msg.content, ["text", "doc_url"]) : msg.content,
+            };
+        });
+        const parameters = Object.assign({ stream, max_tokens: clamp(data.max_tokens, 1, 8000, 2000), temperature: clamp(data.temperature, 0, 2, 1.0), top_p: clamp(data.top_p, 0, 1, 1.0), frequency_penalty: clamp(data.frequency_penalty, -2, 2, 0.0), presence_penalty: clamp(data.presence_penalty, -2, 2, 0.0), n: clamp(data.n, 1, 10, 1), incremental_output: false }, ((data === null || data === void 0 ? void 0 : data.parameters) || {}));
+        return {
+            model: this.modelName,
+            input: {
+                messages,
+            },
+            parameters,
+        };
+    }
+}
+exports.QwenDocTurbo = QwenDocTurbo;
+QwenDocTurbo.BASE_URL = "https://dashscope.aliyuncs.com";
+QwenDocTurbo.SUB_URL = "api/v1/services/aigc/text-generation/generation";

package/lib/models/QwenImage/index.d.ts

@@ -0,0 +1,81 @@
+import type { IAbstractRequest } from "@vectorx/ai-types";
+import type { DoGenerateOutput, DoStreamOutput, ModelRequestOptions, ReqOptions } from "../../model-type";
+import { TokenManager } from "../../tokenManager";
+import { SimpleChatModel } from "../Chat";
+export interface QwenImageParameters {
+    negative_prompt?: string;
+    size?: "1664*928" | "1472*1140" | "1328*1328" | "1140*1472" | "928*1664" | string;
+    n?: 1;
+    prompt_extend?: boolean;
+    watermark?: boolean;
+}
+export interface QwenImageAPIInput {
+    model: string;
+    input: {
+        messages?: Array<{
+            role: "user";
+            content: Array<{
+                text: string;
+            }>;
+        }>;
+        prompt?: string;
+    };
+    parameters?: QwenImageParameters;
+}
+export type QwenImageRequestOptions = QwenImageAPIInput & {
+    parameters?: QwenImageParameters;
+};
+export type QwenImageContentItem = {
+    image?: string;
+    url?: string;
+    image_url?: string | {
+        url: string;
+    };
+    b64_json?: string;
+    [key: string]: any;
+};
+export interface QwenImageAPIResponse {
+    async?: boolean;
+    output: {
+        choices?: Array<{
+            finish_reason: string;
+            message: {
+                role: "assistant" | "user";
+                content: QwenImageContentItem[];
+            };
+        }>;
+        task_status?: string;
+        task_id?: string;
+        task_metric?: {
+            TOTAL: number;
+            FAILED: number;
+            SUCCEEDED: number;
+        };
+    };
+    usage?: {
+        width?: number;
+        height?: number;
+        image_count?: number;
+    };
+    request_id?: string;
+    id?: string;
+    model?: string;
+    created?: number;
+    object?: string;
+    code?: number;
+    error?: string;
+}
+export declare class QwenImageModel extends SimpleChatModel {
+    static BASE_URL: string;
+    static SUB_SYNTHESIS_URL: string;
+    static SUB_GENERATION_URL: string;
+    modelName: string;
+    constructor(req: IAbstractRequest, baseUrl: string, tokenManager: TokenManager);
+    protected normalizeStandardImageCompletion(res: QwenImageAPIResponse, fallbackModel: string): DoGenerateOutput;
+    protected coverModelRequestToQwenInput(data: ModelRequestOptions & {
+        parameters?: QwenImageParameters;
+    }): QwenImageRequestOptions;
+    protected modelRequest(data: QwenImageRequestOptions, options?: ReqOptions): Promise<ReadableStream<Uint8Array> | Promise<unknown>>;
+    doGenerate(data: ModelRequestOptions, options?: ReqOptions): Promise<DoGenerateOutput>;
+    doStream(data: ModelRequestOptions, options?: ReqOptions): Promise<DoStreamOutput>;
+}

package/lib/models/QwenImage/index.js

@@ -0,0 +1,208 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QwenImageModel = void 0;
+const stream_1 = require("../../stream");
+const utils_1 = require("../../utils");
+const Chat_1 = require("../Chat");
+const index_1 = require("../index");
+class QwenImageModel extends Chat_1.SimpleChatModel {
+    constructor(req, baseUrl, tokenManager) {
+        super(req, baseUrl, QwenImageModel.SUB_GENERATION_URL, tokenManager);
+        this.modelName = index_1.modelName[index_1.MultiModalModelName.QwenImage];
+    }
+    normalizeStandardImageCompletion(res, fallbackModel) {
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q;
+        const qOutput = (res === null || res === void 0 ? void 0 : res.output) || {};
+        if ((qOutput === null || qOutput === void 0 ? void 0 : qOutput.task_status) && (qOutput === null || qOutput === void 0 ? void 0 : qOutput.task_id)) {
+            const created = (_a = res === null || res === void 0 ? void 0 : res.created) !== null && _a !== void 0 ? _a : Math.floor(Date.now() / 1000);
+            const id = (res === null || res === void 0 ? void 0 : res.id) || (res === null || res === void 0 ? void 0 : res.request_id) || "";
+            const normalized = {
+                id,
+                object: (_b = res === null || res === void 0 ? void 0 : res.object) !== null && _b !== void 0 ? _b : "chat.completion",
+                created,
+                model: (_c = res === null || res === void 0 ? void 0 : res.model) !== null && _c !== void 0 ? _c : fallbackModel,
+                log_id: id,
+                error: (_d = res === null || res === void 0 ? void 0 : res.error) !== null && _d !== void 0 ? _d : "",
+                code: (_e = res === null || res === void 0 ? void 0 : res.code) !== null && _e !== void 0 ? _e : 0,
+                choices: [
+                    {
+                        index: 0,
+                        message: {
+                            id,
+                            role: "assistant",
+                            type: "async_task",
+                            content: JSON.stringify(Object.assign(Object.assign({}, ((res === null || res === void 0 ? void 0 : res.output) || {})), { request_id: (res === null || res === void 0 ? void 0 : res.request_id) || id })),
+                            reasoning_content: "",
+                        },
+                        finish_reason: "stop",
+                    },
+                ],
+                usage: mapUsageToStandard(res === null || res === void 0 ? void 0 : res.usage),
+            };
+            return normalized;
+        }
+        const first = ((_g = (_f = qOutput === null || qOutput === void 0 ? void 0 : qOutput.choices) === null || _f === void 0 ? void 0 : _f[0]) !== null && _g !== void 0 ? _g : null);
+        const message = (_h = first === null || first === void 0 ? void 0 : first.message) !== null && _h !== void 0 ? _h : {};
+        const contentUrl = Array.isArray(message === null || message === void 0 ? void 0 : message.content) && ((_j = message.content[0]) === null || _j === void 0 ? void 0 : _j.image) ? String(message.content[0].image) : "";
+        const created = (_k = res === null || res === void 0 ? void 0 : res.created) !== null && _k !== void 0 ? _k : Math.floor(Date.now() / 1000);
+        const id = (res === null || res === void 0 ? void 0 : res.id) || (res === null || res === void 0 ? void 0 : res.request_id) || "";
+        const normalized = {
+            id,
+            object: (_l = res === null || res === void 0 ? void 0 : res.object) !== null && _l !== void 0 ? _l : "chat.completion",
+            created,
+            model: (_m = res === null || res === void 0 ? void 0 : res.model) !== null && _m !== void 0 ? _m : fallbackModel,
+            log_id: id,
+            error: (_o = res === null || res === void 0 ? void 0 : res.error) !== null && _o !== void 0 ? _o : "",
+            code: (_p = res === null || res === void 0 ? void 0 : res.code) !== null && _p !== void 0 ? _p : 0,
+            choices: [
+                {
+                    index: 0,
+                    message: {
+                        id,
+                        role: "assistant",
+                        type: "image",
+                        content: contentUrl || "",
+                        reasoning_content: "",
+                    },
+                    finish_reason: (_q = first === null || first === void 0 ? void 0 : first.finish_reason) !== null && _q !== void 0 ? _q : "stop",
+                },
+            ],
+            usage: mapUsageToStandard(res === null || res === void 0 ? void 0 : res.usage),
+        };
+        return normalized;
+    }
+    coverModelRequestToQwenInput(data) {
+        var _a;
+        let text = "";
+        const messages = data.messages || data.history || [];
+        if (Array.isArray(messages) && messages.length > 0) {
+            const firstUser = (_a = messages.find((m) => (m === null || m === void 0 ? void 0 : m.role) === "user")) !== null && _a !== void 0 ? _a : messages[0];
+            const c = firstUser === null || firstUser === void 0 ? void 0 : firstUser.content;
+            if (typeof c === "string" && c.trim()) {
+                text = c.trim();
+            }
+            else if (Array.isArray(c)) {
+                for (const p of c) {
+                    if ((p === null || p === void 0 ? void 0 : p.type) === "text" && typeof p.text === "string" && p.text.trim()) {
+                        text = p.text.trim();
+                        break;
+                    }
+                }
+            }
+        }
+        if (!text && data.msg)
+            text = String(data.msg);
+        if (!text)
+            throw new Error("QwenImage 需要提供一个 text 提示词");
+        const isAsync = !!(data === null || data === void 0 ? void 0 : data.async);
+        return {
+            parameters: data.parameters,
+            model: this.modelName,
+            input: isAsync
+                ? {
+                    prompt: text,
+                }
+                : {
+                    messages: [
+                        {
+                            role: "user",
+                            content: [{ text }],
+                        },
+                    ],
+                },
+        };
+    }
+    modelRequest(data_1) {
+        return __awaiter(this, arguments, void 0, function* (data, options = { timeout: 30 * 1000 }) {
+            const fetchHeaders = yield this.createAuthHeaders(options === null || options === void 0 ? void 0 : options.headers);
+            if (this.subUrl === QwenImageModel.SUB_SYNTHESIS_URL) {
+                fetchHeaders["X-DashScope-Async"] = "enable";
+            }
+            const joinedUrl = `${String(this.baseUrl).replace(/\/+$/, "")}/${String(this.subUrl).replace(/^\/+/, "")}`;
+            const { data: responseData, header } = (yield this.req.fetch({
+                url: joinedUrl,
+                headers: Object.assign({}, fetchHeaders),
+                body: JSON.stringify(data),
+                method: "post",
+                stream: false,
+            }));
+            return (0, utils_1.handleResponseData)(responseData, header);
+        });
+    }
+    doGenerate(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            data.model = this.modelName;
+            if (!Object.prototype.hasOwnProperty.call(data, "async")) {
+                data.async = true;
+            }
+            if (data.async) {
+                this.subUrl = QwenImageModel.SUB_SYNTHESIS_URL;
+            }
+            else {
+                this.subUrl = QwenImageModel.SUB_GENERATION_URL;
+            }
+            const payload = this.coverModelRequestToQwenInput(data);
+            const res = (yield this.modelRequest(payload, options));
+            return this.normalizeStandardImageCompletion(res, this.modelName);
+        });
+    }
+    doStream(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            const nonStream = yield this.doGenerate(Object.assign({}, data), options);
+            const msg = ((_b = (_a = nonStream.choices) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.message) || {};
+            const singleChunk = {
+                id: nonStream.id,
+                object: "chat.completion.chunk",
+                created: nonStream.created,
+                model: nonStream.model,
+                log_id: nonStream.log_id,
+                error: nonStream.error || "",
+                code: nonStream.code || 0,
+                choices: [
+                    {
+                        index: 0,
+                        message: {
+                            id: nonStream.id,
+                            role: "assistant",
+                            type: msg.type || "image",
+                            content: msg.content || "",
+                            reasoning_content: "",
+                        },
+                        finish_reason: "stop",
+                    },
+                ],
+                usage: nonStream.usage,
+            };
+            const stream = new stream_1.ReadableStream({
+                start(controller) {
+                    controller.enqueue(singleChunk);
+                    controller.close();
+                },
+            });
+            return (0, stream_1.createAsyncIterable)(stream);
+        });
+    }
+}
+exports.QwenImageModel = QwenImageModel;
+QwenImageModel.BASE_URL = "https://dashscope.aliyuncs.com";
+QwenImageModel.SUB_SYNTHESIS_URL = "api/v1/services/aigc/text2image/image-synthesis";
+QwenImageModel.SUB_GENERATION_URL = "api/v1/services/aigc/multimodal-generation/generation";
+function mapUsageToStandard(usage) {
+    return {
+        prompt_tokens: 0,
+        completion_tokens: 0,
+        knowledge_tokens: 0,
+        reasoning_tokens: 0,
+        total_tokens: 0,
+    };
+}

package/lib/models/QwenSketchToImage/index.d.ts

@@ -0,0 +1,35 @@
+import type { IAbstractRequest } from "@vectorx/ai-types";
+import type { DoGenerateOutput, DoStreamOutput, ModelRequestOptions, ReqOptions } from "../../model-type";
+import { TokenManager } from "../../tokenManager";
+import { SimpleChatModel } from "../Chat";
+export interface WanxSketchToImageLiteParameters {
+    size?: "768*768" | "720*1280" | "1280*720" | string;
+    n?: number;
+    sketch_weight?: number;
+    style?: "<auto>" | "<3d cartoon>" | "<anime>" | "<oil painting>" | "<watercolor>" | "<sketch>" | "<chinese painting>" | "<flat illustration>" | string;
+    sketch_extraction?: boolean;
+    sketch_color?: number[][];
+}
+export interface WanxSketchToImageLiteAPIInput {
+    model: string;
+    input: {
+        sketch_image_url: string;
+        prompt: string;
+    };
+    parameters?: WanxSketchToImageLiteParameters;
+}
+export type WanxSketchToImageLiteRequestOptions = WanxSketchToImageLiteAPIInput & {
+    parameters?: WanxSketchToImageLiteParameters;
+};
+export declare class WanxSketchToImageLiteModel extends SimpleChatModel {
+    static BASE_URL: string;
+    static SUB_URL: string;
+    modelName: string;
+    constructor(req: IAbstractRequest, baseUrl: string, tokenManager: TokenManager);
+    protected convertToSketchToImageInput(data: ModelRequestOptions & {
+        parameters?: WanxSketchToImageLiteParameters;
+    }): WanxSketchToImageLiteRequestOptions;
+    protected modelRequest(data: WanxSketchToImageLiteRequestOptions, options?: ReqOptions): Promise<ReadableStream<Uint8Array> | Promise<unknown>>;
+    doGenerate(data: ModelRequestOptions, options?: ReqOptions): Promise<DoGenerateOutput>;
+    doStream(data: ModelRequestOptions, options?: ReqOptions): Promise<DoStreamOutput>;
+}

package/lib/models/QwenSketchToImage/index.js

@@ -0,0 +1,155 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WanxSketchToImageLiteModel = void 0;
+const stream_1 = require("../../stream");
+const utils_1 = require("../../utils");
+const Chat_1 = require("../Chat");
+const index_1 = require("../index");
+class WanxSketchToImageLiteModel extends Chat_1.SimpleChatModel {
+    constructor(req, baseUrl, tokenManager) {
+        super(req, baseUrl, WanxSketchToImageLiteModel.SUB_URL, tokenManager);
+        this.modelName = index_1.modelName[index_1.MultiModalModelName.WanxSketchToImageLite];
+    }
+    convertToSketchToImageInput(data) {
+        var _a;
+        const messages = data.messages || [];
+        if (!Array.isArray(messages) || messages.length === 0) {
+            throw new Error("wanx-sketch-to-image-lite 请求需要提供 messages");
+        }
+        const userMessage = messages.find((m) => m.role === "user");
+        if (!userMessage || !Array.isArray(userMessage.content)) {
+            throw new Error("wanx-sketch-to-image-lite 请求需要提供 user 消息");
+        }
+        const imageContent = userMessage.content.find((c) => c.type === "image_url");
+        if (!((_a = imageContent === null || imageContent === void 0 ? void 0 : imageContent.image_url) === null || _a === void 0 ? void 0 : _a.url)) {
+            throw new Error("wanx-sketch-to-image-lite 请求需要提供草图图片 URL");
+        }
+        const sketchImageUrl = imageContent.image_url.url;
+        const textContent = userMessage.content.find((c) => c.type === "text");
+        if (!(textContent === null || textContent === void 0 ? void 0 : textContent.text)) {
+            throw new Error("wanx-sketch-to-image-lite 请求需要提供文本提示词");
+        }
+        const prompt = textContent.text;
+        const modelParams = data.parameters || {};
+        const parameters = Object.assign({ size: "768*768", n: 4, sketch_weight: 10, style: "<auto>", sketch_extraction: false, sketch_color: [] }, modelParams);
+        if (parameters.n < 1 || parameters.n > 4) {
+            throw new Error("生成图片数量 n 必须在 1-4 之间");
+        }
+        if (parameters.sketch_weight < 0 || parameters.sketch_weight > 10) {
+            throw new Error("草图权重 sketch_weight 必须在 0-10 之间");
+        }
+        return {
+            model: this.modelName,
+            input: {
+                sketch_image_url: sketchImageUrl,
+                prompt: prompt,
+            },
+            parameters,
+        };
+    }
+    modelRequest(data_1) {
+        return __awaiter(this, arguments, void 0, function* (data, options = {}) {
+            const fetchHeaders = yield this.createAuthHeaders(options === null || options === void 0 ? void 0 : options.headers);
+            const { data: responseData, header } = (yield this.req.fetch({
+                method: "post",
+                headers: Object.assign(Object.assign({}, fetchHeaders), ((options === null || options === void 0 ? void 0 : options.headers) || {})),
+                body: JSON.stringify(data),
+                url: `${this.baseUrl}/${this.subUrl}`,
+                stream: false,
+            }));
+            return (0, utils_1.handleResponseData)(responseData, header);
+        });
+    }
+    doGenerate(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t;
+            data.model = this.modelName;
+            const body = this.convertToSketchToImageInput(data);
+            const headers = Object.assign(Object.assign({}, ((options === null || options === void 0 ? void 0 : options.headers) || {})), { "X-DashScope-Async": "enable" });
+            const res = (yield this.modelRequest(body, { headers }));
+            const normalized = {
+                id: (_b = (_a = res === null || res === void 0 ? void 0 : res.request_id) !== null && _a !== void 0 ? _a : res === null || res === void 0 ? void 0 : res.id) !== null && _b !== void 0 ? _b : "",
+                object: (_c = res === null || res === void 0 ? void 0 : res.object) !== null && _c !== void 0 ? _c : "chat.completion",
+                created: Math.floor(Date.now() / 1000),
+                model: this.modelName,
+                log_id: (_e = (_d = res === null || res === void 0 ? void 0 : res.request_id) !== null && _d !== void 0 ? _d : res === null || res === void 0 ? void 0 : res.id) !== null && _e !== void 0 ? _e : "",
+                error: "",
+                code: Number((_f = res === null || res === void 0 ? void 0 : res.status_code) !== null && _f !== void 0 ? _f : 0) || 0,
+                choices: [
+                    {
+                        index: 0,
+                        message: {
+                            id: (_h = (_g = res === null || res === void 0 ? void 0 : res.request_id) !== null && _g !== void 0 ? _g : res === null || res === void 0 ? void 0 : res.id) !== null && _h !== void 0 ? _h : "",
+                            role: "assistant",
+                            type: "async_task",
+                            content: JSON.stringify({
+                                task_id: (_j = res === null || res === void 0 ? void 0 : res.output) === null || _j === void 0 ? void 0 : _j.task_id,
+                                task_status: (_k = res === null || res === void 0 ? void 0 : res.output) === null || _k === void 0 ? void 0 : _k.task_status,
+                                results: (_m = (_l = res === null || res === void 0 ? void 0 : res.output) === null || _l === void 0 ? void 0 : _l.results) !== null && _m !== void 0 ? _m : [],
+                            }),
+                            reasoning_content: "",
+                        },
+                        finish_reason: "stop",
+                    },
+                ],
+                usage: {
+                    prompt_tokens: (_p = (_o = res === null || res === void 0 ? void 0 : res.usage) === null || _o === void 0 ? void 0 : _o.prompt_tokens) !== null && _p !== void 0 ? _p : 0,
+                    completion_tokens: (_r = (_q = res === null || res === void 0 ? void 0 : res.usage) === null || _q === void 0 ? void 0 : _q.completion_tokens) !== null && _r !== void 0 ? _r : 0,
+                    total_tokens: (_t = (_s = res === null || res === void 0 ? void 0 : res.usage) === null || _s === void 0 ? void 0 : _s.total_tokens) !== null && _t !== void 0 ? _t : 0,
+                    knowledge_tokens: 0,
+                    reasoning_tokens: 0,
+                },
+            };
+            return normalized;
+        });
+    }
+    doStream(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            const nonStream = yield this.doGenerate(data, options);
+            const msg = ((_b = (_a = nonStream.choices) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.message) || {};
+            const singleChunk = {
+                id: nonStream.id,
+                object: "chat.completion.chunk",
+                created: nonStream.created,
+                model: nonStream.model,
+                log_id: nonStream.log_id,
+                error: nonStream.error || "",
+                code: nonStream.code || 0,
+                choices: [
+                    {
+                        index: 0,
+                        message: {
+                            id: nonStream.id,
+                            role: "assistant",
+                            type: msg.type || "async_task",
+                            content: msg.content || "",
+                            reasoning_content: "",
+                        },
+                        finish_reason: "stop",
+                    },
+                ],
+                usage: nonStream.usage,
+            };
+            const stream = new ReadableStream({
+                start(controller) {
+                    controller.enqueue(singleChunk);
+                    controller.close();
+                },
+            });
+            return (0, stream_1.createAsyncIterable)(stream);
+        });
+    }
+}
+exports.WanxSketchToImageLiteModel = WanxSketchToImageLiteModel;
+WanxSketchToImageLiteModel.BASE_URL = "https://dashscope.aliyuncs.com";
+WanxSketchToImageLiteModel.SUB_URL = "api/v1/services/aigc/image2image/image-synthesis";