@cloudbase/ai 2.8.21-beta.0 → 2.8.23-beta.0

package/dist/cjs/models/Yi/index.js

@@ -47,10 +47,14 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
     }
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.YiModel = void 0;
+exports.YiSimpleModel = void 0;
 var utils_1 = require("../../utils");
-var YiModel = (function () {
-    function YiModel(req, baseUrl, subUrl) {
+function processInput(input) {
+    var messages = input.messages, model = input.model, temperature = input.temperature, tools = input.tools, top_p = input.top_p;
+    return __assign(__assign({}, input), { messages: messages, model: model, tools: tools === null || tools === void 0 ? void 0 : tools.map(function (tool) { return (__assign(__assign({}, tool), { function: __assign(__assign({}, tool.function), { parameters: JSON.parse(tool.function.parameters) }) })); }), top_p: top_p, temperature: temperature });
+}
+var YiSimpleModel = (function () {
+    function YiSimpleModel(req, baseUrl, subUrl) {
         this.req = req;
         this.baseUrl = baseUrl;
         this.subUrl = '01-ai/v1/chat/completions';
@@ -58,108 +62,70 @@ var YiModel = (function () {
             this.subUrl = subUrl;
         }
     }
-    Object.defineProperty(YiModel.prototype, "url", {
+    Object.defineProperty(YiSimpleModel.prototype, "url", {
         get: function () {
             return "".concat(this.baseUrl, "/").concat(this.subUrl);
         },
         enumerable: false,
         configurable: true
     });
-    YiModel.prototype.generateText = function (data) {
-        var _a, _b, _c, _d;
+    YiSimpleModel.prototype.doGenerate = function (data) {
         return __awaiter(this, void 0, void 0, function () {
             var res;
-            return __generator(this, function (_e) {
-                switch (_e.label) {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
                     case 0: return [4, this.req({
                             url: this.url,
-                            data: __assign(__assign({}, data), { stream: false }),
+                            data: __assign(__assign({}, processInput(data)), { stream: false }),
                             stream: false,
                         })];
                     case 1:
-                        res = (_e.sent());
-                        return [2, {
-                                text: (_d = (_c = (_b = (_a = res === null || res === void 0 ? void 0 : res.choices) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.message) === null || _c === void 0 ? void 0 : _c.content) !== null && _d !== void 0 ? _d : '',
-                                rawResponse: res,
-                            }];
+                        res = (_a.sent());
+                        return [2, __assign(__assign({}, res), { rawResponse: res })];
                 }
             });
         });
     };
-    YiModel.prototype.streamText = function (data) {
+    YiSimpleModel.prototype.doStream = function (data) {
         return __awaiter(this, void 0, void 0, function () {
-            var stream;
+            var isToolCall, _stream, stream, arkStream, streamWithRaw;
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4, this.req({
-                            url: this.url,
-                            data: __assign(__assign({}, data), { stream: true }),
-                            stream: true,
-                        })];
+                    case 0:
+                        isToolCall = null;
+                        return [4, this.req({
+                                url: this.url,
+                                data: __assign(__assign({}, processInput(data)), { stream: true }),
+                                stream: true,
+                            })];
                     case 1:
-                        stream = _a.sent();
-                        return [2, new YiModelStreamResult(stream)];
+                        _stream = _a.sent();
+                        stream = (0, utils_1.toPolyfillReadable)(_stream);
+                        arkStream = (0, utils_1.intoStandardStream)(stream);
+                        streamWithRaw = arkStream.pipeThrough(new utils_1.TransformStream({
+                            transform: function (chunk, controller) {
+                                var _a, _b, _c, _d, _e, _f;
+                                if (!(((_c = (_b = (_a = chunk === null || chunk === void 0 ? void 0 : chunk.choices) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.delta) === null || _c === void 0 ? void 0 : _c.content) || ((_f = (_e = (_d = chunk === null || chunk === void 0 ? void 0 : chunk.choices) === null || _d === void 0 ? void 0 : _d[0]) === null || _e === void 0 ? void 0 : _e.delta) === null || _f === void 0 ? void 0 : _f.tool_calls)))
+                                    return;
+                                var newChoices = chunk.choices.map(function (choice) {
+                                    var message = Object.assign(choice.delta, { role: 'assistant' });
+                                    if (isToolCall == null)
+                                        isToolCall = (0, utils_1.isToolCallAssistantMessage)(message);
+                                    if (isToolCall) {
+                                        return __assign(__assign({}, choice), { finish_reason: 'tool_calls', delta: message });
+                                    }
+                                    return __assign(__assign({}, choice), { delta: message });
+                                });
+                                var newChunk = __assign(__assign({}, chunk), { choices: newChoices });
+                                controller.enqueue(__assign(__assign({}, newChunk), { rawResponse: chunk }));
+                            },
+                        }));
+                        return [2, (0, utils_1.createAsyncIterable)(streamWithRaw)];
                 }
             });
         });
     };
-    return YiModel;
-}());
-exports.YiModel = YiModel;
-var YiModelStreamResult = (function () {
-    function YiModelStreamResult(_stream) {
-        var stream = (0, utils_1.toPolyfillReadable)(_stream);
-        this._eventSourceStream = stream
-            .pipeThrough(new utils_1.TextDecoderStream())
-            .pipeThrough((0, utils_1.createEventSourceParserTransformStream)());
-    }
-    Object.defineProperty(YiModelStreamResult.prototype, "teeedStream", {
-        get: function () {
-            var _a = this._eventSourceStream.tee(), s1 = _a[0], s2 = _a[1];
-            this._eventSourceStream = s2;
-            return s1;
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(YiModelStreamResult.prototype, "eventSourceStream", {
-        get: function () {
-            return (0, utils_1.createAsyncIterable)(this.teeedStream);
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(YiModelStreamResult.prototype, "dataStream", {
-        get: function () {
-            return (0, utils_1.createAsyncIterable)(this.eventSourceStream.pipeThrough(new utils_1.TransformStream({
-                transform: function (chunk, controller) {
-                    try {
-                        var data = JSON.parse(chunk.data);
-                        controller.enqueue(data);
-                    }
-                    catch (e) {
-                        if (chunk.data !== '[DONE]') {
-                            console.warn('Error when transforming event source data to json', e);
-                        }
-                    }
-                },
-            })));
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(YiModelStreamResult.prototype, "textStream", {
-        get: function () {
-            return (0, utils_1.createAsyncIterable)(this.dataStream.pipeThrough(new utils_1.TransformStream({
-                transform: function (chunk, controller) {
-                    var _a, _b, _c, _d;
-                    controller.enqueue((_d = (_c = (_b = (_a = chunk === null || chunk === void 0 ? void 0 : chunk.choices) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.delta) === null || _c === void 0 ? void 0 : _c.content) !== null && _d !== void 0 ? _d : '');
-                },
-            })));
-        },
-        enumerable: false,
-        configurable: true
-    });
-    return YiModelStreamResult;
+    return YiSimpleModel;
 }());
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1lpL2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQ0EscUNBTW9CO0FBSXBCO0lBRUUsaUJBQW9CLEdBQWEsRUFBUyxPQUFlLEVBQUUsTUFBZTtRQUF0RCxRQUFHLEdBQUgsR0FBRyxDQUFVO1FBQVMsWUFBTyxHQUFQLE9BQU8sQ0FBUTtRQURsRCxXQUFNLEdBQUcsMkJBQTJCLENBQUE7UUFFekMsSUFBSSxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ2xCLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFBO1NBQ3JCO0lBQ0gsQ0FBQztJQUVELHNCQUFZLHdCQUFHO2FBQWY7WUFDRSxPQUFPLFVBQUcsSUFBSSxDQUFDLE9BQU8sY0FBSSxJQUFJLENBQUMsTUFBTSxDQUFFLENBQUE7UUFDekMsQ0FBQzs7O09BQUE7SUFFSyw4QkFBWSxHQUFsQixVQUFtQixJQUF3Qjs7Ozs7OzRCQUM1QixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzFCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsS0FBSyxHQUNkOzRCQUNELE1BQU0sRUFBRSxLQUFLO3lCQUNkLENBQUMsRUFBQTs7d0JBUEksR0FBRyxHQUFHLENBQUMsU0FPWCxDQUF5Qjt3QkFDM0IsV0FBTztnQ0FDTCxJQUFJLEVBQUUsTUFBQyxNQUFBLE1BQUEsTUFBQSxHQUFHLGFBQUgsR0FBRyx1QkFBSCxHQUFHLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsT0FBTywwQ0FBRSxPQUFrQixtQ0FBSSxFQUFFO2dDQUMzRCxXQUFXLEVBQUUsR0FBRzs2QkFDakIsRUFBQTs7OztLQUNGO0lBRUssNEJBQVUsR0FBaEIsVUFBaUIsSUFBd0I7Ozs7OzRCQUN4QixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzVCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsSUFBSSxHQUNiOzRCQUNELE1BQU0sRUFBRSxJQUFJO3lCQUNiLENBQUMsRUFBQTs7d0JBUEksTUFBTSxHQUFHLFNBT2I7d0JBQ0YsV0FBTyxJQUFJLG1CQUFtQixDQUFDLE1BQU0sQ0FBQyxFQUFBOzs7O0tBQ3ZDO0lBQ0gsY0FBQztBQUFELENBQUMsQUF0Q0QsSUFzQ0M7QUF0Q1ksMEJBQU87QUF3Q3BCO0lBR0UsNkJBQVksT0FBbUM7UUFDN0MsSUFBTSxNQUFNLEdBQUcsSUFBQSwwQkFBa0IsRUFBQyxPQUFPLENBQW1CLENBQUE7UUFDNUQsSUFBSSxDQUFDLGtCQUFrQixHQUFHLE1BQU07YUFDN0IsV0FBVyxDQUFDLElBQUkseUJBQWlCLEVBQUUsQ0FBQzthQUNwQyxXQUFXLENBQUMsSUFBQSw4Q0FBc0MsR0FBRSxDQUFDLENBQUE7SUFDMUQsQ0FBQztJQUVELHNCQUFZLDRDQUFXO2FBQXZCO1lBQ1EsSUFBQSxLQUFXLElBQUksQ0FBQyxrQkFBa0IsQ0FBQyxHQUFHLEVBQUUsRUFBdkMsRUFBRSxRQUFBLEVBQUUsRUFBRSxRQUFpQyxDQUFBO1lBQzlDLElBQUksQ0FBQyxrQkFBa0IsR0FBRyxFQUFFLENBQUE7WUFDNUIsT0FBTyxFQUFFLENBQUE7UUFDWCxDQUFDOzs7T0FBQTtJQUVELHNCQUFJLGtEQUFpQjthQUFyQjtZQUNFLE9BQU8sSUFBQSwyQkFBbUIsRUFBQyxJQUFJLENBQUMsV0FBVyxDQUFDLENBQUE7UUFDOUMsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSwyQ0FBVTthQUFkO1lBQ0UsT0FBTyxJQUFBLDJCQUFtQixFQUFDLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxXQUFXLENBQUMsSUFBSSx1QkFBZSxDQUFrQztnQkFDakgsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVO29CQUN6QixJQUFJO3dCQUNGLElBQU0sSUFBSSxHQUFHLElBQUksQ0FBQyxLQUFLLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBdUIsQ0FBQTt3QkFDekQsVUFBVSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQTtxQkFDekI7b0JBQUMsT0FBTyxDQUFDLEVBQUU7d0JBQ1YsSUFBSSxLQUFLLENBQUMsSUFBSSxLQUFLLFFBQVEsRUFBRTs0QkFDM0IsT0FBTyxDQUFDLElBQUksQ0FBQyxtREFBbUQsRUFBRSxDQUFDLENBQUMsQ0FBQTt5QkFDckU7cUJBQ0Y7Z0JBQ0gsQ0FBQzthQUNGLENBQUMsQ0FBRSxDQUFFLENBQUE7UUFDUixDQUFDOzs7T0FBQTtJQUVELHNCQUFJLDJDQUFVO2FBQWQ7WUFDRSxPQUFPLElBQUEsMkJBQW1CLEVBQUMsSUFBSSxDQUFDLFVBQVUsQ0FBQyxXQUFXLENBQUMsSUFBSSx1QkFBZSxDQUE2QjtnQkFDckcsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVOztvQkFDekIsVUFBVSxDQUFDLE9BQU8sQ0FBQyxNQUFBLE1BQUEsTUFBQSxNQUFBLEtBQUssYUFBTCxLQUFLLHVCQUFMLEtBQUssQ0FBRSxPQUFPLDBDQUFHLENBQUMsQ0FBQywwQ0FBRSxLQUFLLDBDQUFFLE9BQU8sbUNBQUksRUFBRSxDQUFDLENBQUE7Z0JBQy9ELENBQUM7YUFDRixDQUFDLENBQUUsQ0FBRSxDQUFBO1FBQ1IsQ0FBQzs7O09BQUE7SUFDSCwwQkFBQztBQUFELENBQUMsQUExQ0QsSUEwQ0MiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgeyB0eXBlIFBhcnNlZEV2ZW50IH0gZnJvbSAnLi4vLi4vZXZlbnRzb3VyY2VfcGFyc2VyJ1xuaW1wb3J0IHtcbiAgY3JlYXRlQXN5bmNJdGVyYWJsZSxcbiAgVHJhbnNmb3JtU3RyZWFtLFxuICBUZXh0RGVjb2RlclN0cmVhbSxcbiAgY3JlYXRlRXZlbnRTb3VyY2VQYXJzZXJUcmFuc2Zvcm1TdHJlYW0sXG4gIHRvUG9seWZpbGxSZWFkYWJsZSxcbn0gZnJvbSAnLi4vLi4vdXRpbHMnXG5pbXBvcnQgdHlwZSB7IFlpR2VuZXJhdGVUZXh0T3V0cHV0LCBZaVN0cmVhbVRleHRPdXRwdXQgfSBmcm9tICcuL3R5cGUnXG5pbXBvcnQgdHlwZSB7IENoYXRNb2RlbCwgU3RyZWFtVGV4dFJlc3VsdCwgTW9kZWxSZXEsIEJhc2VDaGF0TW9kZWxJbnB1dCB9IGZyb20gJy4uLy4uL3R5cGUnXG5cbmV4cG9ydCBjbGFzcyBZaU1vZGVsIGltcGxlbWVudHMgQ2hhdE1vZGVsIHtcbiAgcHVibGljIHN1YlVybCA9ICcwMS1haS92MS9jaGF0L2NvbXBsZXRpb25zJ1xuICBjb25zdHJ1Y3Rvcihwcml2YXRlIHJlcTogTW9kZWxSZXEsIHB1YmxpYyBiYXNlVXJsOiBzdHJpbmcsIHN1YlVybD86IHN0cmluZykge1xuICAgIGlmIChzdWJVcmwgIT0gbnVsbCkge1xuICAgICAgdGhpcy5zdWJVcmwgPSBzdWJVcmxcbiAgICB9XG4gIH1cblxuICBwcml2YXRlIGdldCB1cmwoKSB7XG4gICAgcmV0dXJuIGAke3RoaXMuYmFzZVVybH0vJHt0aGlzLnN1YlVybH1gXG4gIH1cblxuICBhc3luYyBnZW5lcmF0ZVRleHQoZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KSB7XG4gICAgY29uc3QgcmVzID0gKGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICB9KSkgYXMgWWlHZW5lcmF0ZVRleHRPdXRwdXRcbiAgICByZXR1cm4ge1xuICAgICAgdGV4dDogKHJlcz8uY2hvaWNlcz8uWzBdPy5tZXNzYWdlPy5jb250ZW50IGFzIHN0cmluZykgPz8gJycsXG4gICAgICByYXdSZXNwb25zZTogcmVzLFxuICAgIH1cbiAgfVxuXG4gIGFzeW5jIHN0cmVhbVRleHQoZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KSB7XG4gICAgY29uc3Qgc3RyZWFtID0gYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiB0cnVlLFxuICAgICAgfSxcbiAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICB9KVxuICAgIHJldHVybiBuZXcgWWlNb2RlbFN0cmVhbVJlc3VsdChzdHJlYW0pXG4gIH1cbn1cblxuY2xhc3MgWWlNb2RlbFN0cmVhbVJlc3VsdCBpbXBsZW1lbnRzIFN0cmVhbVRleHRSZXN1bHQge1xuICBwcml2YXRlIF9ldmVudFNvdXJjZVN0cmVhbTogUmVhZGFibGVTdHJlYW08UGFyc2VkRXZlbnQ+XG5cbiAgY29uc3RydWN0b3IoX3N0cmVhbTogUmVhZGFibGVTdHJlYW08VWludDhBcnJheT4pIHtcbiAgICBjb25zdCBzdHJlYW0gPSB0b1BvbHlmaWxsUmVhZGFibGUoX3N0cmVhbSkgYXMgdHlwZW9mIF9zdHJlYW1cbiAgICB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbSA9IHN0cmVhbVxuICAgICAgLnBpcGVUaHJvdWdoKG5ldyBUZXh0RGVjb2RlclN0cmVhbSgpKVxuICAgICAgLnBpcGVUaHJvdWdoKGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtKCkpXG4gIH1cblxuICBwcml2YXRlIGdldCB0ZWVlZFN0cmVhbSgpIHtcbiAgICBjb25zdCBbczEsIHMyXSA9IHRoaXMuX2V2ZW50U291cmNlU3RyZWFtLnRlZSgpXG4gICAgdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0gPSBzMlxuICAgIHJldHVybiBzMVxuICB9XG5cbiAgZ2V0IGV2ZW50U291cmNlU3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMudGVlZWRTdHJlYW0pXG4gIH1cblxuICBnZXQgZGF0YVN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLmV2ZW50U291cmNlU3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08UGFyc2VkRXZlbnQsIFlpU3RyZWFtVGV4dE91dHB1dD4oe1xuICAgICAgdHJhbnNmb3JtKGNodW5rLCBjb250cm9sbGVyKSB7XG4gICAgICAgIHRyeSB7XG4gICAgICAgICAgY29uc3QgZGF0YSA9IEpTT04ucGFyc2UoY2h1bmsuZGF0YSkgYXMgWWlTdHJlYW1UZXh0T3V0cHV0XG4gICAgICAgICAgY29udHJvbGxlci5lbnF1ZXVlKGRhdGEpXG4gICAgICAgIH0gY2F0Y2ggKGUpIHtcbiAgICAgICAgICBpZiAoY2h1bmsuZGF0YSAhPT0gJ1tET05FXScpIHtcbiAgICAgICAgICAgIGNvbnNvbGUud2FybignRXJyb3Igd2hlbiB0cmFuc2Zvcm1pbmcgZXZlbnQgc291cmNlIGRhdGEgdG8ganNvbicsIGUpXG4gICAgICAgICAgfVxuICAgICAgICB9XG4gICAgICB9LFxuICAgIH0pLCksKVxuICB9XG5cbiAgZ2V0IHRleHRTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy5kYXRhU3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08WWlTdHJlYW1UZXh0T3V0cHV0LCBzdHJpbmc+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICBjb250cm9sbGVyLmVucXVldWUoY2h1bms/LmNob2ljZXM/LlswXT8uZGVsdGE/LmNvbnRlbnQgPz8gJycpXG4gICAgICB9LFxuICAgIH0pLCksKVxuICB9XG59XG4iXX0=
+exports.YiSimpleModel = YiSimpleModel;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1lpL2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQUEscUNBTW9CO0FBV3BCLFNBQVMsWUFBWSxDQUFDLEtBQXlCO0lBQ3JDLElBQUEsUUFBUSxHQUF1QyxLQUFLLFNBQTVDLEVBQUUsS0FBSyxHQUFnQyxLQUFLLE1BQXJDLEVBQUUsV0FBVyxHQUFtQixLQUFLLFlBQXhCLEVBQUUsS0FBSyxHQUFZLEtBQUssTUFBakIsRUFBRSxLQUFLLEdBQUssS0FBSyxNQUFWLENBQVU7SUFFNUQsNkJBQ0ssS0FBSyxLQUNSLFFBQVEsVUFBQSxFQUNSLEtBQUssT0FBQSxFQUNMLEtBQUssRUFBRSxLQUFLLGFBQUwsS0FBSyx1QkFBTCxLQUFLLENBQUUsR0FBRyxDQUFDLFVBQUEsSUFBSSxJQUFJLE9BQUEsdUJBQ3JCLElBQUksS0FDUCxRQUFRLHdCQUFPLElBQUksQ0FBQyxRQUFRLEtBQUUsVUFBVSxFQUFFLElBQUksQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLFFBQVEsQ0FBQyxVQUFVLENBQUMsT0FDOUUsRUFId0IsQ0FHeEIsQ0FBQyxFQUNILEtBQUssT0FBQSxFQUNMLFdBQVcsYUFBQSxJQUNaO0FBQ0gsQ0FBQztBQUVEO0lBR0UsdUJBQW9CLEdBQWEsRUFBUyxPQUFlLEVBQUUsTUFBZTtRQUF0RCxRQUFHLEdBQUgsR0FBRyxDQUFVO1FBQVMsWUFBTyxHQUFQLE9BQU8sQ0FBUTtRQUZsRCxXQUFNLEdBQUcsMkJBQTJCLENBQUE7UUFHekMsSUFBSSxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ2xCLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFBO1NBQ3JCO0lBQ0gsQ0FBQztJQUVELHNCQUFZLDhCQUFHO2FBQWY7WUFDRSxPQUFPLFVBQUcsSUFBSSxDQUFDLE9BQU8sY0FBSSxJQUFJLENBQUMsTUFBTSxDQUFFLENBQUE7UUFDekMsQ0FBQzs7O09BQUE7SUFFWSxrQ0FBVSxHQUF2QixVQUF3QixJQUF3Qjs7Ozs7NEJBQ2pDLFdBQU0sSUFBSSxDQUFDLEdBQUcsQ0FBQzs0QkFDMUIsR0FBRyxFQUFFLElBQUksQ0FBQyxHQUFHOzRCQUNiLElBQUksd0JBQ0MsWUFBWSxDQUFDLElBQUksQ0FBQyxLQUNyQixNQUFNLEVBQUUsS0FBSyxHQUNkOzRCQUNELE1BQU0sRUFBRSxLQUFLO3lCQUNkLENBQUMsRUFBQTs7d0JBUEksR0FBRyxHQUFHLENBQUMsU0FPWCxDQUFxQjt3QkFDdkIsaUNBQVksR0FBRyxLQUFFLFdBQVcsRUFBRSxHQUFHLEtBQUU7Ozs7S0FDcEM7SUFFWSxnQ0FBUSxHQUFyQixVQUFzQixJQUF3Qjs7Ozs7O3dCQUN4QyxVQUFVLEdBQW1CLElBQUksQ0FBQTt3QkFDckIsV0FBTSxJQUFJLENBQUMsR0FBRyxDQUFDO2dDQUM3QixHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUc7Z0NBQ2IsSUFBSSx3QkFDQyxZQUFZLENBQUMsSUFBSSxDQUFDLEtBQ3JCLE1BQU0sRUFBRSxJQUFJLEdBQ2I7Z0NBQ0QsTUFBTSxFQUFFLElBQUk7NkJBQ2IsQ0FBQyxFQUFBOzt3QkFQSSxPQUFPLEdBQUcsU0FPZDt3QkFDSSxNQUFNLEdBQUcsSUFBQSwwQkFBa0IsRUFBQyxPQUFPLENBQW1CLENBQUE7d0JBRXRELFNBQVMsR0FBRyxJQUFBLDBCQUFrQixFQUFpQixNQUFNLENBQUMsQ0FBQTt3QkFDdEQsYUFBYSxHQUFHLFNBQVMsQ0FBQyxXQUFXLENBQUMsSUFBSSx1QkFBZSxDQUFrRTs0QkFDL0gsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVOztnQ0FFekIsSUFBSSxDQUFDLENBQUMsQ0FBQSxNQUFBLE1BQUEsTUFBQSxLQUFLLGFBQUwsS0FBSyx1QkFBTCxLQUFLLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsS0FBSywwQ0FBRSxPQUFPLE1BQUksTUFBQSxNQUFBLE1BQUEsS0FBSyxhQUFMLEtBQUssdUJBQUwsS0FBSyxDQUFFLE9BQU8sMENBQUcsQ0FBQyxDQUFDLDBDQUFFLEtBQUssMENBQUUsVUFBVSxDQUFBLENBQUM7b0NBQUUsT0FBTTtnQ0FFNUYsSUFBTSxVQUFVLEdBQUcsS0FBSyxDQUFDLE9BQU8sQ0FBQyxHQUFHLENBQUMsVUFBQyxNQUFNO29DQUMxQyxJQUFNLE9BQU8sR0FBRyxNQUFNLENBQUMsTUFBTSxDQUFDLE1BQU0sQ0FBQyxLQUFLLEVBQUUsRUFBRSxJQUFJLEVBQUUsV0FBVyxFQUFFLENBRWhFLENBQUE7b0NBQ0QsSUFBSSxVQUFVLElBQUksSUFBSTt3Q0FBRSxVQUFVLEdBQUcsSUFBQSxrQ0FBMEIsRUFBQyxPQUFPLENBQUMsQ0FBQTtvQ0FDeEUsSUFBSSxVQUFVLEVBQUU7d0NBQ2QsNkJBQ0ssTUFBTSxLQUNULGFBQWEsRUFBRSxZQUFxQixFQUNwQyxLQUFLLEVBQUUsT0FBTyxJQUNmO3FDQUNGO29DQUNELDZCQUFZLE1BQU0sS0FBRSxLQUFLLEVBQUUsT0FBTyxJQUFFO2dDQUN0QyxDQUFDLENBQUMsQ0FBQTtnQ0FDRixJQUFNLFFBQVEseUJBQVEsS0FBSyxLQUFFLE9BQU8sRUFBRSxVQUFVLEdBQUUsQ0FBQTtnQ0FDbEQsVUFBVSxDQUFDLE9BQU8sdUJBQU0sUUFBUSxLQUFFLFdBQVcsRUFBRSxLQUFLLElBQUcsQ0FBQTs0QkFDekQsQ0FBQzt5QkFDRixDQUFDLENBQUUsQ0FBQTt3QkFFSixXQUFPLElBQUEsMkJBQW1CLEVBQUMsYUFBYSxDQUFDLEVBQUE7Ozs7S0FDMUM7SUFDSCxvQkFBQztBQUFELENBQUMsQUFoRUQsSUFnRUM7QUFoRVksc0NBQWEiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQge1xuICBjcmVhdGVBc3luY0l0ZXJhYmxlLFxuICBUcmFuc2Zvcm1TdHJlYW0sXG4gIHRvUG9seWZpbGxSZWFkYWJsZSxcbiAgaW50b1N0YW5kYXJkU3RyZWFtLFxuICBpc1Rvb2xDYWxsQXNzaXN0YW50TWVzc2FnZSxcbn0gZnJvbSAnLi4vLi4vdXRpbHMnXG5pbXBvcnQgdHlwZSB7IFlpSW5wdXQsIFlpR2VuZXJhdGVPdXRwdXQsIFlpU3RyZWFtT3V0cHV0IH0gZnJvbSAnLi90eXBlJ1xuaW1wb3J0IHR5cGUge1xuICBNb2RlbFJlcSxcbiAgQmFzZUNoYXRNb2RlbElucHV0LFxuICBEb0dlbmVyYXRlT3V0cHV0LFxuICBEb1N0cmVhbU91dHB1dCxcbiAgQmFzZURvU3RyZWFtT3V0cHV0Q2h1bmssXG4gIFNpbXBsZUNoYXRNb2RlbCxcbn0gZnJvbSAnLi4vLi4vdHlwZSdcblxuZnVuY3Rpb24gcHJvY2Vzc0lucHV0KGlucHV0OiBCYXNlQ2hhdE1vZGVsSW5wdXQpOiBZaUlucHV0IHtcbiAgY29uc3QgeyBtZXNzYWdlcywgbW9kZWwsIHRlbXBlcmF0dXJlLCB0b29scywgdG9wX3AgfSA9IGlucHV0XG5cbiAgcmV0dXJuIHtcbiAgICAuLi5pbnB1dCxcbiAgICBtZXNzYWdlcyxcbiAgICBtb2RlbCxcbiAgICB0b29sczogdG9vbHM/Lm1hcCh0b29sID0+ICh7XG4gICAgICAuLi50b29sLFxuICAgICAgZnVuY3Rpb246IHsgLi4udG9vbC5mdW5jdGlvbiwgcGFyYW1ldGVyczogSlNPTi5wYXJzZSh0b29sLmZ1bmN0aW9uLnBhcmFtZXRlcnMpIH0sXG4gICAgfSkpLFxuICAgIHRvcF9wLFxuICAgIHRlbXBlcmF0dXJlLFxuICB9XG59XG5cbmV4cG9ydCBjbGFzcyBZaVNpbXBsZU1vZGVsIGltcGxlbWVudHMgU2ltcGxlQ2hhdE1vZGVsIHtcbiAgcHVibGljIHN1YlVybCA9ICcwMS1haS92MS9jaGF0L2NvbXBsZXRpb25zJ1xuXG4gIGNvbnN0cnVjdG9yKHByaXZhdGUgcmVxOiBNb2RlbFJlcSwgcHVibGljIGJhc2VVcmw6IHN0cmluZywgc3ViVXJsPzogc3RyaW5nKSB7XG4gICAgaWYgKHN1YlVybCAhPSBudWxsKSB7XG4gICAgICB0aGlzLnN1YlVybCA9IHN1YlVybFxuICAgIH1cbiAgfVxuXG4gIHByaXZhdGUgZ2V0IHVybCgpIHtcbiAgICByZXR1cm4gYCR7dGhpcy5iYXNlVXJsfS8ke3RoaXMuc3ViVXJsfWBcbiAgfVxuXG4gIHB1YmxpYyBhc3luYyBkb0dlbmVyYXRlKGRhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCk6IFByb21pc2U8RG9HZW5lcmF0ZU91dHB1dD4ge1xuICAgIGNvbnN0IHJlcyA9IChhd2FpdCB0aGlzLnJlcSh7XG4gICAgICB1cmw6IHRoaXMudXJsLFxuICAgICAgZGF0YToge1xuICAgICAgICAuLi5wcm9jZXNzSW5wdXQoZGF0YSksXG4gICAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICB9KSkgYXMgWWlHZW5lcmF0ZU91dHB1dFxuICAgIHJldHVybiB7IC4uLnJlcywgcmF3UmVzcG9uc2U6IHJlcyB9XG4gIH1cblxuICBwdWJsaWMgYXN5bmMgZG9TdHJlYW0oZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KTogUHJvbWlzZTxEb1N0cmVhbU91dHB1dD4ge1xuICAgIGxldCBpc1Rvb2xDYWxsOiBudWxsIHwgYm9vbGVhbiA9IG51bGxcbiAgICBjb25zdCBfc3RyZWFtID0gYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4ucHJvY2Vzc0lucHV0KGRhdGEpLFxuICAgICAgICBzdHJlYW06IHRydWUsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiB0cnVlLFxuICAgIH0pXG4gICAgY29uc3Qgc3RyZWFtID0gdG9Qb2x5ZmlsbFJlYWRhYmxlKF9zdHJlYW0pIGFzIHR5cGVvZiBfc3RyZWFtXG5cbiAgICBjb25zdCBhcmtTdHJlYW0gPSBpbnRvU3RhbmRhcmRTdHJlYW08WWlTdHJlYW1PdXRwdXQ+KHN0cmVhbSlcbiAgICBjb25zdCBzdHJlYW1XaXRoUmF3ID0gYXJrU3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08WWlTdHJlYW1PdXRwdXQsIEJhc2VEb1N0cmVhbU91dHB1dENodW5rICYgeyByYXdSZXNwb25zZT86IGFueSB9Pih7XG4gICAgICB0cmFuc2Zvcm0oY2h1bmssIGNvbnRyb2xsZXIpIHtcbiAgICAgICAgLy8g6Z2e5bi45aWH6JGpXG4gICAgICAgIGlmICghKGNodW5rPy5jaG9pY2VzPy5bMF0/LmRlbHRhPy5jb250ZW50IHx8IGNodW5rPy5jaG9pY2VzPy5bMF0/LmRlbHRhPy50b29sX2NhbGxzKSkgcmV0dXJuXG5cbiAgICAgICAgY29uc3QgbmV3Q2hvaWNlcyA9IGNodW5rLmNob2ljZXMubWFwKChjaG9pY2UpID0+IHtcbiAgICAgICAgICBjb25zdCBtZXNzYWdlID0gT2JqZWN0LmFzc2lnbihjaG9pY2UuZGVsdGEsIHsgcm9sZTogJ2Fzc2lzdGFudCcgfSkgYXMgdHlwZW9mIGNob2ljZS5kZWx0YSAmIHtcbiAgICAgICAgICAgIHJvbGU6ICdhc3Npc3RhbnQnXG4gICAgICAgICAgfVxuICAgICAgICAgIGlmIChpc1Rvb2xDYWxsID09IG51bGwpIGlzVG9vbENhbGwgPSBpc1Rvb2xDYWxsQXNzaXN0YW50TWVzc2FnZShtZXNzYWdlKVxuICAgICAgICAgIGlmIChpc1Rvb2xDYWxsKSB7XG4gICAgICAgICAgICByZXR1cm4ge1xuICAgICAgICAgICAgICAuLi5jaG9pY2UsXG4gICAgICAgICAgICAgIGZpbmlzaF9yZWFzb246ICd0b29sX2NhbGxzJyBhcyBjb25zdCxcbiAgICAgICAgICAgICAgZGVsdGE6IG1lc3NhZ2UsXG4gICAgICAgICAgICB9XG4gICAgICAgICAgfVxuICAgICAgICAgIHJldHVybiB7IC4uLmNob2ljZSwgZGVsdGE6IG1lc3NhZ2UgfVxuICAgICAgICB9KVxuICAgICAgICBjb25zdCBuZXdDaHVuayA9IHsgLi4uY2h1bmssIGNob2ljZXM6IG5ld0Nob2ljZXMgfVxuICAgICAgICBjb250cm9sbGVyLmVucXVldWUoeyAuLi5uZXdDaHVuaywgcmF3UmVzcG9uc2U6IGNodW5rIH0pXG4gICAgICB9LFxuICAgIH0pLClcblxuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHN0cmVhbVdpdGhSYXcpXG4gIH1cbn1cbiJdfQ==

package/dist/cjs/models/Yi/type.d.ts

@@ -9,6 +9,10 @@ type UserMessage = {
 type AssistantMessage = {
     role: 'assistant';
     content: string;
+} | {
+    role: 'assistant';
+    tool_calls: Array<ToolCall>;
+    content?: string;
 };
 type ToolMessage = {
     role: 'tool';
@@ -16,34 +20,56 @@ type ToolMessage = {
     tool_call_id: string;
 };
 type YiMessage = SystemMessage | UserMessage | AssistantMessage | ToolMessage;
-export type YiInputData = {
+export type YiInput = {
     model: string;
     messages: Array<YiMessage>;
+    temperature?: number;
+    top_p?: number;
     tools?: Array<{
-        type: 'function';
+        type: string;
         function: {
             name: string;
-            description?: string;
+            description: string;
             parameters: object;
         };
     }>;
-    tool_choice?: string | object;
-    max_tokens?: number | null;
-    top_p?: number;
-    temperature?: number;
 };
-export type YiGenerateTextOutput = {
-    choices?: Array<{
-        message?: {
-            content?: string;
+type ToolCall = {
+    id: string;
+    type: string;
+    function: {
+        name: string;
+        arguments: string;
+    };
+};
+export type YiGenerateOutput = {
+    choices: Array<{
+        finish_reason: string;
+        message: {
+            content: string;
+            role: 'assistant';
+            tool_calls?: Array<ToolCall>;
         };
     }>;
+    usage: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
 };
-export type YiStreamTextOutput = {
-    choices?: Array<{
-        delta?: {
-            content?: string;
+export type YiStreamOutput = {
+    choices: Array<{
+        delta: {
+            content: string;
+            tool_calls?: Array<ToolCall>;
+            role?: 'assistant';
         };
+        finish_reason: string;
     }>;
+    usage?: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
 };
 export {};

package/dist/cjs/models/Yi/type.js

@@ -1,3 +1,3 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvWWkvdHlwZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiIiwic291cmNlc0NvbnRlbnQiOlsidHlwZSBTeXN0ZW1NZXNzYWdlID0ge1xuICByb2xlOiAnc3lzdGVtJ1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBVc2VyTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3VzZXInXG4gIGNvbnRlbnQ6IHN0cmluZ1xufVxuXG50eXBlIEFzc2lzdGFudE1lc3NhZ2UgPSB7XG4gIHJvbGU6ICdhc3Npc3RhbnQnXG4gIGNvbnRlbnQ6IHN0cmluZ1xufVxuXG50eXBlIFRvb2xNZXNzYWdlID0ge1xuICByb2xlOiAndG9vbCdcbiAgY29udGVudDogc3RyaW5nXG4gIHRvb2xfY2FsbF9pZDogc3RyaW5nXG59XG5cbnR5cGUgWWlNZXNzYWdlID0gU3lzdGVtTWVzc2FnZSB8IFVzZXJNZXNzYWdlIHwgQXNzaXN0YW50TWVzc2FnZSB8IFRvb2xNZXNzYWdlXG5cbmV4cG9ydCB0eXBlIFlpSW5wdXREYXRhID0ge1xuICBtb2RlbDogc3RyaW5nXG4gIG1lc3NhZ2VzOiBBcnJheTxZaU1lc3NhZ2U+XG4gIHRvb2xzPzogQXJyYXk8e1xuICAgIHR5cGU6ICdmdW5jdGlvbidcbiAgICBmdW5jdGlvbjogeyBuYW1lOiBzdHJpbmc7IGRlc2NyaXB0aW9uPzogc3RyaW5nOyBwYXJhbWV0ZXJzOiBvYmplY3QgfVxuICB9PlxuICB0b29sX2Nob2ljZT86IHN0cmluZyB8IG9iamVjdFxuICBtYXhfdG9rZW5zPzogbnVtYmVyIHwgbnVsbFxuICB0b3BfcD86IG51bWJlclxuICB0ZW1wZXJhdHVyZT86IG51bWJlclxufVxuXG4vLyB0b2RvOiBtb3JlIHByZWNpc2VcbmV4cG9ydCB0eXBlIFlpR2VuZXJhdGVUZXh0T3V0cHV0ID0ge1xuICBjaG9pY2VzPzogQXJyYXk8e1xuICAgIG1lc3NhZ2U/OiB7IGNvbnRlbnQ/OiBzdHJpbmcgfVxuICB9PlxufVxuXG4vLyB0b2RvOiBtb3JlIHByZWNpc2VcbmV4cG9ydCB0eXBlIFlpU3RyZWFtVGV4dE91dHB1dCA9IHtcbiAgY2hvaWNlcz86IEFycmF5PHtcbiAgICBkZWx0YT86IHsgY29udGVudD86IHN0cmluZyB9XG4gIH0+XG59XG4iXX0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvWWkvdHlwZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiIiwic291cmNlc0NvbnRlbnQiOlsidHlwZSBTeXN0ZW1NZXNzYWdlID0ge1xuICByb2xlOiAnc3lzdGVtJ1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBVc2VyTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3VzZXInXG4gIGNvbnRlbnQ6IHN0cmluZ1xufVxuXG50eXBlIEFzc2lzdGFudE1lc3NhZ2UgPVxuICB8IHtcbiAgICByb2xlOiAnYXNzaXN0YW50J1xuICAgIGNvbnRlbnQ6IHN0cmluZ1xuICB9XG4gIHwge1xuICAgIHJvbGU6ICdhc3Npc3RhbnQnXG4gICAgdG9vbF9jYWxsczogQXJyYXk8VG9vbENhbGw+XG4gICAgY29udGVudD86IHN0cmluZ1xuICB9XG5cbnR5cGUgVG9vbE1lc3NhZ2UgPSB7XG4gIHJvbGU6ICd0b29sJ1xuICBjb250ZW50OiBzdHJpbmdcbiAgdG9vbF9jYWxsX2lkOiBzdHJpbmdcbn1cblxudHlwZSBZaU1lc3NhZ2UgPSBTeXN0ZW1NZXNzYWdlIHwgVXNlck1lc3NhZ2UgfCBBc3Npc3RhbnRNZXNzYWdlIHwgVG9vbE1lc3NhZ2VcblxuZXhwb3J0IHR5cGUgWWlJbnB1dCA9IHtcbiAgbW9kZWw6IHN0cmluZ1xuICBtZXNzYWdlczogQXJyYXk8WWlNZXNzYWdlPlxuICB0ZW1wZXJhdHVyZT86IG51bWJlclxuICB0b3BfcD86IG51bWJlclxuICB0b29scz86IEFycmF5PHtcbiAgICB0eXBlOiBzdHJpbmdcbiAgICBmdW5jdGlvbjogeyBuYW1lOiBzdHJpbmc7IGRlc2NyaXB0aW9uOiBzdHJpbmc7IHBhcmFtZXRlcnM6IG9iamVjdCB9XG4gIH0+XG59XG5cbnR5cGUgVG9vbENhbGwgPSB7XG4gIGlkOiBzdHJpbmdcbiAgdHlwZTogc3RyaW5nXG4gIGZ1bmN0aW9uOiB7IG5hbWU6IHN0cmluZzsgYXJndW1lbnRzOiBzdHJpbmcgfVxufVxuXG5leHBvcnQgdHlwZSBZaUdlbmVyYXRlT3V0cHV0ID0ge1xuICBjaG9pY2VzOiBBcnJheTx7XG4gICAgZmluaXNoX3JlYXNvbjogc3RyaW5nXG4gICAgbWVzc2FnZToge1xuICAgICAgY29udGVudDogc3RyaW5nXG4gICAgICByb2xlOiAnYXNzaXN0YW50J1xuICAgICAgdG9vbF9jYWxscz86IEFycmF5PFRvb2xDYWxsPlxuICAgIH1cbiAgfT5cbiAgdXNhZ2U6IHtcbiAgICBwcm9tcHRfdG9rZW5zOiBudW1iZXJcbiAgICBjb21wbGV0aW9uX3Rva2VuczogbnVtYmVyXG4gICAgdG90YWxfdG9rZW5zOiBudW1iZXJcbiAgfVxufVxuXG5leHBvcnQgdHlwZSBZaVN0cmVhbU91dHB1dCA9IHtcbiAgY2hvaWNlczogQXJyYXk8e1xuICAgIGRlbHRhOiB7XG4gICAgICBjb250ZW50OiBzdHJpbmdcbiAgICAgIHRvb2xfY2FsbHM/OiBBcnJheTxUb29sQ2FsbD5cbiAgICAgIHJvbGU/OiAnYXNzaXN0YW50J1xuICAgIH1cbiAgICBmaW5pc2hfcmVhc29uOiBzdHJpbmdcbiAgfT5cbiAgdXNhZ2U/OiB7XG4gICAgcHJvbXB0X3Rva2VuczogbnVtYmVyXG4gICAgY29tcGxldGlvbl90b2tlbnM6IG51bWJlclxuICAgIHRvdGFsX3Rva2VuczogbnVtYmVyXG4gIH1cbn1cbiJdfQ==

package/dist/cjs/models/ZhiPu/index.d.ts

@@ -1,24 +1,10 @@
-import { type ParsedEvent } from '../../eventsource_parser';
-import type { ZhiPuGenerateTextOutput, ZhiPuStreamTextOutput } from './type';
-import type { ChatModel, StreamTextResult, ModelReq, BaseChatModelInput } from '../../type';
-export declare class ZhiPuModel implements ChatModel {
+import type { ModelReq, BaseChatModelInput, SimpleChatModel, DoStreamOutput, DoGenerateOutput } from '../../type';
+export declare class ZhiPuSimpleModel implements SimpleChatModel {
     private req;
     baseUrl: string;
     subUrl: string;
     constructor(req: ModelReq, baseUrl: string, subUrl?: string);
     private get url();
-    generateText(data: BaseChatModelInput): Promise<{
-        text: string;
-        rawResponse: ZhiPuGenerateTextOutput;
-    }>;
-    streamText(data: BaseChatModelInput): Promise<ZhiPuModelStreamResult>;
+    doGenerate(_data: BaseChatModelInput): Promise<DoGenerateOutput>;
+    doStream(_data: BaseChatModelInput): Promise<DoStreamOutput>;
 }
-declare class ZhiPuModelStreamResult implements StreamTextResult {
-    private _eventSourceStream;
-    constructor(_stream: ReadableStream<Uint8Array>);
-    private get teeedStream();
-    get eventSourceStream(): import("../../type").AsyncIterableReadableStream<ParsedEvent>;
-    get dataStream(): import("../../type").AsyncIterableReadableStream<ZhiPuStreamTextOutput>;
-    get textStream(): import("../../type").AsyncIterableReadableStream<string>;
-}
-export {};

package/dist/cjs/models/ZhiPu/index.js

@@ -47,10 +47,20 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
     }
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ZhiPuModel = void 0;
+exports.ZhiPuSimpleModel = void 0;
 var utils_1 = require("../../utils");
-var ZhiPuModel = (function () {
-    function ZhiPuModel(req, baseUrl, subUrl) {
+function processInput(input) {
+    var messages = input.messages, model = input.model, temperature = input.temperature, tool_choice = input.tool_choice, tools = input.tools, top_p = input.top_p;
+    var processToolChoice = function () {
+        if (tool_choice && tool_choice !== 'auto') {
+            console.warn('`tool_choice` is not \'auto\'');
+        }
+        return tool_choice;
+    };
+    return __assign(__assign({}, input), { messages: messages, model: model, temperature: temperature, tool_choice: processToolChoice(), tools: tools === null || tools === void 0 ? void 0 : tools.map(function (tool) { return (__assign(__assign({}, tool), { function: __assign(__assign({}, tool.function), { parameters: JSON.parse(tool.function.parameters) }) })); }), top_p: top_p });
+}
+var ZhiPuSimpleModel = (function () {
+    function ZhiPuSimpleModel(req, baseUrl, subUrl) {
         this.req = req;
         this.baseUrl = baseUrl;
         this.subUrl = 'zhipu/api/paas/v4/chat/completions';
@@ -58,108 +68,70 @@ var ZhiPuModel = (function () {
             this.subUrl = subUrl;
         }
     }
-    Object.defineProperty(ZhiPuModel.prototype, "url", {
+    Object.defineProperty(ZhiPuSimpleModel.prototype, "url", {
         get: function () {
             return "".concat(this.baseUrl, "/").concat(this.subUrl);
         },
         enumerable: false,
         configurable: true
     });
-    ZhiPuModel.prototype.generateText = function (data) {
-        var _a, _b, _c, _d;
+    ZhiPuSimpleModel.prototype.doGenerate = function (_data) {
         return __awaiter(this, void 0, void 0, function () {
-            var res;
-            return __generator(this, function (_e) {
-                switch (_e.label) {
-                    case 0: return [4, this.req({
-                            url: this.url,
-                            data: __assign(__assign({}, data), { stream: false }),
-                            stream: false,
-                        })];
+            var data, res;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        data = processInput(_data);
+                        return [4, this.req({
+                                url: this.url,
+                                data: __assign(__assign({}, data), { stream: false }),
+                                stream: false,
+                            })];
                     case 1:
-                        res = (_e.sent());
-                        return [2, {
-                                text: (_d = (_c = (_b = (_a = res === null || res === void 0 ? void 0 : res.choices) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.message) === null || _c === void 0 ? void 0 : _c.content) !== null && _d !== void 0 ? _d : '',
-                                rawResponse: res,
-                            }];
+                        res = (_a.sent());
+                        return [2, __assign(__assign({}, res), { rawResponse: res })];
                 }
             });
         });
     };
-    ZhiPuModel.prototype.streamText = function (data) {
+    ZhiPuSimpleModel.prototype.doStream = function (_data) {
         return __awaiter(this, void 0, void 0, function () {
-            var stream;
+            var data, isToolCall, _stream, stream, zhipuStream, streamWithRaw;
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4, this.req({
-                            url: this.url,
-                            data: __assign(__assign({}, data), { stream: true }),
-                            stream: true,
-                        })];
+                    case 0:
+                        data = processInput(_data);
+                        isToolCall = null;
+                        return [4, this.req({
+                                url: this.url,
+                                data: __assign(__assign({}, data), { stream: true }),
+                                stream: true,
+                            })];
                     case 1:
-                        stream = _a.sent();
-                        return [2, new ZhiPuModelStreamResult(stream)];
+                        _stream = _a.sent();
+                        stream = (0, utils_1.toPolyfillReadable)(_stream);
+                        zhipuStream = (0, utils_1.intoStandardStream)(stream);
+                        streamWithRaw = zhipuStream.pipeThrough(new utils_1.TransformStream({
+                            transform: function (chunk, controller) {
+                                var newChoices = chunk.choices.map(function (choice) {
+                                    var message = choice.delta;
+                                    if (isToolCall == null)
+                                        isToolCall = (0, utils_1.isToolCallAssistantMessage)(message);
+                                    if (isToolCall) {
+                                        return __assign(__assign({}, choice), { finish_reason: 'tool_calls', delta: message });
+                                    }
+                                    return choice;
+                                });
+                                var newChunk = __assign(__assign({}, chunk), { choices: newChoices });
+                                controller.enqueue(__assign(__assign({}, newChunk), { rawResponse: chunk }));
+                            },
+                        }));
+                        return [2, (0, utils_1.createAsyncIterable)(streamWithRaw)];
                 }
             });
         });
     };
-    return ZhiPuModel;
-}());
-exports.ZhiPuModel = ZhiPuModel;
-var ZhiPuModelStreamResult = (function () {
-    function ZhiPuModelStreamResult(_stream) {
-        var stream = (0, utils_1.toPolyfillReadable)(_stream);
-        this._eventSourceStream = stream
-            .pipeThrough(new utils_1.TextDecoderStream())
-            .pipeThrough((0, utils_1.createEventSourceParserTransformStream)());
-    }
-    Object.defineProperty(ZhiPuModelStreamResult.prototype, "teeedStream", {
-        get: function () {
-            var _a = this._eventSourceStream.tee(), s1 = _a[0], s2 = _a[1];
-            this._eventSourceStream = s2;
-            return s1;
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(ZhiPuModelStreamResult.prototype, "eventSourceStream", {
-        get: function () {
-            return (0, utils_1.createAsyncIterable)(this.teeedStream);
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(ZhiPuModelStreamResult.prototype, "dataStream", {
-        get: function () {
-            return (0, utils_1.createAsyncIterable)(this.eventSourceStream.pipeThrough(new utils_1.TransformStream({
-                transform: function (chunk, controller) {
-                    try {
-                        var data = JSON.parse(chunk.data);
-                        controller.enqueue(data);
-                    }
-                    catch (e) {
-                        if (chunk.data !== '[DONE]') {
-                            console.warn('Error when transforming event source data to json', e);
-                        }
-                    }
-                },
-            })));
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(ZhiPuModelStreamResult.prototype, "textStream", {
-        get: function () {
-            return (0, utils_1.createAsyncIterable)(this.dataStream.pipeThrough(new utils_1.TransformStream({
-                transform: function (chunk, controller) {
-                    var _a, _b, _c, _d;
-                    controller.enqueue((_d = (_c = (_b = (_a = chunk === null || chunk === void 0 ? void 0 : chunk.choices) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.delta) === null || _c === void 0 ? void 0 : _c.content) !== null && _d !== void 0 ? _d : '');
-                },
-            })));
-        },
-        enumerable: false,
-        configurable: true
-    });
-    return ZhiPuModelStreamResult;
+    return ZhiPuSimpleModel;
 }());
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1poaVB1L2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQ0EscUNBTW9CO0FBSXBCO0lBRUUsb0JBQW9CLEdBQWEsRUFBUyxPQUFlLEVBQUUsTUFBZTtRQUF0RCxRQUFHLEdBQUgsR0FBRyxDQUFVO1FBQVMsWUFBTyxHQUFQLE9BQU8sQ0FBUTtRQURsRCxXQUFNLEdBQUcsb0NBQW9DLENBQUE7UUFFbEQsSUFBSSxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ2xCLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFBO1NBQ3JCO0lBQ0gsQ0FBQztJQUVELHNCQUFZLDJCQUFHO2FBQWY7WUFDRSxPQUFPLFVBQUcsSUFBSSxDQUFDLE9BQU8sY0FBSSxJQUFJLENBQUMsTUFBTSxDQUFFLENBQUE7UUFDekMsQ0FBQzs7O09BQUE7SUFFSyxpQ0FBWSxHQUFsQixVQUFtQixJQUF3Qjs7Ozs7OzRCQUM1QixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzFCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsS0FBSyxHQUNkOzRCQUNELE1BQU0sRUFBRSxLQUFLO3lCQUNkLENBQUMsRUFBQTs7d0JBUEksR0FBRyxHQUFHLENBQUMsU0FPWCxDQUE0Qjt3QkFDOUIsV0FBTztnQ0FDTCxJQUFJLEVBQUUsTUFBQyxNQUFBLE1BQUEsTUFBQSxHQUFHLGFBQUgsR0FBRyx1QkFBSCxHQUFHLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsT0FBTywwQ0FBRSxPQUFrQixtQ0FBSSxFQUFFO2dDQUMzRCxXQUFXLEVBQUUsR0FBRzs2QkFDakIsRUFBQTs7OztLQUNGO0lBRUssK0JBQVUsR0FBaEIsVUFBaUIsSUFBd0I7Ozs7OzRCQUN4QixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzVCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsSUFBSSxHQUNiOzRCQUNELE1BQU0sRUFBRSxJQUFJO3lCQUNiLENBQUMsRUFBQTs7d0JBUEksTUFBTSxHQUFHLFNBT2I7d0JBQ0YsV0FBTyxJQUFJLHNCQUFzQixDQUFDLE1BQU0sQ0FBQyxFQUFBOzs7O0tBQzFDO0lBQ0gsaUJBQUM7QUFBRCxDQUFDLEFBdENELElBc0NDO0FBdENZLGdDQUFVO0FBd0N2QjtJQUdFLGdDQUFZLE9BQW1DO1FBQzdDLElBQU0sTUFBTSxHQUFHLElBQUEsMEJBQWtCLEVBQUMsT0FBTyxDQUFtQixDQUFBO1FBQzVELElBQUksQ0FBQyxrQkFBa0IsR0FBRyxNQUFNO2FBQzdCLFdBQVcsQ0FBQyxJQUFJLHlCQUFpQixFQUFFLENBQUM7YUFDcEMsV0FBVyxDQUFDLElBQUEsOENBQXNDLEdBQUUsQ0FBQyxDQUFBO0lBQzFELENBQUM7SUFFRCxzQkFBWSwrQ0FBVzthQUF2QjtZQUNRLElBQUEsS0FBVyxJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxFQUFFLEVBQXZDLEVBQUUsUUFBQSxFQUFFLEVBQUUsUUFBaUMsQ0FBQTtZQUM5QyxJQUFJLENBQUMsa0JBQWtCLEdBQUcsRUFBRSxDQUFBO1lBQzVCLE9BQU8sRUFBRSxDQUFBO1FBQ1gsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSxxREFBaUI7YUFBckI7WUFDRSxPQUFPLElBQUEsMkJBQW1CLEVBQUMsSUFBSSxDQUFDLFdBQVcsQ0FBQyxDQUFBO1FBQzlDLENBQUM7OztPQUFBO0lBRUQsc0JBQUksOENBQVU7YUFBZDtZQUNFLE9BQU8sSUFBQSwyQkFBbUIsRUFBQyxJQUFJLENBQUMsaUJBQWlCLENBQUMsV0FBVyxDQUFDLElBQUksdUJBQWUsQ0FBcUM7Z0JBQ3BILFNBQVMsWUFBQyxLQUFLLEVBQUUsVUFBVTtvQkFDekIsSUFBSTt3QkFDRixJQUFNLElBQUksR0FBRyxJQUFJLENBQUMsS0FBSyxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQTBCLENBQUE7d0JBQzVELFVBQVUsQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUE7cUJBQ3pCO29CQUFDLE9BQU8sQ0FBQyxFQUFFO3dCQUNWLElBQUksS0FBSyxDQUFDLElBQUksS0FBSyxRQUFRLEVBQUU7NEJBQzNCLE9BQU8sQ0FBQyxJQUFJLENBQUMsbURBQW1ELEVBQUUsQ0FBQyxDQUFDLENBQUE7eUJBQ3JFO3FCQUNGO2dCQUNILENBQUM7YUFDRixDQUFDLENBQUUsQ0FBRSxDQUFBO1FBQ1IsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSw4Q0FBVTthQUFkO1lBQ0UsT0FBTyxJQUFBLDJCQUFtQixFQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsV0FBVyxDQUFDLElBQUksdUJBQWUsQ0FBZ0M7Z0JBQ3hHLFNBQVMsWUFBQyxLQUFLLEVBQUUsVUFBVTs7b0JBQ3pCLFVBQVUsQ0FBQyxPQUFPLENBQUMsTUFBQSxNQUFBLE1BQUEsTUFBQSxLQUFLLGFBQUwsS0FBSyx1QkFBTCxLQUFLLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsS0FBSywwQ0FBRSxPQUFPLG1DQUFJLEVBQUUsQ0FBQyxDQUFBO2dCQUMvRCxDQUFDO2FBQ0YsQ0FBQyxDQUFFLENBQUUsQ0FBQTtRQUNSLENBQUM7OztPQUFBO0lBQ0gsNkJBQUM7QUFBRCxDQUFDLEFBMUNELElBMENDIiwic291cmNlc0NvbnRlbnQiOlsiaW1wb3J0IHsgdHlwZSBQYXJzZWRFdmVudCB9IGZyb20gJy4uLy4uL2V2ZW50c291cmNlX3BhcnNlcidcbmltcG9ydCB7XG4gIGNyZWF0ZUFzeW5jSXRlcmFibGUsXG4gIFRyYW5zZm9ybVN0cmVhbSxcbiAgVGV4dERlY29kZXJTdHJlYW0sXG4gIGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtLFxuICB0b1BvbHlmaWxsUmVhZGFibGUsXG59IGZyb20gJy4uLy4uL3V0aWxzJ1xuaW1wb3J0IHR5cGUgeyBaaGlQdUdlbmVyYXRlVGV4dE91dHB1dCwgWmhpUHVTdHJlYW1UZXh0T3V0cHV0IH0gZnJvbSAnLi90eXBlJ1xuaW1wb3J0IHR5cGUgeyBDaGF0TW9kZWwsIFN0cmVhbVRleHRSZXN1bHQsIE1vZGVsUmVxLCBCYXNlQ2hhdE1vZGVsSW5wdXQgfSBmcm9tICcuLi8uLi90eXBlJ1xuXG5leHBvcnQgY2xhc3MgWmhpUHVNb2RlbCBpbXBsZW1lbnRzIENoYXRNb2RlbCB7XG4gIHB1YmxpYyBzdWJVcmwgPSAnemhpcHUvYXBpL3BhYXMvdjQvY2hhdC9jb21wbGV0aW9ucydcbiAgY29uc3RydWN0b3IocHJpdmF0ZSByZXE6IE1vZGVsUmVxLCBwdWJsaWMgYmFzZVVybDogc3RyaW5nLCBzdWJVcmw/OiBzdHJpbmcpIHtcbiAgICBpZiAoc3ViVXJsICE9IG51bGwpIHtcbiAgICAgIHRoaXMuc3ViVXJsID0gc3ViVXJsXG4gICAgfVxuICB9XG5cbiAgcHJpdmF0ZSBnZXQgdXJsKCkge1xuICAgIHJldHVybiBgJHt0aGlzLmJhc2VVcmx9LyR7dGhpcy5zdWJVcmx9YFxuICB9XG5cbiAgYXN5bmMgZ2VuZXJhdGVUZXh0KGRhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCkge1xuICAgIGNvbnN0IHJlcyA9IChhd2FpdCB0aGlzLnJlcSh7XG4gICAgICB1cmw6IHRoaXMudXJsLFxuICAgICAgZGF0YToge1xuICAgICAgICAuLi5kYXRhLFxuICAgICAgICBzdHJlYW06IGZhbHNlLFxuICAgICAgfSxcbiAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgfSkpIGFzIFpoaVB1R2VuZXJhdGVUZXh0T3V0cHV0XG4gICAgcmV0dXJuIHtcbiAgICAgIHRleHQ6IChyZXM/LmNob2ljZXM/LlswXT8ubWVzc2FnZT8uY29udGVudCBhcyBzdHJpbmcpID8/ICcnLFxuICAgICAgcmF3UmVzcG9uc2U6IHJlcyxcbiAgICB9XG4gIH1cblxuICBhc3luYyBzdHJlYW1UZXh0KGRhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCkge1xuICAgIGNvbnN0IHN0cmVhbSA9IGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IHRydWUsXG4gICAgfSlcbiAgICByZXR1cm4gbmV3IFpoaVB1TW9kZWxTdHJlYW1SZXN1bHQoc3RyZWFtKVxuICB9XG59XG5cbmNsYXNzIFpoaVB1TW9kZWxTdHJlYW1SZXN1bHQgaW1wbGVtZW50cyBTdHJlYW1UZXh0UmVzdWx0IHtcbiAgcHJpdmF0ZSBfZXZlbnRTb3VyY2VTdHJlYW06IFJlYWRhYmxlU3RyZWFtPFBhcnNlZEV2ZW50PlxuXG4gIGNvbnN0cnVjdG9yKF9zdHJlYW06IFJlYWRhYmxlU3RyZWFtPFVpbnQ4QXJyYXk+KSB7XG4gICAgY29uc3Qgc3RyZWFtID0gdG9Qb2x5ZmlsbFJlYWRhYmxlKF9zdHJlYW0pIGFzIHR5cGVvZiBfc3RyZWFtXG4gICAgdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0gPSBzdHJlYW1cbiAgICAgIC5waXBlVGhyb3VnaChuZXcgVGV4dERlY29kZXJTdHJlYW0oKSlcbiAgICAgIC5waXBlVGhyb3VnaChjcmVhdGVFdmVudFNvdXJjZVBhcnNlclRyYW5zZm9ybVN0cmVhbSgpKVxuICB9XG5cbiAgcHJpdmF0ZSBnZXQgdGVlZWRTdHJlYW0oKSB7XG4gICAgY29uc3QgW3MxLCBzMl0gPSB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbS50ZWUoKVxuICAgIHRoaXMuX2V2ZW50U291cmNlU3RyZWFtID0gczJcbiAgICByZXR1cm4gczFcbiAgfVxuXG4gIGdldCBldmVudFNvdXJjZVN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLnRlZWVkU3RyZWFtKVxuICB9XG5cbiAgZ2V0IGRhdGFTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy5ldmVudFNvdXJjZVN0cmVhbS5waXBlVGhyb3VnaChuZXcgVHJhbnNmb3JtU3RyZWFtPFBhcnNlZEV2ZW50LCBaaGlQdVN0cmVhbVRleHRPdXRwdXQ+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICB0cnkge1xuICAgICAgICAgIGNvbnN0IGRhdGEgPSBKU09OLnBhcnNlKGNodW5rLmRhdGEpIGFzIFpoaVB1U3RyZWFtVGV4dE91dHB1dFxuICAgICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZShkYXRhKVxuICAgICAgICB9IGNhdGNoIChlKSB7XG4gICAgICAgICAgaWYgKGNodW5rLmRhdGEgIT09ICdbRE9ORV0nKSB7XG4gICAgICAgICAgICBjb25zb2xlLndhcm4oJ0Vycm9yIHdoZW4gdHJhbnNmb3JtaW5nIGV2ZW50IHNvdXJjZSBkYXRhIHRvIGpzb24nLCBlKVxuICAgICAgICAgIH1cbiAgICAgICAgfVxuICAgICAgfSxcbiAgICB9KSwpLClcbiAgfVxuXG4gIGdldCB0ZXh0U3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMuZGF0YVN0cmVhbS5waXBlVGhyb3VnaChuZXcgVHJhbnNmb3JtU3RyZWFtPFpoaVB1U3RyZWFtVGV4dE91dHB1dCwgc3RyaW5nPih7XG4gICAgICB0cmFuc2Zvcm0oY2h1bmssIGNvbnRyb2xsZXIpIHtcbiAgICAgICAgY29udHJvbGxlci5lbnF1ZXVlKGNodW5rPy5jaG9pY2VzPy5bMF0/LmRlbHRhPy5jb250ZW50ID8/ICcnKVxuICAgICAgfSxcbiAgICB9KSwpLClcbiAgfVxufVxuIl19
+exports.ZhiPuSimpleModel = ZhiPuSimpleModel;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1poaVB1L2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQUEscUNBTW9CO0FBV3BCLFNBQVMsWUFBWSxDQUFDLEtBQXlCO0lBQ3JDLElBQUEsUUFBUSxHQUFvRCxLQUFLLFNBQXpELEVBQUUsS0FBSyxHQUE2QyxLQUFLLE1BQWxELEVBQUUsV0FBVyxHQUFnQyxLQUFLLFlBQXJDLEVBQUUsV0FBVyxHQUFtQixLQUFLLFlBQXhCLEVBQUUsS0FBSyxHQUFZLEtBQUssTUFBakIsRUFBRSxLQUFLLEdBQUssS0FBSyxNQUFWLENBQVU7SUFFekUsSUFBTSxpQkFBaUIsR0FBRztRQUN4QixJQUFJLFdBQVcsSUFBSSxXQUFXLEtBQUssTUFBTSxFQUFFO1lBQ3pDLE9BQU8sQ0FBQyxJQUFJLENBQUMsK0JBQStCLENBQUMsQ0FBQTtTQUM5QztRQUNELE9BQU8sV0FBa0IsQ0FBQTtJQUMzQixDQUFDLENBQUE7SUFFRCw2QkFDSyxLQUFLLEtBQ1IsUUFBUSxVQUFBLEVBQ1IsS0FBSyxPQUFBLEVBQ0wsV0FBVyxhQUFBLEVBQ1gsV0FBVyxFQUFFLGlCQUFpQixFQUFFLEVBQ2hDLEtBQUssRUFBRSxLQUFLLGFBQUwsS0FBSyx1QkFBTCxLQUFLLENBQUUsR0FBRyxDQUFDLFVBQUEsSUFBSSxJQUFJLE9BQUEsdUJBQ3JCLElBQUksS0FDUCxRQUFRLHdCQUFPLElBQUksQ0FBQyxRQUFRLEtBQUUsVUFBVSxFQUFFLElBQUksQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLFFBQVEsQ0FBQyxVQUFVLENBQUMsT0FDOUUsRUFId0IsQ0FHeEIsQ0FBQyxFQUNILEtBQUssT0FBQSxJQUNOO0FBQ0gsQ0FBQztBQUVEO0lBRUUsMEJBQW9CLEdBQWEsRUFBUyxPQUFlLEVBQUUsTUFBZTtRQUF0RCxRQUFHLEdBQUgsR0FBRyxDQUFVO1FBQVMsWUFBTyxHQUFQLE9BQU8sQ0FBUTtRQURsRCxXQUFNLEdBQUcsb0NBQW9DLENBQUE7UUFFbEQsSUFBSSxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ2xCLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFBO1NBQ3JCO0lBQ0gsQ0FBQztJQUVELHNCQUFZLGlDQUFHO2FBQWY7WUFDRSxPQUFPLFVBQUcsSUFBSSxDQUFDLE9BQU8sY0FBSSxJQUFJLENBQUMsTUFBTSxDQUFFLENBQUE7UUFDekMsQ0FBQzs7O09BQUE7SUFFWSxxQ0FBVSxHQUF2QixVQUF3QixLQUF5Qjs7Ozs7O3dCQUN6QyxJQUFJLEdBQUcsWUFBWSxDQUFDLEtBQUssQ0FBQyxDQUFBO3dCQUNuQixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7Z0NBQzFCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRztnQ0FDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsS0FBSyxHQUNkO2dDQUNELE1BQU0sRUFBRSxLQUFLOzZCQUNkLENBQUMsRUFBQTs7d0JBUEksR0FBRyxHQUFHLENBQUMsU0FPWCxDQUE0Qjt3QkFDOUIsaUNBQVksR0FBRyxLQUFFLFdBQVcsRUFBRSxHQUFHLEtBQUU7Ozs7S0FDcEM7SUFFWSxtQ0FBUSxHQUFyQixVQUFzQixLQUF5Qjs7Ozs7O3dCQUN2QyxJQUFJLEdBQUcsWUFBWSxDQUFDLEtBQUssQ0FBQyxDQUFBO3dCQUM1QixVQUFVLEdBQW1CLElBQUksQ0FBQTt3QkFDckIsV0FBTSxJQUFJLENBQUMsR0FBRyxDQUFDO2dDQUM3QixHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUc7Z0NBQ2IsSUFBSSx3QkFDQyxJQUFJLEtBQ1AsTUFBTSxFQUFFLElBQUksR0FDYjtnQ0FDRCxNQUFNLEVBQUUsSUFBSTs2QkFDYixDQUFDLEVBQUE7O3dCQVBJLE9BQU8sR0FBRyxTQU9kO3dCQUNJLE1BQU0sR0FBRyxJQUFBLDBCQUFrQixFQUFDLE9BQU8sQ0FBbUIsQ0FBQTt3QkFFdEQsV0FBVyxHQUFHLElBQUEsMEJBQWtCLEVBQXdCLE1BQU0sQ0FBQyxDQUFBO3dCQUMvRCxhQUFhLEdBQUcsV0FBVyxDQUFDLFdBQVcsQ0FBQyxJQUFJLHVCQUFlLENBQXlFOzRCQUN4SSxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7Z0NBQ3pCLElBQU0sVUFBVSxHQUFHLEtBQUssQ0FBQyxPQUFPLENBQUMsR0FBRyxDQUFDLFVBQUMsTUFBTTtvQ0FDMUMsSUFBTSxPQUFPLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQTtvQ0FDNUIsSUFBSSxVQUFVLElBQUksSUFBSTt3Q0FBRSxVQUFVLEdBQUcsSUFBQSxrQ0FBMEIsRUFBQyxPQUFPLENBQUMsQ0FBQTtvQ0FDeEUsSUFBSSxVQUFVLEVBQUU7d0NBQ2QsNkJBQ0ssTUFBTSxLQUNULGFBQWEsRUFBRSxZQUFxQixFQUNwQyxLQUFLLEVBQUUsT0FBTyxJQUNmO3FDQUNGO29DQUNELE9BQU8sTUFBTSxDQUFBO2dDQUNmLENBQUMsQ0FBQyxDQUFBO2dDQUNGLElBQU0sUUFBUSx5QkFBUSxLQUFLLEtBQUUsT0FBTyxFQUFFLFVBQVUsR0FBRSxDQUFBO2dDQUNsRCxVQUFVLENBQUMsT0FBTyx1QkFBTSxRQUFRLEtBQUUsV0FBVyxFQUFFLEtBQUssSUFBRyxDQUFBOzRCQUN6RCxDQUFDO3lCQUNGLENBQUMsQ0FBRSxDQUFBO3dCQUVKLFdBQU8sSUFBQSwyQkFBbUIsRUFBQyxhQUFhLENBQUMsRUFBQTs7OztLQUMxQztJQUNILHVCQUFDO0FBQUQsQ0FBQyxBQTVERCxJQTREQztBQTVEWSw0Q0FBZ0IiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQge1xuICBjcmVhdGVBc3luY0l0ZXJhYmxlLFxuICBUcmFuc2Zvcm1TdHJlYW0sXG4gIHRvUG9seWZpbGxSZWFkYWJsZSxcbiAgaW50b1N0YW5kYXJkU3RyZWFtLFxuICBpc1Rvb2xDYWxsQXNzaXN0YW50TWVzc2FnZSxcbn0gZnJvbSAnLi4vLi4vdXRpbHMnXG5pbXBvcnQgdHlwZSB7IFpoaVB1R2VuZXJhdGVUZXh0T3V0cHV0LCBaaGlQdUlucHV0RGF0YSwgWmhpUHVTdHJlYW1UZXh0T3V0cHV0IH0gZnJvbSAnLi90eXBlJ1xuaW1wb3J0IHR5cGUge1xuICBNb2RlbFJlcSxcbiAgQmFzZUNoYXRNb2RlbElucHV0LFxuICBTaW1wbGVDaGF0TW9kZWwsXG4gIERvU3RyZWFtT3V0cHV0LFxuICBCYXNlRG9TdHJlYW1PdXRwdXRDaHVuayxcbiAgRG9HZW5lcmF0ZU91dHB1dCxcbn0gZnJvbSAnLi4vLi4vdHlwZSdcblxuZnVuY3Rpb24gcHJvY2Vzc0lucHV0KGlucHV0OiBCYXNlQ2hhdE1vZGVsSW5wdXQpOiBaaGlQdUlucHV0RGF0YSB7XG4gIGNvbnN0IHsgbWVzc2FnZXMsIG1vZGVsLCB0ZW1wZXJhdHVyZSwgdG9vbF9jaG9pY2UsIHRvb2xzLCB0b3BfcCB9ID0gaW5wdXRcblxuICBjb25zdCBwcm9jZXNzVG9vbENob2ljZSA9ICgpID0+IHtcbiAgICBpZiAodG9vbF9jaG9pY2UgJiYgdG9vbF9jaG9pY2UgIT09ICdhdXRvJykge1xuICAgICAgY29uc29sZS53YXJuKCdgdG9vbF9jaG9pY2VgIGlzIG5vdCBcXCdhdXRvXFwnJylcbiAgICB9XG4gICAgcmV0dXJuIHRvb2xfY2hvaWNlIGFzIGFueVxuICB9XG5cbiAgcmV0dXJuIHtcbiAgICAuLi5pbnB1dCxcbiAgICBtZXNzYWdlcyxcbiAgICBtb2RlbCxcbiAgICB0ZW1wZXJhdHVyZSxcbiAgICB0b29sX2Nob2ljZTogcHJvY2Vzc1Rvb2xDaG9pY2UoKSxcbiAgICB0b29sczogdG9vbHM/Lm1hcCh0b29sID0+ICh7XG4gICAgICAuLi50b29sLFxuICAgICAgZnVuY3Rpb246IHsgLi4udG9vbC5mdW5jdGlvbiwgcGFyYW1ldGVyczogSlNPTi5wYXJzZSh0b29sLmZ1bmN0aW9uLnBhcmFtZXRlcnMpIH0sXG4gICAgfSkpLFxuICAgIHRvcF9wLFxuICB9XG59XG5cbmV4cG9ydCBjbGFzcyBaaGlQdVNpbXBsZU1vZGVsIGltcGxlbWVudHMgU2ltcGxlQ2hhdE1vZGVsIHtcbiAgcHVibGljIHN1YlVybCA9ICd6aGlwdS9hcGkvcGFhcy92NC9jaGF0L2NvbXBsZXRpb25zJ1xuICBjb25zdHJ1Y3Rvcihwcml2YXRlIHJlcTogTW9kZWxSZXEsIHB1YmxpYyBiYXNlVXJsOiBzdHJpbmcsIHN1YlVybD86IHN0cmluZykge1xuICAgIGlmIChzdWJVcmwgIT0gbnVsbCkge1xuICAgICAgdGhpcy5zdWJVcmwgPSBzdWJVcmxcbiAgICB9XG4gIH1cblxuICBwcml2YXRlIGdldCB1cmwoKSB7XG4gICAgcmV0dXJuIGAke3RoaXMuYmFzZVVybH0vJHt0aGlzLnN1YlVybH1gXG4gIH1cblxuICBwdWJsaWMgYXN5bmMgZG9HZW5lcmF0ZShfZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KTogUHJvbWlzZTxEb0dlbmVyYXRlT3V0cHV0PiB7XG4gICAgY29uc3QgZGF0YSA9IHByb2Nlc3NJbnB1dChfZGF0YSlcbiAgICBjb25zdCByZXMgPSAoYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IGZhbHNlLFxuICAgIH0pKSBhcyBaaGlQdUdlbmVyYXRlVGV4dE91dHB1dFxuICAgIHJldHVybiB7IC4uLnJlcywgcmF3UmVzcG9uc2U6IHJlcyB9XG4gIH1cblxuICBwdWJsaWMgYXN5bmMgZG9TdHJlYW0oX2RhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCk6IFByb21pc2U8RG9TdHJlYW1PdXRwdXQ+IHtcbiAgICBjb25zdCBkYXRhID0gcHJvY2Vzc0lucHV0KF9kYXRhKVxuICAgIGxldCBpc1Rvb2xDYWxsOiBudWxsIHwgYm9vbGVhbiA9IG51bGxcbiAgICBjb25zdCBfc3RyZWFtID0gYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiB0cnVlLFxuICAgICAgfSxcbiAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICB9KVxuICAgIGNvbnN0IHN0cmVhbSA9IHRvUG9seWZpbGxSZWFkYWJsZShfc3RyZWFtKSBhcyB0eXBlb2YgX3N0cmVhbVxuXG4gICAgY29uc3QgemhpcHVTdHJlYW0gPSBpbnRvU3RhbmRhcmRTdHJlYW08WmhpUHVTdHJlYW1UZXh0T3V0cHV0PihzdHJlYW0pXG4gICAgY29uc3Qgc3RyZWFtV2l0aFJhdyA9IHpoaXB1U3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08WmhpUHVTdHJlYW1UZXh0T3V0cHV0LCBCYXNlRG9TdHJlYW1PdXRwdXRDaHVuayAmIHsgcmF3UmVzcG9uc2U/OiBhbnkgfT4oe1xuICAgICAgdHJhbnNmb3JtKGNodW5rLCBjb250cm9sbGVyKSB7XG4gICAgICAgIGNvbnN0IG5ld0Nob2ljZXMgPSBjaHVuay5jaG9pY2VzLm1hcCgoY2hvaWNlKSA9PiB7XG4gICAgICAgICAgY29uc3QgbWVzc2FnZSA9IGNob2ljZS5kZWx0YVxuICAgICAgICAgIGlmIChpc1Rvb2xDYWxsID09IG51bGwpIGlzVG9vbENhbGwgPSBpc1Rvb2xDYWxsQXNzaXN0YW50TWVzc2FnZShtZXNzYWdlKVxuICAgICAgICAgIGlmIChpc1Rvb2xDYWxsKSB7XG4gICAgICAgICAgICByZXR1cm4ge1xuICAgICAgICAgICAgICAuLi5jaG9pY2UsXG4gICAgICAgICAgICAgIGZpbmlzaF9yZWFzb246ICd0b29sX2NhbGxzJyBhcyBjb25zdCxcbiAgICAgICAgICAgICAgZGVsdGE6IG1lc3NhZ2UsXG4gICAgICAgICAgICB9XG4gICAgICAgICAgfVxuICAgICAgICAgIHJldHVybiBjaG9pY2VcbiAgICAgICAgfSlcbiAgICAgICAgY29uc3QgbmV3Q2h1bmsgPSB7IC4uLmNodW5rLCBjaG9pY2VzOiBuZXdDaG9pY2VzIH1cbiAgICAgICAgY29udHJvbGxlci5lbnF1ZXVlKHsgLi4ubmV3Q2h1bmssIHJhd1Jlc3BvbnNlOiBjaHVuayB9KVxuICAgICAgfSxcbiAgICB9KSwpXG5cbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZShzdHJlYW1XaXRoUmF3KVxuICB9XG59XG4iXX0=

package/dist/cjs/models/ZhiPu/type.d.ts

@@ -1,3 +1,4 @@
+import { ChatModelMessage } from '../../type';
 type SystemMessage = {
     role: 'system';
     content: string;
@@ -11,10 +12,10 @@ type AssistantMessage = {
     content?: string;
     tool_calls?: Array<{
         id: string;
-        type: 'web_search' | 'retrieval' | 'function';
+        type: string;
         function?: {
             name: string;
-            auguments: string;
+            arguments: string;
         };
     }>;
 };
@@ -34,38 +35,37 @@ export type ZhiPuInputData = {
     max_tokens?: number;
     stop?: Array<string>;
     tools?: Array<{
-        type: 'web_search' | 'retrieval' | 'function';
+        type: string;
         function: {
             name: string;
             description: string;
             parameters: object;
         };
-        retrieval: {
-            knowledge_id: string;
-            prompt_template?: string;
-            parameters: object;
-        };
-        web_search: {
-            enable?: boolean;
-            search_query?: string;
-            search_result?: boolean;
-        };
     }>;
     tool_choice?: 'auto';
     user_id?: string;
 };
 export type ZhiPuGenerateTextOutput = {
     choices?: Array<{
-        message?: {
-            content?: string;
-        };
+        finish_reason?: 'stop' | 'tool_calls' | 'length' | 'sensitive' | 'network_error';
+        message?: ChatModelMessage;
     }>;
+    usage?: {
+        completion_tokens: number;
+        prompt_tokens: number;
+        total_tokens: number;
+    };
 };
 export type ZhiPuStreamTextOutput = {
     choices?: Array<{
-        delta?: {
-            content?: string;
-        };
+        index: number;
+        finish_reason?: 'stop' | 'tool_calls' | 'length' | 'sensitive' | 'network_error';
+        delta?: ChatModelMessage;
     }>;
+    usage?: {
+        completion_tokens: number;
+        prompt_tokens: number;
+        total_tokens: number;
+    };
 };
 export {};

package/dist/cjs/models/ZhiPu/type.js

@@ -1,3 +1,3 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvWmhpUHUvdHlwZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiIiwic291cmNlc0NvbnRlbnQiOlsidHlwZSBTeXN0ZW1NZXNzYWdlID0ge1xuICByb2xlOiAnc3lzdGVtJ1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBVc2VyTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3VzZXInXG4gIGNvbnRlbnQ6IHN0cmluZ1xufVxuXG50eXBlIEFzc2lzdGFudE1lc3NhZ2UgPSB7XG4gIHJvbGU6ICdhc3Npc3RhbnQnXG4gIGNvbnRlbnQ/OiBzdHJpbmdcbiAgdG9vbF9jYWxscz86IEFycmF5PHtcbiAgICBpZDogc3RyaW5nXG4gICAgdHlwZTogJ3dlYl9zZWFyY2gnIHwgJ3JldHJpZXZhbCcgfCAnZnVuY3Rpb24nXG4gICAgZnVuY3Rpb24/OiB7IG5hbWU6IHN0cmluZzsgYXVndW1lbnRzOiBzdHJpbmcgfVxuICB9PlxufVxuXG50eXBlIFRvb2xNZXNzYWdlID0ge1xuICByb2xlOiAndG9vbCdcbiAgY29udGVudDogc3RyaW5nXG4gIHRvb2xfY2FsbF9pZDogc3RyaW5nXG59XG5cbnR5cGUgWmhpUHVNZXNzYWdlID0gU3lzdGVtTWVzc2FnZSB8IFVzZXJNZXNzYWdlIHwgQXNzaXN0YW50TWVzc2FnZSB8IFRvb2xNZXNzYWdlXG5cbmV4cG9ydCB0eXBlIFpoaVB1SW5wdXREYXRhID0ge1xuICBtb2RlbDogc3RyaW5nXG4gIG1lc3NhZ2VzOiBBcnJheTxaaGlQdU1lc3NhZ2U+XG4gIHJlcXVlc3RfaWQ/OiBzdHJpbmdcbiAgZG9fc2FtcGxlPzogYm9vbGVhblxuICB0ZW1wZXJhdHVyZT86IG51bWJlclxuICB0b3BfcD86IG51bWJlclxuICBtYXhfdG9rZW5zPzogbnVtYmVyXG4gIHN0b3A/OiBBcnJheTxzdHJpbmc+XG4gIHRvb2xzPzogQXJyYXk8e1xuICAgIHR5cGU6ICd3ZWJfc2VhcmNoJyB8ICdyZXRyaWV2YWwnIHwgJ2Z1bmN0aW9uJ1xuICAgIGZ1bmN0aW9uOiB7IG5hbWU6IHN0cmluZzsgZGVzY3JpcHRpb246IHN0cmluZzsgcGFyYW1ldGVyczogb2JqZWN0IH1cbiAgICByZXRyaWV2YWw6IHsga25vd2xlZGdlX2lkOiBzdHJpbmc7IHByb21wdF90ZW1wbGF0ZT86IHN0cmluZzsgcGFyYW1ldGVyczogb2JqZWN0IH1cbiAgICB3ZWJfc2VhcmNoOiB7IGVuYWJsZT86IGJvb2xlYW47IHNlYXJjaF9xdWVyeT86IHN0cmluZzsgc2VhcmNoX3Jlc3VsdD86IGJvb2xlYW4gfVxuICB9PlxuICB0b29sX2Nob2ljZT86ICdhdXRvJ1xuICB1c2VyX2lkPzogc3RyaW5nXG59XG5cbi8vIHRvZG86IG1vcmUgcHJlY2lzZVxuZXhwb3J0IHR5cGUgWmhpUHVHZW5lcmF0ZVRleHRPdXRwdXQgPSB7XG4gIGNob2ljZXM/OiBBcnJheTx7XG4gICAgbWVzc2FnZT86IHsgY29udGVudD86IHN0cmluZyB9XG4gIH0+XG59XG5cbi8vIHRvZG86IG1vcmUgcHJlY2lzZVxuZXhwb3J0IHR5cGUgWmhpUHVTdHJlYW1UZXh0T3V0cHV0ID0ge1xuICBjaG9pY2VzPzogQXJyYXk8e1xuICAgIGRlbHRhPzogeyBjb250ZW50Pzogc3RyaW5nIH1cbiAgfT5cbn1cbiJdfQ==
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvWmhpUHUvdHlwZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiIiwic291cmNlc0NvbnRlbnQiOlsiaW1wb3J0IHsgQ2hhdE1vZGVsTWVzc2FnZSB9IGZyb20gJy4uLy4uL3R5cGUnXG5cbnR5cGUgU3lzdGVtTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3N5c3RlbSdcbiAgY29udGVudDogc3RyaW5nXG59XG5cbnR5cGUgVXNlck1lc3NhZ2UgPSB7XG4gIHJvbGU6ICd1c2VyJ1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBBc3Npc3RhbnRNZXNzYWdlID0ge1xuICByb2xlOiAnYXNzaXN0YW50J1xuICBjb250ZW50Pzogc3RyaW5nXG4gIHRvb2xfY2FsbHM/OiBBcnJheTx7XG4gICAgaWQ6IHN0cmluZ1xuICAgIHR5cGU6IHN0cmluZ1xuICAgIGZ1bmN0aW9uPzogeyBuYW1lOiBzdHJpbmc7IGFyZ3VtZW50czogc3RyaW5nIH1cbiAgfT5cbn1cblxudHlwZSBUb29sTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3Rvb2wnXG4gIGNvbnRlbnQ6IHN0cmluZ1xuICB0b29sX2NhbGxfaWQ6IHN0cmluZ1xufVxuXG50eXBlIFpoaVB1TWVzc2FnZSA9IFN5c3RlbU1lc3NhZ2UgfCBVc2VyTWVzc2FnZSB8IEFzc2lzdGFudE1lc3NhZ2UgfCBUb29sTWVzc2FnZVxuXG5leHBvcnQgdHlwZSBaaGlQdUlucHV0RGF0YSA9IHtcbiAgbW9kZWw6IHN0cmluZ1xuICBtZXNzYWdlczogQXJyYXk8WmhpUHVNZXNzYWdlPlxuICByZXF1ZXN0X2lkPzogc3RyaW5nXG4gIGRvX3NhbXBsZT86IGJvb2xlYW5cbiAgdGVtcGVyYXR1cmU/OiBudW1iZXJcbiAgdG9wX3A/OiBudW1iZXJcbiAgbWF4X3Rva2Vucz86IG51bWJlclxuICBzdG9wPzogQXJyYXk8c3RyaW5nPlxuICB0b29scz86IEFycmF5PHtcbiAgICB0eXBlOiBzdHJpbmdcbiAgICBmdW5jdGlvbjogeyBuYW1lOiBzdHJpbmc7IGRlc2NyaXB0aW9uOiBzdHJpbmc7IHBhcmFtZXRlcnM6IG9iamVjdCB9XG4gIH0+XG4gIHRvb2xfY2hvaWNlPzogJ2F1dG8nXG4gIHVzZXJfaWQ/OiBzdHJpbmdcbn1cblxuZXhwb3J0IHR5cGUgWmhpUHVHZW5lcmF0ZVRleHRPdXRwdXQgPSB7XG4gIGNob2ljZXM/OiBBcnJheTx7XG4gICAgZmluaXNoX3JlYXNvbj86ICdzdG9wJyB8ICd0b29sX2NhbGxzJyB8ICdsZW5ndGgnIHwgJ3NlbnNpdGl2ZScgfCAnbmV0d29ya19lcnJvcidcbiAgICBtZXNzYWdlPzogQ2hhdE1vZGVsTWVzc2FnZVxuICB9PlxuICB1c2FnZT86IHtcbiAgICBjb21wbGV0aW9uX3Rva2VuczogbnVtYmVyXG4gICAgcHJvbXB0X3Rva2VuczogbnVtYmVyXG4gICAgdG90YWxfdG9rZW5zOiBudW1iZXJcbiAgfVxufVxuXG5leHBvcnQgdHlwZSBaaGlQdVN0cmVhbVRleHRPdXRwdXQgPSB7XG4gIGNob2ljZXM/OiBBcnJheTx7XG4gICAgaW5kZXg6IG51bWJlclxuICAgIGZpbmlzaF9yZWFzb24/OiAnc3RvcCcgfCAndG9vbF9jYWxscycgfCAnbGVuZ3RoJyB8ICdzZW5zaXRpdmUnIHwgJ25ldHdvcmtfZXJyb3InXG4gICAgZGVsdGE/OiBDaGF0TW9kZWxNZXNzYWdlXG4gIH0+XG4gIHVzYWdlPzoge1xuICAgIGNvbXBsZXRpb25fdG9rZW5zOiBudW1iZXJcbiAgICBwcm9tcHRfdG9rZW5zOiBudW1iZXJcbiAgICB0b3RhbF90b2tlbnM6IG51bWJlclxuICB9XG59XG4iXX0=