@coze/uniapp-api
Version:
Official Coze UniApp SDK for seamless AI integration into your applications | 扣子官方 UniApp SDK，助您轻松集成 AI 能力到应用中
259 lines (258 loc) • 13.7 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __await = (this && this.__await) || function (v) { return this instanceof __await ? (this.v = v, this) : new __await(v); }
var __asyncGenerator = (this && this.__asyncGenerator) || function (thisArg, _arguments, generator) {
if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
var g = generator.apply(thisArg, _arguments || []), i, q = [];
return i = Object.create((typeof AsyncIterator === "function" ? AsyncIterator : Object).prototype), verb("next"), verb("throw"), verb("return", awaitReturn), i[Symbol.asyncIterator] = function () { return this; }, i;
function awaitReturn(f) { return function (v) { return Promise.resolve(v).then(f, reject); }; }
function verb(n, f) { if (g[n]) { i[n] = function (v) { return new Promise(function (a, b) { q.push([n, v, a, b]) > 1 || resume(n, v); }); }; if (f) i[n] = f(i[n]); } }
function resume(n, v) { try { step(g[n](v)); } catch (e) { settle(q[0][3], e); } }
function step(r) { r.value instanceof __await ? Promise.resolve(r.value.v).then(fulfill, reject) : settle(q[0][2], r); }
function fulfill(value) { resume("next", value); }
function reject(value) { resume("throw", value); }
function settle(f, v) { if (f(v), q.shift(), q.length) resume(q[0][0], q[0][1]); }
};
var __rest = (this && this.__rest) || function (s, e) {
var t = {};
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
t[p] = s[p];
if (s != null && typeof Object.getOwnPropertySymbols === "function")
for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
t[p[i]] = s[p[i]];
}
return t;
};
var __asyncValues = (this && this.__asyncValues) || function (o) {
if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
var m = o[Symbol.asyncIterator], i;
return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
};
var __asyncDelegator = (this && this.__asyncDelegator) || function (o) {
var i, p;
return i = {}, verb("next"), verb("throw", function (e) { throw e; }), verb("return"), i[Symbol.iterator] = function () { return this; }, i;
function verb(n, f) { i[n] = o[n] ? function (v) { return (p = !p) ? { value: __await(o[n](v)), done: false } : f ? f(v) : v; } : f; }
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.handleParameters = exports.handleAdditionalMessages = void 0;
exports.getWorkflowStreamMixin = getWorkflowStreamMixin;
exports.getChatStreamMixin = getChatStreamMixin;
exports.getWorkflowChatStreamMixin = getWorkflowChatStreamMixin;
exports.getUploadFileMixin = getUploadFileMixin;
/* eslint-disable @typescript-eslint/no-magic-numbers -- ignore */
const api_1 = require("@coze/api");
const request_1 = require("../event-source/request");
const uuid = () => {
const a = Math.random();
return (a * new Date().getTime()).toString();
};
const handleAdditionalMessages = (additional_messages) => additional_messages === null || additional_messages === void 0 ? void 0 : additional_messages.map(i => (Object.assign(Object.assign({}, i), { content: typeof i.content === 'object' ? JSON.stringify(i.content) : i.content })));
exports.handleAdditionalMessages = handleAdditionalMessages;
const handleParameters = (parameters) => {
if (parameters) {
for (const [key, value] of Object.entries(parameters)) {
if (typeof value === 'object') {
parameters[key] = JSON.stringify(value);
}
}
}
return parameters;
};
exports.handleParameters = handleParameters;
// workflows.runs.stream
function getWorkflowStreamMixin(api) {
return function stream(params, options) {
return __asyncGenerator(this, arguments, function* stream_1() {
var _a, _b, _c, _d;
yield __await(refreshToken(api, params, options));
const result = {
messages: [],
done: false,
deferred: null,
error: null,
};
(0, request_1.sendRequest)({
url: `${(_a = api.options.baseURL) !== null && _a !== void 0 ? _a : ''}/v1/workflow/stream_run`,
method: 'POST',
data: params,
headers: Object.assign({
Authorization: `Bearer ${api.options.token}`,
}, api.options.headers, options === null || options === void 0 ? void 0 : options.headers),
signal: options === null || options === void 0 ? void 0 : options.signal,
timeout: (_b = options === null || options === void 0 ? void 0 : options.timeout) !== null && _b !== void 0 ? _b : (_c = api.axiosOptions) === null || _c === void 0 ? void 0 : _c.timeout,
}, result);
while (true) {
if (result.done) {
break;
}
if (!result.messages.length) {
yield __await(((_d = result.deferred) === null || _d === void 0 ? void 0 : _d.promise));
if (result.error) {
throw result.error;
}
}
let message = result.messages.shift();
while (message) {
if (message.event === api_1.WorkflowEventType.DONE) {
yield yield __await(new api_1.WorkflowEvent(Number(message.id), api_1.WorkflowEventType.DONE));
}
else {
yield yield __await(new api_1.WorkflowEvent(Number(message.id), message.event, JSON.parse(message.data)));
}
message = result.messages.shift();
}
}
});
};
}
// chat.stream
function getChatStreamMixin(api) {
return function stream(params, options) {
return __asyncGenerator(this, arguments, function* stream_2() {
var _a, _b, _c;
yield __await(refreshToken(api, params, options));
const result = {
messages: [],
done: false,
deferred: null,
error: null,
};
if (!params.user_id) {
params.user_id = uuid();
}
const { conversation_id } = params, rest = __rest(params, ["conversation_id"]);
(0, request_1.sendRequest)({
url: `${(_a = api.options.baseURL) !== null && _a !== void 0 ? _a : ''}/v3/chat${conversation_id ? `?conversation_id=${conversation_id}` : ''}`,
method: 'POST',
data: Object.assign(Object.assign({ stream: true }, rest), { additional_messages: (0, exports.handleAdditionalMessages)(params.additional_messages), shortcut_command: params.shortcut_command
? Object.assign(Object.assign({}, params.shortcut_command), { parameters: (0, exports.handleParameters)(params.shortcut_command.parameters) }) : undefined }),
headers: Object.assign({
Authorization: `Bearer ${api.options.token}`,
}, api.options.headers, options === null || options === void 0 ? void 0 : options.headers),
signal: options === null || options === void 0 ? void 0 : options.signal,
timeout: (_b = options === null || options === void 0 ? void 0 : options.timeout) !== null && _b !== void 0 ? _b : (_c = api.axiosOptions) === null || _c === void 0 ? void 0 : _c.timeout,
}, result);
yield __await(yield* __asyncDelegator(__asyncValues(handleStreamMessages(result))));
});
};
}
// workflows.chat.stream
function getWorkflowChatStreamMixin(api) {
return function stream(params, options) {
return __asyncGenerator(this, arguments, function* stream_3() {
var _a, _b, _c;
yield __await(refreshToken(api, params, options));
const result = {
messages: [],
done: false,
deferred: null,
error: null,
};
(0, request_1.sendRequest)({
url: `${(_a = api.options.baseURL) !== null && _a !== void 0 ? _a : ''}/v1/workflows/chat`,
method: 'POST',
data: Object.assign(Object.assign({}, params), { additional_messages: (0, exports.handleAdditionalMessages)(params.additional_messages) }),
headers: Object.assign({
Authorization: `Bearer ${api.options.token}`,
}, api.options.headers, options === null || options === void 0 ? void 0 : options.headers),
signal: options === null || options === void 0 ? void 0 : options.signal,
timeout: (_b = options === null || options === void 0 ? void 0 : options.timeout) !== null && _b !== void 0 ? _b : (_c = api.axiosOptions) === null || _c === void 0 ? void 0 : _c.timeout,
}, result);
yield __await(yield* __asyncDelegator(__asyncValues(handleStreamMessages(result))));
});
};
}
// files.upload
function getUploadFileMixin(api) {
return (params, options) => __awaiter(this, void 0, void 0, function* () {
yield refreshToken(api, params, options);
return new Promise((resolve, reject) => {
var _a, _b, _c, _d, _e;
const task = uni.uploadFile({
url: `${(_a = api.options.baseURL) !== null && _a !== void 0 ? _a : ''}/v1/files/upload`,
header: Object.assign({
Authorization: `Bearer ${api.options.token}`,
}, api.options.headers, options === null || options === void 0 ? void 0 : options.headers),
filePath: params.file.filePath,
name: 'file',
timeout: (_b = options === null || options === void 0 ? void 0 : options.timeout) !== null && _b !== void 0 ? _b : (_c = api.axiosOptions) === null || _c === void 0 ? void 0 : _c.timeout,
success(res) {
if (res.statusCode !== 200) {
reject(new Error(res.data));
}
else {
resolve(JSON.parse(res.data).data);
}
},
fail(res) {
reject(new Error(res.errMsg));
},
});
if (options === null || options === void 0 ? void 0 : options.signal) {
(_e = (_d = options.signal).addEventListener) === null || _e === void 0 ? void 0 : _e.call(_d, 'abort', () => {
task.abort();
});
}
});
});
}
function refreshToken(api, params, options) {
return __awaiter(this, void 0, void 0, function* () {
var _a;
if (api.options.onBeforeAPICall) {
const config = (_a = (yield api.options.onBeforeAPICall(Object.assign(Object.assign({}, options), { data: params })))) !== null && _a !== void 0 ? _a : {};
if (config.token) {
api.options.token = config.token;
api.token = config.token;
}
}
});
}
function handleStreamMessages(result) {
return __asyncGenerator(this, arguments, function* handleStreamMessages_1() {
var _a;
while (true) {
if (result.done) {
break;
}
if (!result.messages.length) {
yield __await(((_a = result.deferred) === null || _a === void 0 ? void 0 : _a.promise));
if (result.error) {
throw result.error;
}
}
let message = result.messages.shift();
while (message) {
try {
if (message.event === api_1.ChatEventType.DONE) {
yield yield __await({
event: message.event,
data: '[DONE]',
});
}
else {
yield yield __await({
event: message.event,
data: JSON.parse(message.data),
});
}
}
catch (e) {
console.error('Failed to parse json data from event-stream', e);
}
message = result.messages.shift();
}
}
});
}