@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,160 @@
1
+ var __assign = (this && this.__assign) || function () {
2
+ __assign = Object.assign || function(t) {
3
+ for (var s, i = 1, n = arguments.length; i < n; i++) {
4
+ s = arguments[i];
5
+ for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
6
+ t[p] = s[p];
7
+ }
8
+ return t;
9
+ };
10
+ return __assign.apply(this, arguments);
11
+ };
12
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
13
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
14
+ return new (P || (P = Promise))(function (resolve, reject) {
15
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
16
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
17
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
18
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
19
+ });
20
+ };
21
+ var __generator = (this && this.__generator) || function (thisArg, body) {
22
+ var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
23
+ return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
24
+ function verb(n) { return function (v) { return step([n, v]); }; }
25
+ function step(op) {
26
+ if (f) throw new TypeError("Generator is already executing.");
27
+ while (g && (g = 0, op[0] && (_ = 0)), _) try {
28
+ 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;
29
+ if (y = 0, t) op = [op[0] & 2, t.value];
30
+ switch (op[0]) {
31
+ case 0: case 1: t = op; break;
32
+ case 4: _.label++; return { value: op[1], done: false };
33
+ case 5: _.label++; y = op[1]; op = [0]; continue;
34
+ case 7: op = _.ops.pop(); _.trys.pop(); continue;
35
+ default:
36
+ if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
37
+ if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
38
+ if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
39
+ if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
40
+ if (t[2]) _.ops.pop();
41
+ _.trys.pop(); continue;
42
+ }
43
+ op = body.call(thisArg, _);
44
+ } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
45
+ if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
46
+ }
47
+ };
48
+ import { createAsyncIterable, TransformStream, TextDecoderStream, createEventSourceParserTransformStream, toPolyfillReadable, } from '../../utils';
49
+ var LingJiModel = (function () {
50
+ function LingJiModel(req, baseUrl, subUrl) {
51
+ this.req = req;
52
+ this.baseUrl = baseUrl;
53
+ this.subUrl = 'ali/compatible-mode/v1/chat/completions';
54
+ if (subUrl != null) {
55
+ this.subUrl = subUrl;
56
+ }
57
+ }
58
+ Object.defineProperty(LingJiModel.prototype, "url", {
59
+ get: function () {
60
+ return "".concat(this.baseUrl, "/").concat(this.subUrl);
61
+ },
62
+ enumerable: false,
63
+ configurable: true
64
+ });
65
+ LingJiModel.prototype.generateText = function (data) {
66
+ var _a, _b, _c, _d;
67
+ return __awaiter(this, void 0, void 0, function () {
68
+ var res;
69
+ return __generator(this, function (_e) {
70
+ switch (_e.label) {
71
+ case 0: return [4, this.req({
72
+ url: this.url,
73
+ data: __assign(__assign({}, data), { stream: false }),
74
+ stream: false,
75
+ })];
76
+ case 1:
77
+ res = (_e.sent());
78
+ return [2, {
79
+ 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 : '',
80
+ rawResponse: res,
81
+ }];
82
+ }
83
+ });
84
+ });
85
+ };
86
+ LingJiModel.prototype.streamText = function (data) {
87
+ return __awaiter(this, void 0, void 0, function () {
88
+ var stream;
89
+ return __generator(this, function (_a) {
90
+ switch (_a.label) {
91
+ case 0: return [4, this.req({
92
+ url: this.url,
93
+ data: __assign(__assign({}, data), { stream: true }),
94
+ stream: true,
95
+ })];
96
+ case 1:
97
+ stream = _a.sent();
98
+ return [2, new LingJiModelStreamResult(stream)];
99
+ }
100
+ });
101
+ });
102
+ };
103
+ return LingJiModel;
104
+ }());
105
+ export { LingJiModel };
106
+ var LingJiModelStreamResult = (function () {
107
+ function LingJiModelStreamResult(_stream) {
108
+ var stream = toPolyfillReadable(_stream);
109
+ this._eventSourceStream = stream
110
+ .pipeThrough(new TextDecoderStream())
111
+ .pipeThrough(createEventSourceParserTransformStream());
112
+ }
113
+ Object.defineProperty(LingJiModelStreamResult.prototype, "teeedStream", {
114
+ get: function () {
115
+ var _a = this._eventSourceStream.tee(), s1 = _a[0], s2 = _a[1];
116
+ this._eventSourceStream = s2;
117
+ return s1;
118
+ },
119
+ enumerable: false,
120
+ configurable: true
121
+ });
122
+ Object.defineProperty(LingJiModelStreamResult.prototype, "eventSourceStream", {
123
+ get: function () {
124
+ return createAsyncIterable(this.teeedStream);
125
+ },
126
+ enumerable: false,
127
+ configurable: true
128
+ });
129
+ Object.defineProperty(LingJiModelStreamResult.prototype, "dataStream", {
130
+ get: function () {
131
+ return createAsyncIterable(this.eventSourceStream.pipeThrough(new TransformStream({
132
+ transform: function (chunk, controller) {
133
+ try {
134
+ var data = JSON.parse(chunk.data);
135
+ controller.enqueue(data);
136
+ }
137
+ catch (e) {
138
+ console.warn('Error when transforming event source data to json', e);
139
+ }
140
+ },
141
+ })));
142
+ },
143
+ enumerable: false,
144
+ configurable: true
145
+ });
146
+ Object.defineProperty(LingJiModelStreamResult.prototype, "textStream", {
147
+ get: function () {
148
+ return createAsyncIterable(this.dataStream.pipeThrough(new TransformStream({
149
+ transform: function (chunk, controller) {
150
+ var _a, _b, _c, _d;
151
+ 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 : '');
152
+ },
153
+ })));
154
+ },
155
+ enumerable: false,
156
+ configurable: true
157
+ });
158
+ return LingJiModelStreamResult;
159
+ }());
160
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL0xpbmdKaS9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUNBLE9BQU8sRUFDTCxtQkFBbUIsRUFDbkIsZUFBZSxFQUNmLGlCQUFpQixFQUNqQixzQ0FBc0MsRUFDdEMsa0JBQWtCLEdBQ25CLE1BQU0sYUFBYSxDQUFBO0FBSXBCO0lBRUUscUJBQW9CLEdBQWEsRUFBUyxPQUFlLEVBQUUsTUFBZTtRQUF0RCxRQUFHLEdBQUgsR0FBRyxDQUFVO1FBQVMsWUFBTyxHQUFQLE9BQU8sQ0FBUTtRQURsRCxXQUFNLEdBQUcseUNBQXlDLENBQUE7UUFFdkQsSUFBSSxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ2xCLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFBO1NBQ3JCO0lBQ0gsQ0FBQztJQUVELHNCQUFZLDRCQUFHO2FBQWY7WUFDRSxPQUFPLFVBQUcsSUFBSSxDQUFDLE9BQU8sY0FBSSxJQUFJLENBQUMsTUFBTSxDQUFFLENBQUE7UUFDekMsQ0FBQzs7O09BQUE7SUFFSyxrQ0FBWSxHQUFsQixVQUFtQixJQUFxQjs7Ozs7OzRCQUN6QixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzFCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsS0FBSyxHQUNkOzRCQUNELE1BQU0sRUFBRSxLQUFLO3lCQUNkLENBQUMsRUFBQTs7d0JBUEksR0FBRyxHQUFHLENBQUMsU0FPWCxDQUE2Qjt3QkFDL0IsV0FBTztnQ0FDTCxJQUFJLEVBQUUsTUFBQyxNQUFBLE1BQUEsTUFBQSxHQUFHLGFBQUgsR0FBRyx1QkFBSCxHQUFHLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsT0FBTywwQ0FBRSxPQUFrQixtQ0FBSSxFQUFFO2dDQUMzRCxXQUFXLEVBQUUsR0FBRzs2QkFDakIsRUFBQTs7OztLQUNGO0lBRUssZ0NBQVUsR0FBaEIsVUFBaUIsSUFBcUI7Ozs7OzRCQUNyQixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzVCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsSUFBSSxHQUNiOzRCQUNELE1BQU0sRUFBRSxJQUFJO3lCQUNiLENBQUMsRUFBQTs7d0JBUEksTUFBTSxHQUFHLFNBT2I7d0JBQ0YsV0FBTyxJQUFJLHVCQUF1QixDQUFDLE1BQU0sQ0FBQyxFQUFBOzs7O0tBQzNDO0lBQ0gsa0JBQUM7QUFBRCxDQUFDLEFBdENELElBc0NDOztBQUVEO0lBR0UsaUNBQVksT0FBbUM7UUFDN0MsSUFBTSxNQUFNLEdBQUcsa0JBQWtCLENBQUMsT0FBTyxDQUFtQixDQUFBO1FBQzVELElBQUksQ0FBQyxrQkFBa0IsR0FBRyxNQUFNO2FBQzdCLFdBQVcsQ0FBQyxJQUFJLGlCQUFpQixFQUFFLENBQUM7YUFDcEMsV0FBVyxDQUFDLHNDQUFzQyxFQUFFLENBQUMsQ0FBQTtJQUMxRCxDQUFDO0lBRUQsc0JBQVksZ0RBQVc7YUFBdkI7WUFDUSxJQUFBLEtBQVcsSUFBSSxDQUFDLGtCQUFrQixDQUFDLEdBQUcsRUFBRSxFQUF2QyxFQUFFLFFBQUEsRUFBRSxFQUFFLFFBQWlDLENBQUE7WUFDOUMsSUFBSSxDQUFDLGtCQUFrQixHQUFHLEVBQUUsQ0FBQTtZQUM1QixPQUFPLEVBQUUsQ0FBQTtRQUNYLENBQUM7OztPQUFBO0lBRUQsc0JBQUksc0RBQWlCO2FBQXJCO1lBQ0UsT0FBTyxtQkFBbUIsQ0FBQyxJQUFJLENBQUMsV0FBVyxDQUFDLENBQUE7UUFDOUMsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSwrQ0FBVTthQUFkO1lBQ0UsT0FBTyxtQkFBbUIsQ0FBQyxJQUFJLENBQUMsaUJBQWlCLENBQUMsV0FBVyxDQUFDLElBQUksZUFBZSxDQUFzQztnQkFDckgsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVO29CQUN6QixJQUFJO3dCQUNGLElBQU0sSUFBSSxHQUFHLElBQUksQ0FBQyxLQUFLLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBMkIsQ0FBQTt3QkFDN0QsVUFBVSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQTtxQkFDekI7b0JBQUMsT0FBTyxDQUFDLEVBQUU7d0JBQ1YsT0FBTyxDQUFDLElBQUksQ0FBQyxtREFBbUQsRUFBRSxDQUFDLENBQUMsQ0FBQTtxQkFDckU7Z0JBQ0gsQ0FBQzthQUNGLENBQUMsQ0FBRSxDQUFFLENBQUE7UUFDUixDQUFDOzs7T0FBQTtJQUVELHNCQUFJLCtDQUFVO2FBQWQ7WUFDRSxPQUFPLG1CQUFtQixDQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsV0FBVyxDQUFDLElBQUksZUFBZSxDQUFpQztnQkFDekcsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVOztvQkFDekIsVUFBVSxDQUFDLE9BQU8sQ0FBQyxNQUFBLE1BQUEsTUFBQSxNQUFBLEtBQUssYUFBTCxLQUFLLHVCQUFMLEtBQUssQ0FBRSxPQUFPLDBDQUFHLENBQUMsQ0FBQywwQ0FBRSxLQUFLLDBDQUFFLE9BQU8sbUNBQUksRUFBRSxDQUFDLENBQUE7Z0JBQy9ELENBQUM7YUFDRixDQUFDLENBQUUsQ0FBRSxDQUFBO1FBQ1IsQ0FBQzs7O09BQUE7SUFDSCw4QkFBQztBQUFELENBQUMsQUF4Q0QsSUF3Q0MiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgeyB0eXBlIFBhcnNlZEV2ZW50IH0gZnJvbSAnZXZlbnRzb3VyY2UtcGFyc2VyL3N0cmVhbSdcbmltcG9ydCB7XG4gIGNyZWF0ZUFzeW5jSXRlcmFibGUsXG4gIFRyYW5zZm9ybVN0cmVhbSxcbiAgVGV4dERlY29kZXJTdHJlYW0sXG4gIGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtLFxuICB0b1BvbHlmaWxsUmVhZGFibGUsXG59IGZyb20gJy4uLy4uL3V0aWxzJ1xuaW1wb3J0IHR5cGUgeyBMaW5nSmlJbnB1dERhdGEsIExpbmdKaUdlbmVyYXRlVGV4dE91dHB1dCwgTGluZ0ppU3RyZWFtVGV4dE91dHB1dCB9IGZyb20gJy4vdHlwZSdcbmltcG9ydCB0eXBlIHsgQ2hhdE1vZGVsLCBTdHJlYW1UZXh0UmVzdWx0LCBNb2RlbFJlcSB9IGZyb20gJy4uLy4uL3R5cGUnXG5cbmV4cG9ydCBjbGFzcyBMaW5nSmlNb2RlbCBpbXBsZW1lbnRzIENoYXRNb2RlbCB7XG4gIHB1YmxpYyBzdWJVcmwgPSAnYWxpL2NvbXBhdGlibGUtbW9kZS92MS9jaGF0L2NvbXBsZXRpb25zJ1xuICBjb25zdHJ1Y3Rvcihwcml2YXRlIHJlcTogTW9kZWxSZXEsIHB1YmxpYyBiYXNlVXJsOiBzdHJpbmcsIHN1YlVybD86IHN0cmluZykge1xuICAgIGlmIChzdWJVcmwgIT0gbnVsbCkge1xuICAgICAgdGhpcy5zdWJVcmwgPSBzdWJVcmxcbiAgICB9XG4gIH1cblxuICBwcml2YXRlIGdldCB1cmwoKSB7XG4gICAgcmV0dXJuIGAke3RoaXMuYmFzZVVybH0vJHt0aGlzLnN1YlVybH1gXG4gIH1cblxuICBhc3luYyBnZW5lcmF0ZVRleHQoZGF0YTogTGluZ0ppSW5wdXREYXRhKSB7XG4gICAgY29uc3QgcmVzID0gKGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICB9KSkgYXMgTGluZ0ppR2VuZXJhdGVUZXh0T3V0cHV0XG4gICAgcmV0dXJuIHtcbiAgICAgIHRleHQ6IChyZXM/LmNob2ljZXM/LlswXT8ubWVzc2FnZT8uY29udGVudCBhcyBzdHJpbmcpID8/ICcnLFxuICAgICAgcmF3UmVzcG9uc2U6IHJlcyxcbiAgICB9XG4gIH1cblxuICBhc3luYyBzdHJlYW1UZXh0KGRhdGE6IExpbmdKaUlucHV0RGF0YSkge1xuICAgIGNvbnN0IHN0cmVhbSA9IGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IHRydWUsXG4gICAgfSlcbiAgICByZXR1cm4gbmV3IExpbmdKaU1vZGVsU3RyZWFtUmVzdWx0KHN0cmVhbSlcbiAgfVxufVxuXG5jbGFzcyBMaW5nSmlNb2RlbFN0cmVhbVJlc3VsdCBpbXBsZW1lbnRzIFN0cmVhbVRleHRSZXN1bHQge1xuICBwcml2YXRlIF9ldmVudFNvdXJjZVN0cmVhbTogUmVhZGFibGVTdHJlYW08UGFyc2VkRXZlbnQ+XG5cbiAgY29uc3RydWN0b3IoX3N0cmVhbTogUmVhZGFibGVTdHJlYW08VWludDhBcnJheT4pIHtcbiAgICBjb25zdCBzdHJlYW0gPSB0b1BvbHlmaWxsUmVhZGFibGUoX3N0cmVhbSkgYXMgdHlwZW9mIF9zdHJlYW1cbiAgICB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbSA9IHN0cmVhbVxuICAgICAgLnBpcGVUaHJvdWdoKG5ldyBUZXh0RGVjb2RlclN0cmVhbSgpKVxuICAgICAgLnBpcGVUaHJvdWdoKGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtKCkpXG4gIH1cblxuICBwcml2YXRlIGdldCB0ZWVlZFN0cmVhbSgpIHtcbiAgICBjb25zdCBbczEsIHMyXSA9IHRoaXMuX2V2ZW50U291cmNlU3RyZWFtLnRlZSgpXG4gICAgdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0gPSBzMlxuICAgIHJldHVybiBzMVxuICB9XG5cbiAgZ2V0IGV2ZW50U291cmNlU3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMudGVlZWRTdHJlYW0pXG4gIH1cblxuICBnZXQgZGF0YVN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLmV2ZW50U291cmNlU3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08UGFyc2VkRXZlbnQsIExpbmdKaVN0cmVhbVRleHRPdXRwdXQ+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICB0cnkge1xuICAgICAgICAgIGNvbnN0IGRhdGEgPSBKU09OLnBhcnNlKGNodW5rLmRhdGEpIGFzIExpbmdKaVN0cmVhbVRleHRPdXRwdXRcbiAgICAgICAgICBjb250cm9sbGVyLmVucXVldWUoZGF0YSlcbiAgICAgICAgfSBjYXRjaCAoZSkge1xuICAgICAgICAgIGNvbnNvbGUud2FybignRXJyb3Igd2hlbiB0cmFuc2Zvcm1pbmcgZXZlbnQgc291cmNlIGRhdGEgdG8ganNvbicsIGUpXG4gICAgICAgIH1cbiAgICAgIH0sXG4gICAgfSksKSwpXG4gIH1cblxuICBnZXQgdGV4dFN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLmRhdGFTdHJlYW0ucGlwZVRocm91Z2gobmV3IFRyYW5zZm9ybVN0cmVhbTxMaW5nSmlTdHJlYW1UZXh0T3V0cHV0LCBzdHJpbmc+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICBjb250cm9sbGVyLmVucXVldWUoY2h1bms/LmNob2ljZXM/LlswXT8uZGVsdGE/LmNvbnRlbnQgPz8gJycpXG4gICAgICB9LFxuICAgIH0pLCksKVxuICB9XG59XG4iXX0=
@@ -0,0 +1,43 @@
1
+ type LingJiMessage = {
2
+ role: 'user' | 'system' | 'assistant';
3
+ content: string;
4
+ };
5
+ export type LingJiInputData = {
6
+ model: string;
7
+ messages: Array<LingJiMessage>;
8
+ temperature?: number;
9
+ top_p?: number;
10
+ presence_penalty?: number;
11
+ max_tokens?: number;
12
+ seed?: number;
13
+ stop?: string | Array<string>;
14
+ request_id?: string;
15
+ do_sample?: boolean;
16
+ tools?: Array<{
17
+ type: 'function';
18
+ function: {
19
+ name: string;
20
+ description: string;
21
+ parameters: object;
22
+ };
23
+ }>;
24
+ stream_options?: {
25
+ include_usage: boolean;
26
+ };
27
+ enable_search?: boolean;
28
+ };
29
+ export type LingJiGenerateTextOutput = {
30
+ choices?: Array<{
31
+ message?: {
32
+ content?: string;
33
+ };
34
+ }>;
35
+ };
36
+ export type LingJiStreamTextOutput = {
37
+ choices?: Array<{
38
+ delta?: {
39
+ content?: string;
40
+ };
41
+ }>;
42
+ };
43
+ export {};
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvTGluZ0ppL3R5cGUudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IiIsInNvdXJjZXNDb250ZW50IjpbInR5cGUgTGluZ0ppTWVzc2FnZSA9IHsgcm9sZTogJ3VzZXInIHwgJ3N5c3RlbScgfCAnYXNzaXN0YW50JzsgY29udGVudDogc3RyaW5nIH1cblxuZXhwb3J0IHR5cGUgTGluZ0ppSW5wdXREYXRhID0ge1xuICBtb2RlbDogc3RyaW5nXG4gIG1lc3NhZ2VzOiBBcnJheTxMaW5nSmlNZXNzYWdlPlxuICB0ZW1wZXJhdHVyZT86IG51bWJlclxuICB0b3BfcD86IG51bWJlclxuICBwcmVzZW5jZV9wZW5hbHR5PzogbnVtYmVyXG4gIG1heF90b2tlbnM/OiBudW1iZXJcbiAgc2VlZD86IG51bWJlclxuICBzdG9wPzogc3RyaW5nIHwgQXJyYXk8c3RyaW5nPlxuICByZXF1ZXN0X2lkPzogc3RyaW5nXG4gIGRvX3NhbXBsZT86IGJvb2xlYW5cbiAgdG9vbHM/OiBBcnJheTx7XG4gICAgdHlwZTogJ2Z1bmN0aW9uJ1xuICAgIGZ1bmN0aW9uOiB7IG5hbWU6IHN0cmluZzsgZGVzY3JpcHRpb246IHN0cmluZzsgcGFyYW1ldGVyczogb2JqZWN0IH1cbiAgfT5cbiAgc3RyZWFtX29wdGlvbnM/OiB7IGluY2x1ZGVfdXNhZ2U6IGJvb2xlYW4gfVxuICBlbmFibGVfc2VhcmNoPzogYm9vbGVhblxufVxuXG4vLyB0b2RvOiBtb3JlIHByZWNpc2VcbmV4cG9ydCB0eXBlIExpbmdKaUdlbmVyYXRlVGV4dE91dHB1dCA9IHtcbiAgY2hvaWNlcz86IEFycmF5PHtcbiAgICBtZXNzYWdlPzogeyBjb250ZW50Pzogc3RyaW5nIH1cbiAgfT5cbn1cblxuLy8gdG9kbzogbW9yZSBwcmVjaXNlXG5leHBvcnQgdHlwZSBMaW5nSmlTdHJlYW1UZXh0T3V0cHV0ID0ge1xuICBjaG9pY2VzPzogQXJyYXk8e1xuICAgIGRlbHRhPzogeyBjb250ZW50Pzogc3RyaW5nIH1cbiAgfT5cbn1cbiJdfQ==
@@ -0,0 +1,24 @@
1
+ import { type ParsedEvent } from 'eventsource-parser/stream';
2
+ import type { LingYiGenerateTextOutput, LingYiStreamTextOutput } from './type';
3
+ import type { ChatModel, StreamTextResult, ModelReq, BaseChatModelInput } from '../../type';
4
+ export declare class LingYiModel 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: LingYiGenerateTextOutput;
13
+ }>;
14
+ streamText(data: BaseChatModelInput): Promise<LingYiModelStreamResult>;
15
+ }
16
+ declare class LingYiModelStreamResult 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<LingYiStreamTextOutput>;
22
+ get textStream(): import("../../type").AsyncIterableReadableStream<string>;
23
+ }
24
+ export {};
@@ -0,0 +1,160 @@
1
+ var __assign = (this && this.__assign) || function () {
2
+ __assign = Object.assign || function(t) {
3
+ for (var s, i = 1, n = arguments.length; i < n; i++) {
4
+ s = arguments[i];
5
+ for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
6
+ t[p] = s[p];
7
+ }
8
+ return t;
9
+ };
10
+ return __assign.apply(this, arguments);
11
+ };
12
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
13
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
14
+ return new (P || (P = Promise))(function (resolve, reject) {
15
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
16
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
17
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
18
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
19
+ });
20
+ };
21
+ var __generator = (this && this.__generator) || function (thisArg, body) {
22
+ var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
23
+ return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
24
+ function verb(n) { return function (v) { return step([n, v]); }; }
25
+ function step(op) {
26
+ if (f) throw new TypeError("Generator is already executing.");
27
+ while (g && (g = 0, op[0] && (_ = 0)), _) try {
28
+ 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;
29
+ if (y = 0, t) op = [op[0] & 2, t.value];
30
+ switch (op[0]) {
31
+ case 0: case 1: t = op; break;
32
+ case 4: _.label++; return { value: op[1], done: false };
33
+ case 5: _.label++; y = op[1]; op = [0]; continue;
34
+ case 7: op = _.ops.pop(); _.trys.pop(); continue;
35
+ default:
36
+ if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
37
+ if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
38
+ if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
39
+ if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
40
+ if (t[2]) _.ops.pop();
41
+ _.trys.pop(); continue;
42
+ }
43
+ op = body.call(thisArg, _);
44
+ } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
45
+ if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
46
+ }
47
+ };
48
+ import { createAsyncIterable, TransformStream, TextDecoderStream, createEventSourceParserTransformStream, toPolyfillReadable, } from '../../utils';
49
+ var LingYiModel = (function () {
50
+ function LingYiModel(req, baseUrl, subUrl) {
51
+ this.req = req;
52
+ this.baseUrl = baseUrl;
53
+ this.subUrl = 'lingyiwanwu/v1/chat/completion';
54
+ if (subUrl != null) {
55
+ this.subUrl = subUrl;
56
+ }
57
+ }
58
+ Object.defineProperty(LingYiModel.prototype, "url", {
59
+ get: function () {
60
+ return "".concat(this.baseUrl, "/").concat(this.subUrl);
61
+ },
62
+ enumerable: false,
63
+ configurable: true
64
+ });
65
+ LingYiModel.prototype.generateText = function (data) {
66
+ var _a, _b, _c, _d;
67
+ return __awaiter(this, void 0, void 0, function () {
68
+ var res;
69
+ return __generator(this, function (_e) {
70
+ switch (_e.label) {
71
+ case 0: return [4, this.req({
72
+ url: this.url,
73
+ data: __assign(__assign({}, data), { stream: false }),
74
+ stream: false,
75
+ })];
76
+ case 1:
77
+ res = (_e.sent());
78
+ return [2, {
79
+ 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 : '',
80
+ rawResponse: res,
81
+ }];
82
+ }
83
+ });
84
+ });
85
+ };
86
+ LingYiModel.prototype.streamText = function (data) {
87
+ return __awaiter(this, void 0, void 0, function () {
88
+ var stream;
89
+ return __generator(this, function (_a) {
90
+ switch (_a.label) {
91
+ case 0: return [4, this.req({
92
+ url: this.url,
93
+ data: __assign(__assign({}, data), { stream: true }),
94
+ stream: true,
95
+ })];
96
+ case 1:
97
+ stream = _a.sent();
98
+ return [2, new LingYiModelStreamResult(stream)];
99
+ }
100
+ });
101
+ });
102
+ };
103
+ return LingYiModel;
104
+ }());
105
+ export { LingYiModel };
106
+ var LingYiModelStreamResult = (function () {
107
+ function LingYiModelStreamResult(_stream) {
108
+ var stream = toPolyfillReadable(_stream);
109
+ this._eventSourceStream = stream
110
+ .pipeThrough(new TextDecoderStream())
111
+ .pipeThrough(createEventSourceParserTransformStream());
112
+ }
113
+ Object.defineProperty(LingYiModelStreamResult.prototype, "teeedStream", {
114
+ get: function () {
115
+ var _a = this._eventSourceStream.tee(), s1 = _a[0], s2 = _a[1];
116
+ this._eventSourceStream = s2;
117
+ return s1;
118
+ },
119
+ enumerable: false,
120
+ configurable: true
121
+ });
122
+ Object.defineProperty(LingYiModelStreamResult.prototype, "eventSourceStream", {
123
+ get: function () {
124
+ return createAsyncIterable(this.teeedStream);
125
+ },
126
+ enumerable: false,
127
+ configurable: true
128
+ });
129
+ Object.defineProperty(LingYiModelStreamResult.prototype, "dataStream", {
130
+ get: function () {
131
+ return createAsyncIterable(this.eventSourceStream.pipeThrough(new TransformStream({
132
+ transform: function (chunk, controller) {
133
+ try {
134
+ var data = JSON.parse(chunk.data);
135
+ controller.enqueue(data);
136
+ }
137
+ catch (e) {
138
+ console.warn('Error when transforming event source data to json', e);
139
+ }
140
+ },
141
+ })));
142
+ },
143
+ enumerable: false,
144
+ configurable: true
145
+ });
146
+ Object.defineProperty(LingYiModelStreamResult.prototype, "textStream", {
147
+ get: function () {
148
+ return createAsyncIterable(this.dataStream.pipeThrough(new TransformStream({
149
+ transform: function (chunk, controller) {
150
+ var _a, _b, _c, _d;
151
+ 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 : '');
152
+ },
153
+ })));
154
+ },
155
+ enumerable: false,
156
+ configurable: true
157
+ });
158
+ return LingYiModelStreamResult;
159
+ }());
160
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL0xpbmdZaS9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUNBLE9BQU8sRUFDTCxtQkFBbUIsRUFDbkIsZUFBZSxFQUNmLGlCQUFpQixFQUNqQixzQ0FBc0MsRUFDdEMsa0JBQWtCLEdBQ25CLE1BQU0sYUFBYSxDQUFBO0FBSXBCO0lBRUUscUJBQW9CLEdBQWEsRUFBUyxPQUFlLEVBQUUsTUFBZTtRQUF0RCxRQUFHLEdBQUgsR0FBRyxDQUFVO1FBQVMsWUFBTyxHQUFQLE9BQU8sQ0FBUTtRQURsRCxXQUFNLEdBQUcsZ0NBQWdDLENBQUE7UUFFOUMsSUFBSSxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ2xCLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFBO1NBQ3JCO0lBQ0gsQ0FBQztJQUVELHNCQUFZLDRCQUFHO2FBQWY7WUFDRSxPQUFPLFVBQUcsSUFBSSxDQUFDLE9BQU8sY0FBSSxJQUFJLENBQUMsTUFBTSxDQUFFLENBQUE7UUFDekMsQ0FBQzs7O09BQUE7SUFFSyxrQ0FBWSxHQUFsQixVQUFtQixJQUF3Qjs7Ozs7OzRCQUM1QixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzFCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsS0FBSyxHQUNkOzRCQUNELE1BQU0sRUFBRSxLQUFLO3lCQUNkLENBQUMsRUFBQTs7d0JBUEksR0FBRyxHQUFHLENBQUMsU0FPWCxDQUE2Qjt3QkFDL0IsV0FBTztnQ0FDTCxJQUFJLEVBQUUsTUFBQyxNQUFBLE1BQUEsTUFBQSxHQUFHLGFBQUgsR0FBRyx1QkFBSCxHQUFHLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsT0FBTywwQ0FBRSxPQUFrQixtQ0FBSSxFQUFFO2dDQUMzRCxXQUFXLEVBQUUsR0FBRzs2QkFDakIsRUFBQTs7OztLQUNGO0lBRUssZ0NBQVUsR0FBaEIsVUFBaUIsSUFBd0I7Ozs7OzRCQUN4QixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzVCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsSUFBSSxHQUNiOzRCQUNELE1BQU0sRUFBRSxJQUFJO3lCQUNiLENBQUMsRUFBQTs7d0JBUEksTUFBTSxHQUFHLFNBT2I7d0JBQ0YsV0FBTyxJQUFJLHVCQUF1QixDQUFDLE1BQU0sQ0FBQyxFQUFBOzs7O0tBQzNDO0lBQ0gsa0JBQUM7QUFBRCxDQUFDLEFBdENELElBc0NDOztBQUVEO0lBR0UsaUNBQVksT0FBbUM7UUFDN0MsSUFBTSxNQUFNLEdBQUcsa0JBQWtCLENBQUMsT0FBTyxDQUFtQixDQUFBO1FBQzVELElBQUksQ0FBQyxrQkFBa0IsR0FBRyxNQUFNO2FBQzdCLFdBQVcsQ0FBQyxJQUFJLGlCQUFpQixFQUFFLENBQUM7YUFDcEMsV0FBVyxDQUFDLHNDQUFzQyxFQUFFLENBQUMsQ0FBQTtJQUMxRCxDQUFDO0lBRUQsc0JBQVksZ0RBQVc7YUFBdkI7WUFDUSxJQUFBLEtBQVcsSUFBSSxDQUFDLGtCQUFrQixDQUFDLEdBQUcsRUFBRSxFQUF2QyxFQUFFLFFBQUEsRUFBRSxFQUFFLFFBQWlDLENBQUE7WUFDOUMsSUFBSSxDQUFDLGtCQUFrQixHQUFHLEVBQUUsQ0FBQTtZQUM1QixPQUFPLEVBQUUsQ0FBQTtRQUNYLENBQUM7OztPQUFBO0lBRUQsc0JBQUksc0RBQWlCO2FBQXJCO1lBQ0UsT0FBTyxtQkFBbUIsQ0FBQyxJQUFJLENBQUMsV0FBVyxDQUFDLENBQUE7UUFDOUMsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSwrQ0FBVTthQUFkO1lBQ0UsT0FBTyxtQkFBbUIsQ0FBQyxJQUFJLENBQUMsaUJBQWlCLENBQUMsV0FBVyxDQUFDLElBQUksZUFBZSxDQUFzQztnQkFDckgsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVO29CQUN6QixJQUFJO3dCQUNGLElBQU0sSUFBSSxHQUFHLElBQUksQ0FBQyxLQUFLLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBMkIsQ0FBQTt3QkFDN0QsVUFBVSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQTtxQkFDekI7b0JBQUMsT0FBTyxDQUFDLEVBQUU7d0JBQ1YsT0FBTyxDQUFDLElBQUksQ0FBQyxtREFBbUQsRUFBRSxDQUFDLENBQUMsQ0FBQTtxQkFDckU7Z0JBQ0gsQ0FBQzthQUNGLENBQUMsQ0FBRSxDQUFFLENBQUE7UUFDUixDQUFDOzs7T0FBQTtJQUVELHNCQUFJLCtDQUFVO2FBQWQ7WUFDRSxPQUFPLG1CQUFtQixDQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsV0FBVyxDQUFDLElBQUksZUFBZSxDQUFpQztnQkFDekcsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVOztvQkFDekIsVUFBVSxDQUFDLE9BQU8sQ0FBQyxNQUFBLE1BQUEsTUFBQSxNQUFBLEtBQUssYUFBTCxLQUFLLHVCQUFMLEtBQUssQ0FBRSxPQUFPLDBDQUFHLENBQUMsQ0FBQywwQ0FBRSxLQUFLLDBDQUFFLE9BQU8sbUNBQUksRUFBRSxDQUFDLENBQUE7Z0JBQy9ELENBQUM7YUFDRixDQUFDLENBQUUsQ0FBRSxDQUFBO1FBQ1IsQ0FBQzs7O09BQUE7SUFDSCw4QkFBQztBQUFELENBQUMsQUF4Q0QsSUF3Q0MiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgeyB0eXBlIFBhcnNlZEV2ZW50IH0gZnJvbSAnZXZlbnRzb3VyY2UtcGFyc2VyL3N0cmVhbSdcbmltcG9ydCB7XG4gIGNyZWF0ZUFzeW5jSXRlcmFibGUsXG4gIFRyYW5zZm9ybVN0cmVhbSxcbiAgVGV4dERlY29kZXJTdHJlYW0sXG4gIGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtLFxuICB0b1BvbHlmaWxsUmVhZGFibGUsXG59IGZyb20gJy4uLy4uL3V0aWxzJ1xuaW1wb3J0IHR5cGUgeyBMaW5nWWlHZW5lcmF0ZVRleHRPdXRwdXQsIExpbmdZaVN0cmVhbVRleHRPdXRwdXQgfSBmcm9tICcuL3R5cGUnXG5pbXBvcnQgdHlwZSB7IENoYXRNb2RlbCwgU3RyZWFtVGV4dFJlc3VsdCwgTW9kZWxSZXEsIEJhc2VDaGF0TW9kZWxJbnB1dCB9IGZyb20gJy4uLy4uL3R5cGUnXG5cbmV4cG9ydCBjbGFzcyBMaW5nWWlNb2RlbCBpbXBsZW1lbnRzIENoYXRNb2RlbCB7XG4gIHB1YmxpYyBzdWJVcmwgPSAnbGluZ3lpd2Fud3UvdjEvY2hhdC9jb21wbGV0aW9uJ1xuICBjb25zdHJ1Y3Rvcihwcml2YXRlIHJlcTogTW9kZWxSZXEsIHB1YmxpYyBiYXNlVXJsOiBzdHJpbmcsIHN1YlVybD86IHN0cmluZykge1xuICAgIGlmIChzdWJVcmwgIT0gbnVsbCkge1xuICAgICAgdGhpcy5zdWJVcmwgPSBzdWJVcmxcbiAgICB9XG4gIH1cblxuICBwcml2YXRlIGdldCB1cmwoKSB7XG4gICAgcmV0dXJuIGAke3RoaXMuYmFzZVVybH0vJHt0aGlzLnN1YlVybH1gXG4gIH1cblxuICBhc3luYyBnZW5lcmF0ZVRleHQoZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KSB7XG4gICAgY29uc3QgcmVzID0gKGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICB9KSkgYXMgTGluZ1lpR2VuZXJhdGVUZXh0T3V0cHV0XG4gICAgcmV0dXJuIHtcbiAgICAgIHRleHQ6IChyZXM/LmNob2ljZXM/LlswXT8ubWVzc2FnZT8uY29udGVudCBhcyBzdHJpbmcpID8/ICcnLFxuICAgICAgcmF3UmVzcG9uc2U6IHJlcyxcbiAgICB9XG4gIH1cblxuICBhc3luYyBzdHJlYW1UZXh0KGRhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCkge1xuICAgIGNvbnN0IHN0cmVhbSA9IGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IHRydWUsXG4gICAgfSlcbiAgICByZXR1cm4gbmV3IExpbmdZaU1vZGVsU3RyZWFtUmVzdWx0KHN0cmVhbSlcbiAgfVxufVxuXG5jbGFzcyBMaW5nWWlNb2RlbFN0cmVhbVJlc3VsdCBpbXBsZW1lbnRzIFN0cmVhbVRleHRSZXN1bHQge1xuICBwcml2YXRlIF9ldmVudFNvdXJjZVN0cmVhbTogUmVhZGFibGVTdHJlYW08UGFyc2VkRXZlbnQ+XG5cbiAgY29uc3RydWN0b3IoX3N0cmVhbTogUmVhZGFibGVTdHJlYW08VWludDhBcnJheT4pIHtcbiAgICBjb25zdCBzdHJlYW0gPSB0b1BvbHlmaWxsUmVhZGFibGUoX3N0cmVhbSkgYXMgdHlwZW9mIF9zdHJlYW1cbiAgICB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbSA9IHN0cmVhbVxuICAgICAgLnBpcGVUaHJvdWdoKG5ldyBUZXh0RGVjb2RlclN0cmVhbSgpKVxuICAgICAgLnBpcGVUaHJvdWdoKGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtKCkpXG4gIH1cblxuICBwcml2YXRlIGdldCB0ZWVlZFN0cmVhbSgpIHtcbiAgICBjb25zdCBbczEsIHMyXSA9IHRoaXMuX2V2ZW50U291cmNlU3RyZWFtLnRlZSgpXG4gICAgdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0gPSBzMlxuICAgIHJldHVybiBzMVxuICB9XG5cbiAgZ2V0IGV2ZW50U291cmNlU3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMudGVlZWRTdHJlYW0pXG4gIH1cblxuICBnZXQgZGF0YVN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLmV2ZW50U291cmNlU3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08UGFyc2VkRXZlbnQsIExpbmdZaVN0cmVhbVRleHRPdXRwdXQ+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICB0cnkge1xuICAgICAgICAgIGNvbnN0IGRhdGEgPSBKU09OLnBhcnNlKGNodW5rLmRhdGEpIGFzIExpbmdZaVN0cmVhbVRleHRPdXRwdXRcbiAgICAgICAgICBjb250cm9sbGVyLmVucXVldWUoZGF0YSlcbiAgICAgICAgfSBjYXRjaCAoZSkge1xuICAgICAgICAgIGNvbnNvbGUud2FybignRXJyb3Igd2hlbiB0cmFuc2Zvcm1pbmcgZXZlbnQgc291cmNlIGRhdGEgdG8ganNvbicsIGUpXG4gICAgICAgIH1cbiAgICAgIH0sXG4gICAgfSksKSwpXG4gIH1cblxuICBnZXQgdGV4dFN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLmRhdGFTdHJlYW0ucGlwZVRocm91Z2gobmV3IFRyYW5zZm9ybVN0cmVhbTxMaW5nWWlTdHJlYW1UZXh0T3V0cHV0LCBzdHJpbmc+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICBjb250cm9sbGVyLmVucXVldWUoY2h1bms/LmNob2ljZXM/LlswXT8uZGVsdGE/LmNvbnRlbnQgPz8gJycpXG4gICAgICB9LFxuICAgIH0pLCksKVxuICB9XG59XG4iXX0=
@@ -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,2 @@
1
+ export {};
2
+ //# 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 {};