@vectorx/ai-sdk 0.1.3 → 0.3.0

package/lib/models/QwenVlMax/index.js

@@ -0,0 +1,120 @@
+"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 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: 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,47 @@
+import { DefaultSimpleModel } from "./Default";
+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",
+    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-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, ReActModel, toolMap, WanxSketchToImageLiteModel, QwenImageModel, QwenVlMax, QwenStyleRepaintV1Model, };

package/lib/models/index.js

@@ -1,10 +1,18 @@
 "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.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 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 +24,17 @@ 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["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 +43,23 @@ 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.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;

package/{types → lib}/stream.d.ts

@@ -1,17 +1,11 @@
 import { type ParsedEvent } from "./eventsource_parser";
-import type { BaseDoGenerateOutput, BaseDoStreamOutputChunk, ChatModelMessage, ModelTool, SimpleChatModel, ToolCallAssistantMessage } from "./types";
+import type { ChatModelMessage, ModelTool, ToolCallAssistantMessage } from "./model-type";
 export type FunctionTool = {
     name: string;
     description: string;
     fn: CallableFunction;
     parameters: object;
 };
-type RawResponse = {
-    rawResponse?: any;
-};
-export type DoGenerateOutput = BaseDoGenerateOutput & RawResponse;
-export type DoStreamOutput = AsyncIterableReadableStream<BaseDoStreamOutputChunk & RawResponse>;
-export type ChatModelConstructor = typeof SimpleChatModel;
 export type AsyncIterableReadableStream<T> = ReadableStream<T> & {
     [Symbol.asyncIterator]: () => {
         next(): Promise<IteratorResult<T>>;
@@ -51,4 +45,3 @@ export declare function createPromise<T = unknown>(): {
 };
 export declare function isToolCallAssistantMessage(message: ChatModelMessage): message is ToolCallAssistantMessage;
 export declare function functionToolToModelTool(tool: FunctionTool): ModelTool;
-export {};

package/lib/tokenManager.d.ts

@@ -0,0 +1,36 @@
+import type { IAbstractRequest } from "@vectorx/ai-types";
+export interface TokenResponse {
+    code: number;
+    msg: string;
+    data: {
+        token: string;
+        expires_at: number | null;
+    };
+    success: boolean;
+}
+export interface TokenError {
+    code: string;
+    message: string;
+    request_id: string;
+}
+export declare class TokenManager {
+    private currentToken;
+    private tokenExpiresAt;
+    private readonly baseUrl;
+    private readonly subUrl;
+    private readonly defaultExpireSeconds;
+    private readonly refreshBufferSeconds;
+    private req;
+    constructor(req: IAbstractRequest, baseUrl: string);
+    get fullUrl(): string;
+    getValidToken(): Promise<string>;
+    private refreshToken;
+    isTokenValid(): boolean;
+    getTokenRemainingTime(): number;
+    clearToken(): void;
+    getTokenInfo(): {
+        token: string | null;
+        expiresAt: number;
+        remainingTime: number;
+    };
+}

package/lib/tokenManager.js

@@ -0,0 +1,89 @@
+"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.TokenManager = void 0;
+class TokenManager {
+    constructor(req, baseUrl) {
+        this.currentToken = null;
+        this.tokenExpiresAt = 0;
+        this.subUrl = "agent/token";
+        this.defaultExpireSeconds = 600;
+        this.refreshBufferSeconds = 300;
+        this.req = req;
+        this.baseUrl = baseUrl;
+    }
+    get fullUrl() {
+        return `${this.baseUrl}/${this.subUrl}`;
+    }
+    getValidToken() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const now = Math.floor(Date.now() / 1000);
+            if (this.currentToken && this.tokenExpiresAt > now + this.refreshBufferSeconds) {
+                return this.currentToken;
+            }
+            yield this.refreshToken();
+            return this.currentToken;
+        });
+    }
+    refreshToken(expireInSeconds) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d;
+            const expireSeconds = expireInSeconds || this.defaultExpireSeconds;
+            const url = `${this.fullUrl}?expire_in_seconds=${expireSeconds}`;
+            try {
+                const response = yield this.req.post({
+                    url,
+                    headers: {
+                        "Content-Type": "application/json",
+                    },
+                });
+                if (response.statusCode !== 200) {
+                    throw new Error(`获取临时 Token 失败: ${(_a = response.data) === null || _a === void 0 ? void 0 : _a.msg} (${(_b = response.data) === null || _b === void 0 ? void 0 : _b.code})`);
+                }
+                if (!((_d = (_c = response.data) === null || _c === void 0 ? void 0 : _c.data) === null || _d === void 0 ? void 0 : _d.token)) {
+                    throw new Error("获取临时 Token 失败: 响应数据无效");
+                }
+                this.currentToken = response.data.data.token;
+                this.tokenExpiresAt = response.data.data.expires_at || 0;
+            }
+            catch (error) {
+                console.error("刷新临时 Token 失败:", error);
+                throw error;
+            }
+        });
+    }
+    isTokenValid() {
+        if (!this.currentToken) {
+            return false;
+        }
+        const now = Math.floor(Date.now() / 1000);
+        return this.tokenExpiresAt > now + this.refreshBufferSeconds;
+    }
+    getTokenRemainingTime() {
+        if (!this.currentToken) {
+            return 0;
+        }
+        const now = Math.floor(Date.now() / 1000);
+        return Math.max(0, this.tokenExpiresAt - now);
+    }
+    clearToken() {
+        this.currentToken = null;
+        this.tokenExpiresAt = 0;
+    }
+    getTokenInfo() {
+        return {
+            token: this.currentToken,
+            expiresAt: this.tokenExpiresAt,
+            remainingTime: this.getTokenRemainingTime(),
+        };
+    }
+}
+exports.TokenManager = TokenManager;

package/lib/utils.js

@@ -10,21 +10,20 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.handleResponseData = handleResponseData;
-const GO_TO_AI_TEXT = "请检查调用方式，或前往云开发 AI+ 首页查看文档：https://tcb.cloud.tencent.com/dev#/ai";
 function handleResponseData(responseData, header) {
     return __awaiter(this, void 0, void 0, function* () {
         var _a, _b;
         if (typeof responseData === "object" && responseData && "then" in responseData) {
             const json = (yield responseData);
             if (typeof json === "object" && json && "code" in json && json.code !== 0) {
-                throw new Error(`小红书 ModelRequest 请求出错，错误码：${json.code}，错误信息：${json.message}\n${GO_TO_AI_TEXT}\n${JSON.stringify(json, null, 2)}`);
+                throw new Error(`ModelRequest 请求出错，错误码：${json.code}，错误信息：${json.message}\n${JSON.stringify(json, null, 2)}`);
             }
             return responseData;
         }
         if ((_b = (_a = header === null || header === void 0 ? void 0 : header.get) === null || _a === void 0 ? void 0 : _a.call(header, "content-type")) === null || _b === void 0 ? void 0 : _b.includes("application/json")) {
             const json = yield readableStream2JsonObject(responseData);
             if (typeof json === "object" && json && "code" in json && json.code !== 0) {
-                throw new Error(`小红书 ModelRequest 请求出错，错误码：${json.code}，错误信息：${json.message}\n${GO_TO_AI_TEXT}\n${JSON.stringify(json, null, 2)}`);
+                throw new Error(`ModelRequest 请求出错，错误码：${json.code}，错误信息：${json.message}\n${JSON.stringify(json, null, 2)}`);
             }
         }
         return responseData;

package/package.json

@@ -1,9 +1,8 @@
 {
   "name": "@vectorx/ai-sdk",
-  "version": "0.1.3",
+  "version": "0.3.0",
   "description": "Cloud AI SDK",
   "main": "lib/index.js",
-  "types": "types/index.d.ts",
   "sideEffects": false,
   "files": [
     "lib",
@@ -22,9 +21,10 @@
     "node": ">=18.0.0"
   },
   "dependencies": {
-    "@vectorx/ai-types": "0.1.3",
-    "@vectorx/agent-runtime": "0.1.3",
     "@mattiasbuelens/web-streams-adapter": "^0.1.0",
+    "@vectorx/ai-types": "0.3.0",
+    "langfuse": "^3.38.4",
+    "openai": "^4.103.0",
     "text-encoding-shim": "^1.0.5",
     "web-streams-polyfill": "^4.1.0",
     "zod": "^3.24.2"
@@ -36,7 +36,6 @@
     "@typescript-eslint/parser": "^7.1.0",
     "eslint": "^8.57.0",
     "jest": "^29.7.0",
-    "openai": "^4.103.0",
     "ts-node-dev": "^2.0.0",
     "typescript": "^5.3.3"
   },

package/lib/models/model-types.js

@@ -1,6 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SimpleChatModel = void 0;
-class SimpleChatModel {
-}
-exports.SimpleChatModel = SimpleChatModel;

package/types/models/default.d.ts

@@ -1,13 +0,0 @@
-import type { DoGenerateOutput, DoStreamOutput, ModelReq, ModelRequestOptions, ReqOptions } from "../types";
-import type { ModelName } from "./index";
-export declare abstract class SimpleChatModel {
-    abstract doGenerate(data: ModelRequestOptions, options?: ReqOptions): Promise<DoGenerateOutput>;
-    abstract doStream(data: ModelRequestOptions, options?: ReqOptions): Promise<DoStreamOutput>;
-}
-export declare class DefaultSimpleModel implements SimpleChatModel {
-    private req;
-    modelName: ModelName;
-    constructor(req: ModelReq, modelName: ModelName);
-    doGenerate(data: ModelRequestOptions, options?: ReqOptions): Promise<DoGenerateOutput>;
-    doStream(data: ModelRequestOptions, options?: ReqOptions): Promise<DoStreamOutput>;
-}

package/types/models/index.d.ts

@@ -1,23 +0,0 @@
-import { DefaultSimpleModel } from "./default";
-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 const modelName: {
-    "deepseek-r1": string;
-    "deepseek-v3": string;
-    "qwen-max": string;
-    "qwen-long": string;
-    "qwen-plus": string;
-    "qwen-turbo": string;
-    "qwen-vl-ocr": string;
-};
-declare const toolMap: Map<string, Function>;
-export { DefaultSimpleModel, ReactModel, toolMap };

package/types/models/model-types.d.ts

@@ -1,131 +0,0 @@
-export declare abstract class SimpleChatModel {
-    abstract doGenerate(data: ModelRequestOptions, options?: ReqOptions): Promise<DoGenerateOutput>;
-    abstract doStream(data: ModelRequestOptions, options?: ReqOptions): Promise<DoStreamOutput>;
-}
-export interface ModelRequestOptions {
-    model: string;
-    max_tokens?: number;
-    temperature?: number;
-    top_p?: number;
-    n?: number;
-    conversation_id?: string;
-    frequency_penalty?: number;
-    presence_penalty?: number;
-    stream?: boolean;
-    platform_tools?: Array<{
-        plantform_tool_id: string;
-        payload: any;
-    }>;
-    historys?: Array<{
-        role: string;
-        content: string;
-    }>;
-    knowledge_base?: Array<{
-        knowledge_base_id: string;
-        search_mode?: "vector" | "full_text" | "hybrid";
-        limit?: number;
-        score_threshold?: number;
-    }>;
-    db_base?: any[];
-    enable_thinking?: boolean;
-    enable_search?: boolean;
-}
-type RawResponse = {
-    rawResponse?: any;
-};
-export type DoGenerateOutput = BaseDoGenerateOutput & RawResponse;
-export type DoStreamOutput = AsyncIterableReadableStream<BaseDoStreamOutputChunk & RawResponse>;
-export type ChatModelConstructor = typeof SimpleChatModel;
-export type AsyncIterableReadableStream<T> = ReadableStream<T> & {
-    [Symbol.asyncIterator]: () => {
-        next(): Promise<IteratorResult<T>>;
-    };
-};
-export interface IModelReqInput {
-    url: string;
-    headers?: Record<string, string>;
-    data?: Object;
-    stream?: boolean;
-    timeout?: number;
-}
-export type ModelReq = <T extends IModelReqInput>(props: T) => T["stream"] extends true ? Promise<ReadableStream<Uint8Array>> : Promise<Object>;
-export interface IAgentReqInput {
-    url: string;
-    method: string;
-    headers?: Record<string, string>;
-    data?: Object;
-    stream?: boolean;
-    timeout?: number;
-}
-export interface ReqOptions {
-    timeout?: number;
-}
-export type AgentReq = <T extends IAgentReqInput>(props: T) => T["stream"] extends true ? Promise<ReadableStream<Uint8Array>> : Promise<Object>;
-export type UserMessage = {
-    role: "user";
-    content: string;
-};
-export type SystemMessage = {
-    role: "system";
-    content: string;
-};
-export type AssistantMessage = PlainAssistantMessage | ToolCallAssistantMessage;
-export type PlainAssistantMessage = {
-    role: "assistant";
-    content: string;
-};
-export type ToolCallAssistantMessage = {
-    role: "assistant";
-    tool_calls: Array<ToolCall>;
-    content?: string;
-};
-export type ToolMessage = {
-    role: "tool";
-    tool_call_id: string;
-    content: string;
-};
-export type ChatModelMessage = UserMessage | SystemMessage | AssistantMessage | ToolMessage;
-export type FunctionTool = {
-    name: string;
-    description: string;
-    fn: CallableFunction;
-    parameters: object;
-};
-export type ModelTool = {
-    type: string;
-    function: ModelToolFunction;
-};
-export type ModelToolFunction = {
-    name: string;
-    description: string;
-    parameters: object;
-};
-export type ToolCall = {
-    id: string;
-    type: string;
-    function: {
-        name: string;
-        arguments: string;
-    };
-};
-type FinishReason = "tool_calls" | (string & {});
-export type Usage = {
-    completion_tokens: number;
-    prompt_tokens: number;
-    total_tokens: number;
-};
-export interface BaseDoGenerateOutput {
-    choices?: Array<{
-        finish_reason?: FinishReason;
-        message?: ChatModelMessage;
-    }>;
-    usage?: Usage;
-}
-export interface BaseDoStreamOutputChunk {
-    choices?: Array<{
-        finish_reason?: FinishReason;
-        delta?: ChatModelMessage;
-    }>;
-    usage?: Usage;
-}
-export {};