@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,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 YueZhiModel = (function () {
|
|
50
|
+
function YueZhiModel(req, baseUrl, subUrl) {
|
|
51
|
+
this.req = req;
|
|
52
|
+
this.baseUrl = baseUrl;
|
|
53
|
+
this.subUrl = 'yuezhianmian/v1/chat/completions';
|
|
54
|
+
if (subUrl != null) {
|
|
55
|
+
this.subUrl = subUrl;
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
Object.defineProperty(YueZhiModel.prototype, "url", {
|
|
59
|
+
get: function () {
|
|
60
|
+
return "".concat(this.baseUrl, "/").concat(this.subUrl);
|
|
61
|
+
},
|
|
62
|
+
enumerable: false,
|
|
63
|
+
configurable: true
|
|
64
|
+
});
|
|
65
|
+
YueZhiModel.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
|
+
YueZhiModel.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 YueZhiModelStreamResult(stream)];
|
|
99
|
+
}
|
|
100
|
+
});
|
|
101
|
+
});
|
|
102
|
+
};
|
|
103
|
+
return YueZhiModel;
|
|
104
|
+
}());
|
|
105
|
+
export { YueZhiModel };
|
|
106
|
+
var YueZhiModelStreamResult = (function () {
|
|
107
|
+
function YueZhiModelStreamResult(_stream) {
|
|
108
|
+
var stream = toPolyfillReadable(_stream);
|
|
109
|
+
this._eventSourceStream = stream
|
|
110
|
+
.pipeThrough(new TextDecoderStream())
|
|
111
|
+
.pipeThrough(createEventSourceParserTransformStream());
|
|
112
|
+
}
|
|
113
|
+
Object.defineProperty(YueZhiModelStreamResult.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(YueZhiModelStreamResult.prototype, "eventSourceStream", {
|
|
123
|
+
get: function () {
|
|
124
|
+
return createAsyncIterable(this.teeedStream);
|
|
125
|
+
},
|
|
126
|
+
enumerable: false,
|
|
127
|
+
configurable: true
|
|
128
|
+
});
|
|
129
|
+
Object.defineProperty(YueZhiModelStreamResult.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(YueZhiModelStreamResult.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 YueZhiModelStreamResult;
|
|
159
|
+
}());
|
|
160
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1l1ZVpoaS9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUNBLE9BQU8sRUFDTCxtQkFBbUIsRUFDbkIsZUFBZSxFQUNmLGlCQUFpQixFQUNqQixzQ0FBc0MsRUFDdEMsa0JBQWtCLEdBQ25CLE1BQU0sYUFBYSxDQUFBO0FBSXBCO0lBRUUscUJBQW9CLEdBQWEsRUFBUyxPQUFlLEVBQUUsTUFBZTtRQUF0RCxRQUFHLEdBQUgsR0FBRyxDQUFVO1FBQVMsWUFBTyxHQUFQLE9BQU8sQ0FBUTtRQURsRCxXQUFNLEdBQUcsa0NBQWtDLENBQUE7UUFFaEQsSUFBSSxNQUFNLElBQUksSUFBSSxFQUFFO1lBQ2xCLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFBO1NBQ3JCO0lBQ0gsQ0FBQztJQUVELHNCQUFZLDRCQUFHO2FBQWY7WUFDRSxPQUFPLFVBQUcsSUFBSSxDQUFDLE9BQU8sY0FBSSxJQUFJLENBQUMsTUFBTSxDQUFFLENBQUE7UUFDekMsQ0FBQzs7O09BQUE7SUFFSyxrQ0FBWSxHQUFsQixVQUFtQixJQUFxQjs7Ozs7OzRCQUN6QixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzFCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsS0FBSyxHQUNkOzRCQUNELE1BQU0sRUFBRSxLQUFLO3lCQUNkLENBQUMsRUFBQTs7d0JBUEksR0FBRyxHQUFHLENBQUMsU0FPWCxDQUE2Qjt3QkFDL0IsV0FBTztnQ0FDTCxJQUFJLEVBQUUsTUFBQyxNQUFBLE1BQUEsTUFBQSxHQUFHLGFBQUgsR0FBRyx1QkFBSCxHQUFHLENBQUUsT0FBTywwQ0FBRyxDQUFDLENBQUMsMENBQUUsT0FBTywwQ0FBRSxPQUFrQixtQ0FBSSxFQUFFO2dDQUMzRCxXQUFXLEVBQUUsR0FBRzs2QkFDakIsRUFBQTs7OztLQUNGO0lBRUssZ0NBQVUsR0FBaEIsVUFBaUIsSUFBcUI7Ozs7OzRCQUNyQixXQUFNLElBQUksQ0FBQyxHQUFHLENBQUM7NEJBQzVCLEdBQUcsRUFBRSxJQUFJLENBQUMsR0FBRzs0QkFDYixJQUFJLHdCQUNDLElBQUksS0FDUCxNQUFNLEVBQUUsSUFBSSxHQUNiOzRCQUNELE1BQU0sRUFBRSxJQUFJO3lCQUNiLENBQUMsRUFBQTs7d0JBUEksTUFBTSxHQUFHLFNBT2I7d0JBQ0YsV0FBTyxJQUFJLHVCQUF1QixDQUFDLE1BQU0sQ0FBQyxFQUFBOzs7O0tBQzNDO0lBQ0gsa0JBQUM7QUFBRCxDQUFDLEFBdENELElBc0NDOztBQUVEO0lBR0UsaUNBQVksT0FBbUM7UUFDN0MsSUFBTSxNQUFNLEdBQUcsa0JBQWtCLENBQUMsT0FBTyxDQUFtQixDQUFBO1FBQzVELElBQUksQ0FBQyxrQkFBa0IsR0FBRyxNQUFNO2FBQzdCLFdBQVcsQ0FBQyxJQUFJLGlCQUFpQixFQUFFLENBQUM7YUFDcEMsV0FBVyxDQUFDLHNDQUFzQyxFQUFFLENBQUMsQ0FBQTtJQUMxRCxDQUFDO0lBRUQsc0JBQVksZ0RBQVc7YUFBdkI7WUFDUSxJQUFBLEtBQVcsSUFBSSxDQUFDLGtCQUFrQixDQUFDLEdBQUcsRUFBRSxFQUF2QyxFQUFFLFFBQUEsRUFBRSxFQUFFLFFBQWlDLENBQUE7WUFDOUMsSUFBSSxDQUFDLGtCQUFrQixHQUFHLEVBQUUsQ0FBQTtZQUM1QixPQUFPLEVBQUUsQ0FBQTtRQUNYLENBQUM7OztPQUFBO0lBRUQsc0JBQUksc0RBQWlCO2FBQXJCO1lBQ0UsT0FBTyxtQkFBbUIsQ0FBQyxJQUFJLENBQUMsV0FBVyxDQUFDLENBQUE7UUFDOUMsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSwrQ0FBVTthQUFkO1lBQ0UsT0FBTyxtQkFBbUIsQ0FBQyxJQUFJLENBQUMsaUJBQWlCLENBQUMsV0FBVyxDQUFDLElBQUksZUFBZSxDQUFzQztnQkFDckgsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVO29CQUN6QixJQUFJO3dCQUNGLElBQU0sSUFBSSxHQUFHLElBQUksQ0FBQyxLQUFLLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBMkIsQ0FBQTt3QkFDN0QsVUFBVSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQTtxQkFDekI7b0JBQUMsT0FBTyxDQUFDLEVBQUU7d0JBQ1YsT0FBTyxDQUFDLElBQUksQ0FBQyxtREFBbUQsRUFBRSxDQUFDLENBQUMsQ0FBQTtxQkFDckU7Z0JBQ0gsQ0FBQzthQUNGLENBQUMsQ0FBRSxDQUFFLENBQUE7UUFDUixDQUFDOzs7T0FBQTtJQUVELHNCQUFJLCtDQUFVO2FBQWQ7WUFDRSxPQUFPLG1CQUFtQixDQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsV0FBVyxDQUFDLElBQUksZUFBZSxDQUFpQztnQkFDekcsU0FBUyxZQUFDLEtBQUssRUFBRSxVQUFVOztvQkFDekIsVUFBVSxDQUFDLE9BQU8sQ0FBQyxNQUFBLE1BQUEsTUFBQSxNQUFBLEtBQUssYUFBTCxLQUFLLHVCQUFMLEtBQUssQ0FBRSxPQUFPLDBDQUFHLENBQUMsQ0FBQywwQ0FBRSxLQUFLLDBDQUFFLE9BQU8sbUNBQUksRUFBRSxDQUFDLENBQUE7Z0JBQy9ELENBQUM7YUFDRixDQUFDLENBQUUsQ0FBRSxDQUFBO1FBQ1IsQ0FBQzs7O09BQUE7SUFDSCw4QkFBQztBQUFELENBQUMsQUF4Q0QsSUF3Q0MiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgeyB0eXBlIFBhcnNlZEV2ZW50IH0gZnJvbSAnZXZlbnRzb3VyY2UtcGFyc2VyL3N0cmVhbSdcbmltcG9ydCB7XG4gIGNyZWF0ZUFzeW5jSXRlcmFibGUsXG4gIFRyYW5zZm9ybVN0cmVhbSxcbiAgVGV4dERlY29kZXJTdHJlYW0sXG4gIGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtLFxuICB0b1BvbHlmaWxsUmVhZGFibGUsXG59IGZyb20gJy4uLy4uL3V0aWxzJ1xuaW1wb3J0IHR5cGUgeyBZdWVaaGlJbnB1dERhdGEsIFl1ZVpoaUdlbmVyYXRlVGV4dE91dHB1dCwgWXVlWmhpU3RyZWFtVGV4dE91dHB1dCB9IGZyb20gJy4vdHlwZSdcbmltcG9ydCB0eXBlIHsgQ2hhdE1vZGVsLCBTdHJlYW1UZXh0UmVzdWx0LCBNb2RlbFJlcSB9IGZyb20gJy4uLy4uL3R5cGUnXG5cbmV4cG9ydCBjbGFzcyBZdWVaaGlNb2RlbCBpbXBsZW1lbnRzIENoYXRNb2RlbCB7XG4gIHB1YmxpYyBzdWJVcmwgPSAneXVlemhpYW5taWFuL3YxL2NoYXQvY29tcGxldGlvbnMnXG4gIGNvbnN0cnVjdG9yKHByaXZhdGUgcmVxOiBNb2RlbFJlcSwgcHVibGljIGJhc2VVcmw6IHN0cmluZywgc3ViVXJsPzogc3RyaW5nKSB7XG4gICAgaWYgKHN1YlVybCAhPSBudWxsKSB7XG4gICAgICB0aGlzLnN1YlVybCA9IHN1YlVybFxuICAgIH1cbiAgfVxuXG4gIHByaXZhdGUgZ2V0IHVybCgpIHtcbiAgICByZXR1cm4gYCR7dGhpcy5iYXNlVXJsfS8ke3RoaXMuc3ViVXJsfWBcbiAgfVxuXG4gIGFzeW5jIGdlbmVyYXRlVGV4dChkYXRhOiBZdWVaaGlJbnB1dERhdGEpIHtcbiAgICBjb25zdCByZXMgPSAoYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICAgIH0sXG4gICAgICBzdHJlYW06IGZhbHNlLFxuICAgIH0pKSBhcyBZdWVaaGlHZW5lcmF0ZVRleHRPdXRwdXRcbiAgICByZXR1cm4ge1xuICAgICAgdGV4dDogKHJlcz8uY2hvaWNlcz8uWzBdPy5tZXNzYWdlPy5jb250ZW50IGFzIHN0cmluZykgPz8gJycsXG4gICAgICByYXdSZXNwb25zZTogcmVzLFxuICAgIH1cbiAgfVxuXG4gIGFzeW5jIHN0cmVhbVRleHQoZGF0YTogWXVlWmhpSW5wdXREYXRhKSB7XG4gICAgY29uc3Qgc3RyZWFtID0gYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiB0cnVlLFxuICAgICAgfSxcbiAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICB9KVxuICAgIHJldHVybiBuZXcgWXVlWmhpTW9kZWxTdHJlYW1SZXN1bHQoc3RyZWFtKVxuICB9XG59XG5cbmNsYXNzIFl1ZVpoaU1vZGVsU3RyZWFtUmVzdWx0IGltcGxlbWVudHMgU3RyZWFtVGV4dFJlc3VsdCB7XG4gIHByaXZhdGUgX2V2ZW50U291cmNlU3RyZWFtOiBSZWFkYWJsZVN0cmVhbTxQYXJzZWRFdmVudD5cblxuICBjb25zdHJ1Y3Rvcihfc3RyZWFtOiBSZWFkYWJsZVN0cmVhbTxVaW50OEFycmF5Pikge1xuICAgIGNvbnN0IHN0cmVhbSA9IHRvUG9seWZpbGxSZWFkYWJsZShfc3RyZWFtKSBhcyB0eXBlb2YgX3N0cmVhbVxuICAgIHRoaXMuX2V2ZW50U291cmNlU3RyZWFtID0gc3RyZWFtXG4gICAgICAucGlwZVRocm91Z2gobmV3IFRleHREZWNvZGVyU3RyZWFtKCkpXG4gICAgICAucGlwZVRocm91Z2goY3JlYXRlRXZlbnRTb3VyY2VQYXJzZXJUcmFuc2Zvcm1TdHJlYW0oKSlcbiAgfVxuXG4gIHByaXZhdGUgZ2V0IHRlZWVkU3RyZWFtKCkge1xuICAgIGNvbnN0IFtzMSwgczJdID0gdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0udGVlKClcbiAgICB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbSA9IHMyXG4gICAgcmV0dXJuIHMxXG4gIH1cblxuICBnZXQgZXZlbnRTb3VyY2VTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy50ZWVlZFN0cmVhbSlcbiAgfVxuXG4gIGdldCBkYXRhU3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMuZXZlbnRTb3VyY2VTdHJlYW0ucGlwZVRocm91Z2gobmV3IFRyYW5zZm9ybVN0cmVhbTxQYXJzZWRFdmVudCwgWXVlWmhpU3RyZWFtVGV4dE91dHB1dD4oe1xuICAgICAgdHJhbnNmb3JtKGNodW5rLCBjb250cm9sbGVyKSB7XG4gICAgICAgIHRyeSB7XG4gICAgICAgICAgY29uc3QgZGF0YSA9IEpTT04ucGFyc2UoY2h1bmsuZGF0YSkgYXMgWXVlWmhpU3RyZWFtVGV4dE91dHB1dFxuICAgICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZShkYXRhKVxuICAgICAgICB9IGNhdGNoIChlKSB7XG4gICAgICAgICAgY29uc29sZS53YXJuKCdFcnJvciB3aGVuIHRyYW5zZm9ybWluZyBldmVudCBzb3VyY2UgZGF0YSB0byBqc29uJywgZSlcbiAgICAgICAgfVxuICAgICAgfSxcbiAgICB9KSwpLClcbiAgfVxuXG4gIGdldCB0ZXh0U3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMuZGF0YVN0cmVhbS5waXBlVGhyb3VnaChuZXcgVHJhbnNmb3JtU3RyZWFtPFl1ZVpoaVN0cmVhbVRleHRPdXRwdXQsIHN0cmluZz4oe1xuICAgICAgdHJhbnNmb3JtKGNodW5rLCBjb250cm9sbGVyKSB7XG4gICAgICAgIGNvbnRyb2xsZXIuZW5xdWV1ZShjaHVuaz8uY2hvaWNlcz8uWzBdPy5kZWx0YT8uY29udGVudCA/PyAnJylcbiAgICAgIH0sXG4gICAgfSksKSwpXG4gIH1cbn1cbiJdfQ==
|
|
@@ -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,2 @@
|
|
|
1
|
+
export {};
|
|
2
|
+
//# 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 {};
|
|
@@ -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 ZhiPuModel = (function () {
|
|
50
|
+
function ZhiPuModel(req, baseUrl, subUrl) {
|
|
51
|
+
this.req = req;
|
|
52
|
+
this.baseUrl = baseUrl;
|
|
53
|
+
this.subUrl = 'zhipu/api/paas/v4/chat/completions';
|
|
54
|
+
if (subUrl != null) {
|
|
55
|
+
this.subUrl = subUrl;
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
Object.defineProperty(ZhiPuModel.prototype, "url", {
|
|
59
|
+
get: function () {
|
|
60
|
+
return "".concat(this.baseUrl, "/").concat(this.subUrl);
|
|
61
|
+
},
|
|
62
|
+
enumerable: false,
|
|
63
|
+
configurable: true
|
|
64
|
+
});
|
|
65
|
+
ZhiPuModel.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
|
+
ZhiPuModel.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 ZhiPuModelStreamResult(stream)];
|
|
99
|
+
}
|
|
100
|
+
});
|
|
101
|
+
});
|
|
102
|
+
};
|
|
103
|
+
return ZhiPuModel;
|
|
104
|
+
}());
|
|
105
|
+
export { ZhiPuModel };
|
|
106
|
+
var ZhiPuModelStreamResult = (function () {
|
|
107
|
+
function ZhiPuModelStreamResult(_stream) {
|
|
108
|
+
var stream = toPolyfillReadable(_stream);
|
|
109
|
+
this._eventSourceStream = stream
|
|
110
|
+
.pipeThrough(new TextDecoderStream())
|
|
111
|
+
.pipeThrough(createEventSourceParserTransformStream());
|
|
112
|
+
}
|
|
113
|
+
Object.defineProperty(ZhiPuModelStreamResult.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(ZhiPuModelStreamResult.prototype, "eventSourceStream", {
|
|
123
|
+
get: function () {
|
|
124
|
+
return createAsyncIterable(this.teeedStream);
|
|
125
|
+
},
|
|
126
|
+
enumerable: false,
|
|
127
|
+
configurable: true
|
|
128
|
+
});
|
|
129
|
+
Object.defineProperty(ZhiPuModelStreamResult.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(ZhiPuModelStreamResult.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 ZhiPuModelStreamResult;
|
|
159
|
+
}());
|
|
160
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi8uLi9zcmMvbW9kZWxzL1poaVB1L2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQ0EsT0FBTyxFQUNMLG1CQUFtQixFQUNuQixlQUFlLEVBQ2YsaUJBQWlCLEVBQ2pCLHNDQUFzQyxFQUN0QyxrQkFBa0IsR0FDbkIsTUFBTSxhQUFhLENBQUE7QUFJcEI7SUFFRSxvQkFBb0IsR0FBYSxFQUFTLE9BQWUsRUFBRSxNQUFlO1FBQXRELFFBQUcsR0FBSCxHQUFHLENBQVU7UUFBUyxZQUFPLEdBQVAsT0FBTyxDQUFRO1FBRGxELFdBQU0sR0FBRyxvQ0FBb0MsQ0FBQTtRQUVsRCxJQUFJLE1BQU0sSUFBSSxJQUFJLEVBQUU7WUFDbEIsSUFBSSxDQUFDLE1BQU0sR0FBRyxNQUFNLENBQUE7U0FDckI7SUFDSCxDQUFDO0lBRUQsc0JBQVksMkJBQUc7YUFBZjtZQUNFLE9BQU8sVUFBRyxJQUFJLENBQUMsT0FBTyxjQUFJLElBQUksQ0FBQyxNQUFNLENBQUUsQ0FBQTtRQUN6QyxDQUFDOzs7T0FBQTtJQUVLLGlDQUFZLEdBQWxCLFVBQW1CLElBQXdCOzs7Ozs7NEJBQzVCLFdBQU0sSUFBSSxDQUFDLEdBQUcsQ0FBQzs0QkFDMUIsR0FBRyxFQUFFLElBQUksQ0FBQyxHQUFHOzRCQUNiLElBQUksd0JBQ0MsSUFBSSxLQUNQLE1BQU0sRUFBRSxLQUFLLEdBQ2Q7NEJBQ0QsTUFBTSxFQUFFLEtBQUs7eUJBQ2QsQ0FBQyxFQUFBOzt3QkFQSSxHQUFHLEdBQUcsQ0FBQyxTQU9YLENBQTRCO3dCQUM5QixXQUFPO2dDQUNMLElBQUksRUFBRSxNQUFDLE1BQUEsTUFBQSxNQUFBLEdBQUcsYUFBSCxHQUFHLHVCQUFILEdBQUcsQ0FBRSxPQUFPLDBDQUFHLENBQUMsQ0FBQywwQ0FBRSxPQUFPLDBDQUFFLE9BQWtCLG1DQUFJLEVBQUU7Z0NBQzNELFdBQVcsRUFBRSxHQUFHOzZCQUNqQixFQUFBOzs7O0tBQ0Y7SUFFSywrQkFBVSxHQUFoQixVQUFpQixJQUF3Qjs7Ozs7NEJBQ3hCLFdBQU0sSUFBSSxDQUFDLEdBQUcsQ0FBQzs0QkFDNUIsR0FBRyxFQUFFLElBQUksQ0FBQyxHQUFHOzRCQUNiLElBQUksd0JBQ0MsSUFBSSxLQUNQLE1BQU0sRUFBRSxJQUFJLEdBQ2I7NEJBQ0QsTUFBTSxFQUFFLElBQUk7eUJBQ2IsQ0FBQyxFQUFBOzt3QkFQSSxNQUFNLEdBQUcsU0FPYjt3QkFDRixXQUFPLElBQUksc0JBQXNCLENBQUMsTUFBTSxDQUFDLEVBQUE7Ozs7S0FDMUM7SUFDSCxpQkFBQztBQUFELENBQUMsQUF0Q0QsSUFzQ0M7O0FBRUQ7SUFHRSxnQ0FBWSxPQUFtQztRQUM3QyxJQUFNLE1BQU0sR0FBRyxrQkFBa0IsQ0FBQyxPQUFPLENBQW1CLENBQUE7UUFDNUQsSUFBSSxDQUFDLGtCQUFrQixHQUFHLE1BQU07YUFDN0IsV0FBVyxDQUFDLElBQUksaUJBQWlCLEVBQUUsQ0FBQzthQUNwQyxXQUFXLENBQUMsc0NBQXNDLEVBQUUsQ0FBQyxDQUFBO0lBQzFELENBQUM7SUFFRCxzQkFBWSwrQ0FBVzthQUF2QjtZQUNRLElBQUEsS0FBVyxJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxFQUFFLEVBQXZDLEVBQUUsUUFBQSxFQUFFLEVBQUUsUUFBaUMsQ0FBQTtZQUM5QyxJQUFJLENBQUMsa0JBQWtCLEdBQUcsRUFBRSxDQUFBO1lBQzVCLE9BQU8sRUFBRSxDQUFBO1FBQ1gsQ0FBQzs7O09BQUE7SUFFRCxzQkFBSSxxREFBaUI7YUFBckI7WUFDRSxPQUFPLG1CQUFtQixDQUFDLElBQUksQ0FBQyxXQUFXLENBQUMsQ0FBQTtRQUM5QyxDQUFDOzs7T0FBQTtJQUVELHNCQUFJLDhDQUFVO2FBQWQ7WUFDRSxPQUFPLG1CQUFtQixDQUFDLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxXQUFXLENBQUMsSUFBSSxlQUFlLENBQXFDO2dCQUNwSCxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7b0JBQ3pCLElBQUk7d0JBQ0YsSUFBTSxJQUFJLEdBQUcsSUFBSSxDQUFDLEtBQUssQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUEwQixDQUFBO3dCQUM1RCxVQUFVLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFBO3FCQUN6QjtvQkFBQyxPQUFPLENBQUMsRUFBRTt3QkFDVixPQUFPLENBQUMsSUFBSSxDQUFDLG1EQUFtRCxFQUFFLENBQUMsQ0FBQyxDQUFBO3FCQUNyRTtnQkFDSCxDQUFDO2FBQ0YsQ0FBQyxDQUFFLENBQUUsQ0FBQTtRQUNSLENBQUM7OztPQUFBO0lBRUQsc0JBQUksOENBQVU7YUFBZDtZQUNFLE9BQU8sbUJBQW1CLENBQUMsSUFBSSxDQUFDLFVBQVUsQ0FBQyxXQUFXLENBQUMsSUFBSSxlQUFlLENBQWdDO2dCQUN4RyxTQUFTLFlBQUMsS0FBSyxFQUFFLFVBQVU7O29CQUN6QixVQUFVLENBQUMsT0FBTyxDQUFDLE1BQUEsTUFBQSxNQUFBLE1BQUEsS0FBSyxhQUFMLEtBQUssdUJBQUwsS0FBSyxDQUFFLE9BQU8sMENBQUcsQ0FBQyxDQUFDLDBDQUFFLEtBQUssMENBQUUsT0FBTyxtQ0FBSSxFQUFFLENBQUMsQ0FBQTtnQkFDL0QsQ0FBQzthQUNGLENBQUMsQ0FBRSxDQUFFLENBQUE7UUFDUixDQUFDOzs7T0FBQTtJQUNILDZCQUFDO0FBQUQsQ0FBQyxBQXhDRCxJQXdDQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7IHR5cGUgUGFyc2VkRXZlbnQgfSBmcm9tICdldmVudHNvdXJjZS1wYXJzZXIvc3RyZWFtJ1xuaW1wb3J0IHtcbiAgY3JlYXRlQXN5bmNJdGVyYWJsZSxcbiAgVHJhbnNmb3JtU3RyZWFtLFxuICBUZXh0RGVjb2RlclN0cmVhbSxcbiAgY3JlYXRlRXZlbnRTb3VyY2VQYXJzZXJUcmFuc2Zvcm1TdHJlYW0sXG4gIHRvUG9seWZpbGxSZWFkYWJsZSxcbn0gZnJvbSAnLi4vLi4vdXRpbHMnXG5pbXBvcnQgdHlwZSB7IFpoaVB1R2VuZXJhdGVUZXh0T3V0cHV0LCBaaGlQdVN0cmVhbVRleHRPdXRwdXQgfSBmcm9tICcuL3R5cGUnXG5pbXBvcnQgdHlwZSB7IENoYXRNb2RlbCwgU3RyZWFtVGV4dFJlc3VsdCwgTW9kZWxSZXEsIEJhc2VDaGF0TW9kZWxJbnB1dCB9IGZyb20gJy4uLy4uL3R5cGUnXG5cbmV4cG9ydCBjbGFzcyBaaGlQdU1vZGVsIGltcGxlbWVudHMgQ2hhdE1vZGVsIHtcbiAgcHVibGljIHN1YlVybCA9ICd6aGlwdS9hcGkvcGFhcy92NC9jaGF0L2NvbXBsZXRpb25zJ1xuICBjb25zdHJ1Y3Rvcihwcml2YXRlIHJlcTogTW9kZWxSZXEsIHB1YmxpYyBiYXNlVXJsOiBzdHJpbmcsIHN1YlVybD86IHN0cmluZykge1xuICAgIGlmIChzdWJVcmwgIT0gbnVsbCkge1xuICAgICAgdGhpcy5zdWJVcmwgPSBzdWJVcmxcbiAgICB9XG4gIH1cblxuICBwcml2YXRlIGdldCB1cmwoKSB7XG4gICAgcmV0dXJuIGAke3RoaXMuYmFzZVVybH0vJHt0aGlzLnN1YlVybH1gXG4gIH1cblxuICBhc3luYyBnZW5lcmF0ZVRleHQoZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KSB7XG4gICAgY29uc3QgcmVzID0gKGF3YWl0IHRoaXMucmVxKHtcbiAgICAgIHVybDogdGhpcy51cmwsXG4gICAgICBkYXRhOiB7XG4gICAgICAgIC4uLmRhdGEsXG4gICAgICAgIHN0cmVhbTogZmFsc2UsXG4gICAgICB9LFxuICAgICAgc3RyZWFtOiBmYWxzZSxcbiAgICB9KSkgYXMgWmhpUHVHZW5lcmF0ZVRleHRPdXRwdXRcbiAgICByZXR1cm4ge1xuICAgICAgdGV4dDogKHJlcz8uY2hvaWNlcz8uWzBdPy5tZXNzYWdlPy5jb250ZW50IGFzIHN0cmluZykgPz8gJycsXG4gICAgICByYXdSZXNwb25zZTogcmVzLFxuICAgIH1cbiAgfVxuXG4gIGFzeW5jIHN0cmVhbVRleHQoZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KSB7XG4gICAgY29uc3Qgc3RyZWFtID0gYXdhaXQgdGhpcy5yZXEoe1xuICAgICAgdXJsOiB0aGlzLnVybCxcbiAgICAgIGRhdGE6IHtcbiAgICAgICAgLi4uZGF0YSxcbiAgICAgICAgc3RyZWFtOiB0cnVlLFxuICAgICAgfSxcbiAgICAgIHN0cmVhbTogdHJ1ZSxcbiAgICB9KVxuICAgIHJldHVybiBuZXcgWmhpUHVNb2RlbFN0cmVhbVJlc3VsdChzdHJlYW0pXG4gIH1cbn1cblxuY2xhc3MgWmhpUHVNb2RlbFN0cmVhbVJlc3VsdCBpbXBsZW1lbnRzIFN0cmVhbVRleHRSZXN1bHQge1xuICBwcml2YXRlIF9ldmVudFNvdXJjZVN0cmVhbTogUmVhZGFibGVTdHJlYW08UGFyc2VkRXZlbnQ+XG5cbiAgY29uc3RydWN0b3IoX3N0cmVhbTogUmVhZGFibGVTdHJlYW08VWludDhBcnJheT4pIHtcbiAgICBjb25zdCBzdHJlYW0gPSB0b1BvbHlmaWxsUmVhZGFibGUoX3N0cmVhbSkgYXMgdHlwZW9mIF9zdHJlYW1cbiAgICB0aGlzLl9ldmVudFNvdXJjZVN0cmVhbSA9IHN0cmVhbVxuICAgICAgLnBpcGVUaHJvdWdoKG5ldyBUZXh0RGVjb2RlclN0cmVhbSgpKVxuICAgICAgLnBpcGVUaHJvdWdoKGNyZWF0ZUV2ZW50U291cmNlUGFyc2VyVHJhbnNmb3JtU3RyZWFtKCkpXG4gIH1cblxuICBwcml2YXRlIGdldCB0ZWVlZFN0cmVhbSgpIHtcbiAgICBjb25zdCBbczEsIHMyXSA9IHRoaXMuX2V2ZW50U291cmNlU3RyZWFtLnRlZSgpXG4gICAgdGhpcy5fZXZlbnRTb3VyY2VTdHJlYW0gPSBzMlxuICAgIHJldHVybiBzMVxuICB9XG5cbiAgZ2V0IGV2ZW50U291cmNlU3RyZWFtKCkge1xuICAgIHJldHVybiBjcmVhdGVBc3luY0l0ZXJhYmxlKHRoaXMudGVlZWRTdHJlYW0pXG4gIH1cblxuICBnZXQgZGF0YVN0cmVhbSgpIHtcbiAgICByZXR1cm4gY3JlYXRlQXN5bmNJdGVyYWJsZSh0aGlzLmV2ZW50U291cmNlU3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08UGFyc2VkRXZlbnQsIFpoaVB1U3RyZWFtVGV4dE91dHB1dD4oe1xuICAgICAgdHJhbnNmb3JtKGNodW5rLCBjb250cm9sbGVyKSB7XG4gICAgICAgIHRyeSB7XG4gICAgICAgICAgY29uc3QgZGF0YSA9IEpTT04ucGFyc2UoY2h1bmsuZGF0YSkgYXMgWmhpUHVTdHJlYW1UZXh0T3V0cHV0XG4gICAgICAgICAgY29udHJvbGxlci5lbnF1ZXVlKGRhdGEpXG4gICAgICAgIH0gY2F0Y2ggKGUpIHtcbiAgICAgICAgICBjb25zb2xlLndhcm4oJ0Vycm9yIHdoZW4gdHJhbnNmb3JtaW5nIGV2ZW50IHNvdXJjZSBkYXRhIHRvIGpzb24nLCBlKVxuICAgICAgICB9XG4gICAgICB9LFxuICAgIH0pLCksKVxuICB9XG5cbiAgZ2V0IHRleHRTdHJlYW0oKSB7XG4gICAgcmV0dXJuIGNyZWF0ZUFzeW5jSXRlcmFibGUodGhpcy5kYXRhU3RyZWFtLnBpcGVUaHJvdWdoKG5ldyBUcmFuc2Zvcm1TdHJlYW08WmhpUHVTdHJlYW1UZXh0T3V0cHV0LCBzdHJpbmc+KHtcbiAgICAgIHRyYW5zZm9ybShjaHVuaywgY29udHJvbGxlcikge1xuICAgICAgICBjb250cm9sbGVyLmVucXVldWUoY2h1bms/LmNob2ljZXM/LlswXT8uZGVsdGE/LmNvbnRlbnQgPz8gJycpXG4gICAgICB9LFxuICAgIH0pLCksKVxuICB9XG59XG4iXX0=
|
|
@@ -0,0 +1,71 @@
|
|
|
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
|
+
tool_calls?: Array<{
|
|
13
|
+
id: string;
|
|
14
|
+
type: 'web_search' | 'retrieval' | 'function';
|
|
15
|
+
function?: {
|
|
16
|
+
name: string;
|
|
17
|
+
auguments: string;
|
|
18
|
+
};
|
|
19
|
+
}>;
|
|
20
|
+
};
|
|
21
|
+
type ToolMessage = {
|
|
22
|
+
role: 'tool';
|
|
23
|
+
content: string;
|
|
24
|
+
tool_call_id: string;
|
|
25
|
+
};
|
|
26
|
+
type ZhiPuMessage = SystemMessage | UserMessage | AssistantMessage | ToolMessage;
|
|
27
|
+
export type ZhiPuInputData = {
|
|
28
|
+
model: string;
|
|
29
|
+
messages: Array<ZhiPuMessage>;
|
|
30
|
+
request_id?: string;
|
|
31
|
+
do_sample?: boolean;
|
|
32
|
+
temperature?: number;
|
|
33
|
+
top_p?: number;
|
|
34
|
+
max_tokens?: number;
|
|
35
|
+
stop?: Array<string>;
|
|
36
|
+
tools?: Array<{
|
|
37
|
+
type: 'web_search' | 'retrieval' | 'function';
|
|
38
|
+
function: {
|
|
39
|
+
name: string;
|
|
40
|
+
description: string;
|
|
41
|
+
parameters: object;
|
|
42
|
+
};
|
|
43
|
+
retrieval: {
|
|
44
|
+
knowledge_id: string;
|
|
45
|
+
prompt_template?: string;
|
|
46
|
+
parameters: object;
|
|
47
|
+
};
|
|
48
|
+
web_search: {
|
|
49
|
+
enable?: boolean;
|
|
50
|
+
search_query?: string;
|
|
51
|
+
search_result?: boolean;
|
|
52
|
+
};
|
|
53
|
+
}>;
|
|
54
|
+
tool_choice?: 'auto';
|
|
55
|
+
user_id?: string;
|
|
56
|
+
};
|
|
57
|
+
export type ZhiPuGenerateTextOutput = {
|
|
58
|
+
choices?: Array<{
|
|
59
|
+
message?: {
|
|
60
|
+
content?: string;
|
|
61
|
+
};
|
|
62
|
+
}>;
|
|
63
|
+
};
|
|
64
|
+
export type ZhiPuStreamTextOutput = {
|
|
65
|
+
choices?: Array<{
|
|
66
|
+
delta?: {
|
|
67
|
+
content?: string;
|
|
68
|
+
};
|
|
69
|
+
}>;
|
|
70
|
+
};
|
|
71
|
+
export {};
|
|
@@ -0,0 +1,2 @@
|
|
|
1
|
+
export {};
|
|
2
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uL3NyYy9tb2RlbHMvWmhpUHUvdHlwZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiIiwic291cmNlc0NvbnRlbnQiOlsidHlwZSBTeXN0ZW1NZXNzYWdlID0ge1xuICByb2xlOiAnc3lzdGVtJ1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxudHlwZSBVc2VyTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3VzZXInXG4gIGNvbnRlbnQ6IHN0cmluZ1xufVxuXG50eXBlIEFzc2lzdGFudE1lc3NhZ2UgPSB7XG4gIHJvbGU6ICdhc3Npc3RhbnQnXG4gIGNvbnRlbnQ/OiBzdHJpbmdcbiAgdG9vbF9jYWxscz86IEFycmF5PHtcbiAgICBpZDogc3RyaW5nXG4gICAgdHlwZTogJ3dlYl9zZWFyY2gnIHwgJ3JldHJpZXZhbCcgfCAnZnVuY3Rpb24nXG4gICAgZnVuY3Rpb24/OiB7IG5hbWU6IHN0cmluZzsgYXVndW1lbnRzOiBzdHJpbmcgfVxuICB9PlxufVxuXG50eXBlIFRvb2xNZXNzYWdlID0ge1xuICByb2xlOiAndG9vbCdcbiAgY29udGVudDogc3RyaW5nXG4gIHRvb2xfY2FsbF9pZDogc3RyaW5nXG59XG5cbnR5cGUgWmhpUHVNZXNzYWdlID0gU3lzdGVtTWVzc2FnZSB8IFVzZXJNZXNzYWdlIHwgQXNzaXN0YW50TWVzc2FnZSB8IFRvb2xNZXNzYWdlXG5cbmV4cG9ydCB0eXBlIFpoaVB1SW5wdXREYXRhID0ge1xuICBtb2RlbDogc3RyaW5nXG4gIG1lc3NhZ2VzOiBBcnJheTxaaGlQdU1lc3NhZ2U+XG4gIHJlcXVlc3RfaWQ/OiBzdHJpbmdcbiAgZG9fc2FtcGxlPzogYm9vbGVhblxuICB0ZW1wZXJhdHVyZT86IG51bWJlclxuICB0b3BfcD86IG51bWJlclxuICBtYXhfdG9rZW5zPzogbnVtYmVyXG4gIHN0b3A/OiBBcnJheTxzdHJpbmc+XG4gIHRvb2xzPzogQXJyYXk8e1xuICAgIHR5cGU6ICd3ZWJfc2VhcmNoJyB8ICdyZXRyaWV2YWwnIHwgJ2Z1bmN0aW9uJ1xuICAgIGZ1bmN0aW9uOiB7IG5hbWU6IHN0cmluZzsgZGVzY3JpcHRpb246IHN0cmluZzsgcGFyYW1ldGVyczogb2JqZWN0IH1cbiAgICByZXRyaWV2YWw6IHsga25vd2xlZGdlX2lkOiBzdHJpbmc7IHByb21wdF90ZW1wbGF0ZT86IHN0cmluZzsgcGFyYW1ldGVyczogb2JqZWN0IH1cbiAgICB3ZWJfc2VhcmNoOiB7IGVuYWJsZT86IGJvb2xlYW47IHNlYXJjaF9xdWVyeT86IHN0cmluZzsgc2VhcmNoX3Jlc3VsdD86IGJvb2xlYW4gfVxuICB9PlxuICB0b29sX2Nob2ljZT86ICdhdXRvJ1xuICB1c2VyX2lkPzogc3RyaW5nXG59XG5cbi8vIHRvZG86IG1vcmUgcHJlY2lzZVxuZXhwb3J0IHR5cGUgWmhpUHVHZW5lcmF0ZVRleHRPdXRwdXQgPSB7XG4gIGNob2ljZXM/OiBBcnJheTx7XG4gICAgbWVzc2FnZT86IHsgY29udGVudD86IHN0cmluZyB9XG4gIH0+XG59XG5cbi8vIHRvZG86IG1vcmUgcHJlY2lzZVxuZXhwb3J0IHR5cGUgWmhpUHVTdHJlYW1UZXh0T3V0cHV0ID0ge1xuICBjaG9pY2VzPzogQXJyYXk8e1xuICAgIGRlbHRhPzogeyBjb250ZW50Pzogc3RyaW5nIH1cbiAgfT5cbn1cbiJdfQ==
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { HunYuanModel } from './HunYuan';
|
|
2
|
+
import { HuoShanModel } from './HuoShan';
|
|
3
|
+
import { LingJiModel } from './LingJi';
|
|
4
|
+
import { LingYiModel } from './LingYi';
|
|
5
|
+
import { YueZhiModel } from './YueZhi';
|
|
6
|
+
import { ZhiPuModel } from './ZhiPu';
|
|
7
|
+
import { HunYuanPreviewModel } from './HunYuanPreview';
|
|
8
|
+
export declare const MODELS: {
|
|
9
|
+
HunYuan: typeof HunYuanModel;
|
|
10
|
+
HunYuanPreview: typeof HunYuanPreviewModel;
|
|
11
|
+
HuoShan: typeof HuoShanModel;
|
|
12
|
+
LingJi: typeof LingJiModel;
|
|
13
|
+
LingYi: typeof LingYiModel;
|
|
14
|
+
YueZhi: typeof YueZhiModel;
|
|
15
|
+
ZhiPu: typeof ZhiPuModel;
|
|
16
|
+
};
|
|
17
|
+
export { HunYuanModel, HunYuanPreviewModel, HuoShanModel, LingJiModel, LingYiModel, YueZhiModel, ZhiPuModel };
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import { HunYuanModel } from './HunYuan';
|
|
2
|
+
import { HuoShanModel } from './HuoShan';
|
|
3
|
+
import { LingJiModel } from './LingJi';
|
|
4
|
+
import { LingYiModel } from './LingYi';
|
|
5
|
+
import { YueZhiModel } from './YueZhi';
|
|
6
|
+
import { ZhiPuModel } from './ZhiPu';
|
|
7
|
+
import { HunYuanPreviewModel } from './HunYuanPreview';
|
|
8
|
+
export var MODELS = {
|
|
9
|
+
HunYuan: HunYuanModel,
|
|
10
|
+
HunYuanPreview: HunYuanPreviewModel,
|
|
11
|
+
HuoShan: HuoShanModel,
|
|
12
|
+
LingJi: LingJiModel,
|
|
13
|
+
LingYi: LingYiModel,
|
|
14
|
+
YueZhi: YueZhiModel,
|
|
15
|
+
ZhiPu: ZhiPuModel,
|
|
16
|
+
};
|
|
17
|
+
export { HunYuanModel, HunYuanPreviewModel, HuoShanModel, LingJiModel, LingYiModel, YueZhiModel, ZhiPuModel };
|
|
18
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi9zcmMvbW9kZWxzL2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxZQUFZLEVBQUUsTUFBTSxXQUFXLENBQUE7QUFDeEMsT0FBTyxFQUFFLFlBQVksRUFBRSxNQUFNLFdBQVcsQ0FBQTtBQUN4QyxPQUFPLEVBQUUsV0FBVyxFQUFFLE1BQU0sVUFBVSxDQUFBO0FBQ3RDLE9BQU8sRUFBRSxXQUFXLEVBQUUsTUFBTSxVQUFVLENBQUE7QUFDdEMsT0FBTyxFQUFFLFdBQVcsRUFBRSxNQUFNLFVBQVUsQ0FBQTtBQUN0QyxPQUFPLEVBQUUsVUFBVSxFQUFFLE1BQU0sU0FBUyxDQUFBO0FBQ3BDLE9BQU8sRUFBRSxtQkFBbUIsRUFBRSxNQUFNLGtCQUFrQixDQUFBO0FBRXRELE1BQU0sQ0FBQyxJQUFNLE1BQU0sR0FBRztJQUNwQixPQUFPLEVBQUUsWUFBWTtJQUNyQixjQUFjLEVBQUUsbUJBQW1CO0lBQ25DLE9BQU8sRUFBRSxZQUFZO0lBQ3JCLE1BQU0sRUFBRSxXQUFXO0lBQ25CLE1BQU0sRUFBRSxXQUFXO0lBQ25CLE1BQU0sRUFBRSxXQUFXO0lBQ25CLEtBQUssRUFBRSxVQUFVO0NBQ2xCLENBQUE7QUFFRCxPQUFPLEVBQUUsWUFBWSxFQUFFLG1CQUFtQixFQUFFLFlBQVksRUFBRSxXQUFXLEVBQUUsV0FBVyxFQUFFLFdBQVcsRUFBRSxVQUFVLEVBQUUsQ0FBQSIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7IEh1bll1YW5Nb2RlbCB9IGZyb20gJy4vSHVuWXVhbidcbmltcG9ydCB7IEh1b1NoYW5Nb2RlbCB9IGZyb20gJy4vSHVvU2hhbidcbmltcG9ydCB7IExpbmdKaU1vZGVsIH0gZnJvbSAnLi9MaW5nSmknXG5pbXBvcnQgeyBMaW5nWWlNb2RlbCB9IGZyb20gJy4vTGluZ1lpJ1xuaW1wb3J0IHsgWXVlWmhpTW9kZWwgfSBmcm9tICcuL1l1ZVpoaSdcbmltcG9ydCB7IFpoaVB1TW9kZWwgfSBmcm9tICcuL1poaVB1J1xuaW1wb3J0IHsgSHVuWXVhblByZXZpZXdNb2RlbCB9IGZyb20gJy4vSHVuWXVhblByZXZpZXcnXG5cbmV4cG9ydCBjb25zdCBNT0RFTFMgPSB7XG4gIEh1bll1YW46IEh1bll1YW5Nb2RlbCxcbiAgSHVuWXVhblByZXZpZXc6IEh1bll1YW5QcmV2aWV3TW9kZWwsXG4gIEh1b1NoYW46IEh1b1NoYW5Nb2RlbCxcbiAgTGluZ0ppOiBMaW5nSmlNb2RlbCxcbiAgTGluZ1lpOiBMaW5nWWlNb2RlbCxcbiAgWXVlWmhpOiBZdWVaaGlNb2RlbCxcbiAgWmhpUHU6IFpoaVB1TW9kZWwsXG59XG5cbmV4cG9ydCB7IEh1bll1YW5Nb2RlbCwgSHVuWXVhblByZXZpZXdNb2RlbCwgSHVvU2hhbk1vZGVsLCBMaW5nSmlNb2RlbCwgTGluZ1lpTW9kZWwsIFl1ZVpoaU1vZGVsLCBaaGlQdU1vZGVsIH1cbiJdfQ==
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import { type ParsedEvent } from 'eventsource-parser/stream';
|
|
2
|
+
export declare abstract class ChatModel {
|
|
3
|
+
abstract generateText(data: BaseChatModelInput): Promise<{
|
|
4
|
+
text: string;
|
|
5
|
+
rawResponse: any;
|
|
6
|
+
}>;
|
|
7
|
+
abstract streamText(data: BaseChatModelInput): Promise<StreamTextResult>;
|
|
8
|
+
}
|
|
9
|
+
export type ChatModelConstructor = typeof ChatModel;
|
|
10
|
+
export declare abstract class StreamTextResult {
|
|
11
|
+
abstract eventSourceStream: AsyncIterableReadableStream<ParsedEvent>;
|
|
12
|
+
abstract dataStream: AsyncIterableReadableStream<any>;
|
|
13
|
+
abstract textStream: AsyncIterableReadableStream<string>;
|
|
14
|
+
}
|
|
15
|
+
export type AsyncIterableReadableStream<T> = ReadableStream<T> & {
|
|
16
|
+
[Symbol.asyncIterator]: () => {
|
|
17
|
+
next(): Promise<IteratorResult<T>>;
|
|
18
|
+
};
|
|
19
|
+
};
|
|
20
|
+
export interface IModelReqInput {
|
|
21
|
+
url: string;
|
|
22
|
+
headers?: Record<string, string>;
|
|
23
|
+
data?: Object;
|
|
24
|
+
stream?: boolean;
|
|
25
|
+
}
|
|
26
|
+
export type ModelReq = <T extends IModelReqInput>(props: T) => T['stream'] extends true ? Promise<ReadableStream<Uint8Array>> : Promise<Object>;
|
|
27
|
+
type ChatModelMessage = {
|
|
28
|
+
role: 'user' | 'system' | 'assistant';
|
|
29
|
+
content: string;
|
|
30
|
+
};
|
|
31
|
+
export interface BaseChatModelInput {
|
|
32
|
+
model: string;
|
|
33
|
+
messages: Array<ChatModelMessage>;
|
|
34
|
+
temperature?: number;
|
|
35
|
+
top_p?: number;
|
|
36
|
+
}
|
|
37
|
+
export {};
|
package/dist/esm/type.js
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
var ChatModel = (function () {
|
|
2
|
+
function ChatModel() {
|
|
3
|
+
}
|
|
4
|
+
return ChatModel;
|
|
5
|
+
}());
|
|
6
|
+
export { ChatModel };
|
|
7
|
+
var StreamTextResult = (function () {
|
|
8
|
+
function StreamTextResult() {
|
|
9
|
+
}
|
|
10
|
+
return StreamTextResult;
|
|
11
|
+
}());
|
|
12
|
+
export { StreamTextResult };
|
|
13
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHlwZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy90eXBlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUVBO0lBQUE7SUFJQSxDQUFDO0lBQUQsZ0JBQUM7QUFBRCxDQUFDLEFBSkQsSUFJQzs7QUFJRDtJQUFBO0lBSUEsQ0FBQztJQUFELHVCQUFDO0FBQUQsQ0FBQyxBQUpELElBSUMiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgeyB0eXBlIFBhcnNlZEV2ZW50IH0gZnJvbSAnZXZlbnRzb3VyY2UtcGFyc2VyL3N0cmVhbSdcblxuZXhwb3J0IGFic3RyYWN0IGNsYXNzIENoYXRNb2RlbCB7XG4gIHB1YmxpYyBhYnN0cmFjdCBnZW5lcmF0ZVRleHQoZGF0YTogQmFzZUNoYXRNb2RlbElucHV0KTogUHJvbWlzZTx7IHRleHQ6IHN0cmluZzsgcmF3UmVzcG9uc2U6IGFueSB9PlxuXG4gIHB1YmxpYyBhYnN0cmFjdCBzdHJlYW1UZXh0KGRhdGE6IEJhc2VDaGF0TW9kZWxJbnB1dCk6IFByb21pc2U8U3RyZWFtVGV4dFJlc3VsdD5cbn1cblxuZXhwb3J0IHR5cGUgQ2hhdE1vZGVsQ29uc3RydWN0b3IgPSB0eXBlb2YgQ2hhdE1vZGVsXG5cbmV4cG9ydCBhYnN0cmFjdCBjbGFzcyBTdHJlYW1UZXh0UmVzdWx0IHtcbiAgcHVibGljIGFic3RyYWN0IGV2ZW50U291cmNlU3RyZWFtOiBBc3luY0l0ZXJhYmxlUmVhZGFibGVTdHJlYW08UGFyc2VkRXZlbnQ+XG4gIHB1YmxpYyBhYnN0cmFjdCBkYXRhU3RyZWFtOiBBc3luY0l0ZXJhYmxlUmVhZGFibGVTdHJlYW08YW55PlxuICBwdWJsaWMgYWJzdHJhY3QgdGV4dFN0cmVhbTogQXN5bmNJdGVyYWJsZVJlYWRhYmxlU3RyZWFtPHN0cmluZz5cbn1cblxuZXhwb3J0IHR5cGUgQXN5bmNJdGVyYWJsZVJlYWRhYmxlU3RyZWFtPFQ+ID0gUmVhZGFibGVTdHJlYW08VD4gJiB7XG4gIFtTeW1ib2wuYXN5bmNJdGVyYXRvcl06ICgpID0+IHsgbmV4dCgpOiBQcm9taXNlPEl0ZXJhdG9yUmVzdWx0PFQ+PiB9XG59XG5cbmV4cG9ydCBpbnRlcmZhY2UgSU1vZGVsUmVxSW5wdXQge1xuICB1cmw6IHN0cmluZ1xuICBoZWFkZXJzPzogUmVjb3JkPHN0cmluZywgc3RyaW5nPlxuICBkYXRhPzogT2JqZWN0XG4gIHN0cmVhbT86IGJvb2xlYW5cbn1cblxuZXhwb3J0IHR5cGUgTW9kZWxSZXEgPSA8VCBleHRlbmRzIElNb2RlbFJlcUlucHV0PihcbiAgcHJvcHM6IFQsXG4pID0+IFRbJ3N0cmVhbSddIGV4dGVuZHMgdHJ1ZSA/IFByb21pc2U8UmVhZGFibGVTdHJlYW08VWludDhBcnJheT4+IDogUHJvbWlzZTxPYmplY3Q+XG5cbnR5cGUgQ2hhdE1vZGVsTWVzc2FnZSA9IHtcbiAgcm9sZTogJ3VzZXInIHwgJ3N5c3RlbScgfCAnYXNzaXN0YW50J1xuICBjb250ZW50OiBzdHJpbmdcbn1cblxuZXhwb3J0IGludGVyZmFjZSBCYXNlQ2hhdE1vZGVsSW5wdXQge1xuICBtb2RlbDogc3RyaW5nXG4gIG1lc3NhZ2VzOiBBcnJheTxDaGF0TW9kZWxNZXNzYWdlPlxuICB0ZW1wZXJhdHVyZT86IG51bWJlclxuICB0b3BfcD86IG51bWJlclxufVxuIl19
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { type ParsedEvent } from 'eventsource-parser';
|
|
2
|
+
import type { AsyncIterableReadableStream } from './type';
|
|
3
|
+
export declare const ReadableStream: {
|
|
4
|
+
new (underlyingSource: UnderlyingByteSource, strategy?: {
|
|
5
|
+
highWaterMark?: number;
|
|
6
|
+
}): ReadableStream<Uint8Array>;
|
|
7
|
+
new <R = any>(underlyingSource: UnderlyingDefaultSource<R>, strategy?: QueuingStrategy<R>): ReadableStream<R>;
|
|
8
|
+
new <R_1 = any>(underlyingSource?: UnderlyingSource<R_1>, strategy?: QueuingStrategy<R_1>): ReadableStream<R_1>;
|
|
9
|
+
prototype: ReadableStream;
|
|
10
|
+
};
|
|
11
|
+
export declare const TransformStream: {
|
|
12
|
+
new <I = any, O = any>(transformer?: Transformer<I, O>, writableStrategy?: QueuingStrategy<I>, readableStrategy?: QueuingStrategy<O>): TransformStream<I, O>;
|
|
13
|
+
prototype: TransformStream;
|
|
14
|
+
};
|
|
15
|
+
export declare const createEventSourceParserTransformStream: () => TransformStream<string, ParsedEvent>;
|
|
16
|
+
export declare const toPolyfillReadable: import("@mattiasbuelens/web-streams-adapter").ReadableStreamWrapper;
|
|
17
|
+
export declare class TextDecoderStream {
|
|
18
|
+
private handle;
|
|
19
|
+
private transform;
|
|
20
|
+
constructor(encoding?: string, options?: TextDecoderOptions);
|
|
21
|
+
get encoding(): string;
|
|
22
|
+
get fatal(): boolean;
|
|
23
|
+
get ignoreBOM(): boolean;
|
|
24
|
+
get readable(): ReadableStream<any>;
|
|
25
|
+
get writable(): WritableStream<any>;
|
|
26
|
+
get [Symbol.toStringTag](): string;
|
|
27
|
+
}
|
|
28
|
+
export declare function createAsyncIterable<T>(stream: ReadableStream<T>): AsyncIterableReadableStream<T>;
|