@cloudbase/ai 2.7.3-beta.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (96) hide show
  1. package/.eslintignore +2 -0
  2. package/.eslintrc.js +15 -0
  3. package/LICENSE +202 -0
  4. package/dist/cjs/index.d.ts +40 -0
  5. package/dist/cjs/index.js +171 -0
  6. package/dist/cjs/models/HunYuan/index.d.ts +25 -0
  7. package/dist/cjs/models/HunYuan/index.js +191 -0
  8. package/dist/cjs/models/HunYuan/type.d.ts +52 -0
  9. package/dist/cjs/models/HunYuan/type.js +3 -0
  10. package/dist/cjs/models/HunYuanPreview/index.d.ts +25 -0
  11. package/dist/cjs/models/HunYuanPreview/index.js +191 -0
  12. package/dist/cjs/models/HunYuanPreview/type.d.ts +52 -0
  13. package/dist/cjs/models/HunYuanPreview/type.js +3 -0
  14. package/dist/cjs/models/HuoShan/index.d.ts +24 -0
  15. package/dist/cjs/models/HuoShan/index.js +163 -0
  16. package/dist/cjs/models/HuoShan/type.d.ts +38 -0
  17. package/dist/cjs/models/HuoShan/type.js +3 -0
  18. package/dist/cjs/models/LingJi/index.d.ts +24 -0
  19. package/dist/cjs/models/LingJi/index.js +163 -0
  20. package/dist/cjs/models/LingJi/type.d.ts +43 -0
  21. package/dist/cjs/models/LingJi/type.js +3 -0
  22. package/dist/cjs/models/LingYi/index.d.ts +24 -0
  23. package/dist/cjs/models/LingYi/index.js +163 -0
  24. package/dist/cjs/models/LingYi/type.d.ts +49 -0
  25. package/dist/cjs/models/LingYi/type.js +3 -0
  26. package/dist/cjs/models/YueZhi/index.d.ts +24 -0
  27. package/dist/cjs/models/YueZhi/index.js +163 -0
  28. package/dist/cjs/models/YueZhi/type.d.ts +31 -0
  29. package/dist/cjs/models/YueZhi/type.js +3 -0
  30. package/dist/cjs/models/ZhiPu/index.d.ts +24 -0
  31. package/dist/cjs/models/ZhiPu/index.js +163 -0
  32. package/dist/cjs/models/ZhiPu/type.d.ts +71 -0
  33. package/dist/cjs/models/ZhiPu/type.js +3 -0
  34. package/dist/cjs/models/index.d.ts +17 -0
  35. package/dist/cjs/models/index.js +27 -0
  36. package/dist/cjs/type.d.ts +37 -0
  37. package/dist/cjs/type.js +16 -0
  38. package/dist/cjs/utils.d.ts +28 -0
  39. package/dist/cjs/utils.js +152 -0
  40. package/dist/esm/index.d.ts +40 -0
  41. package/dist/esm/index.js +141 -0
  42. package/dist/esm/models/HunYuan/index.d.ts +25 -0
  43. package/dist/esm/models/HunYuan/index.js +188 -0
  44. package/dist/esm/models/HunYuan/type.d.ts +52 -0
  45. package/dist/esm/models/HunYuan/type.js +2 -0
  46. package/dist/esm/models/HunYuanPreview/index.d.ts +25 -0
  47. package/dist/esm/models/HunYuanPreview/index.js +188 -0
  48. package/dist/esm/models/HunYuanPreview/type.d.ts +52 -0
  49. package/dist/esm/models/HunYuanPreview/type.js +2 -0
  50. package/dist/esm/models/HuoShan/index.d.ts +24 -0
  51. package/dist/esm/models/HuoShan/index.js +160 -0
  52. package/dist/esm/models/HuoShan/type.d.ts +38 -0
  53. package/dist/esm/models/HuoShan/type.js +2 -0
  54. package/dist/esm/models/LingJi/index.d.ts +24 -0
  55. package/dist/esm/models/LingJi/index.js +160 -0
  56. package/dist/esm/models/LingJi/type.d.ts +43 -0
  57. package/dist/esm/models/LingJi/type.js +2 -0
  58. package/dist/esm/models/LingYi/index.d.ts +24 -0
  59. package/dist/esm/models/LingYi/index.js +160 -0
  60. package/dist/esm/models/LingYi/type.d.ts +49 -0
  61. package/dist/esm/models/LingYi/type.js +2 -0
  62. package/dist/esm/models/YueZhi/index.d.ts +24 -0
  63. package/dist/esm/models/YueZhi/index.js +160 -0
  64. package/dist/esm/models/YueZhi/type.d.ts +31 -0
  65. package/dist/esm/models/YueZhi/type.js +2 -0
  66. package/dist/esm/models/ZhiPu/index.d.ts +24 -0
  67. package/dist/esm/models/ZhiPu/index.js +160 -0
  68. package/dist/esm/models/ZhiPu/type.d.ts +71 -0
  69. package/dist/esm/models/ZhiPu/type.js +2 -0
  70. package/dist/esm/models/index.d.ts +17 -0
  71. package/dist/esm/models/index.js +18 -0
  72. package/dist/esm/type.d.ts +37 -0
  73. package/dist/esm/type.js +13 -0
  74. package/dist/esm/utils.d.ts +28 -0
  75. package/dist/esm/utils.js +147 -0
  76. package/package.json +36 -0
  77. package/src/index.ts +77 -0
  78. package/src/models/HunYuan/index.ts +110 -0
  79. package/src/models/HunYuan/type.ts +61 -0
  80. package/src/models/HunYuanPreview/index.ts +110 -0
  81. package/src/models/HunYuanPreview/type.ts +61 -0
  82. package/src/models/HuoShan/index.ts +92 -0
  83. package/src/models/HuoShan/type.ts +41 -0
  84. package/src/models/LingJi/index.ts +92 -0
  85. package/src/models/LingJi/type.ts +34 -0
  86. package/src/models/LingYi/index.ts +92 -0
  87. package/src/models/LingYi/type.ts +49 -0
  88. package/src/models/YueZhi/index.ts +92 -0
  89. package/src/models/YueZhi/type.ts +28 -0
  90. package/src/models/ZhiPu/index.ts +92 -0
  91. package/src/models/ZhiPu/type.ts +60 -0
  92. package/src/models/index.ts +19 -0
  93. package/src/type.ts +42 -0
  94. package/src/utils.ts +110 -0
  95. package/tsconfig.esm.json +16 -0
  96. package/tsconfig.json +16 -0
@@ -0,0 +1,163 @@
1
+ "use strict";
2
+ var __assign = (this && this.__assign) || function () {
3
+ __assign = Object.assign || function(t) {
4
+ for (var s, i = 1, n = arguments.length; i < n; i++) {
5
+ s = arguments[i];
6
+ for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
7
+ t[p] = s[p];
8
+ }
9
+ return t;
10
+ };
11
+ return __assign.apply(this, arguments);
12
+ };
13
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
14
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
15
+ return new (P || (P = Promise))(function (resolve, reject) {
16
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
17
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
18
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
19
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
20
+ });
21
+ };
22
+ var __generator = (this && this.__generator) || function (thisArg, body) {
23
+ var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
24
+ return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
25
+ function verb(n) { return function (v) { return step([n, v]); }; }
26
+ function step(op) {
27
+ if (f) throw new TypeError("Generator is already executing.");
28
+ while (g && (g = 0, op[0] && (_ = 0)), _) try {
29
+ if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
30
+ if (y = 0, t) op = [op[0] & 2, t.value];
31
+ switch (op[0]) {
32
+ case 0: case 1: t = op; break;
33
+ case 4: _.label++; return { value: op[1], done: false };
34
+ case 5: _.label++; y = op[1]; op = [0]; continue;
35
+ case 7: op = _.ops.pop(); _.trys.pop(); continue;
36
+ default:
37
+ if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
38
+ if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
39
+ if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
40
+ if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
41
+ if (t[2]) _.ops.pop();
42
+ _.trys.pop(); continue;
43
+ }
44
+ op = body.call(thisArg, _);
45
+ } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
46
+ if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
47
+ }
48
+ };
49
+ Object.defineProperty(exports, "__esModule", { value: true });
50
+ exports.LingYiModel = void 0;
51
+ var utils_1 = require("../../utils");
52
+ var LingYiModel = (function () {
53
+ function LingYiModel(req, baseUrl, subUrl) {
54
+ this.req = req;
55
+ this.baseUrl = baseUrl;
56
+ this.subUrl = 'lingyiwanwu/v1/chat/completion';
57
+ if (subUrl != null) {
58
+ this.subUrl = subUrl;
59
+ }
60
+ }
61
+ Object.defineProperty(LingYiModel.prototype, "url", {
62
+ get: function () {
63
+ return "".concat(this.baseUrl, "/").concat(this.subUrl);
64
+ },
65
+ enumerable: false,
66
+ configurable: true
67
+ });
68
+ LingYiModel.prototype.generateText = function (data) {
69
+ var _a, _b, _c, _d;
70
+ return __awaiter(this, void 0, void 0, function () {
71
+ var res;
72
+ return __generator(this, function (_e) {
73
+ switch (_e.label) {
74
+ case 0: return [4, this.req({
75
+ url: this.url,
76
+ data: __assign(__assign({}, data), { stream: false }),
77
+ stream: false,
78
+ })];
79
+ case 1:
80
+ res = (_e.sent());
81
+ return [2, {
82
+ 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 : '',
83
+ rawResponse: res,
84
+ }];
85
+ }
86
+ });
87
+ });
88
+ };
89
+ LingYiModel.prototype.streamText = function (data) {
90
+ return __awaiter(this, void 0, void 0, function () {
91
+ var stream;
92
+ return __generator(this, function (_a) {
93
+ switch (_a.label) {
94
+ case 0: return [4, this.req({
95
+ url: this.url,
96
+ data: __assign(__assign({}, data), { stream: true }),
97
+ stream: true,
98
+ })];
99
+ case 1:
100
+ stream = _a.sent();
101
+ return [2, new LingYiModelStreamResult(stream)];
102
+ }
103
+ });
104
+ });
105
+ };
106
+ return LingYiModel;
107
+ }());
108
+ exports.LingYiModel = LingYiModel;
109
+ var LingYiModelStreamResult = (function () {
110
+ function LingYiModelStreamResult(_stream) {
111
+ var stream = (0, utils_1.toPolyfillReadable)(_stream);
112
+ this._eventSourceStream = stream
113
+ .pipeThrough(new utils_1.TextDecoderStream())
114
+ .pipeThrough((0, utils_1.createEventSourceParserTransformStream)());
115
+ }
116
+ Object.defineProperty(LingYiModelStreamResult.prototype, "teeedStream", {
117
+ get: function () {
118
+ var _a = this._eventSourceStream.tee(), s1 = _a[0], s2 = _a[1];
119
+ this._eventSourceStream = s2;
120
+ return s1;
121
+ },
122
+ enumerable: false,
123
+ configurable: true
124
+ });
125
+ Object.defineProperty(LingYiModelStreamResult.prototype, "eventSourceStream", {
126
+ get: function () {
127
+ return (0, utils_1.createAsyncIterable)(this.teeedStream);
128
+ },
129
+ enumerable: false,
130
+ configurable: true
131
+ });
132
+ Object.defineProperty(LingYiModelStreamResult.prototype, "dataStream", {
133
+ get: function () {
134
+ return (0, utils_1.createAsyncIterable)(this.eventSourceStream.pipeThrough(new utils_1.TransformStream({
135
+ transform: function (chunk, controller) {
136
+ try {
137
+ var data = JSON.parse(chunk.data);
138
+ controller.enqueue(data);
139
+ }
140
+ catch (e) {
141
+ console.warn('Error when transforming event source data to json', e);
142
+ }
143
+ },
144
+ })));
145
+ },
146
+ enumerable: false,
147
+ configurable: true
148
+ });
149
+ Object.defineProperty(LingYiModelStreamResult.prototype, "textStream", {
150
+ get: function () {
151
+ return (0, utils_1.createAsyncIterable)(this.dataStream.pipeThrough(new utils_1.TransformStream({
152
+ transform: function (chunk, controller) {
153
+ var _a, _b, _c, _d;
154
+ 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 : '');
155
+ },
156
+ })));
157
+ },
158
+ enumerable: false,
159
+ configurable: true
160
+ });
161
+ return LingYiModelStreamResult;
162
+ }());
163
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL0xpbmdZaS9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUNBLHFDQU1vQjtBQUlwQjtJQUVFLHFCQUFvQixHQUFhLEVBQVMsT0FBZSxFQUFFLE1BQWU7UUFBdEQsUUFBRyxHQUFILEdBQUcsQ0FBVTtRQUFTLFlBQU8sR0FBUCxPQUFPLENBQVE7UUFEbEQsV0FBTSxHQUFHLGdDQUFnQyxDQUFBO1FBRTlDLElBQUksTUFBTSxJQUFJLElBQUksRUFBRTtZQUNsQixJQUFJLENBQUMsTUFBTSxHQUFHLE1BQU0sQ0FBQTtTQUNyQjtJQUNILENBQUM7SUFFRCxzQkFBWSw0QkFBRzthQUFmO1lBQ0UsT0FBTyxVQUFHLElBQUksQ0FBQyxPQUFPLGNBQUksSUFBSSxDQUFDLE1BQU0sQ0FBRSxDQUFBO1FBQ3pDLENBQUM7OztPQUFBO0lBRUssa0NBQVksR0FBbEIsVUFBbUIsSUFBd0I7Ozs7Ozs0QkFDNUIsV0FBTSxJQUFJLENBQUMsR0FBRyxDQUFDOzRCQUMxQixHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUc7NEJBQ2IsSUFBSSx3QkFDQyxJQUFJLEtBQ1AsTUFBTSxFQUFFLEtBQUssR0FDZDs0QkFDRCxNQUFNLEVBQUUsS0FBSzt5QkFDZCxDQUFDLEVBQUE7O3dCQVBJLEdBQUcsR0FBRyxDQUFDLFNBT1gsQ0FBNkI7d0JBQy9CLFdBQU87Z0NBQ0wsSUFBSSxFQUFFLE1BQUMsTUFBQSxNQUFBLE1BQUEsR0FBRyxhQUFILEdBQUcsdUJBQUgsR0FBRyxDQUFFLE9BQU8sMENBQUcsQ0FBQyxDQUFDLDBDQUFFLE9BQU8sMENBQUUsT0FBa0IsbUNBQUksRUFBRTtnQ0FDM0QsV0FBVyxFQUFFLEdBQUc7NkJBQ2pCLEVBQUE7Ozs7S0FDRjtJQUVLLGdDQUFVLEdBQWhCLFVBQWlCLElBQXdCOzs7Ozs0QkFDeEIsV0FBTSxJQUFJLENBQUMsR0FBRyxDQUFDOzRCQUM1QixHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUc7NEJBQ2IsSUFBSSx3QkFDQyxJQUFJLEtBQ1AsTUFBTSxFQUFFLElBQUksR0FDYjs0QkFDRCxNQUFNLEVBQUUsSUFBSTt5QkFDYixDQUFDLEVBQUE7O3dCQVBJLE1BQU0sR0FBRyxTQU9iO3dCQUNGLFdBQU8sSUFBSSx1QkFBdUIsQ0FBQyxNQUFNLENBQUMsRUFBQTs7OztLQUMzQztJQUNILGtCQUFDO0FBQUQsQ0FBQyxBQXRDRCxJQXNDQztBQXRDWSxrQ0FBVztBQXdDeEI7SUFHRSxpQ0FBWSxPQUFtQztRQUM3QyxJQUFNLE1BQU0sR0FBRyxJQUFBLDBCQUFrQixFQUFDLE9BQU8sQ0FBbUIsQ0FBQTtRQUM1RCxJQUFJLENBQUMsa0JBQWtCLEdBQUcsTUFBTTthQUM3QixXQUFXLENBQUMsSUFBSSx5QkFBaUIsRUFBRSxDQUFDO2FBQ3BDLFdBQVcsQ0FBQyxJQUFBLDhDQUFzQyxHQUFFLENBQUMsQ0FBQTtJQUMxRCxDQUFDO0lBRUQsc0JBQVksZ0RBQVc7YUFBdkI7WUFDUSxJQUFBLEtBQVcsSUFBSSxDQUFDLGtCQUFrQixDQUFDLEdBQUcsRUFBRSxFQUF2QyxFQUFFLFFBQUEsRUFBRSxFQUFFLFFBQWlDLENBQUE7WUFDOUMsSUFBSSxDQUFDLGtCQUFrQixHQUFHLEVBQUUsQ0FBQTtZQUM1QixPQUFPLEVBQUUsQ0FBQTtRQUNYLENBQUM7OztPQUFBO0lBRUQsc0JBQUksc0RBQWlCO2FBQXJCO1lBQ0UsT0FBTyxJQUFBLDJCQUFtQixFQUFDLElBQUksQ0FBQyxXQUFXLENBQUMsQ0FBQTtRQUM5QyxDQUFDOzs7T0FBQTtJQUVELHNCQUFJLCtDQUFVO2FBQWQ7WUFDRSxPQUFPLElBQUEsMkJBQW1CLEVBQUMsSUFBSSxDQUFDLGlCQUFpQixDQUFDLFdBQVcsQ0FBQyxJQUFJLHVCQUFlLENBQXNDO2dCQUNySCxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7b0JBQ3pCLElBQUk7d0JBQ0YsSUFBTSxJQUFJLEdBQUcsSUFBSSxDQUFDLEtBQUssQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUEyQixDQUFBO3dCQUM3RCxVQUFVLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFBO3FCQUN6QjtvQkFBQyxPQUFPLENBQUMsRUFBRTt3QkFDVixPQUFPLENBQUMsSUFBSSxDQUFDLG1EQUFtRCxFQUFFLENBQUMsQ0FBQyxDQUFBO3FCQUNyRTtnQkFDSCxDQUFDO2FBQ0YsQ0FBQyxDQUFFLENBQUUsQ0FBQTtRQUNSLENBQUM7OztPQUFBO0lBRUQsc0JBQUksK0NBQVU7YUFBZDtZQUNFLE9BQU8sSUFBQSwyQkFBbUIsRUFBQyxJQUFJLENBQUMsVUFBVSxDQUFDLFdBQVcsQ0FBQyxJQUFJLHVCQUFlLENBQWlDO2dCQUN6RyxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7O29CQUN6QixVQUFVLENBQUMsT0FBTyxDQUFDLE1BQUEsTUFBQSxNQUFBLE1BQUEsS0FBSyxhQUFMLEtBQUssdUJBQUwsS0FBSyxDQUFFLE9BQU8sMENBQUcsQ0FBQyxDQUFDLDBDQUFFLEtBQUssMENBQUUsT0FBTyxtQ0FBSSxFQUFFLENBQUMsQ0FBQTtnQkFDL0QsQ0FBQzthQUNGLENBQUMsQ0FBRSxDQUFFLENBQUE7UUFDUixDQUFDOzs7T0FBQTtJQUNILDhCQUFDO0FBQUQsQ0FBQyxBQXhDRCxJQXdDQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7IHR5cGUgUGFyc2VkRXZlbnQgfSBmcm9tICdldmVudHNvdXJjZS1wYXJzZXIvc3RyZWFtJ1xuaW1wb3J0IHtcbiAgY3JlYXRlQXN5bmNJdGVyYWJsZSxcbiAgVHJhbnNmb3JtU3RyZWFtLFxuICBUZXh0RGVjb2RlclN0cmVhbSxcbiAgY3JlYXRlRXZlbnRTb3VyY2VQYXJzZXJUcmFuc2Zvcm1TdHJlYW0sXG4gIHRvUG9seWZpbGxSZWFkYWJsZSxcbn0gZnJvbSAnLi4vLi4vdXRpbHMnXG5pbXBvcnQgdHlwZSB7IExpbmdZaUdlbmVyYXRlVGV4dE91dHB1dCwgTGluZ1lpU3RyZWFtVGV4dE91dHB1dCB9IGZyb20gJy4vdHlwZSdcbmltcG9ydCB0eXBlIHsgQ2hhdE1vZGVsLCBTdHJlYW1UZXh0UmVzdWx0LCBNb2RlbFJlcSwgQmFzZUNoYXRNb2RlbElucHV0IH0gZnJvbSAnLi4vLi4vdHlwZSdcblxuZXhwb3J0IGNsYXNzIExpbmdZaU1vZGVsIGltcGxlbWVudHMgQ2hhdE1vZGVsIHtcbiAgcHVibGljIHN1YlVybCA9ICdsaW5neWl3YW53dS92MS9jaGF0L2NvbXBsZXRpb24nXG4gIGNvbnN0cnVjdG9yKHByaXZhdGUgcmVxOiBNb2RlbFJlcSwgcHVibGljIGJhc2VVcmw6IHN0cmluZywgc3ViVXJsPzogc3RyaW5nKSB7XG4gICAgaWYgKHN1YlVybCAhPSBudWxsKSB7XG4gICAgICB0aGlzLnN1YlVybCA9IHN1YlVybFxuICAgIH1cbiAgfVxuXG4gIHByaXZhdGUgZ2V0IHVybCgpIHtcbiAgICByZXR1cm4gYCR7dGhpcy5iYXNlVXJsfS8ke3RoaXMuc3ViVXJsfWBcbiAgfVxuXG4gIGFzeW5jIGdlbmVyYXRlVGV4dChkYXRhOiBCYXNlQ2hhdE1vZGVsSW5wdXQpIHtcbiAgICBjb25zdCByZXMgPSAoYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IGZhbHNlLFxuICAgIH0pKSBhcyBMaW5nWWlHZW5lcmF0ZVRleHRPdXRwdXRcbiAgICByZXR1cm4ge1xuICAgICAgdGV4dDogKHJlcz8uY2hvaWNlcz8uWzBdPy5tZXNzYWdlPy5jb250ZW50IGFzIHN0cmluZykgPz8gJycsXG4gICAgICByYXdSZXNwb25zZTogcmVzLFxuICAgIH1cbiAgfVxuXG4gIGFzeW5jIHN0cmVhbVRleHQoZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KSB7XG4gICAgY29uc3Qgc3RyZWFtID0gYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiB0cnVlLFxuICAgICAgfSxcbiAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICB9KVxuICAgIHJldHVybiBuZXcgTGluZ1lpTW9kZWxTdHJlYW1SZXN1bHQoc3RyZWFtKVxuICB9XG59XG5cbmNsYXNzIExpbmdZaU1vZGVsU3RyZWFtUmVzdWx0IGltcGxlbWVudHMgU3RyZWFtVGV4dFJlc3VsdCB7XG4gIHByaXZhdGUgX2V2ZW50U291cmNlU3RyZWFtOiBSZWFkYWJsZVN0cmVhbTxQYXJzZWRFdmVudD5cblxuICBjb25zdHJ1Y3Rvcihfc3RyZWFtOiBSZWFkYWJsZVN0cmVhbTxVaW50OEFycmF5Pikge1xuICAgIGNvbnN0IHN0cmVhbSA9IHRvUG9seWZpbGxSZWFkYWJsZShfc3RyZWFtKSBhcyB0eXBlb2YgX3N0cmVhbVxuICAgIHRoaXMuX2V2ZW50U291cmNlU3RyZWFtID0gc3RyZWFtXG4gICAgICAucGlwZVRocm91Z2gobmV3IFRleHREZWNvZGVyU3RyZWFtKCkpXG4gICAgICAucGlwZVRocm91Z2goY3JlYXRlRXZlbnRTb3VyY2VQYXJzZXJUcmFuc2Zvcm1TdHJlYW0oKSlcbiAgfVxuXG4gIHByaXZhdGUgZ2V0IHRlZWVkU3RyZWFtKCkge1xuICAgIGNvbnN0IFtzMSwgczJdID0gdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0udGVlKClcbiAgICB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbSA9IHMyXG4gICAgcmV0dXJuIHMxXG4gIH1cblxuICBnZXQgZXZlbnRTb3VyY2VTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy50ZWVlZFN0cmVhbSlcbiAgfVxuXG4gIGdldCBkYXRhU3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMuZXZlbnRTb3VyY2VTdHJlYW0ucGlwZVRocm91Z2gobmV3IFRyYW5zZm9ybVN0cmVhbTxQYXJzZWRFdmVudCwgTGluZ1lpU3RyZWFtVGV4dE91dHB1dD4oe1xuICAgICAgdHJhbnNmb3JtKGNodW5rLCBjb250cm9sbGVyKSB7XG4gICAgICAgIHRyeSB7XG4gICAgICAgICAgY29uc3QgZGF0YSA9IEpTT04ucGFyc2UoY2h1bmsuZGF0YSkgYXMgTGluZ1lpU3RyZWFtVGV4dE91dHB1dFxuICAgICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZShkYXRhKVxuICAgICAgICB9IGNhdGNoIChlKSB7XG4gICAgICAgICAgY29uc29sZS53YXJuKCdFcnJvciB3aGVuIHRyYW5zZm9ybWluZyBldmVudCBzb3VyY2UgZGF0YSB0byBqc29uJywgZSlcbiAgICAgICAgfVxuICAgICAgfSxcbiAgICB9KSwpLClcbiAgfVxuXG4gIGdldCB0ZXh0U3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMuZGF0YVN0cmVhbS5waXBlVGhyb3VnaChuZXcgVHJhbnNmb3JtU3RyZWFtPExpbmdZaVN0cmVhbVRleHRPdXRwdXQsIHN0cmluZz4oe1xuICAgICAgdHJhbnNmb3JtKGNodW5rLCBjb250cm9sbGVyKSB7XG4gICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZShjaHVuaz8uY2hvaWNlcz8uWzBdPy5kZWx0YT8uY29udGVudCA/PyAnJylcbiAgICAgIH0sXG4gICAgfSksKSwpXG4gIH1cbn1cbiJdfQ==
@@ -0,0 +1,49 @@
1
+ type SystemMessage = {
2
+ role: 'system';
3
+ content: string;
4
+ };
5
+ type UserMessage = {
6
+ role: 'user';
7
+ content: string;
8
+ };
9
+ type AssistantMessage = {
10
+ role: 'assistant';
11
+ content: string;
12
+ };
13
+ type ToolMessage = {
14
+ role: 'tool';
15
+ content: string;
16
+ tool_call_id: string;
17
+ };
18
+ type LingYiMessage = SystemMessage | UserMessage | AssistantMessage | ToolMessage;
19
+ export type LingYiInputData = {
20
+ model: string;
21
+ messages: Array<LingYiMessage>;
22
+ tools?: Array<{
23
+ type: 'function';
24
+ function: {
25
+ name: string;
26
+ description?: string;
27
+ parameters: object;
28
+ };
29
+ }>;
30
+ tool_choice?: string | object;
31
+ max_tokens?: number | null;
32
+ top_p?: number;
33
+ temperature?: number;
34
+ };
35
+ export type LingYiGenerateTextOutput = {
36
+ choices?: Array<{
37
+ message?: {
38
+ content?: string;
39
+ };
40
+ }>;
41
+ };
42
+ export type LingYiStreamTextOutput = {
43
+ choices?: Array<{
44
+ delta?: {
45
+ content?: string;
46
+ };
47
+ }>;
48
+ };
49
+ export {};
@@ -0,0 +1,3 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvTGluZ1lpL3R5cGUudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IiIsInNvdXJjZXNDb250ZW50IjpbInR5cGUgU3lzdGVtTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3N5c3RlbSdcbiAgY29udGVudDogc3RyaW5nXG59XG5cbnR5cGUgVXNlck1lc3NhZ2UgPSB7XG4gIHJvbGU6ICd1c2VyJ1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBBc3Npc3RhbnRNZXNzYWdlID0ge1xuICByb2xlOiAnYXNzaXN0YW50J1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBUb29sTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3Rvb2wnXG4gIGNvbnRlbnQ6IHN0cmluZ1xuICB0b29sX2NhbGxfaWQ6IHN0cmluZ1xufVxuXG50eXBlIExpbmdZaU1lc3NhZ2UgPSBTeXN0ZW1NZXNzYWdlIHwgVXNlck1lc3NhZ2UgfCBBc3Npc3RhbnRNZXNzYWdlIHwgVG9vbE1lc3NhZ2VcblxuZXhwb3J0IHR5cGUgTGluZ1lpSW5wdXREYXRhID0ge1xuICBtb2RlbDogc3RyaW5nXG4gIG1lc3NhZ2VzOiBBcnJheTxMaW5nWWlNZXNzYWdlPlxuICB0b29scz86IEFycmF5PHtcbiAgICB0eXBlOiAnZnVuY3Rpb24nXG4gICAgZnVuY3Rpb246IHsgbmFtZTogc3RyaW5nOyBkZXNjcmlwdGlvbj86IHN0cmluZzsgcGFyYW1ldGVyczogb2JqZWN0IH1cbiAgfT5cbiAgdG9vbF9jaG9pY2U/OiBzdHJpbmcgfCBvYmplY3RcbiAgbWF4X3Rva2Vucz86IG51bWJlciB8IG51bGxcbiAgdG9wX3A/OiBudW1iZXJcbiAgdGVtcGVyYXR1cmU/OiBudW1iZXJcbn1cblxuLy8gdG9kbzogbW9yZSBwcmVjaXNlXG5leHBvcnQgdHlwZSBMaW5nWWlHZW5lcmF0ZVRleHRPdXRwdXQgPSB7XG4gIGNob2ljZXM/OiBBcnJheTx7XG4gICAgbWVzc2FnZT86IHsgY29udGVudD86IHN0cmluZyB9XG4gIH0+XG59XG5cbi8vIHRvZG86IG1vcmUgcHJlY2lzZVxuZXhwb3J0IHR5cGUgTGluZ1lpU3RyZWFtVGV4dE91dHB1dCA9IHtcbiAgY2hvaWNlcz86IEFycmF5PHtcbiAgICBkZWx0YT86IHsgY29udGVudD86IHN0cmluZyB9XG4gIH0+XG59XG4iXX0=
@@ -0,0 +1,24 @@
1
+ import { type ParsedEvent } from 'eventsource-parser/stream';
2
+ import type { YueZhiInputData, YueZhiGenerateTextOutput, YueZhiStreamTextOutput } from './type';
3
+ import type { ChatModel, StreamTextResult, ModelReq } from '../../type';
4
+ export declare class YueZhiModel implements ChatModel {
5
+ private req;
6
+ baseUrl: string;
7
+ subUrl: string;
8
+ constructor(req: ModelReq, baseUrl: string, subUrl?: string);
9
+ private get url();
10
+ generateText(data: YueZhiInputData): Promise<{
11
+ text: string;
12
+ rawResponse: YueZhiGenerateTextOutput;
13
+ }>;
14
+ streamText(data: YueZhiInputData): Promise<YueZhiModelStreamResult>;
15
+ }
16
+ declare class YueZhiModelStreamResult implements StreamTextResult {
17
+ private _eventSourceStream;
18
+ constructor(_stream: ReadableStream<Uint8Array>);
19
+ private get teeedStream();
20
+ get eventSourceStream(): import("../../type").AsyncIterableReadableStream<ParsedEvent>;
21
+ get dataStream(): import("../../type").AsyncIterableReadableStream<YueZhiStreamTextOutput>;
22
+ get textStream(): import("../../type").AsyncIterableReadableStream<string>;
23
+ }
24
+ export {};
@@ -0,0 +1,163 @@
1
+ "use strict";
2
+ var __assign = (this && this.__assign) || function () {
3
+ __assign = Object.assign || function(t) {
4
+ for (var s, i = 1, n = arguments.length; i < n; i++) {
5
+ s = arguments[i];
6
+ for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
7
+ t[p] = s[p];
8
+ }
9
+ return t;
10
+ };
11
+ return __assign.apply(this, arguments);
12
+ };
13
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
14
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
15
+ return new (P || (P = Promise))(function (resolve, reject) {
16
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
17
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
18
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
19
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
20
+ });
21
+ };
22
+ var __generator = (this && this.__generator) || function (thisArg, body) {
23
+ var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
24
+ return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
25
+ function verb(n) { return function (v) { return step([n, v]); }; }
26
+ function step(op) {
27
+ if (f) throw new TypeError("Generator is already executing.");
28
+ while (g && (g = 0, op[0] && (_ = 0)), _) try {
29
+ if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
30
+ if (y = 0, t) op = [op[0] & 2, t.value];
31
+ switch (op[0]) {
32
+ case 0: case 1: t = op; break;
33
+ case 4: _.label++; return { value: op[1], done: false };
34
+ case 5: _.label++; y = op[1]; op = [0]; continue;
35
+ case 7: op = _.ops.pop(); _.trys.pop(); continue;
36
+ default:
37
+ if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
38
+ if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
39
+ if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
40
+ if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
41
+ if (t[2]) _.ops.pop();
42
+ _.trys.pop(); continue;
43
+ }
44
+ op = body.call(thisArg, _);
45
+ } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
46
+ if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
47
+ }
48
+ };
49
+ Object.defineProperty(exports, "__esModule", { value: true });
50
+ exports.YueZhiModel = void 0;
51
+ var utils_1 = require("../../utils");
52
+ var YueZhiModel = (function () {
53
+ function YueZhiModel(req, baseUrl, subUrl) {
54
+ this.req = req;
55
+ this.baseUrl = baseUrl;
56
+ this.subUrl = 'yuezhianmian/v1/chat/completions';
57
+ if (subUrl != null) {
58
+ this.subUrl = subUrl;
59
+ }
60
+ }
61
+ Object.defineProperty(YueZhiModel.prototype, "url", {
62
+ get: function () {
63
+ return "".concat(this.baseUrl, "/").concat(this.subUrl);
64
+ },
65
+ enumerable: false,
66
+ configurable: true
67
+ });
68
+ YueZhiModel.prototype.generateText = function (data) {
69
+ var _a, _b, _c, _d;
70
+ return __awaiter(this, void 0, void 0, function () {
71
+ var res;
72
+ return __generator(this, function (_e) {
73
+ switch (_e.label) {
74
+ case 0: return [4, this.req({
75
+ url: this.url,
76
+ data: __assign(__assign({}, data), { stream: false }),
77
+ stream: false,
78
+ })];
79
+ case 1:
80
+ res = (_e.sent());
81
+ return [2, {
82
+ 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 : '',
83
+ rawResponse: res,
84
+ }];
85
+ }
86
+ });
87
+ });
88
+ };
89
+ YueZhiModel.prototype.streamText = function (data) {
90
+ return __awaiter(this, void 0, void 0, function () {
91
+ var stream;
92
+ return __generator(this, function (_a) {
93
+ switch (_a.label) {
94
+ case 0: return [4, this.req({
95
+ url: this.url,
96
+ data: __assign(__assign({}, data), { stream: true }),
97
+ stream: true,
98
+ })];
99
+ case 1:
100
+ stream = _a.sent();
101
+ return [2, new YueZhiModelStreamResult(stream)];
102
+ }
103
+ });
104
+ });
105
+ };
106
+ return YueZhiModel;
107
+ }());
108
+ exports.YueZhiModel = YueZhiModel;
109
+ var YueZhiModelStreamResult = (function () {
110
+ function YueZhiModelStreamResult(_stream) {
111
+ var stream = (0, utils_1.toPolyfillReadable)(_stream);
112
+ this._eventSourceStream = stream
113
+ .pipeThrough(new utils_1.TextDecoderStream())
114
+ .pipeThrough((0, utils_1.createEventSourceParserTransformStream)());
115
+ }
116
+ Object.defineProperty(YueZhiModelStreamResult.prototype, "teeedStream", {
117
+ get: function () {
118
+ var _a = this._eventSourceStream.tee(), s1 = _a[0], s2 = _a[1];
119
+ this._eventSourceStream = s2;
120
+ return s1;
121
+ },
122
+ enumerable: false,
123
+ configurable: true
124
+ });
125
+ Object.defineProperty(YueZhiModelStreamResult.prototype, "eventSourceStream", {
126
+ get: function () {
127
+ return (0, utils_1.createAsyncIterable)(this.teeedStream);
128
+ },
129
+ enumerable: false,
130
+ configurable: true
131
+ });
132
+ Object.defineProperty(YueZhiModelStreamResult.prototype, "dataStream", {
133
+ get: function () {
134
+ return (0, utils_1.createAsyncIterable)(this.eventSourceStream.pipeThrough(new utils_1.TransformStream({
135
+ transform: function (chunk, controller) {
136
+ try {
137
+ var data = JSON.parse(chunk.data);
138
+ controller.enqueue(data);
139
+ }
140
+ catch (e) {
141
+ console.warn('Error when transforming event source data to json', e);
142
+ }
143
+ },
144
+ })));
145
+ },
146
+ enumerable: false,
147
+ configurable: true
148
+ });
149
+ Object.defineProperty(YueZhiModelStreamResult.prototype, "textStream", {
150
+ get: function () {
151
+ return (0, utils_1.createAsyncIterable)(this.dataStream.pipeThrough(new utils_1.TransformStream({
152
+ transform: function (chunk, controller) {
153
+ var _a, _b, _c, _d;
154
+ 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 : '');
155
+ },
156
+ })));
157
+ },
158
+ enumerable: false,
159
+ configurable: true
160
+ });
161
+ return YueZhiModelStreamResult;
162
+ }());
163
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1l1ZVpoaS9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUNBLHFDQU1vQjtBQUlwQjtJQUVFLHFCQUFvQixHQUFhLEVBQVMsT0FBZSxFQUFFLE1BQWU7UUFBdEQsUUFBRyxHQUFILEdBQUcsQ0FBVTtRQUFTLFlBQU8sR0FBUCxPQUFPLENBQVE7UUFEbEQsV0FBTSxHQUFHLGtDQUFrQyxDQUFBO1FBRWhELElBQUksTUFBTSxJQUFJLElBQUksRUFBRTtZQUNsQixJQUFJLENBQUMsTUFBTSxHQUFHLE1BQU0sQ0FBQTtTQUNyQjtJQUNILENBQUM7SUFFRCxzQkFBWSw0QkFBRzthQUFmO1lBQ0UsT0FBTyxVQUFHLElBQUksQ0FBQyxPQUFPLGNBQUksSUFBSSxDQUFDLE1BQU0sQ0FBRSxDQUFBO1FBQ3pDLENBQUM7OztPQUFBO0lBRUssa0NBQVksR0FBbEIsVUFBbUIsSUFBcUI7Ozs7Ozs0QkFDekIsV0FBTSxJQUFJLENBQUMsR0FBRyxDQUFDOzRCQUMxQixHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUc7NEJBQ2IsSUFBSSx3QkFDQyxJQUFJLEtBQ1AsTUFBTSxFQUFFLEtBQUssR0FDZDs0QkFDRCxNQUFNLEVBQUUsS0FBSzt5QkFDZCxDQUFDLEVBQUE7O3dCQVBJLEdBQUcsR0FBRyxDQUFDLFNBT1gsQ0FBNkI7d0JBQy9CLFdBQU87Z0NBQ0wsSUFBSSxFQUFFLE1BQUMsTUFBQSxNQUFBLE1BQUEsR0FBRyxhQUFILEdBQUcsdUJBQUgsR0FBRyxDQUFFLE9BQU8sMENBQUcsQ0FBQyxDQUFDLDBDQUFFLE9BQU8sMENBQUUsT0FBa0IsbUNBQUksRUFBRTtnQ0FDM0QsV0FBVyxFQUFFLEdBQUc7NkJBQ2pCLEVBQUE7Ozs7S0FDRjtJQUVLLGdDQUFVLEdBQWhCLFVBQWlCLElBQXFCOzs7Ozs0QkFDckIsV0FBTSxJQUFJLENBQUMsR0FBRyxDQUFDOzRCQUM1QixHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUc7NEJBQ2IsSUFBSSx3QkFDQyxJQUFJLEtBQ1AsTUFBTSxFQUFFLElBQUksR0FDYjs0QkFDRCxNQUFNLEVBQUUsSUFBSTt5QkFDYixDQUFDLEVBQUE7O3dCQVBJLE1BQU0sR0FBRyxTQU9iO3dCQUNGLFdBQU8sSUFBSSx1QkFBdUIsQ0FBQyxNQUFNLENBQUMsRUFBQTs7OztLQUMzQztJQUNILGtCQUFDO0FBQUQsQ0FBQyxBQXRDRCxJQXNDQztBQXRDWSxrQ0FBVztBQXdDeEI7SUFHRSxpQ0FBWSxPQUFtQztRQUM3QyxJQUFNLE1BQU0sR0FBRyxJQUFBLDBCQUFrQixFQUFDLE9BQU8sQ0FBbUIsQ0FBQTtRQUM1RCxJQUFJLENBQUMsa0JBQWtCLEdBQUcsTUFBTTthQUM3QixXQUFXLENBQUMsSUFBSSx5QkFBaUIsRUFBRSxDQUFDO2FBQ3BDLFdBQVcsQ0FBQyxJQUFBLDhDQUFzQyxHQUFFLENBQUMsQ0FBQTtJQUMxRCxDQUFDO0lBRUQsc0JBQVksZ0RBQVc7YUFBdkI7WUFDUSxJQUFBLEtBQVcsSUFBSSxDQUFDLGtCQUFrQixDQUFDLEdBQUcsRUFBRSxFQUF2QyxFQUFFLFFBQUEsRUFBRSxFQUFFLFFBQWlDLENBQUE7WUFDOUMsSUFBSSxDQUFDLGtCQUFrQixHQUFHLEVBQUUsQ0FBQTtZQUM1QixPQUFPLEVBQUUsQ0FBQTtRQUNYLENBQUM7OztPQUFBO0lBRUQsc0JBQUksc0RBQWlCO2FBQXJCO1lBQ0UsT0FBTyxJQUFBLDJCQUFtQixFQUFDLElBQUksQ0FBQyxXQUFXLENBQUMsQ0FBQTtRQUM5QyxDQUFDOzs7T0FBQTtJQUVELHNCQUFJLCtDQUFVO2FBQWQ7WUFDRSxPQUFPLElBQUEsMkJBQW1CLEVBQUMsSUFBSSxDQUFDLGlCQUFpQixDQUFDLFdBQVcsQ0FBQyxJQUFJLHVCQUFlLENBQXNDO2dCQUNySCxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7b0JBQ3pCLElBQUk7d0JBQ0YsSUFBTSxJQUFJLEdBQUcsSUFBSSxDQUFDLEtBQUssQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUEyQixDQUFBO3dCQUM3RCxVQUFVLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFBO3FCQUN6QjtvQkFBQyxPQUFPLENBQUMsRUFBRTt3QkFDVixPQUFPLENBQUMsSUFBSSxDQUFDLG1EQUFtRCxFQUFFLENBQUMsQ0FBQyxDQUFBO3FCQUNyRTtnQkFDSCxDQUFDO2FBQ0YsQ0FBQyxDQUFFLENBQUUsQ0FBQTtRQUNSLENBQUM7OztPQUFBO0lBRUQsc0JBQUksK0NBQVU7YUFBZDtZQUNFLE9BQU8sSUFBQSwyQkFBbUIsRUFBQyxJQUFJLENBQUMsVUFBVSxDQUFDLFdBQVcsQ0FBQyxJQUFJLHVCQUFlLENBQWlDO2dCQUN6RyxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7O29CQUN6QixVQUFVLENBQUMsT0FBTyxDQUFDLE1BQUEsTUFBQSxNQUFBLE1BQUEsS0FBSyxhQUFMLEtBQUssdUJBQUwsS0FBSyxDQUFFLE9BQU8sMENBQUcsQ0FBQyxDQUFDLDBDQUFFLEtBQUssMENBQUUsT0FBTyxtQ0FBSSxFQUFFLENBQUMsQ0FBQTtnQkFDL0QsQ0FBQzthQUNGLENBQUMsQ0FBRSxDQUFFLENBQUE7UUFDUixDQUFDOzs7T0FBQTtJQUNILDhCQUFDO0FBQUQsQ0FBQyxBQXhDRCxJQXdDQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7IHR5cGUgUGFyc2VkRXZlbnQgfSBmcm9tICdldmVudHNvdXJjZS1wYXJzZXIvc3RyZWFtJ1xuaW1wb3J0IHtcbiAgY3JlYXRlQXN5bmNJdGVyYWJsZSxcbiAgVHJhbnNmb3JtU3RyZWFtLFxuICBUZXh0RGVjb2RlclN0cmVhbSxcbiAgY3JlYXRlRXZlbnRTb3VyY2VQYXJzZXJUcmFuc2Zvcm1TdHJlYW0sXG4gIHRvUG9seWZpbGxSZWFkYWJsZSxcbn0gZnJvbSAnLi4vLi4vdXRpbHMnXG5pbXBvcnQgdHlwZSB7IFl1ZVpoaUlucHV0RGF0YSwgWXVlWmhpR2VuZXJhdGVUZXh0T3V0cHV0LCBZdWVaaGlTdHJlYW1UZXh0T3V0cHV0IH0gZnJvbSAnLi90eXBlJ1xuaW1wb3J0IHR5cGUgeyBDaGF0TW9kZWwsIFN0cmVhbVRleHRSZXN1bHQsIE1vZGVsUmVxIH0gZnJvbSAnLi4vLi4vdHlwZSdcblxuZXhwb3J0IGNsYXNzIFl1ZVpoaU1vZGVsIGltcGxlbWVudHMgQ2hhdE1vZGVsIHtcbiAgcHVibGljIHN1YlVybCA9ICd5dWV6aGlhbm1pYW4vdjEvY2hhdC9jb21wbGV0aW9ucydcbiAgY29uc3RydWN0b3IocHJpdmF0ZSByZXE6IE1vZGVsUmVxLCBwdWJsaWMgYmFzZVVybDogc3RyaW5nLCBzdWJVcmw/OiBzdHJpbmcpIHtcbiAgICBpZiAoc3ViVXJsICE9IG51bGwpIHtcbiAgICAgIHRoaXMuc3ViVXJsID0gc3ViVXJsXG4gICAgfVxuICB9XG5cbiAgcHJpdmF0ZSBnZXQgdXJsKCkge1xuICAgIHJldHVybiBgJHt0aGlzLmJhc2VVcmx9LyR7dGhpcy5zdWJVcmx9YFxuICB9XG5cbiAgYXN5bmMgZ2VuZXJhdGVUZXh0KGRhdGE6IFl1ZVpoaUlucHV0RGF0YSkge1xuICAgIGNvbnN0IHJlcyA9IChhd2FpdCB0aGlzLnJlcSh7XG4gICAgICB1cmw6IHRoaXMudXJsLFxuICAgICAgZGF0YToge1xuICAgICAgICAuLi5kYXRhLFxuICAgICAgICBzdHJlYW06IGZhbHNlLFxuICAgICAgfSxcbiAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgfSkpIGFzIFl1ZVpoaUdlbmVyYXRlVGV4dE91dHB1dFxuICAgIHJldHVybiB7XG4gICAgICB0ZXh0OiAocmVzPy5jaG9pY2VzPy5bMF0/Lm1lc3NhZ2U/LmNvbnRlbnQgYXMgc3RyaW5nKSA/PyAnJyxcbiAgICAgIHJhd1Jlc3BvbnNlOiByZXMsXG4gICAgfVxuICB9XG5cbiAgYXN5bmMgc3RyZWFtVGV4dChkYXRhOiBZdWVaaGlJbnB1dERhdGEpIHtcbiAgICBjb25zdCBzdHJlYW0gPSBhd2FpdCB0aGlzLnJlcSh7XG4gICAgICB1cmw6IHRoaXMudXJsLFxuICAgICAgZGF0YToge1xuICAgICAgICAuLi5kYXRhLFxuICAgICAgICBzdHJlYW06IHRydWUsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiB0cnVlLFxuICAgIH0pXG4gICAgcmV0dXJuIG5ldyBZdWVaaGlNb2RlbFN0cmVhbVJlc3VsdChzdHJlYW0pXG4gIH1cbn1cblxuY2xhc3MgWXVlWmhpTW9kZWxTdHJlYW1SZXN1bHQgaW1wbGVtZW50cyBTdHJlYW1UZXh0UmVzdWx0IHtcbiAgcHJpdmF0ZSBfZXZlbnRTb3VyY2VTdHJlYW06IFJlYWRhYmxlU3RyZWFtPFBhcnNlZEV2ZW50PlxuXG4gIGNvbnN0cnVjdG9yKF9zdHJlYW06IFJlYWRhYmxlU3RyZWFtPFVpbnQ4QXJyYXk+KSB7XG4gICAgY29uc3Qgc3RyZWFtID0gdG9Qb2x5ZmlsbFJlYWRhYmxlKF9zdHJlYW0pIGFzIHR5cGVvZiBfc3RyZWFtXG4gICAgdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0gPSBzdHJlYW1cbiAgICAgIC5waXBlVGhyb3VnaChuZXcgVGV4dERlY29kZXJTdHJlYW0oKSlcbiAgICAgIC5waXBlVGhyb3VnaChjcmVhdGVFdmVudFNvdXJjZVBhcnNlclRyYW5zZm9ybVN0cmVhbSgpKVxuICB9XG5cbiAgcHJpdmF0ZSBnZXQgdGVlZWRTdHJlYW0oKSB7XG4gICAgY29uc3QgW3MxLCBzMl0gPSB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbS50ZWUoKVxuICAgIHRoaXMuX2V2ZW50U291cmNlU3RyZWFtID0gczJcbiAgICByZXR1cm4gczFcbiAgfVxuXG4gIGdldCBldmVudFNvdXJjZVN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLnRlZWVkU3RyZWFtKVxuICB9XG5cbiAgZ2V0IGRhdGFTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy5ldmVudFNvdXJjZVN0cmVhbS5waXBlVGhyb3VnaChuZXcgVHJhbnNmb3JtU3RyZWFtPFBhcnNlZEV2ZW50LCBZdWVaaGlTdHJlYW1UZXh0T3V0cHV0Pih7XG4gICAgICB0cmFuc2Zvcm0oY2h1bmssIGNvbnRyb2xsZXIpIHtcbiAgICAgICAgdHJ5IHtcbiAgICAgICAgICBjb25zdCBkYXRhID0gSlNPTi5wYXJzZShjaHVuay5kYXRhKSBhcyBZdWVaaGlTdHJlYW1UZXh0T3V0cHV0XG4gICAgICAgICAgY29udHJvbGxlci5lbnF1ZXVlKGRhdGEpXG4gICAgICAgIH0gY2F0Y2ggKGUpIHtcbiAgICAgICAgICBjb25zb2xlLndhcm4oJ0Vycm9yIHdoZW4gdHJhbnNmb3JtaW5nIGV2ZW50IHNvdXJjZSBkYXRhIHRvIGpzb24nLCBlKVxuICAgICAgICB9XG4gICAgICB9LFxuICAgIH0pLCksKVxuICB9XG5cbiAgZ2V0IHRleHRTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy5kYXRhU3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08WXVlWmhpU3RyZWFtVGV4dE91dHB1dCwgc3RyaW5nPih7XG4gICAgICB0cmFuc2Zvcm0oY2h1bmssIGNvbnRyb2xsZXIpIHtcbiAgICAgICAgY29udHJvbGxlci5lbnF1ZXVlKGNodW5rPy5jaG9pY2VzPy5bMF0/LmRlbHRhPy5jb250ZW50ID8/ICcnKVxuICAgICAgfSxcbiAgICB9KSwpLClcbiAgfVxufVxuIl19
@@ -0,0 +1,31 @@
1
+ type YueZhiMessage = {
2
+ role: 'user' | 'system' | 'assistant';
3
+ content: string;
4
+ };
5
+ export type YueZhiInputData = {
6
+ messages: Array<YueZhiMessage>;
7
+ model: 'moonshot-v1-8k' | 'moonshot-v1-32k' | 'moonshot-v1-128k';
8
+ max_tokens?: number;
9
+ temperature?: number;
10
+ top_p?: number;
11
+ n?: number;
12
+ presence_penalty?: number;
13
+ frequency_penalty?: number;
14
+ response_format?: object;
15
+ stop?: string | Array<string>;
16
+ };
17
+ export type YueZhiGenerateTextOutput = {
18
+ choices?: Array<{
19
+ message?: {
20
+ content?: string;
21
+ };
22
+ }>;
23
+ };
24
+ export type YueZhiStreamTextOutput = {
25
+ choices?: Array<{
26
+ delta?: {
27
+ content?: string;
28
+ };
29
+ }>;
30
+ };
31
+ export {};
@@ -0,0 +1,3 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvWXVlWmhpL3R5cGUudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IiIsInNvdXJjZXNDb250ZW50IjpbInR5cGUgWXVlWmhpTWVzc2FnZSA9IHsgcm9sZTogJ3VzZXInIHwgJ3N5c3RlbScgfCAnYXNzaXN0YW50JzsgY29udGVudDogc3RyaW5nIH1cblxuZXhwb3J0IHR5cGUgWXVlWmhpSW5wdXREYXRhID0ge1xuICBtZXNzYWdlczogQXJyYXk8WXVlWmhpTWVzc2FnZT5cbiAgbW9kZWw6ICdtb29uc2hvdC12MS04aycgfCAnbW9vbnNob3QtdjEtMzJrJyB8ICdtb29uc2hvdC12MS0xMjhrJ1xuICBtYXhfdG9rZW5zPzogbnVtYmVyXG4gIHRlbXBlcmF0dXJlPzogbnVtYmVyXG4gIHRvcF9wPzogbnVtYmVyXG4gIG4/OiBudW1iZXJcbiAgcHJlc2VuY2VfcGVuYWx0eT86IG51bWJlclxuICBmcmVxdWVuY3lfcGVuYWx0eT86IG51bWJlclxuICByZXNwb25zZV9mb3JtYXQ/OiBvYmplY3RcbiAgc3RvcD86IHN0cmluZyB8IEFycmF5PHN0cmluZz5cbn1cblxuLy8gdG9kbzogbW9yZSBwcmVjaXNlXG5leHBvcnQgdHlwZSBZdWVaaGlHZW5lcmF0ZVRleHRPdXRwdXQgPSB7XG4gIGNob2ljZXM/OiBBcnJheTx7XG4gICAgbWVzc2FnZT86IHsgY29udGVudD86IHN0cmluZyB9XG4gIH0+XG59XG5cbi8vIHRvZG86IG1vcmUgcHJlY2lzZVxuZXhwb3J0IHR5cGUgWXVlWmhpU3RyZWFtVGV4dE91dHB1dCA9IHtcbiAgY2hvaWNlcz86IEFycmF5PHtcbiAgICBkZWx0YT86IHsgY29udGVudD86IHN0cmluZyB9XG4gIH0+XG59XG4iXX0=
@@ -0,0 +1,24 @@
1
+ import { type ParsedEvent } from 'eventsource-parser/stream';
2
+ import type { ZhiPuGenerateTextOutput, ZhiPuStreamTextOutput } from './type';
3
+ import type { ChatModel, StreamTextResult, ModelReq, BaseChatModelInput } from '../../type';
4
+ export declare class ZhiPuModel implements ChatModel {
5
+ private req;
6
+ baseUrl: string;
7
+ subUrl: string;
8
+ constructor(req: ModelReq, baseUrl: string, subUrl?: string);
9
+ private get url();
10
+ generateText(data: BaseChatModelInput): Promise<{
11
+ text: string;
12
+ rawResponse: ZhiPuGenerateTextOutput;
13
+ }>;
14
+ streamText(data: BaseChatModelInput): Promise<ZhiPuModelStreamResult>;
15
+ }
16
+ declare class ZhiPuModelStreamResult implements StreamTextResult {
17
+ private _eventSourceStream;
18
+ constructor(_stream: ReadableStream<Uint8Array>);
19
+ private get teeedStream();
20
+ get eventSourceStream(): import("../../type").AsyncIterableReadableStream<ParsedEvent>;
21
+ get dataStream(): import("../../type").AsyncIterableReadableStream<ZhiPuStreamTextOutput>;
22
+ get textStream(): import("../../type").AsyncIterableReadableStream<string>;
23
+ }
24
+ export {};