@cloudbase/ai 2.8.20-beta.0 → 2.8.22-beta.0

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

@@ -45,9 +45,13 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
         if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
     }
 };
-import { createAsyncIterable, TransformStream, TextDecoderStream, createEventSourceParserTransformStream, toPolyfillReadable, } from '../../utils';
-var YiModel = (function () {
-    function YiModel(req, baseUrl, subUrl) {
+import { createAsyncIterable, TransformStream, toPolyfillReadable, intoStandardStream, isToolCallAssistantMessage, } from '../../utils';
+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';
@@ -55,108 +59,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 = toPolyfillReadable(_stream);
+                        arkStream = intoStandardStream(stream);
+                        streamWithRaw = arkStream.pipeThrough(new 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 = 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, createAsyncIterable(streamWithRaw)];
                 }
             });
         });
     };
-    return YiModel;
-}());
-export { YiModel };
-var YiModelStreamResult = (function () {
-    function YiModelStreamResult(_stream) {
-        var stream = toPolyfillReadable(_stream);
-        this._eventSourceStream = stream
-            .pipeThrough(new TextDecoderStream())
-            .pipeThrough(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 createAsyncIterable(this.teeedStream);
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(YiModelStreamResult.prototype, "dataStream", {
-        get: function () {
-            return createAsyncIterable(this.eventSourceStream.pipeThrough(new 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 createAsyncIterable(this.dataStream.pipeThrough(new 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,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1lpL2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQ0EsT0FBTyxFQUNMLG1CQUFtQixFQUNuQixlQUFlLEVBQ2YsaUJBQWlCLEVBQ2pCLHNDQUFzQyxFQUN0QyxrQkFBa0IsR0FDbkIsTUFBTSxhQUFhLENBQUE7QUFJcEI7SUFFRSxpQkFBb0IsR0FBYSxFQUFTLE9BQWUsRUFBRSxNQUFlO1FBQXRELFFBQUcsR0FBSCxHQUFHLENBQVU7UUFBUyxZQUFPLEdBQVAsT0FBTyxDQUFRO1FBRGxELFdBQU0sR0FBRywyQkFBMkIsQ0FBQTtRQUV6QyxJQUFJLE1BQU0sSUFBSSxJQUFJLEVBQUU7WUFDbEIsSUFBSSxDQUFDLE1BQU0sR0FBRyxNQUFNLENBQUE7U0FDckI7SUFDSCxDQUFDO0lBRUQsc0JBQVksd0JBQUc7YUFBZjtZQUNFLE9BQU8sVUFBRyxJQUFJLENBQUMsT0FBTyxjQUFJLElBQUksQ0FBQyxNQUFNLENBQUUsQ0FBQTtRQUN6QyxDQUFDOzs7T0FBQTtJQUVLLDhCQUFZLEdBQWxCLFVBQW1CLElBQXdCOzs7Ozs7NEJBQzVCLFdBQU0sSUFBSSxDQUFDLEdBQUcsQ0FBQzs0QkFDMUIsR0FBRyxFQUFFLElBQUksQ0FBQyxHQUFHOzRCQUNiLElBQUksd0JBQ0MsSUFBSSxLQUNQLE1BQU0sRUFBRSxLQUFLLEdBQ2Q7NEJBQ0QsTUFBTSxFQUFFLEtBQUs7eUJBQ2QsQ0FBQyxFQUFBOzt3QkFQSSxHQUFHLEdBQUcsQ0FBQyxTQU9YLENBQXlCO3dCQUMzQixXQUFPO2dDQUNMLElBQUksRUFBRSxNQUFDLE1BQUEsTUFBQSxNQUFBLEdBQUcsYUFBSCxHQUFHLHVCQUFILEdBQUcsQ0FBRSxPQUFPLDBDQUFHLENBQUMsQ0FBQywwQ0FBRSxPQUFPLDBDQUFFLE9BQWtCLG1DQUFJLEVBQUU7Z0NBQzNELFdBQVcsRUFBRSxHQUFHOzZCQUNqQixFQUFBOzs7O0tBQ0Y7SUFFSyw0QkFBVSxHQUFoQixVQUFpQixJQUF3Qjs7Ozs7NEJBQ3hCLFdBQU0sSUFBSSxDQUFDLEdBQUcsQ0FBQzs0QkFDNUIsR0FBRyxFQUFFLElBQUksQ0FBQyxHQUFHOzRCQUNiLElBQUksd0JBQ0MsSUFBSSxLQUNQLE1BQU0sRUFBRSxJQUFJLEdBQ2I7NEJBQ0QsTUFBTSxFQUFFLElBQUk7eUJBQ2IsQ0FBQyxFQUFBOzt3QkFQSSxNQUFNLEdBQUcsU0FPYjt3QkFDRixXQUFPLElBQUksbUJBQW1CLENBQUMsTUFBTSxDQUFDLEVBQUE7Ozs7S0FDdkM7SUFDSCxjQUFDO0FBQUQsQ0FBQyxBQXRDRCxJQXNDQzs7QUFFRDtJQUdFLDZCQUFZLE9BQW1DO1FBQzdDLElBQU0sTUFBTSxHQUFHLGtCQUFrQixDQUFDLE9BQU8sQ0FBbUIsQ0FBQTtRQUM1RCxJQUFJLENBQUMsa0JBQWtCLEdBQUcsTUFBTTthQUM3QixXQUFXLENBQUMsSUFBSSxpQkFBaUIsRUFBRSxDQUFDO2FBQ3BDLFdBQVcsQ0FBQyxzQ0FBc0MsRUFBRSxDQUFDLENBQUE7SUFDMUQsQ0FBQztJQUVELHNCQUFZLDRDQUFXO2FBQXZCO1lBQ1EsSUFBQSxLQUFXLElBQUksQ0FBQyxrQkFBa0IsQ0FBQyxHQUFHLEVBQUUsRUFBdkMsRUFBRSxRQUFBLEVBQUUsRUFBRSxRQUFpQyxDQUFBO1lBQzlDLElBQUksQ0FBQyxrQkFBa0IsR0FBRyxFQUFFLENBQUE7WUFDNUIsT0FBTyxFQUFFLENBQUE7UUFDWCxDQUFDOzs7T0FBQTtJQUVELHNCQUFJLGtEQUFpQjthQUFyQjtZQUNFLE9BQU8sbUJBQW1CLENBQUMsSUFBSSxDQUFDLFdBQVcsQ0FBQyxDQUFBO1FBQzlDLENBQUM7OztPQUFBO0lBRUQsc0JBQUksMkNBQVU7YUFBZDtZQUNFLE9BQU8sbUJBQW1CLENBQUMsSUFBSSxDQUFDLGlCQUFpQixDQUFDLFdBQVcsQ0FBQyxJQUFJLGVBQWUsQ0FBa0M7Z0JBQ2pILFNBQVMsWUFBQyxLQUFLLEVBQUUsVUFBVTtvQkFDekIsSUFBSTt3QkFDRixJQUFNLElBQUksR0FBRyxJQUFJLENBQUMsS0FBSyxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQXVCLENBQUE7d0JBQ3pELFVBQVUsQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUE7cUJBQ3pCO29CQUFDLE9BQU8sQ0FBQyxFQUFFO3dCQUNWLElBQUksS0FBSyxDQUFDLElBQUksS0FBSyxRQUFRLEVBQUU7NEJBQzNCLE9BQU8sQ0FBQyxJQUFJLENBQUMsbURBQW1ELEVBQUUsQ0FBQyxDQUFDLENBQUE7eUJBQ3JFO3FCQUNGO2dCQUNILENBQUM7YUFDRixDQUFDLENBQUUsQ0FBRSxDQUFBO1FBQ1IsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSwyQ0FBVTthQUFkO1lBQ0UsT0FBTyxtQkFBbUIsQ0FBQyxJQUFJLENBQUMsVUFBVSxDQUFDLFdBQVcsQ0FBQyxJQUFJLGVBQWUsQ0FBNkI7Z0JBQ3JHLFNBQVMsWUFBQyxLQUFLLEVBQUUsVUFBVTs7b0JBQ3pCLFVBQVUsQ0FBQyxPQUFPLENBQUMsTUFBQSxNQUFBLE1BQUEsTUFBQSxLQUFLLGFBQUwsS0FBSyx1QkFBTCxLQUFLLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsS0FBSywwQ0FBRSxPQUFPLG1DQUFJLEVBQUUsQ0FBQyxDQUFBO2dCQUMvRCxDQUFDO2FBQ0YsQ0FBQyxDQUFFLENBQUUsQ0FBQTtRQUNSLENBQUM7OztPQUFBO0lBQ0gsMEJBQUM7QUFBRCxDQUFDLEFBMUNELElBMENDIiwic291cmNlc0NvbnRlbnQiOlsiaW1wb3J0IHsgdHlwZSBQYXJzZWRFdmVudCB9IGZyb20gJy4uLy4uL2V2ZW50c291cmNlX3BhcnNlcidcbmltcG9ydCB7XG4gIGNyZWF0ZUFzeW5jSXRlcmFibGUsXG4gIFRyYW5zZm9ybVN0cmVhbSxcbiAgVGV4dERlY29kZXJTdHJlYW0sXG4gIGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtLFxuICB0b1BvbHlmaWxsUmVhZGFibGUsXG59IGZyb20gJy4uLy4uL3V0aWxzJ1xuaW1wb3J0IHR5cGUgeyBZaUdlbmVyYXRlVGV4dE91dHB1dCwgWWlTdHJlYW1UZXh0T3V0cHV0IH0gZnJvbSAnLi90eXBlJ1xuaW1wb3J0IHR5cGUgeyBDaGF0TW9kZWwsIFN0cmVhbVRleHRSZXN1bHQsIE1vZGVsUmVxLCBCYXNlQ2hhdE1vZGVsSW5wdXQgfSBmcm9tICcuLi8uLi90eXBlJ1xuXG5leHBvcnQgY2xhc3MgWWlNb2RlbCBpbXBsZW1lbnRzIENoYXRNb2RlbCB7XG4gIHB1YmxpYyBzdWJVcmwgPSAnMDEtYWkvdjEvY2hhdC9jb21wbGV0aW9ucydcbiAgY29uc3RydWN0b3IocHJpdmF0ZSByZXE6IE1vZGVsUmVxLCBwdWJsaWMgYmFzZVVybDogc3RyaW5nLCBzdWJVcmw/OiBzdHJpbmcpIHtcbiAgICBpZiAoc3ViVXJsICE9IG51bGwpIHtcbiAgICAgIHRoaXMuc3ViVXJsID0gc3ViVXJsXG4gICAgfVxuICB9XG5cbiAgcHJpdmF0ZSBnZXQgdXJsKCkge1xuICAgIHJldHVybiBgJHt0aGlzLmJhc2VVcmx9LyR7dGhpcy5zdWJVcmx9YFxuICB9XG5cbiAgYXN5bmMgZ2VuZXJhdGVUZXh0KGRhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCkge1xuICAgIGNvbnN0IHJlcyA9IChhd2FpdCB0aGlzLnJlcSh7XG4gICAgICB1cmw6IHRoaXMudXJsLFxuICAgICAgZGF0YToge1xuICAgICAgICAuLi5kYXRhLFxuICAgICAgICBzdHJlYW06IGZhbHNlLFxuICAgICAgfSxcbiAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgfSkpIGFzIFlpR2VuZXJhdGVUZXh0T3V0cHV0XG4gICAgcmV0dXJuIHtcbiAgICAgIHRleHQ6IChyZXM/LmNob2ljZXM/LlswXT8ubWVzc2FnZT8uY29udGVudCBhcyBzdHJpbmcpID8/ICcnLFxuICAgICAgcmF3UmVzcG9uc2U6IHJlcyxcbiAgICB9XG4gIH1cblxuICBhc3luYyBzdHJlYW1UZXh0KGRhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCkge1xuICAgIGNvbnN0IHN0cmVhbSA9IGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IHRydWUsXG4gICAgfSlcbiAgICByZXR1cm4gbmV3IFlpTW9kZWxTdHJlYW1SZXN1bHQoc3RyZWFtKVxuICB9XG59XG5cbmNsYXNzIFlpTW9kZWxTdHJlYW1SZXN1bHQgaW1wbGVtZW50cyBTdHJlYW1UZXh0UmVzdWx0IHtcbiAgcHJpdmF0ZSBfZXZlbnRTb3VyY2VTdHJlYW06IFJlYWRhYmxlU3RyZWFtPFBhcnNlZEV2ZW50PlxuXG4gIGNvbnN0cnVjdG9yKF9zdHJlYW06IFJlYWRhYmxlU3RyZWFtPFVpbnQ4QXJyYXk+KSB7XG4gICAgY29uc3Qgc3RyZWFtID0gdG9Qb2x5ZmlsbFJlYWRhYmxlKF9zdHJlYW0pIGFzIHR5cGVvZiBfc3RyZWFtXG4gICAgdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0gPSBzdHJlYW1cbiAgICAgIC5waXBlVGhyb3VnaChuZXcgVGV4dERlY29kZXJTdHJlYW0oKSlcbiAgICAgIC5waXBlVGhyb3VnaChjcmVhdGVFdmVudFNvdXJjZVBhcnNlclRyYW5zZm9ybVN0cmVhbSgpKVxuICB9XG5cbiAgcHJpdmF0ZSBnZXQgdGVlZWRTdHJlYW0oKSB7XG4gICAgY29uc3QgW3MxLCBzMl0gPSB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbS50ZWUoKVxuICAgIHRoaXMuX2V2ZW50U291cmNlU3RyZWFtID0gczJcbiAgICByZXR1cm4gczFcbiAgfVxuXG4gIGdldCBldmVudFNvdXJjZVN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLnRlZWVkU3RyZWFtKVxuICB9XG5cbiAgZ2V0IGRhdGFTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy5ldmVudFNvdXJjZVN0cmVhbS5waXBlVGhyb3VnaChuZXcgVHJhbnNmb3JtU3RyZWFtPFBhcnNlZEV2ZW50LCBZaVN0cmVhbVRleHRPdXRwdXQ+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICB0cnkge1xuICAgICAgICAgIGNvbnN0IGRhdGEgPSBKU09OLnBhcnNlKGNodW5rLmRhdGEpIGFzIFlpU3RyZWFtVGV4dE91dHB1dFxuICAgICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZShkYXRhKVxuICAgICAgICB9IGNhdGNoIChlKSB7XG4gICAgICAgICAgaWYgKGNodW5rLmRhdGEgIT09ICdbRE9ORV0nKSB7XG4gICAgICAgICAgICBjb25zb2xlLndhcm4oJ0Vycm9yIHdoZW4gdHJhbnNmb3JtaW5nIGV2ZW50IHNvdXJjZSBkYXRhIHRvIGpzb24nLCBlKVxuICAgICAgICAgIH1cbiAgICAgICAgfVxuICAgICAgfSxcbiAgICB9KSwpLClcbiAgfVxuXG4gIGdldCB0ZXh0U3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMuZGF0YVN0cmVhbS5waXBlVGhyb3VnaChuZXcgVHJhbnNmb3JtU3RyZWFtPFlpU3RyZWFtVGV4dE91dHB1dCwgc3RyaW5nPih7XG4gICAgICB0cmFuc2Zvcm0oY2h1bmssIGNvbnRyb2xsZXIpIHtcbiAgICAgICAgY29udHJvbGxlci5lbnF1ZXVlKGNodW5rPy5jaG9pY2VzPy5bMF0/LmRlbHRhPy5jb250ZW50ID8/ICcnKVxuICAgICAgfSxcbiAgICB9KSwpLClcbiAgfVxufVxuIl19
+export { YiSimpleModel };
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1lpL2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsT0FBTyxFQUNMLG1CQUFtQixFQUNuQixlQUFlLEVBQ2Ysa0JBQWtCLEVBQ2xCLGtCQUFrQixFQUNsQiwwQkFBMEIsR0FDM0IsTUFBTSxhQUFhLENBQUE7QUFXcEIsU0FBUyxZQUFZLENBQUMsS0FBeUI7SUFDckMsSUFBQSxRQUFRLEdBQXVDLEtBQUssU0FBNUMsRUFBRSxLQUFLLEdBQWdDLEtBQUssTUFBckMsRUFBRSxXQUFXLEdBQW1CLEtBQUssWUFBeEIsRUFBRSxLQUFLLEdBQVksS0FBSyxNQUFqQixFQUFFLEtBQUssR0FBSyxLQUFLLE1BQVYsQ0FBVTtJQUU1RCw2QkFDSyxLQUFLLEtBQ1IsUUFBUSxVQUFBLEVBQ1IsS0FBSyxPQUFBLEVBQ0wsS0FBSyxFQUFFLEtBQUssYUFBTCxLQUFLLHVCQUFMLEtBQUssQ0FBRSxHQUFHLENBQUMsVUFBQSxJQUFJLElBQUksT0FBQSx1QkFDckIsSUFBSSxLQUNQLFFBQVEsd0JBQU8sSUFBSSxDQUFDLFFBQVEsS0FBRSxVQUFVLEVBQUUsSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsUUFBUSxDQUFDLFVBQVUsQ0FBQyxPQUM5RSxFQUh3QixDQUd4QixDQUFDLEVBQ0gsS0FBSyxPQUFBLEVBQ0wsV0FBVyxhQUFBLElBQ1o7QUFDSCxDQUFDO0FBRUQ7SUFHRSx1QkFBb0IsR0FBYSxFQUFTLE9BQWUsRUFBRSxNQUFlO1FBQXRELFFBQUcsR0FBSCxHQUFHLENBQVU7UUFBUyxZQUFPLEdBQVAsT0FBTyxDQUFRO1FBRmxELFdBQU0sR0FBRywyQkFBMkIsQ0FBQTtRQUd6QyxJQUFJLE1BQU0sSUFBSSxJQUFJLEVBQUU7WUFDbEIsSUFBSSxDQUFDLE1BQU0sR0FBRyxNQUFNLENBQUE7U0FDckI7SUFDSCxDQUFDO0lBRUQsc0JBQVksOEJBQUc7YUFBZjtZQUNFLE9BQU8sVUFBRyxJQUFJLENBQUMsT0FBTyxjQUFJLElBQUksQ0FBQyxNQUFNLENBQUUsQ0FBQTtRQUN6QyxDQUFDOzs7T0FBQTtJQUVZLGtDQUFVLEdBQXZCLFVBQXdCLElBQXdCOzs7Ozs0QkFDakMsV0FBTSxJQUFJLENBQUMsR0FBRyxDQUFDOzRCQUMxQixHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUc7NEJBQ2IsSUFBSSx3QkFDQyxZQUFZLENBQUMsSUFBSSxDQUFDLEtBQ3JCLE1BQU0sRUFBRSxLQUFLLEdBQ2Q7NEJBQ0QsTUFBTSxFQUFFLEtBQUs7eUJBQ2QsQ0FBQyxFQUFBOzt3QkFQSSxHQUFHLEdBQUcsQ0FBQyxTQU9YLENBQXFCO3dCQUN2QixpQ0FBWSxHQUFHLEtBQUUsV0FBVyxFQUFFLEdBQUcsS0FBRTs7OztLQUNwQztJQUVZLGdDQUFRLEdBQXJCLFVBQXNCLElBQXdCOzs7Ozs7d0JBQ3hDLFVBQVUsR0FBbUIsSUFBSSxDQUFBO3dCQUNyQixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7Z0NBQzdCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRztnQ0FDYixJQUFJLHdCQUNDLFlBQVksQ0FBQyxJQUFJLENBQUMsS0FDckIsTUFBTSxFQUFFLElBQUksR0FDYjtnQ0FDRCxNQUFNLEVBQUUsSUFBSTs2QkFDYixDQUFDLEVBQUE7O3dCQVBJLE9BQU8sR0FBRyxTQU9kO3dCQUNJLE1BQU0sR0FBRyxrQkFBa0IsQ0FBQyxPQUFPLENBQW1CLENBQUE7d0JBRXRELFNBQVMsR0FBRyxrQkFBa0IsQ0FBaUIsTUFBTSxDQUFDLENBQUE7d0JBQ3RELGFBQWEsR0FBRyxTQUFTLENBQUMsV0FBVyxDQUFDLElBQUksZUFBZSxDQUFrRTs0QkFDL0gsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVOztnQ0FFekIsSUFBSSxDQUFDLENBQUMsQ0FBQSxNQUFBLE1BQUEsTUFBQSxLQUFLLGFBQUwsS0FBSyx1QkFBTCxLQUFLLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsS0FBSywwQ0FBRSxPQUFPLE1BQUksTUFBQSxNQUFBLE1BQUEsS0FBSyxhQUFMLEtBQUssdUJBQUwsS0FBSyxDQUFFLE9BQU8sMENBQUcsQ0FBQyxDQUFDLDBDQUFFLEtBQUssMENBQUUsVUFBVSxDQUFBLENBQUM7b0NBQUUsT0FBTTtnQ0FFNUYsSUFBTSxVQUFVLEdBQUcsS0FBSyxDQUFDLE9BQU8sQ0FBQyxHQUFHLENBQUMsVUFBQyxNQUFNO29DQUMxQyxJQUFNLE9BQU8sR0FBRyxNQUFNLENBQUMsTUFBTSxDQUFDLE1BQU0sQ0FBQyxLQUFLLEVBQUUsRUFBRSxJQUFJLEVBQUUsV0FBVyxFQUFFLENBRWhFLENBQUE7b0NBQ0QsSUFBSSxVQUFVLElBQUksSUFBSTt3Q0FBRSxVQUFVLEdBQUcsMEJBQTBCLENBQUMsT0FBTyxDQUFDLENBQUE7b0NBQ3hFLElBQUksVUFBVSxFQUFFO3dDQUNkLDZCQUNLLE1BQU0sS0FDVCxhQUFhLEVBQUUsWUFBcUIsRUFDcEMsS0FBSyxFQUFFLE9BQU8sSUFDZjtxQ0FDRjtvQ0FDRCw2QkFBWSxNQUFNLEtBQUUsS0FBSyxFQUFFLE9BQU8sSUFBRTtnQ0FDdEMsQ0FBQyxDQUFDLENBQUE7Z0NBQ0YsSUFBTSxRQUFRLHlCQUFRLEtBQUssS0FBRSxPQUFPLEVBQUUsVUFBVSxHQUFFLENBQUE7Z0NBQ2xELFVBQVUsQ0FBQyxPQUFPLHVCQUFNLFFBQVEsS0FBRSxXQUFXLEVBQUUsS0FBSyxJQUFHLENBQUE7NEJBQ3pELENBQUM7eUJBQ0YsQ0FBQyxDQUFFLENBQUE7d0JBRUosV0FBTyxtQkFBbUIsQ0FBQyxhQUFhLENBQUMsRUFBQTs7OztLQUMxQztJQUNILG9CQUFDO0FBQUQsQ0FBQyxBQWhFRCxJQWdFQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7XG4gIGNyZWF0ZUFzeW5jSXRlcmFibGUsXG4gIFRyYW5zZm9ybVN0cmVhbSxcbiAgdG9Qb2x5ZmlsbFJlYWRhYmxlLFxuICBpbnRvU3RhbmRhcmRTdHJlYW0sXG4gIGlzVG9vbENhbGxBc3Npc3RhbnRNZXNzYWdlLFxufSBmcm9tICcuLi8uLi91dGlscydcbmltcG9ydCB0eXBlIHsgWWlJbnB1dCwgWWlHZW5lcmF0ZU91dHB1dCwgWWlTdHJlYW1PdXRwdXQgfSBmcm9tICcuL3R5cGUnXG5pbXBvcnQgdHlwZSB7XG4gIE1vZGVsUmVxLFxuICBCYXNlQ2hhdE1vZGVsSW5wdXQsXG4gIERvR2VuZXJhdGVPdXRwdXQsXG4gIERvU3RyZWFtT3V0cHV0LFxuICBCYXNlRG9TdHJlYW1PdXRwdXRDaHVuayxcbiAgU2ltcGxlQ2hhdE1vZGVsLFxufSBmcm9tICcuLi8uLi90eXBlJ1xuXG5mdW5jdGlvbiBwcm9jZXNzSW5wdXQoaW5wdXQ6IEJhc2VDaGF0TW9kZWxJbnB1dCk6IFlpSW5wdXQge1xuICBjb25zdCB7IG1lc3NhZ2VzLCBtb2RlbCwgdGVtcGVyYXR1cmUsIHRvb2xzLCB0b3BfcCB9ID0gaW5wdXRcblxuICByZXR1cm4ge1xuICAgIC4uLmlucHV0LFxuICAgIG1lc3NhZ2VzLFxuICAgIG1vZGVsLFxuICAgIHRvb2xzOiB0b29scz8ubWFwKHRvb2wgPT4gKHtcbiAgICAgIC4uLnRvb2wsXG4gICAgICBmdW5jdGlvbjogeyAuLi50b29sLmZ1bmN0aW9uLCBwYXJhbWV0ZXJzOiBKU09OLnBhcnNlKHRvb2wuZnVuY3Rpb24ucGFyYW1ldGVycykgfSxcbiAgICB9KSksXG4gICAgdG9wX3AsXG4gICAgdGVtcGVyYXR1cmUsXG4gIH1cbn1cblxuZXhwb3J0IGNsYXNzIFlpU2ltcGxlTW9kZWwgaW1wbGVtZW50cyBTaW1wbGVDaGF0TW9kZWwge1xuICBwdWJsaWMgc3ViVXJsID0gJzAxLWFpL3YxL2NoYXQvY29tcGxldGlvbnMnXG5cbiAgY29uc3RydWN0b3IocHJpdmF0ZSByZXE6IE1vZGVsUmVxLCBwdWJsaWMgYmFzZVVybDogc3RyaW5nLCBzdWJVcmw/OiBzdHJpbmcpIHtcbiAgICBpZiAoc3ViVXJsICE9IG51bGwpIHtcbiAgICAgIHRoaXMuc3ViVXJsID0gc3ViVXJsXG4gICAgfVxuICB9XG5cbiAgcHJpdmF0ZSBnZXQgdXJsKCkge1xuICAgIHJldHVybiBgJHt0aGlzLmJhc2VVcmx9LyR7dGhpcy5zdWJVcmx9YFxuICB9XG5cbiAgcHVibGljIGFzeW5jIGRvR2VuZXJhdGUoZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KTogUHJvbWlzZTxEb0dlbmVyYXRlT3V0cHV0PiB7XG4gICAgY29uc3QgcmVzID0gKGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLnByb2Nlc3NJbnB1dChkYXRhKSxcbiAgICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IGZhbHNlLFxuICAgIH0pKSBhcyBZaUdlbmVyYXRlT3V0cHV0XG4gICAgcmV0dXJuIHsgLi4ucmVzLCByYXdSZXNwb25zZTogcmVzIH1cbiAgfVxuXG4gIHB1YmxpYyBhc3luYyBkb1N0cmVhbShkYXRhOiBCYXNlQ2hhdE1vZGVsSW5wdXQpOiBQcm9taXNlPERvU3RyZWFtT3V0cHV0PiB7XG4gICAgbGV0IGlzVG9vbENhbGw6IG51bGwgfCBib29sZWFuID0gbnVsbFxuICAgIGNvbnN0IF9zdHJlYW0gPSBhd2FpdCB0aGlzLnJlcSh7XG4gICAgICB1cmw6IHRoaXMudXJsLFxuICAgICAgZGF0YToge1xuICAgICAgICAuLi5wcm9jZXNzSW5wdXQoZGF0YSksXG4gICAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IHRydWUsXG4gICAgfSlcbiAgICBjb25zdCBzdHJlYW0gPSB0b1BvbHlmaWxsUmVhZGFibGUoX3N0cmVhbSkgYXMgdHlwZW9mIF9zdHJlYW1cblxuICAgIGNvbnN0IGFya1N0cmVhbSA9IGludG9TdGFuZGFyZFN0cmVhbTxZaVN0cmVhbU91dHB1dD4oc3RyZWFtKVxuICAgIGNvbnN0IHN0cmVhbVdpdGhSYXcgPSBhcmtTdHJlYW0ucGlwZVRocm91Z2gobmV3IFRyYW5zZm9ybVN0cmVhbTxZaVN0cmVhbU91dHB1dCwgQmFzZURvU3RyZWFtT3V0cHV0Q2h1bmsgJiB7IHJhd1Jlc3BvbnNlPzogYW55IH0+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICAvLyDpnZ7luLjlpYfokalcbiAgICAgICAgaWYgKCEoY2h1bms/LmNob2ljZXM/LlswXT8uZGVsdGE/LmNvbnRlbnQgfHwgY2h1bms/LmNob2ljZXM/LlswXT8uZGVsdGE/LnRvb2xfY2FsbHMpKSByZXR1cm5cblxuICAgICAgICBjb25zdCBuZXdDaG9pY2VzID0gY2h1bmsuY2hvaWNlcy5tYXAoKGNob2ljZSkgPT4ge1xuICAgICAgICAgIGNvbnN0IG1lc3NhZ2UgPSBPYmplY3QuYXNzaWduKGNob2ljZS5kZWx0YSwgeyByb2xlOiAnYXNzaXN0YW50JyB9KSBhcyB0eXBlb2YgY2hvaWNlLmRlbHRhICYge1xuICAgICAgICAgICAgcm9sZTogJ2Fzc2lzdGFudCdcbiAgICAgICAgICB9XG4gICAgICAgICAgaWYgKGlzVG9vbENhbGwgPT0gbnVsbCkgaXNUb29sQ2FsbCA9IGlzVG9vbENhbGxBc3Npc3RhbnRNZXNzYWdlKG1lc3NhZ2UpXG4gICAgICAgICAgaWYgKGlzVG9vbENhbGwpIHtcbiAgICAgICAgICAgIHJldHVybiB7XG4gICAgICAgICAgICAgIC4uLmNob2ljZSxcbiAgICAgICAgICAgICAgZmluaXNoX3JlYXNvbjogJ3Rvb2xfY2FsbHMnIGFzIGNvbnN0LFxuICAgICAgICAgICAgICBkZWx0YTogbWVzc2FnZSxcbiAgICAgICAgICAgIH1cbiAgICAgICAgICB9XG4gICAgICAgICAgcmV0dXJuIHsgLi4uY2hvaWNlLCBkZWx0YTogbWVzc2FnZSB9XG4gICAgICAgIH0pXG4gICAgICAgIGNvbnN0IG5ld0NodW5rID0geyAuLi5jaHVuaywgY2hvaWNlczogbmV3Q2hvaWNlcyB9XG4gICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZSh7IC4uLm5ld0NodW5rLCByYXdSZXNwb25zZTogY2h1bmsgfSlcbiAgICAgIH0sXG4gICAgfSksKVxuXG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUoc3RyZWFtV2l0aFJhdylcbiAgfVxufVxuIl19

package/dist/esm/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/esm/models/Yi/type.js

@@ -1,2 +1,2 @@
 export {};
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvWWkvdHlwZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiIiwic291cmNlc0NvbnRlbnQiOlsidHlwZSBTeXN0ZW1NZXNzYWdlID0ge1xuICByb2xlOiAnc3lzdGVtJ1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBVc2VyTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3VzZXInXG4gIGNvbnRlbnQ6IHN0cmluZ1xufVxuXG50eXBlIEFzc2lzdGFudE1lc3NhZ2UgPSB7XG4gIHJvbGU6ICdhc3Npc3RhbnQnXG4gIGNvbnRlbnQ6IHN0cmluZ1xufVxuXG50eXBlIFRvb2xNZXNzYWdlID0ge1xuICByb2xlOiAndG9vbCdcbiAgY29udGVudDogc3RyaW5nXG4gIHRvb2xfY2FsbF9pZDogc3RyaW5nXG59XG5cbnR5cGUgWWlNZXNzYWdlID0gU3lzdGVtTWVzc2FnZSB8IFVzZXJNZXNzYWdlIHwgQXNzaXN0YW50TWVzc2FnZSB8IFRvb2xNZXNzYWdlXG5cbmV4cG9ydCB0eXBlIFlpSW5wdXREYXRhID0ge1xuICBtb2RlbDogc3RyaW5nXG4gIG1lc3NhZ2VzOiBBcnJheTxZaU1lc3NhZ2U+XG4gIHRvb2xzPzogQXJyYXk8e1xuICAgIHR5cGU6ICdmdW5jdGlvbidcbiAgICBmdW5jdGlvbjogeyBuYW1lOiBzdHJpbmc7IGRlc2NyaXB0aW9uPzogc3RyaW5nOyBwYXJhbWV0ZXJzOiBvYmplY3QgfVxuICB9PlxuICB0b29sX2Nob2ljZT86IHN0cmluZyB8IG9iamVjdFxuICBtYXhfdG9rZW5zPzogbnVtYmVyIHwgbnVsbFxuICB0b3BfcD86IG51bWJlclxuICB0ZW1wZXJhdHVyZT86IG51bWJlclxufVxuXG4vLyB0b2RvOiBtb3JlIHByZWNpc2VcbmV4cG9ydCB0eXBlIFlpR2VuZXJhdGVUZXh0T3V0cHV0ID0ge1xuICBjaG9pY2VzPzogQXJyYXk8e1xuICAgIG1lc3NhZ2U/OiB7IGNvbnRlbnQ/OiBzdHJpbmcgfVxuICB9PlxufVxuXG4vLyB0b2RvOiBtb3JlIHByZWNpc2VcbmV4cG9ydCB0eXBlIFlpU3RyZWFtVGV4dE91dHB1dCA9IHtcbiAgY2hvaWNlcz86IEFycmF5PHtcbiAgICBkZWx0YT86IHsgY29udGVudD86IHN0cmluZyB9XG4gIH0+XG59XG4iXX0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvWWkvdHlwZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiIiwic291cmNlc0NvbnRlbnQiOlsidHlwZSBTeXN0ZW1NZXNzYWdlID0ge1xuICByb2xlOiAnc3lzdGVtJ1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBVc2VyTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3VzZXInXG4gIGNvbnRlbnQ6IHN0cmluZ1xufVxuXG50eXBlIEFzc2lzdGFudE1lc3NhZ2UgPVxuICB8IHtcbiAgICByb2xlOiAnYXNzaXN0YW50J1xuICAgIGNvbnRlbnQ6IHN0cmluZ1xuICB9XG4gIHwge1xuICAgIHJvbGU6ICdhc3Npc3RhbnQnXG4gICAgdG9vbF9jYWxsczogQXJyYXk8VG9vbENhbGw+XG4gICAgY29udGVudD86IHN0cmluZ1xuICB9XG5cbnR5cGUgVG9vbE1lc3NhZ2UgPSB7XG4gIHJvbGU6ICd0b29sJ1xuICBjb250ZW50OiBzdHJpbmdcbiAgdG9vbF9jYWxsX2lkOiBzdHJpbmdcbn1cblxudHlwZSBZaU1lc3NhZ2UgPSBTeXN0ZW1NZXNzYWdlIHwgVXNlck1lc3NhZ2UgfCBBc3Npc3RhbnRNZXNzYWdlIHwgVG9vbE1lc3NhZ2VcblxuZXhwb3J0IHR5cGUgWWlJbnB1dCA9IHtcbiAgbW9kZWw6IHN0cmluZ1xuICBtZXNzYWdlczogQXJyYXk8WWlNZXNzYWdlPlxuICB0ZW1wZXJhdHVyZT86IG51bWJlclxuICB0b3BfcD86IG51bWJlclxuICB0b29scz86IEFycmF5PHtcbiAgICB0eXBlOiBzdHJpbmdcbiAgICBmdW5jdGlvbjogeyBuYW1lOiBzdHJpbmc7IGRlc2NyaXB0aW9uOiBzdHJpbmc7IHBhcmFtZXRlcnM6IG9iamVjdCB9XG4gIH0+XG59XG5cbnR5cGUgVG9vbENhbGwgPSB7XG4gIGlkOiBzdHJpbmdcbiAgdHlwZTogc3RyaW5nXG4gIGZ1bmN0aW9uOiB7IG5hbWU6IHN0cmluZzsgYXJndW1lbnRzOiBzdHJpbmcgfVxufVxuXG5leHBvcnQgdHlwZSBZaUdlbmVyYXRlT3V0cHV0ID0ge1xuICBjaG9pY2VzOiBBcnJheTx7XG4gICAgZmluaXNoX3JlYXNvbjogc3RyaW5nXG4gICAgbWVzc2FnZToge1xuICAgICAgY29udGVudDogc3RyaW5nXG4gICAgICByb2xlOiAnYXNzaXN0YW50J1xuICAgICAgdG9vbF9jYWxscz86IEFycmF5PFRvb2xDYWxsPlxuICAgIH1cbiAgfT5cbiAgdXNhZ2U6IHtcbiAgICBwcm9tcHRfdG9rZW5zOiBudW1iZXJcbiAgICBjb21wbGV0aW9uX3Rva2VuczogbnVtYmVyXG4gICAgdG90YWxfdG9rZW5zOiBudW1iZXJcbiAgfVxufVxuXG5leHBvcnQgdHlwZSBZaVN0cmVhbU91dHB1dCA9IHtcbiAgY2hvaWNlczogQXJyYXk8e1xuICAgIGRlbHRhOiB7XG4gICAgICBjb250ZW50OiBzdHJpbmdcbiAgICAgIHRvb2xfY2FsbHM/OiBBcnJheTxUb29sQ2FsbD5cbiAgICAgIHJvbGU/OiAnYXNzaXN0YW50J1xuICAgIH1cbiAgICBmaW5pc2hfcmVhc29uOiBzdHJpbmdcbiAgfT5cbiAgdXNhZ2U/OiB7XG4gICAgcHJvbXB0X3Rva2VuczogbnVtYmVyXG4gICAgY29tcGxldGlvbl90b2tlbnM6IG51bWJlclxuICAgIHRvdGFsX3Rva2VuczogbnVtYmVyXG4gIH1cbn1cbiJdfQ==

package/dist/esm/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/esm/models/ZhiPu/index.js

@@ -45,9 +45,19 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
         if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
     }
 };
-import { createAsyncIterable, TransformStream, TextDecoderStream, createEventSourceParserTransformStream, toPolyfillReadable, } from '../../utils';
-var ZhiPuModel = (function () {
-    function ZhiPuModel(req, baseUrl, subUrl) {
+import { createAsyncIterable, TransformStream, toPolyfillReadable, intoStandardStream, isToolCallAssistantMessage, } from '../../utils';
+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';
@@ -55,108 +65,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 = toPolyfillReadable(_stream);
+                        zhipuStream = intoStandardStream(stream);
+                        streamWithRaw = zhipuStream.pipeThrough(new TransformStream({
+                            transform: function (chunk, controller) {
+                                var newChoices = chunk.choices.map(function (choice) {
+                                    var message = choice.delta;
+                                    if (isToolCall == null)
+                                        isToolCall = 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, createAsyncIterable(streamWithRaw)];
                 }
             });
         });
     };
-    return ZhiPuModel;
-}());
-export { ZhiPuModel };
-var ZhiPuModelStreamResult = (function () {
-    function ZhiPuModelStreamResult(_stream) {
-        var stream = toPolyfillReadable(_stream);
-        this._eventSourceStream = stream
-            .pipeThrough(new TextDecoderStream())
-            .pipeThrough(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 createAsyncIterable(this.teeedStream);
-        },
-        enumerable: false,
-        configurable: true
-    });
-    Object.defineProperty(ZhiPuModelStreamResult.prototype, "dataStream", {
-        get: function () {
-            return createAsyncIterable(this.eventSourceStream.pipeThrough(new 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 createAsyncIterable(this.dataStream.pipeThrough(new 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,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1poaVB1L2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQ0EsT0FBTyxFQUNMLG1CQUFtQixFQUNuQixlQUFlLEVBQ2YsaUJBQWlCLEVBQ2pCLHNDQUFzQyxFQUN0QyxrQkFBa0IsR0FDbkIsTUFBTSxhQUFhLENBQUE7QUFJcEI7SUFFRSxvQkFBb0IsR0FBYSxFQUFTLE9BQWUsRUFBRSxNQUFlO1FBQXRELFFBQUcsR0FBSCxHQUFHLENBQVU7UUFBUyxZQUFPLEdBQVAsT0FBTyxDQUFRO1FBRGxELFdBQU0sR0FBRyxvQ0FBb0MsQ0FBQTtRQUVsRCxJQUFJLE1BQU0sSUFBSSxJQUFJLEVBQUU7WUFDbEIsSUFBSSxDQUFDLE1BQU0sR0FBRyxNQUFNLENBQUE7U0FDckI7SUFDSCxDQUFDO0lBRUQsc0JBQVksMkJBQUc7YUFBZjtZQUNFLE9BQU8sVUFBRyxJQUFJLENBQUMsT0FBTyxjQUFJLElBQUksQ0FBQyxNQUFNLENBQUUsQ0FBQTtRQUN6QyxDQUFDOzs7T0FBQTtJQUVLLGlDQUFZLEdBQWxCLFVBQW1CLElBQXdCOzs7Ozs7NEJBQzVCLFdBQU0sSUFBSSxDQUFDLEdBQUcsQ0FBQzs0QkFDMUIsR0FBRyxFQUFFLElBQUksQ0FBQyxHQUFHOzRCQUNiLElBQUksd0JBQ0MsSUFBSSxLQUNQLE1BQU0sRUFBRSxLQUFLLEdBQ2Q7NEJBQ0QsTUFBTSxFQUFFLEtBQUs7eUJBQ2QsQ0FBQyxFQUFBOzt3QkFQSSxHQUFHLEdBQUcsQ0FBQyxTQU9YLENBQTRCO3dCQUM5QixXQUFPO2dDQUNMLElBQUksRUFBRSxNQUFDLE1BQUEsTUFBQSxNQUFBLEdBQUcsYUFBSCxHQUFHLHVCQUFILEdBQUcsQ0FBRSxPQUFPLDBDQUFHLENBQUMsQ0FBQywwQ0FBRSxPQUFPLDBDQUFFLE9BQWtCLG1DQUFJLEVBQUU7Z0NBQzNELFdBQVcsRUFBRSxHQUFHOzZCQUNqQixFQUFBOzs7O0tBQ0Y7SUFFSywrQkFBVSxHQUFoQixVQUFpQixJQUF3Qjs7Ozs7NEJBQ3hCLFdBQU0sSUFBSSxDQUFDLEdBQUcsQ0FBQzs0QkFDNUIsR0FBRyxFQUFFLElBQUksQ0FBQyxHQUFHOzRCQUNiLElBQUksd0JBQ0MsSUFBSSxLQUNQLE1BQU0sRUFBRSxJQUFJLEdBQ2I7NEJBQ0QsTUFBTSxFQUFFLElBQUk7eUJBQ2IsQ0FBQyxFQUFBOzt3QkFQSSxNQUFNLEdBQUcsU0FPYjt3QkFDRixXQUFPLElBQUksc0JBQXNCLENBQUMsTUFBTSxDQUFDLEVBQUE7Ozs7S0FDMUM7SUFDSCxpQkFBQztBQUFELENBQUMsQUF0Q0QsSUFzQ0M7O0FBRUQ7SUFHRSxnQ0FBWSxPQUFtQztRQUM3QyxJQUFNLE1BQU0sR0FBRyxrQkFBa0IsQ0FBQyxPQUFPLENBQW1CLENBQUE7UUFDNUQsSUFBSSxDQUFDLGtCQUFrQixHQUFHLE1BQU07YUFDN0IsV0FBVyxDQUFDLElBQUksaUJBQWlCLEVBQUUsQ0FBQzthQUNwQyxXQUFXLENBQUMsc0NBQXNDLEVBQUUsQ0FBQyxDQUFBO0lBQzFELENBQUM7SUFFRCxzQkFBWSwrQ0FBVzthQUF2QjtZQUNRLElBQUEsS0FBVyxJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxFQUFFLEVBQXZDLEVBQUUsUUFBQSxFQUFFLEVBQUUsUUFBaUMsQ0FBQTtZQUM5QyxJQUFJLENBQUMsa0JBQWtCLEdBQUcsRUFBRSxDQUFBO1lBQzVCLE9BQU8sRUFBRSxDQUFBO1FBQ1gsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSxxREFBaUI7YUFBckI7WUFDRSxPQUFPLG1CQUFtQixDQUFDLElBQUksQ0FBQyxXQUFXLENBQUMsQ0FBQTtRQUM5QyxDQUFDOzs7T0FBQTtJQUVELHNCQUFJLDhDQUFVO2FBQWQ7WUFDRSxPQUFPLG1CQUFtQixDQUFDLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxXQUFXLENBQUMsSUFBSSxlQUFlLENBQXFDO2dCQUNwSCxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7b0JBQ3pCLElBQUk7d0JBQ0YsSUFBTSxJQUFJLEdBQUcsSUFBSSxDQUFDLEtBQUssQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUEwQixDQUFBO3dCQUM1RCxVQUFVLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFBO3FCQUN6QjtvQkFBQyxPQUFPLENBQUMsRUFBRTt3QkFDVixJQUFJLEtBQUssQ0FBQyxJQUFJLEtBQUssUUFBUSxFQUFFOzRCQUMzQixPQUFPLENBQUMsSUFBSSxDQUFDLG1EQUFtRCxFQUFFLENBQUMsQ0FBQyxDQUFBO3lCQUNyRTtxQkFDRjtnQkFDSCxDQUFDO2FBQ0YsQ0FBQyxDQUFFLENBQUUsQ0FBQTtRQUNSLENBQUM7OztPQUFBO0lBRUQsc0JBQUksOENBQVU7YUFBZDtZQUNFLE9BQU8sbUJBQW1CLENBQUMsSUFBSSxDQUFDLFVBQVUsQ0FBQyxXQUFXLENBQUMsSUFBSSxlQUFlLENBQWdDO2dCQUN4RyxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7O29CQUN6QixVQUFVLENBQUMsT0FBTyxDQUFDLE1BQUEsTUFBQSxNQUFBLE1BQUEsS0FBSyxhQUFMLEtBQUssdUJBQUwsS0FBSyxDQUFFLE9BQU8sMENBQUcsQ0FBQyxDQUFDLDBDQUFFLEtBQUssMENBQUUsT0FBTyxtQ0FBSSxFQUFFLENBQUMsQ0FBQTtnQkFDL0QsQ0FBQzthQUNGLENBQUMsQ0FBRSxDQUFFLENBQUE7UUFDUixDQUFDOzs7T0FBQTtJQUNILDZCQUFDO0FBQUQsQ0FBQyxBQTFDRCxJQTBDQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7IHR5cGUgUGFyc2VkRXZlbnQgfSBmcm9tICcuLi8uLi9ldmVudHNvdXJjZV9wYXJzZXInXG5pbXBvcnQge1xuICBjcmVhdGVBc3luY0l0ZXJhYmxlLFxuICBUcmFuc2Zvcm1TdHJlYW0sXG4gIFRleHREZWNvZGVyU3RyZWFtLFxuICBjcmVhdGVFdmVudFNvdXJjZVBhcnNlclRyYW5zZm9ybVN0cmVhbSxcbiAgdG9Qb2x5ZmlsbFJlYWRhYmxlLFxufSBmcm9tICcuLi8uLi91dGlscydcbmltcG9ydCB0eXBlIHsgWmhpUHVHZW5lcmF0ZVRleHRPdXRwdXQsIFpoaVB1U3RyZWFtVGV4dE91dHB1dCB9IGZyb20gJy4vdHlwZSdcbmltcG9ydCB0eXBlIHsgQ2hhdE1vZGVsLCBTdHJlYW1UZXh0UmVzdWx0LCBNb2RlbFJlcSwgQmFzZUNoYXRNb2RlbElucHV0IH0gZnJvbSAnLi4vLi4vdHlwZSdcblxuZXhwb3J0IGNsYXNzIFpoaVB1TW9kZWwgaW1wbGVtZW50cyBDaGF0TW9kZWwge1xuICBwdWJsaWMgc3ViVXJsID0gJ3poaXB1L2FwaS9wYWFzL3Y0L2NoYXQvY29tcGxldGlvbnMnXG4gIGNvbnN0cnVjdG9yKHByaXZhdGUgcmVxOiBNb2RlbFJlcSwgcHVibGljIGJhc2VVcmw6IHN0cmluZywgc3ViVXJsPzogc3RyaW5nKSB7XG4gICAgaWYgKHN1YlVybCAhPSBudWxsKSB7XG4gICAgICB0aGlzLnN1YlVybCA9IHN1YlVybFxuICAgIH1cbiAgfVxuXG4gIHByaXZhdGUgZ2V0IHVybCgpIHtcbiAgICByZXR1cm4gYCR7dGhpcy5iYXNlVXJsfS8ke3RoaXMuc3ViVXJsfWBcbiAgfVxuXG4gIGFzeW5jIGdlbmVyYXRlVGV4dChkYXRhOiBCYXNlQ2hhdE1vZGVsSW5wdXQpIHtcbiAgICBjb25zdCByZXMgPSAoYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IGZhbHNlLFxuICAgIH0pKSBhcyBaaGlQdUdlbmVyYXRlVGV4dE91dHB1dFxuICAgIHJldHVybiB7XG4gICAgICB0ZXh0OiAocmVzPy5jaG9pY2VzPy5bMF0/Lm1lc3NhZ2U/LmNvbnRlbnQgYXMgc3RyaW5nKSA/PyAnJyxcbiAgICAgIHJhd1Jlc3BvbnNlOiByZXMsXG4gICAgfVxuICB9XG5cbiAgYXN5bmMgc3RyZWFtVGV4dChkYXRhOiBCYXNlQ2hhdE1vZGVsSW5wdXQpIHtcbiAgICBjb25zdCBzdHJlYW0gPSBhd2FpdCB0aGlzLnJlcSh7XG4gICAgICB1cmw6IHRoaXMudXJsLFxuICAgICAgZGF0YToge1xuICAgICAgICAuLi5kYXRhLFxuICAgICAgICBzdHJlYW06IHRydWUsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiB0cnVlLFxuICAgIH0pXG4gICAgcmV0dXJuIG5ldyBaaGlQdU1vZGVsU3RyZWFtUmVzdWx0KHN0cmVhbSlcbiAgfVxufVxuXG5jbGFzcyBaaGlQdU1vZGVsU3RyZWFtUmVzdWx0IGltcGxlbWVudHMgU3RyZWFtVGV4dFJlc3VsdCB7XG4gIHByaXZhdGUgX2V2ZW50U291cmNlU3RyZWFtOiBSZWFkYWJsZVN0cmVhbTxQYXJzZWRFdmVudD5cblxuICBjb25zdHJ1Y3Rvcihfc3RyZWFtOiBSZWFkYWJsZVN0cmVhbTxVaW50OEFycmF5Pikge1xuICAgIGNvbnN0IHN0cmVhbSA9IHRvUG9seWZpbGxSZWFkYWJsZShfc3RyZWFtKSBhcyB0eXBlb2YgX3N0cmVhbVxuICAgIHRoaXMuX2V2ZW50U291cmNlU3RyZWFtID0gc3RyZWFtXG4gICAgICAucGlwZVRocm91Z2gobmV3IFRleHREZWNvZGVyU3RyZWFtKCkpXG4gICAgICAucGlwZVRocm91Z2goY3JlYXRlRXZlbnRTb3VyY2VQYXJzZXJUcmFuc2Zvcm1TdHJlYW0oKSlcbiAgfVxuXG4gIHByaXZhdGUgZ2V0IHRlZWVkU3RyZWFtKCkge1xuICAgIGNvbnN0IFtzMSwgczJdID0gdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0udGVlKClcbiAgICB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbSA9IHMyXG4gICAgcmV0dXJuIHMxXG4gIH1cblxuICBnZXQgZXZlbnRTb3VyY2VTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy50ZWVlZFN0cmVhbSlcbiAgfVxuXG4gIGdldCBkYXRhU3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMuZXZlbnRTb3VyY2VTdHJlYW0ucGlwZVRocm91Z2gobmV3IFRyYW5zZm9ybVN0cmVhbTxQYXJzZWRFdmVudCwgWmhpUHVTdHJlYW1UZXh0T3V0cHV0Pih7XG4gICAgICB0cmFuc2Zvcm0oY2h1bmssIGNvbnRyb2xsZXIpIHtcbiAgICAgICAgdHJ5IHtcbiAgICAgICAgICBjb25zdCBkYXRhID0gSlNPTi5wYXJzZShjaHVuay5kYXRhKSBhcyBaaGlQdVN0cmVhbVRleHRPdXRwdXRcbiAgICAgICAgICBjb250cm9sbGVyLmVucXVldWUoZGF0YSlcbiAgICAgICAgfSBjYXRjaCAoZSkge1xuICAgICAgICAgIGlmIChjaHVuay5kYXRhICE9PSAnW0RPTkVdJykge1xuICAgICAgICAgICAgY29uc29sZS53YXJuKCdFcnJvciB3aGVuIHRyYW5zZm9ybWluZyBldmVudCBzb3VyY2UgZGF0YSB0byBqc29uJywgZSlcbiAgICAgICAgICB9XG4gICAgICAgIH1cbiAgICAgIH0sXG4gICAgfSksKSwpXG4gIH1cblxuICBnZXQgdGV4dFN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLmRhdGFTdHJlYW0ucGlwZVRocm91Z2gobmV3IFRyYW5zZm9ybVN0cmVhbTxaaGlQdVN0cmVhbVRleHRPdXRwdXQsIHN0cmluZz4oe1xuICAgICAgdHJhbnNmb3JtKGNodW5rLCBjb250cm9sbGVyKSB7XG4gICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZShjaHVuaz8uY2hvaWNlcz8uWzBdPy5kZWx0YT8uY29udGVudCA/PyAnJylcbiAgICAgIH0sXG4gICAgfSksKSwpXG4gIH1cbn1cbiJdfQ==
+export { ZhiPuSimpleModel };
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1poaVB1L2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsT0FBTyxFQUNMLG1CQUFtQixFQUNuQixlQUFlLEVBQ2Ysa0JBQWtCLEVBQ2xCLGtCQUFrQixFQUNsQiwwQkFBMEIsR0FDM0IsTUFBTSxhQUFhLENBQUE7QUFXcEIsU0FBUyxZQUFZLENBQUMsS0FBeUI7SUFDckMsSUFBQSxRQUFRLEdBQW9ELEtBQUssU0FBekQsRUFBRSxLQUFLLEdBQTZDLEtBQUssTUFBbEQsRUFBRSxXQUFXLEdBQWdDLEtBQUssWUFBckMsRUFBRSxXQUFXLEdBQW1CLEtBQUssWUFBeEIsRUFBRSxLQUFLLEdBQVksS0FBSyxNQUFqQixFQUFFLEtBQUssR0FBSyxLQUFLLE1BQVYsQ0FBVTtJQUV6RSxJQUFNLGlCQUFpQixHQUFHO1FBQ3hCLElBQUksV0FBVyxJQUFJLFdBQVcsS0FBSyxNQUFNLEVBQUU7WUFDekMsT0FBTyxDQUFDLElBQUksQ0FBQywrQkFBK0IsQ0FBQyxDQUFBO1NBQzlDO1FBQ0QsT0FBTyxXQUFrQixDQUFBO0lBQzNCLENBQUMsQ0FBQTtJQUVELDZCQUNLLEtBQUssS0FDUixRQUFRLFVBQUEsRUFDUixLQUFLLE9BQUEsRUFDTCxXQUFXLGFBQUEsRUFDWCxXQUFXLEVBQUUsaUJBQWlCLEVBQUUsRUFDaEMsS0FBSyxFQUFFLEtBQUssYUFBTCxLQUFLLHVCQUFMLEtBQUssQ0FBRSxHQUFHLENBQUMsVUFBQSxJQUFJLElBQUksT0FBQSx1QkFDckIsSUFBSSxLQUNQLFFBQVEsd0JBQU8sSUFBSSxDQUFDLFFBQVEsS0FBRSxVQUFVLEVBQUUsSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsUUFBUSxDQUFDLFVBQVUsQ0FBQyxPQUM5RSxFQUh3QixDQUd4QixDQUFDLEVBQ0gsS0FBSyxPQUFBLElBQ047QUFDSCxDQUFDO0FBRUQ7SUFFRSwwQkFBb0IsR0FBYSxFQUFTLE9BQWUsRUFBRSxNQUFlO1FBQXRELFFBQUcsR0FBSCxHQUFHLENBQVU7UUFBUyxZQUFPLEdBQVAsT0FBTyxDQUFRO1FBRGxELFdBQU0sR0FBRyxvQ0FBb0MsQ0FBQTtRQUVsRCxJQUFJLE1BQU0sSUFBSSxJQUFJLEVBQUU7WUFDbEIsSUFBSSxDQUFDLE1BQU0sR0FBRyxNQUFNLENBQUE7U0FDckI7SUFDSCxDQUFDO0lBRUQsc0JBQVksaUNBQUc7YUFBZjtZQUNFLE9BQU8sVUFBRyxJQUFJLENBQUMsT0FBTyxjQUFJLElBQUksQ0FBQyxNQUFNLENBQUUsQ0FBQTtRQUN6QyxDQUFDOzs7T0FBQTtJQUVZLHFDQUFVLEdBQXZCLFVBQXdCLEtBQXlCOzs7Ozs7d0JBQ3pDLElBQUksR0FBRyxZQUFZLENBQUMsS0FBSyxDQUFDLENBQUE7d0JBQ25CLFdBQU0sSUFBSSxDQUFDLEdBQUcsQ0FBQztnQ0FDMUIsR0FBRyxFQUFFLElBQUksQ0FBQyxHQUFHO2dDQUNiLElBQUksd0JBQ0MsSUFBSSxLQUNQLE1BQU0sRUFBRSxLQUFLLEdBQ2Q7Z0NBQ0QsTUFBTSxFQUFFLEtBQUs7NkJBQ2QsQ0FBQyxFQUFBOzt3QkFQSSxHQUFHLEdBQUcsQ0FBQyxTQU9YLENBQTRCO3dCQUM5QixpQ0FBWSxHQUFHLEtBQUUsV0FBVyxFQUFFLEdBQUcsS0FBRTs7OztLQUNwQztJQUVZLG1DQUFRLEdBQXJCLFVBQXNCLEtBQXlCOzs7Ozs7d0JBQ3ZDLElBQUksR0FBRyxZQUFZLENBQUMsS0FBSyxDQUFDLENBQUE7d0JBQzVCLFVBQVUsR0FBbUIsSUFBSSxDQUFBO3dCQUNyQixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7Z0NBQzdCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRztnQ0FDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsSUFBSSxHQUNiO2dDQUNELE1BQU0sRUFBRSxJQUFJOzZCQUNiLENBQUMsRUFBQTs7d0JBUEksT0FBTyxHQUFHLFNBT2Q7d0JBQ0ksTUFBTSxHQUFHLGtCQUFrQixDQUFDLE9BQU8sQ0FBbUIsQ0FBQTt3QkFFdEQsV0FBVyxHQUFHLGtCQUFrQixDQUF3QixNQUFNLENBQUMsQ0FBQTt3QkFDL0QsYUFBYSxHQUFHLFdBQVcsQ0FBQyxXQUFXLENBQUMsSUFBSSxlQUFlLENBQXlFOzRCQUN4SSxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7Z0NBQ3pCLElBQU0sVUFBVSxHQUFHLEtBQUssQ0FBQyxPQUFPLENBQUMsR0FBRyxDQUFDLFVBQUMsTUFBTTtvQ0FDMUMsSUFBTSxPQUFPLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQTtvQ0FDNUIsSUFBSSxVQUFVLElBQUksSUFBSTt3Q0FBRSxVQUFVLEdBQUcsMEJBQTBCLENBQUMsT0FBTyxDQUFDLENBQUE7b0NBQ3hFLElBQUksVUFBVSxFQUFFO3dDQUNkLDZCQUNLLE1BQU0sS0FDVCxhQUFhLEVBQUUsWUFBcUIsRUFDcEMsS0FBSyxFQUFFLE9BQU8sSUFDZjtxQ0FDRjtvQ0FDRCxPQUFPLE1BQU0sQ0FBQTtnQ0FDZixDQUFDLENBQUMsQ0FBQTtnQ0FDRixJQUFNLFFBQVEseUJBQVEsS0FBSyxLQUFFLE9BQU8sRUFBRSxVQUFVLEdBQUUsQ0FBQTtnQ0FDbEQsVUFBVSxDQUFDLE9BQU8sdUJBQU0sUUFBUSxLQUFFLFdBQVcsRUFBRSxLQUFLLElBQUcsQ0FBQTs0QkFDekQsQ0FBQzt5QkFDRixDQUFDLENBQUUsQ0FBQTt3QkFFSixXQUFPLG1CQUFtQixDQUFDLGFBQWEsQ0FBQyxFQUFBOzs7O0tBQzFDO0lBQ0gsdUJBQUM7QUFBRCxDQUFDLEFBNURELElBNERDIiwic291cmNlc0NvbnRlbnQiOlsiaW1wb3J0IHtcbiAgY3JlYXRlQXN5bmNJdGVyYWJsZSxcbiAgVHJhbnNmb3JtU3RyZWFtLFxuICB0b1BvbHlmaWxsUmVhZGFibGUsXG4gIGludG9TdGFuZGFyZFN0cmVhbSxcbiAgaXNUb29sQ2FsbEFzc2lzdGFudE1lc3NhZ2UsXG59IGZyb20gJy4uLy4uL3V0aWxzJ1xuaW1wb3J0IHR5cGUgeyBaaGlQdUdlbmVyYXRlVGV4dE91dHB1dCwgWmhpUHVJbnB1dERhdGEsIFpoaVB1U3RyZWFtVGV4dE91dHB1dCB9IGZyb20gJy4vdHlwZSdcbmltcG9ydCB0eXBlIHtcbiAgTW9kZWxSZXEsXG4gIEJhc2VDaGF0TW9kZWxJbnB1dCxcbiAgU2ltcGxlQ2hhdE1vZGVsLFxuICBEb1N0cmVhbU91dHB1dCxcbiAgQmFzZURvU3RyZWFtT3V0cHV0Q2h1bmssXG4gIERvR2VuZXJhdGVPdXRwdXQsXG59IGZyb20gJy4uLy4uL3R5cGUnXG5cbmZ1bmN0aW9uIHByb2Nlc3NJbnB1dChpbnB1dDogQmFzZUNoYXRNb2RlbElucHV0KTogWmhpUHVJbnB1dERhdGEge1xuICBjb25zdCB7IG1lc3NhZ2VzLCBtb2RlbCwgdGVtcGVyYXR1cmUsIHRvb2xfY2hvaWNlLCB0b29scywgdG9wX3AgfSA9IGlucHV0XG5cbiAgY29uc3QgcHJvY2Vzc1Rvb2xDaG9pY2UgPSAoKSA9PiB7XG4gICAgaWYgKHRvb2xfY2hvaWNlICYmIHRvb2xfY2hvaWNlICE9PSAnYXV0bycpIHtcbiAgICAgIGNvbnNvbGUud2FybignYHRvb2xfY2hvaWNlYCBpcyBub3QgXFwnYXV0b1xcJycpXG4gICAgfVxuICAgIHJldHVybiB0b29sX2Nob2ljZSBhcyBhbnlcbiAgfVxuXG4gIHJldHVybiB7XG4gICAgLi4uaW5wdXQsXG4gICAgbWVzc2FnZXMsXG4gICAgbW9kZWwsXG4gICAgdGVtcGVyYXR1cmUsXG4gICAgdG9vbF9jaG9pY2U6IHByb2Nlc3NUb29sQ2hvaWNlKCksXG4gICAgdG9vbHM6IHRvb2xzPy5tYXAodG9vbCA9PiAoe1xuICAgICAgLi4udG9vbCxcbiAgICAgIGZ1bmN0aW9uOiB7IC4uLnRvb2wuZnVuY3Rpb24sIHBhcmFtZXRlcnM6IEpTT04ucGFyc2UodG9vbC5mdW5jdGlvbi5wYXJhbWV0ZXJzKSB9LFxuICAgIH0pKSxcbiAgICB0b3BfcCxcbiAgfVxufVxuXG5leHBvcnQgY2xhc3MgWmhpUHVTaW1wbGVNb2RlbCBpbXBsZW1lbnRzIFNpbXBsZUNoYXRNb2RlbCB7XG4gIHB1YmxpYyBzdWJVcmwgPSAnemhpcHUvYXBpL3BhYXMvdjQvY2hhdC9jb21wbGV0aW9ucydcbiAgY29uc3RydWN0b3IocHJpdmF0ZSByZXE6IE1vZGVsUmVxLCBwdWJsaWMgYmFzZVVybDogc3RyaW5nLCBzdWJVcmw/OiBzdHJpbmcpIHtcbiAgICBpZiAoc3ViVXJsICE9IG51bGwpIHtcbiAgICAgIHRoaXMuc3ViVXJsID0gc3ViVXJsXG4gICAgfVxuICB9XG5cbiAgcHJpdmF0ZSBnZXQgdXJsKCkge1xuICAgIHJldHVybiBgJHt0aGlzLmJhc2VVcmx9LyR7dGhpcy5zdWJVcmx9YFxuICB9XG5cbiAgcHVibGljIGFzeW5jIGRvR2VuZXJhdGUoX2RhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCk6IFByb21pc2U8RG9HZW5lcmF0ZU91dHB1dD4ge1xuICAgIGNvbnN0IGRhdGEgPSBwcm9jZXNzSW5wdXQoX2RhdGEpXG4gICAgY29uc3QgcmVzID0gKGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICB9KSkgYXMgWmhpUHVHZW5lcmF0ZVRleHRPdXRwdXRcbiAgICByZXR1cm4geyAuLi5yZXMsIHJhd1Jlc3BvbnNlOiByZXMgfVxuICB9XG5cbiAgcHVibGljIGFzeW5jIGRvU3RyZWFtKF9kYXRhOiBCYXNlQ2hhdE1vZGVsSW5wdXQpOiBQcm9taXNlPERvU3RyZWFtT3V0cHV0PiB7XG4gICAgY29uc3QgZGF0YSA9IHByb2Nlc3NJbnB1dChfZGF0YSlcbiAgICBsZXQgaXNUb29sQ2FsbDogbnVsbCB8IGJvb2xlYW4gPSBudWxsXG4gICAgY29uc3QgX3N0cmVhbSA9IGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IHRydWUsXG4gICAgfSlcbiAgICBjb25zdCBzdHJlYW0gPSB0b1BvbHlmaWxsUmVhZGFibGUoX3N0cmVhbSkgYXMgdHlwZW9mIF9zdHJlYW1cblxuICAgIGNvbnN0IHpoaXB1U3RyZWFtID0gaW50b1N0YW5kYXJkU3RyZWFtPFpoaVB1U3RyZWFtVGV4dE91dHB1dD4oc3RyZWFtKVxuICAgIGNvbnN0IHN0cmVhbVdpdGhSYXcgPSB6aGlwdVN0cmVhbS5waXBlVGhyb3VnaChuZXcgVHJhbnNmb3JtU3RyZWFtPFpoaVB1U3RyZWFtVGV4dE91dHB1dCwgQmFzZURvU3RyZWFtT3V0cHV0Q2h1bmsgJiB7IHJhd1Jlc3BvbnNlPzogYW55IH0+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICBjb25zdCBuZXdDaG9pY2VzID0gY2h1bmsuY2hvaWNlcy5tYXAoKGNob2ljZSkgPT4ge1xuICAgICAgICAgIGNvbnN0IG1lc3NhZ2UgPSBjaG9pY2UuZGVsdGFcbiAgICAgICAgICBpZiAoaXNUb29sQ2FsbCA9PSBudWxsKSBpc1Rvb2xDYWxsID0gaXNUb29sQ2FsbEFzc2lzdGFudE1lc3NhZ2UobWVzc2FnZSlcbiAgICAgICAgICBpZiAoaXNUb29sQ2FsbCkge1xuICAgICAgICAgICAgcmV0dXJuIHtcbiAgICAgICAgICAgICAgLi4uY2hvaWNlLFxuICAgICAgICAgICAgICBmaW5pc2hfcmVhc29uOiAndG9vbF9jYWxscycgYXMgY29uc3QsXG4gICAgICAgICAgICAgIGRlbHRhOiBtZXNzYWdlLFxuICAgICAgICAgICAgfVxuICAgICAgICAgIH1cbiAgICAgICAgICByZXR1cm4gY2hvaWNlXG4gICAgICAgIH0pXG4gICAgICAgIGNvbnN0IG5ld0NodW5rID0geyAuLi5jaHVuaywgY2hvaWNlczogbmV3Q2hvaWNlcyB9XG4gICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZSh7IC4uLm5ld0NodW5rLCByYXdSZXNwb25zZTogY2h1bmsgfSlcbiAgICAgIH0sXG4gICAgfSksKVxuXG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUoc3RyZWFtV2l0aFJhdylcbiAgfVxufVxuIl19

package/dist/esm/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/esm/models/ZhiPu/type.js

@@ -1,2 +1,2 @@
 export {};
-//# 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=