@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.
- package/.eslintignore +2 -0
- package/.eslintrc.js +15 -0
- package/LICENSE +202 -0
- package/dist/cjs/index.d.ts +40 -0
- package/dist/cjs/index.js +171 -0
- package/dist/cjs/models/HunYuan/index.d.ts +25 -0
- package/dist/cjs/models/HunYuan/index.js +191 -0
- package/dist/cjs/models/HunYuan/type.d.ts +52 -0
- package/dist/cjs/models/HunYuan/type.js +3 -0
- package/dist/cjs/models/HunYuanPreview/index.d.ts +25 -0
- package/dist/cjs/models/HunYuanPreview/index.js +191 -0
- package/dist/cjs/models/HunYuanPreview/type.d.ts +52 -0
- package/dist/cjs/models/HunYuanPreview/type.js +3 -0
- package/dist/cjs/models/HuoShan/index.d.ts +24 -0
- package/dist/cjs/models/HuoShan/index.js +163 -0
- package/dist/cjs/models/HuoShan/type.d.ts +38 -0
- package/dist/cjs/models/HuoShan/type.js +3 -0
- package/dist/cjs/models/LingJi/index.d.ts +24 -0
- package/dist/cjs/models/LingJi/index.js +163 -0
- package/dist/cjs/models/LingJi/type.d.ts +43 -0
- package/dist/cjs/models/LingJi/type.js +3 -0
- package/dist/cjs/models/LingYi/index.d.ts +24 -0
- package/dist/cjs/models/LingYi/index.js +163 -0
- package/dist/cjs/models/LingYi/type.d.ts +49 -0
- package/dist/cjs/models/LingYi/type.js +3 -0
- package/dist/cjs/models/YueZhi/index.d.ts +24 -0
- package/dist/cjs/models/YueZhi/index.js +163 -0
- package/dist/cjs/models/YueZhi/type.d.ts +31 -0
- package/dist/cjs/models/YueZhi/type.js +3 -0
- package/dist/cjs/models/ZhiPu/index.d.ts +24 -0
- package/dist/cjs/models/ZhiPu/index.js +163 -0
- package/dist/cjs/models/ZhiPu/type.d.ts +71 -0
- package/dist/cjs/models/ZhiPu/type.js +3 -0
- package/dist/cjs/models/index.d.ts +17 -0
- package/dist/cjs/models/index.js +27 -0
- package/dist/cjs/type.d.ts +37 -0
- package/dist/cjs/type.js +16 -0
- package/dist/cjs/utils.d.ts +28 -0
- package/dist/cjs/utils.js +152 -0
- package/dist/esm/index.d.ts +40 -0
- package/dist/esm/index.js +141 -0
- package/dist/esm/models/HunYuan/index.d.ts +25 -0
- package/dist/esm/models/HunYuan/index.js +188 -0
- package/dist/esm/models/HunYuan/type.d.ts +52 -0
- package/dist/esm/models/HunYuan/type.js +2 -0
- package/dist/esm/models/HunYuanPreview/index.d.ts +25 -0
- package/dist/esm/models/HunYuanPreview/index.js +188 -0
- package/dist/esm/models/HunYuanPreview/type.d.ts +52 -0
- package/dist/esm/models/HunYuanPreview/type.js +2 -0
- package/dist/esm/models/HuoShan/index.d.ts +24 -0
- package/dist/esm/models/HuoShan/index.js +160 -0
- package/dist/esm/models/HuoShan/type.d.ts +38 -0
- package/dist/esm/models/HuoShan/type.js +2 -0
- package/dist/esm/models/LingJi/index.d.ts +24 -0
- package/dist/esm/models/LingJi/index.js +160 -0
- package/dist/esm/models/LingJi/type.d.ts +43 -0
- package/dist/esm/models/LingJi/type.js +2 -0
- package/dist/esm/models/LingYi/index.d.ts +24 -0
- package/dist/esm/models/LingYi/index.js +160 -0
- package/dist/esm/models/LingYi/type.d.ts +49 -0
- package/dist/esm/models/LingYi/type.js +2 -0
- package/dist/esm/models/YueZhi/index.d.ts +24 -0
- package/dist/esm/models/YueZhi/index.js +160 -0
- package/dist/esm/models/YueZhi/type.d.ts +31 -0
- package/dist/esm/models/YueZhi/type.js +2 -0
- package/dist/esm/models/ZhiPu/index.d.ts +24 -0
- package/dist/esm/models/ZhiPu/index.js +160 -0
- package/dist/esm/models/ZhiPu/type.d.ts +71 -0
- package/dist/esm/models/ZhiPu/type.js +2 -0
- package/dist/esm/models/index.d.ts +17 -0
- package/dist/esm/models/index.js +18 -0
- package/dist/esm/type.d.ts +37 -0
- package/dist/esm/type.js +13 -0
- package/dist/esm/utils.d.ts +28 -0
- package/dist/esm/utils.js +147 -0
- package/package.json +36 -0
- package/src/index.ts +77 -0
- package/src/models/HunYuan/index.ts +110 -0
- package/src/models/HunYuan/type.ts +61 -0
- package/src/models/HunYuanPreview/index.ts +110 -0
- package/src/models/HunYuanPreview/type.ts +61 -0
- package/src/models/HuoShan/index.ts +92 -0
- package/src/models/HuoShan/type.ts +41 -0
- package/src/models/LingJi/index.ts +92 -0
- package/src/models/LingJi/type.ts +34 -0
- package/src/models/LingYi/index.ts +92 -0
- package/src/models/LingYi/type.ts +49 -0
- package/src/models/YueZhi/index.ts +92 -0
- package/src/models/YueZhi/type.ts +28 -0
- package/src/models/ZhiPu/index.ts +92 -0
- package/src/models/ZhiPu/type.ts +60 -0
- package/src/models/index.ts +19 -0
- package/src/type.ts +42 -0
- package/src/utils.ts +110 -0
- package/tsconfig.esm.json +16 -0
- 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 {};
|