@vectorx/ai-sdk 0.1.3 → 0.4.0

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

@@ -0,0 +1,114 @@
+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 declare const STYLE_INDEX: {
+    readonly CUSTOM: -1;
+    readonly RETRO_COMIC: 0;
+    readonly FAIRY_TALE_3D: 1;
+    readonly ANIME_2D: 2;
+    readonly FRESH_STYLE: 3;
+    readonly FUTURE_TECH: 4;
+    readonly CHINESE_PAINTING: 5;
+    readonly GENERAL_BATTLE: 6;
+    readonly COLORFUL_CARTOON: 7;
+    readonly ELEGANT_CHINESE: 8;
+    readonly NEW_YEAR: 9;
+    readonly GONGBI_STYLE: 14;
+    readonly NEW_YEAR_GREET: 15;
+    readonly FAIRY_WORLD: 30;
+    readonly CLAY_WORLD: 31;
+    readonly PIXEL_WORLD: 32;
+    readonly ADVENTURE_WORLD: 33;
+    readonly JAPANESE_ANIME: 34;
+    readonly WORLD_3D: 35;
+    readonly ANIME_WORLD: 36;
+    readonly HAND_DRAWN: 37;
+    readonly CRAYON_WORLD: 38;
+    readonly FRIDGE_MAGNET: 39;
+    readonly BAJIQUAN_WORLD: 40;
+};
+export type StyleIndex = (typeof STYLE_INDEX)[keyof typeof STYLE_INDEX];
+export interface QwenStyleRepaintV1Parameters {
+    style_index?: StyleIndex;
+    style_ref_url?: string;
+}
+export interface QwenStyleRepaintV1StandardInput {
+    messages: Array<{
+        role: "user" | "assistant" | "system";
+        content: Array<{
+            type: "image_url";
+            image_url: {
+                url: string;
+            };
+        } | {
+            type: "text";
+            text: string;
+        }>;
+    }>;
+    parameters?: QwenStyleRepaintV1Parameters;
+}
+export interface QwenStyleRepaintV1APIInput {
+    model: string;
+    input: {
+        image_url: string;
+        style_index: number;
+        style_ref_url?: string;
+    };
+}
+export type QwenStyleRepaintV1RequestOptions = QwenStyleRepaintV1APIInput;
+export type QwenStyleRepaintV1ContentItem = {
+    image?: string;
+    url?: string;
+    image_url?: string | {
+        url: string;
+    };
+    b64_json?: string;
+    [key: string]: any;
+};
+export interface QwenStyleRepaintV1APIResponse {
+    async?: boolean;
+    output: {
+        choices?: Array<{
+            finish_reason: string;
+            message: {
+                role: "assistant" | "user";
+                content: QwenStyleRepaintV1ContentItem[];
+            };
+        }>;
+        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 QwenStyleRepaintV1Model extends SimpleChatModel {
+    static BASE_URL: string;
+    static SUB_URL: string;
+    modelName: string;
+    constructor(req: IAbstractRequest, baseUrl: string, tokenManager: TokenManager);
+    protected normalizeStandardImageCompletion(res: QwenStyleRepaintV1APIResponse, fallbackModel: string): DoGenerateOutput;
+    protected convertToStyleRepaintInput(data: ModelRequestOptions & {
+        parameters?: QwenStyleRepaintV1Parameters;
+    }): QwenStyleRepaintV1APIInput;
+    protected modelRequest(data: QwenStyleRepaintV1APIInput, options?: ReqOptions): Promise<ReadableStream<Uint8Array> | Promise<unknown>>;
+    doGenerate(data: ModelRequestOptions & {
+        parameters?: QwenStyleRepaintV1Parameters;
+    }, options?: ReqOptions): Promise<DoGenerateOutput>;
+    doStream(data: ModelRequestOptions, options?: ReqOptions): Promise<DoStreamOutput>;
+}

package/lib/models/QwenStyleRepaintV1/index.js

@@ -0,0 +1,213 @@
+"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.QwenStyleRepaintV1Model = exports.STYLE_INDEX = void 0;
+const stream_1 = require("../../stream");
+const utils_1 = require("../../utils");
+const Chat_1 = require("../Chat");
+const index_1 = require("../index");
+exports.STYLE_INDEX = {
+    CUSTOM: -1,
+    RETRO_COMIC: 0,
+    FAIRY_TALE_3D: 1,
+    ANIME_2D: 2,
+    FRESH_STYLE: 3,
+    FUTURE_TECH: 4,
+    CHINESE_PAINTING: 5,
+    GENERAL_BATTLE: 6,
+    COLORFUL_CARTOON: 7,
+    ELEGANT_CHINESE: 8,
+    NEW_YEAR: 9,
+    GONGBI_STYLE: 14,
+    NEW_YEAR_GREET: 15,
+    FAIRY_WORLD: 30,
+    CLAY_WORLD: 31,
+    PIXEL_WORLD: 32,
+    ADVENTURE_WORLD: 33,
+    JAPANESE_ANIME: 34,
+    WORLD_3D: 35,
+    ANIME_WORLD: 36,
+    HAND_DRAWN: 37,
+    CRAYON_WORLD: 38,
+    FRIDGE_MAGNET: 39,
+    BAJIQUAN_WORLD: 40,
+};
+class QwenStyleRepaintV1Model extends Chat_1.SimpleChatModel {
+    constructor(req, baseUrl, tokenManager) {
+        super(req, baseUrl, QwenStyleRepaintV1Model.SUB_URL, tokenManager);
+        this.modelName = index_1.modelName[index_1.MultiModalModelName.QwenStyleRepaintV1];
+    }
+    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;
+    }
+    convertToStyleRepaintInput(data) {
+        var _a, _b;
+        const messages = data.messages || [];
+        if (!Array.isArray(messages) || messages.length === 0) {
+            throw new Error("wanx-style-repaint-v1 请求需要提供 messages");
+        }
+        const userMessage = messages.find((m) => m.role === "user");
+        if (!userMessage || !Array.isArray(userMessage.content)) {
+            throw new Error("wanx-style-repaint-v1 请求需要提供 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-style-repaint-v1 请求需要提供图片 URL");
+        }
+        const imageUrl = imageContent.image_url.url;
+        const modelParams = data.parameters || {};
+        const styleIndex = (_b = modelParams.style_index) !== null && _b !== void 0 ? _b : exports.STYLE_INDEX.FRESH_STYLE;
+        const styleRefUrl = modelParams.style_ref_url;
+        if (styleIndex === exports.STYLE_INDEX.CUSTOM && !styleRefUrl) {
+            throw new Error("使用自定义风格时（style_index: -1），必须提供 style_ref_url");
+        }
+        const input = {
+            image_url: imageUrl,
+            style_index: styleIndex,
+        };
+        if (styleIndex === exports.STYLE_INDEX.CUSTOM && styleRefUrl) {
+            input.style_ref_url = styleRefUrl;
+        }
+        return {
+            model: this.modelName,
+            input,
+        };
+    }
+    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* () {
+            data.model = this.modelName;
+            const body = this.convertToStyleRepaintInput(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, Object.assign(Object.assign({}, options), { headers })));
+            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(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 stream_1.ReadableStream({
+                start(controller) {
+                    controller.enqueue(singleChunk);
+                    controller.close();
+                },
+            });
+            return (0, stream_1.createAsyncIterable)(stream);
+        });
+    }
+}
+exports.QwenStyleRepaintV1Model = QwenStyleRepaintV1Model;
+QwenStyleRepaintV1Model.BASE_URL = "https://dashscope.aliyuncs.com";
+QwenStyleRepaintV1Model.SUB_URL = "api/v1/services/aigc/image-generation/generation";
+function mapUsageToStandard(usage) {
+    return {
+        prompt_tokens: 0,
+        completion_tokens: 0,
+        knowledge_tokens: 0,
+        reasoning_tokens: 0,
+        total_tokens: 0,
+    };
+}

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

@@ -0,0 +1,78 @@
+import type { IAbstractRequest } from "@vectorx/ai-types";
+import type { DoGenerateOutput, DoStreamOutput, ImageUrlContent, ModelRequestOptions, ReqOptions, TextContent } from "../../model-type";
+import type { TokenManager } from "../../tokenManager";
+import { SimpleChatModel } from "../Chat";
+import type { MultiModalModelName } from "../index";
+export interface QwenVlMaxMessage {
+    role: "system" | "user" | "assistant";
+    content: string | Array<TextContent | ImageUrlContent>;
+}
+export interface QwenVlMaxParameters {
+    seed?: number;
+    temperature?: number;
+    top_p?: number;
+    max_tokens?: number;
+    n?: number;
+    stream?: boolean;
+    stop?: string | string[];
+    frequency_penalty?: number;
+    presence_penalty?: number;
+    user?: string;
+    enable_search?: boolean;
+    enable_thinking?: boolean;
+}
+export interface QwenVlMaxAPIInput extends QwenVlMaxParameters {
+    model: string;
+    messages: QwenVlMaxMessage[];
+    stream?: boolean;
+}
+export interface QwenVlMaxResponse {
+    id: string;
+    object: string;
+    created: number;
+    model: string;
+    choices: Array<{
+        index: number;
+        message: {
+            role: string;
+            content: string;
+        };
+        finish_reason: string;
+    }>;
+    usage: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+        output_tokens?: number;
+        input_tokens?: number;
+    };
+}
+export interface QwenVlMaxStreamChunk {
+    id: string;
+    object: string;
+    created: number;
+    model: string;
+    choices: Array<{
+        index: number;
+        delta: {
+            role?: string;
+            content?: string;
+        };
+        finish_reason: string | null;
+    }>;
+    usage?: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+}
+export declare class QwenVlMax extends SimpleChatModel {
+    static BASE_URL: string;
+    modelName: MultiModalModelName;
+    constructor(req: IAbstractRequest, baseUrl: string, modelName: MultiModalModelName, tokenManager: TokenManager);
+    protected modelRequest(data: QwenVlMaxAPIInput, options?: ReqOptions): Promise<ReadableStream<Uint8Array> | Promise<unknown>>;
+    protected normalizeResponse(response: QwenVlMaxResponse): DoGenerateOutput;
+    doGenerate(data: ModelRequestOptions, options?: ReqOptions): Promise<DoGenerateOutput>;
+    doStream(data: ModelRequestOptions, options?: ReqOptions): Promise<DoStreamOutput>;
+    private convertToQwenVlMaxRequestOptions;
+}

package/lib/models/QwenVlMax/index.js

@@ -0,0 +1,121 @@
+"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.QwenVlMax = void 0;
+const model_type_1 = require("../../model-type");
+const stream_1 = require("../../stream");
+const utils_1 = require("../../utils");
+const Chat_1 = require("../Chat");
+const defaultParameters = {
+    max_tokens: 2000,
+    temperature: 1.0,
+    top_p: 1.0,
+    frequency_penalty: 0.0,
+    presence_penalty: 0.0,
+    stream: false,
+    enable_thinking: false,
+    enable_search: false,
+    n: 1,
+};
+class QwenVlMax extends Chat_1.SimpleChatModel {
+    constructor(req, baseUrl, modelName, tokenManager) {
+        super(req, baseUrl, "compatible-mode/v1/chat/completions", tokenManager);
+        this.modelName = modelName;
+    }
+    modelRequest(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            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({}, fetchHeaders),
+                body: JSON.stringify(data),
+                url: `${this.baseUrl}/${this.subUrl}`,
+                stream: Boolean(data.stream),
+            }));
+            return (0, utils_1.handleResponseData)(responseData, header);
+        });
+    }
+    normalizeResponse(response) {
+        var _a, _b, _c, _d;
+        const choice = (_a = response.choices) === null || _a === void 0 ? void 0 : _a[0];
+        const message = choice === null || choice === void 0 ? void 0 : choice.message;
+        return {
+            id: response.id,
+            object: response.object,
+            created: response.created,
+            model: response.model,
+            log_id: response.id,
+            error: "",
+            code: 0,
+            choices: [
+                {
+                    index: (choice === null || choice === void 0 ? void 0 : choice.index) || 0,
+                    message: {
+                        id: response.id,
+                        role: (message === null || message === void 0 ? void 0 : message.role) || "assistant",
+                        type: "answer",
+                        content: (message === null || message === void 0 ? void 0 : message.content) || "",
+                        reasoning_content: "",
+                    },
+                    finish_reason: (choice === null || choice === void 0 ? void 0 : choice.finish_reason) || "stop",
+                },
+            ],
+            usage: {
+                prompt_tokens: ((_b = response.usage) === null || _b === void 0 ? void 0 : _b.prompt_tokens) || 0,
+                completion_tokens: ((_c = response.usage) === null || _c === void 0 ? void 0 : _c.completion_tokens) || 0,
+                knowledge_tokens: 0,
+                reasoning_tokens: 0,
+                total_tokens: ((_d = response.usage) === null || _d === void 0 ? void 0 : _d.total_tokens) || 0,
+            },
+        };
+    }
+    doGenerate(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const qwenVlMaxData = this.convertToQwenVlMaxRequestOptions(data);
+            const requestData = Object.assign(Object.assign({}, qwenVlMaxData), { stream: false });
+            const res = yield this.modelRequest(requestData, options);
+            return this.normalizeResponse(res);
+        });
+    }
+    doStream(data, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const qwenVlMaxData = this.convertToQwenVlMaxRequestOptions(data);
+            const requestData = Object.assign(Object.assign({}, qwenVlMaxData), { stream: true });
+            const _stream = (yield this.modelRequest(requestData, options));
+            const stream = (0, stream_1.toPolyfillReadable)(_stream);
+            const standardStream = (0, stream_1.intoStandardStream)(stream);
+            return (0, stream_1.createAsyncIterable)(standardStream);
+        });
+    }
+    convertToQwenVlMaxRequestOptions(data) {
+        var _a, _b, _c;
+        const clamp = (value, min, max, defaultValue) => value !== undefined ? Math.max(min, Math.min(max, value)) : defaultValue;
+        const messages = (data.messages || []).map((msg) => ({
+            role: msg.role,
+            content: Array.isArray(msg.content) ? (0, model_type_1.filterContentByTypes)(msg.content, ["text", "image_url"]) : msg.content,
+        }));
+        return {
+            model: this.modelName,
+            messages,
+            temperature: clamp(data.temperature, 0, 2, defaultParameters.temperature),
+            top_p: clamp(data.top_p, 0, 1, defaultParameters.top_p),
+            frequency_penalty: clamp(data.frequency_penalty, -2, 2, defaultParameters.frequency_penalty),
+            presence_penalty: clamp(data.presence_penalty, -2, 2, defaultParameters.presence_penalty),
+            max_tokens: clamp(data.max_tokens, 1, Number.POSITIVE_INFINITY, defaultParameters.max_tokens),
+            n: clamp(data.n, 1, Number.POSITIVE_INFINITY, defaultParameters.n),
+            stream: (_a = data.stream) !== null && _a !== void 0 ? _a : defaultParameters.stream,
+            enable_search: (_b = data.enable_search) !== null && _b !== void 0 ? _b : defaultParameters.enable_search,
+            enable_thinking: (_c = data.enable_thinking) !== null && _c !== void 0 ? _c : defaultParameters.enable_thinking,
+        };
+    }
+}
+exports.QwenVlMax = QwenVlMax;
+QwenVlMax.BASE_URL = "https://dashscope.aliyuncs.com";

package/lib/models/index.d.ts

@@ -0,0 +1,50 @@
+import { DefaultSimpleModel } from "./Default";
+import { QwenDocTurbo } from "./QwenDocTurbo";
+import { QwenImageModel } from "./QwenImage";
+import { WanxSketchToImageLiteModel } from "./QwenSketchToImage";
+import { QwenStyleRepaintV1Model } from "./QwenStyleRepaintV1";
+import { QwenVlMax } from "./QwenVlMax";
+import { ReActModel } from "./react";
+export declare enum ModelName {
+    DeepSeekR1 = "deepseek-r1",
+    DeepSeekV3 = "deepseek-v3",
+    QwenMax = "qwen-max",
+    QwenLong = "qwen-long",
+    QwenPlus = "qwen-plus",
+    QwenTurbo = "qwen-turbo",
+    QwenVlOcr = "qwen-vl-ocr",
+    Wanx21T2iTurbo = "wanx2.1-t2i-turbo"
+}
+export declare enum MultiModalModelName {
+    QvqMaxLatest = "qvq-max-latest",
+    QwenVlMax = "qwen-vl-max",
+    QwenDocTurbo = "qwen-doc-turbo",
+    QwenOmniTurboRealtime = "qwen-omni-turbo-realtime",
+    Wanx21T2iPlus = "wanx2.1-t2i-plus",
+    Wanx21T2iTurbo = "wanx2.1-t2i-turbo",
+    QwenImage = "qwen-image",
+    WanxSketchToImageLite = "wanx-sketch-to-image-lite",
+    QwenStyleRepaintV1 = "wanx-style-repaint-v1"
+}
+export declare const modelName: {
+    "deepseek-r1": string;
+    "deepseek-v3": string;
+    "qwen-max": string;
+    "qwen-long": string;
+    "qwen-plus": string;
+    "qwen-turbo": string;
+    "qwen-vl-ocr": string;
+    "qvq-max-latest": string;
+    "qwen-vl-max": string;
+    "qwen-doc-turbo": string;
+    "qwen-omni-turbo-realtime": string;
+    "wanx2.1-t2i-plus": string;
+    "wanx2.1-t2i-turbo": string;
+    "qwen-image": string;
+    "wanx-sketch-to-image-lite": string;
+    "wanx-style-repaint-v1": string;
+};
+export declare const isValidModel: (model: ModelName | MultiModalModelName) => model is ModelName | MultiModalModelName;
+export declare const isMultiModalModel: (model: ModelName | MultiModalModelName) => model is MultiModalModelName;
+declare const toolMap: Map<string, Function>;
+export { DefaultSimpleModel, QwenDocTurbo, ReActModel, toolMap, WanxSketchToImageLiteModel, QwenImageModel, QwenVlMax, QwenStyleRepaintV1Model, };

package/lib/models/index.js

@@ -1,10 +1,20 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.toolMap = exports.ReactModel = exports.DefaultSimpleModel = exports.modelName = exports.ModelName = void 0;
-const default_1 = require("./default");
-Object.defineProperty(exports, "DefaultSimpleModel", { enumerable: true, get: function () { return default_1.DefaultSimpleModel; } });
+exports.QwenStyleRepaintV1Model = exports.QwenVlMax = exports.QwenImageModel = exports.WanxSketchToImageLiteModel = exports.toolMap = exports.ReActModel = exports.QwenDocTurbo = exports.DefaultSimpleModel = exports.isMultiModalModel = exports.isValidModel = exports.modelName = exports.MultiModalModelName = exports.ModelName = void 0;
+const Default_1 = require("./Default");
+Object.defineProperty(exports, "DefaultSimpleModel", { enumerable: true, get: function () { return Default_1.DefaultSimpleModel; } });
+const QwenDocTurbo_1 = require("./QwenDocTurbo");
+Object.defineProperty(exports, "QwenDocTurbo", { enumerable: true, get: function () { return QwenDocTurbo_1.QwenDocTurbo; } });
+const QwenImage_1 = require("./QwenImage");
+Object.defineProperty(exports, "QwenImageModel", { enumerable: true, get: function () { return QwenImage_1.QwenImageModel; } });
+const QwenSketchToImage_1 = require("./QwenSketchToImage");
+Object.defineProperty(exports, "WanxSketchToImageLiteModel", { enumerable: true, get: function () { return QwenSketchToImage_1.WanxSketchToImageLiteModel; } });
+const QwenStyleRepaintV1_1 = require("./QwenStyleRepaintV1");
+Object.defineProperty(exports, "QwenStyleRepaintV1Model", { enumerable: true, get: function () { return QwenStyleRepaintV1_1.QwenStyleRepaintV1Model; } });
+const QwenVlMax_1 = require("./QwenVlMax");
+Object.defineProperty(exports, "QwenVlMax", { enumerable: true, get: function () { return QwenVlMax_1.QwenVlMax; } });
 const react_1 = require("./react");
-Object.defineProperty(exports, "ReactModel", { enumerable: true, get: function () { return react_1.ReactModel; } });
+Object.defineProperty(exports, "ReActModel", { enumerable: true, get: function () { return react_1.ReActModel; } });
 var ModelName;
 (function (ModelName) {
     ModelName["DeepSeekR1"] = "deepseek-r1";
@@ -16,6 +26,18 @@ var ModelName;
     ModelName["QwenVlOcr"] = "qwen-vl-ocr";
     ModelName["Wanx21T2iTurbo"] = "wanx2.1-t2i-turbo";
 })(ModelName || (exports.ModelName = ModelName = {}));
+var MultiModalModelName;
+(function (MultiModalModelName) {
+    MultiModalModelName["QvqMaxLatest"] = "qvq-max-latest";
+    MultiModalModelName["QwenVlMax"] = "qwen-vl-max";
+    MultiModalModelName["QwenDocTurbo"] = "qwen-doc-turbo";
+    MultiModalModelName["QwenOmniTurboRealtime"] = "qwen-omni-turbo-realtime";
+    MultiModalModelName["Wanx21T2iPlus"] = "wanx2.1-t2i-plus";
+    MultiModalModelName["Wanx21T2iTurbo"] = "wanx2.1-t2i-turbo";
+    MultiModalModelName["QwenImage"] = "qwen-image";
+    MultiModalModelName["WanxSketchToImageLite"] = "wanx-sketch-to-image-lite";
+    MultiModalModelName["QwenStyleRepaintV1"] = "wanx-style-repaint-v1";
+})(MultiModalModelName || (exports.MultiModalModelName = MultiModalModelName = {}));
 exports.modelName = {
     [ModelName.DeepSeekR1]: "deepseek-r1",
     [ModelName.DeepSeekV3]: "deepseek-v3",
@@ -24,6 +46,24 @@ exports.modelName = {
     [ModelName.QwenPlus]: "qwen-plus",
     [ModelName.QwenTurbo]: "qwen-turbo",
     [ModelName.QwenVlOcr]: "qwen-vl-ocr",
+    [MultiModalModelName.QvqMaxLatest]: "qvq-max-latest",
+    [MultiModalModelName.QwenVlMax]: "qwen-vl-max",
+    [MultiModalModelName.QwenDocTurbo]: "qwen-doc-turbo",
+    [MultiModalModelName.QwenOmniTurboRealtime]: "qwen-omni-turbo-realtime",
+    [MultiModalModelName.Wanx21T2iPlus]: "wanx2.1-t2i-plus",
+    [MultiModalModelName.Wanx21T2iTurbo]: "wanx2.1-t2i-turbo",
+    [MultiModalModelName.QwenImage]: "qwen-image",
+    [MultiModalModelName.WanxSketchToImageLite]: "wanx-sketch-to-image-lite",
+    [MultiModalModelName.QwenStyleRepaintV1]: "wanx-style-repaint-v1",
 };
+const isValidModel = (model) => {
+    const modelNames = Object.values(exports.modelName);
+    return modelNames.includes(model);
+};
+exports.isValidModel = isValidModel;
+const isMultiModalModel = (model) => {
+    return Object.values(MultiModalModelName).includes(model);
+};
+exports.isMultiModalModel = isMultiModalModel;
 const toolMap = new Map();
 exports.toolMap = toolMap;

package/{types → lib}/models/react.d.ts

@@ -1,5 +1,6 @@
-import type { DoGenerateOutput, DoStreamOutput, ModelRequestOptions, ReqOptions, SimpleChatModel } from "../types";
-export declare class ReactModel {
+import type { DoGenerateOutput, DoStreamOutput, ModelRequestOptions, ReqOptions } from "../model-type";
+import type { SimpleChatModel } from "./Chat";
+export declare class ReActModel {
     private model;
     constructor(model: SimpleChatModel);
     generateText(_input: ModelRequestOptions, options?: ReqOptions): Promise<DoGenerateOutput>;

package/lib/models/react.js

@@ -9,8 +9,8 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ReactModel = void 0;
-class ReactModel {
+exports.ReActModel = void 0;
+class ReActModel {
     constructor(model) {
         this.model = model;
     }
@@ -25,4 +25,4 @@ class ReactModel {
         });
     }
 }
-exports.ReactModel = ReactModel;
+exports.ReActModel = ReActModel;