@leikeduntech/leiai-js 2.3.6 → 2.5.0

package/build/index.d.ts

@@ -510,7 +510,6 @@ declare class ChatGPTAPI {
     set apiOrg(apiOrg: string);
     protected _buildMessages(text: string, opts: SendMessageOptions): Promise<{
         messages: openai.ChatCompletionRequestMessage[];
-        aliyunMessage: any;
         maxTokens: number;
         numTokens: number;
         errorMessage: string;

package/build/index.js

@@ -108,6 +108,37 @@ async function fetchSSE(url, options, fetch2 = fetch, manufacturer = "OpenAI") {
 
 // src/chatgpt-api.ts
 import { Spark } from "@leikeduntech/spark-nodejs";
+
+// src/utils.ts
+import CryptoJS from "crypto-js";
+var uuidv4Re = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i;
+function isValidUUIDv4(str) {
+  return str && uuidv4Re.test(str);
+}
+function signTencentHunyuan(bodyData, url, keyList) {
+  const sortedObj = {};
+  Object.keys(bodyData).sort().forEach((key) => {
+    sortedObj[key] = bodyData[key];
+  });
+  let signStr = "";
+  for (let key in sortedObj) {
+    if (signStr) {
+      if (typeof sortedObj[key] === "object") {
+        signStr += `&${key}=${JSON.stringify(sortedObj[key])}`;
+      } else {
+        signStr += `&${key}=${sortedObj[key]}`;
+      }
+    } else {
+      signStr += `${key}=${sortedObj[key]}`;
+    }
+  }
+  signStr = `${url.replace("https://", "")}?${signStr}`;
+  const hmac = CryptoJS.HmacSHA1(signStr, keyList[2]);
+  const signBase64 = CryptoJS.enc.Base64.stringify(hmac);
+  return signBase64;
+}
+
+// src/chatgpt-api.ts
 var CHATGPT_MODEL = "gpt-3.5-turbo";
 var USER_LABEL_DEFAULT = "User";
 var ASSISTANT_LABEL_DEFAULT = "ChatGPT";
@@ -155,12 +186,6 @@ var ChatGPTAPI = class {
       ...completionParams
     };
     this._systemMessage = systemMessage;
-    if (this._systemMessage === void 0) {
-      const currentDate = (/* @__PURE__ */ new Date()).toISOString().split("T")[0];
-      this._systemMessage = `You are ChatGPT, a large language model trained by ${this._manufacturer}. Answer as concisely as possible.
-Knowledge cutoff: 2021-09-01
-Current date: ${currentDate}`;
-    }
     this._maxModelTokens = maxModelTokens;
     this._maxResponseTokens = maxResponseTokens;
     this._getMessageById = getMessageById ?? this._defaultGetMessageById;
@@ -205,6 +230,13 @@ Current date: ${currentDate}`;
    * @returns The response from ChatGPT
    */
   async sendMessage(text, opts = {}) {
+    var _a;
+    if (this._systemMessage === void 0 && ["openai", "azure"].indexOf((_a = this._manufacturer) == null ? void 0 : _a.toLowerCase()) > -1) {
+      const currentDate = (/* @__PURE__ */ new Date()).toISOString().split("T")[0];
+      this._systemMessage = `You are ChatGPT, a large language model trained by ${this._manufacturer}. Answer as concisely as possible.
+Knowledge cutoff: 2021-09-01
+Current date: ${currentDate}`;
+    }
     const {
       parentMessageId,
       messageId = uuidv4(),
@@ -228,7 +260,7 @@ Current date: ${currentDate}`;
       text
     };
     const latestQuestion = message;
-    const { messages, aliyunMessage, maxTokens, numTokens, errorMessage } = await this._buildMessages(
+    const { messages, maxTokens, numTokens, errorMessage } = await this._buildMessages(
       text,
       opts
     );
@@ -255,7 +287,7 @@ Current date: ${currentDate}`;
     };
     const responseP = new Promise(
       async (resolve, reject) => {
-        var _a, _b, _c, _d, _e;
+        var _a2, _b, _c, _d, _e;
         let url = `${this._apiBaseUrl}/chat/completions`;
         const headers = {
           "Content-Type": "application/json",
@@ -289,20 +321,28 @@ Current date: ${currentDate}`;
           stream
         };
         if (this._manufacturer.toLowerCase() === "aliyun") {
+          body = Object.assign(body, { parameters: { result_format: "message" }, input: { messages } });
           delete body.messages;
-          body = Object.assign(body, { input: aliyunMessage });
         } else if (this._manufacturer.toLowerCase() === "zhipu") {
           delete body.messages;
           body = Object.assign(body, { prompt: messages });
+        } else if (this._manufacturer.toLowerCase() === "tencent") {
+          url = this._apiBaseUrl;
+          const timestamp = Math.ceil((/* @__PURE__ */ new Date()).getTime() / 1e3) + 1;
+          const keyList = this._apiKey.split(".");
+          body = Object.assign(body, { app_id: parseInt(keyList[0]), secret_id: keyList[1], timestamp, expired: timestamp + 24 * 60 * 60, stream: stream ? 1 : 0 });
+          delete body.model;
+          delete body.max_tokens;
+          headers["Authorization"] = signTencentHunyuan(body, url, keyList);
         }
         if (this._apiOrg && this._manufacturer.toLowerCase() === "openai") {
           headers["OpenAI-Organization"] = this._apiOrg;
         }
         if (this._debug) {
-          console.log(`api url (${url}`);
-          console.log(`api header (${JSON.stringify(headers)}`);
+          console.log(`api url ${url}`);
+          console.log(`api header ${JSON.stringify(headers)}`);
           console.log(`sendMessage (${numTokens} tokens) body: `, body);
-          console.log(`sendMessage (${numTokens} tokens) message : `, aliyunMessage || messages);
+          console.log(`sendMessage (${numTokens} tokens) message : `, messages, typeof messages);
         }
         if (this._manufacturer.toLowerCase() === "xunfei") {
           const self = this;
@@ -351,7 +391,7 @@ Current date: ${currentDate}`;
                 body: JSON.stringify(body),
                 signal: abortSignal,
                 onMessage: (data) => {
-                  var _a2, _b2, _c2, _d2, _e2, _f, _g, _h, _i, _j, _k;
+                  var _a3, _b2, _c2, _d2, _e2, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v;
                   if (this._debug) {
                   }
                   if (data === "[DONE]") {
@@ -360,6 +400,8 @@ Current date: ${currentDate}`;
                   }
                   try {
                     const response = JSON.parse(data);
+                    if (this._debug) {
+                    }
                     if (this._manufacturer.toLowerCase() === "baidu") {
                       if ((response == null ? void 0 : response.is_end) === true) {
                         result.text += response.result.trim();
@@ -367,13 +409,13 @@ Current date: ${currentDate}`;
                         return resolve(result);
                       }
                     } else if (this._manufacturer.toLowerCase() === "azure") {
-                      if (((_a2 = response.choices[0]) == null ? void 0 : _a2.finish_reason) === "stop") {
+                      if (((_a3 = response.choices[0]) == null ? void 0 : _a3.finish_reason) === "stop") {
                         result.text = result.text.trim();
                         return resolve(result);
                       }
                     } else if (this._manufacturer.toLowerCase() === "aliyun") {
-                      if (((_b2 = response == null ? void 0 : response.output) == null ? void 0 : _b2.finish_reason) === "stop") {
-                        result.text = response == null ? void 0 : response.output.text.trim();
+                      if (["stop", "length"].indexOf((_c2 = (_b2 = response == null ? void 0 : response.output) == null ? void 0 : _b2.choices[0]) == null ? void 0 : _c2.finish_reason) > -1) {
+                        result.text = (_f = (_e2 = (_d2 = response == null ? void 0 : response.output) == null ? void 0 : _d2.choices[0]) == null ? void 0 : _e2.message) == null ? void 0 : _f.content.trim();
                         return resolve(result);
                       }
                     } else if (this._manufacturer.toLowerCase() === "zhipu") {
@@ -381,6 +423,11 @@ Current date: ${currentDate}`;
                         result.text += response == null ? void 0 : response.data.trim();
                         return resolve(result);
                       }
+                    } else if (this._manufacturer.toLowerCase() === "tencent") {
+                      if (((_g = response.choices[0]) == null ? void 0 : _g.finish_reason) === "stop") {
+                        result.text += (_i = (_h = response == null ? void 0 : response.choices[0]) == null ? void 0 : _h.delta) == null ? void 0 : _i.content.trim();
+                        return resolve(result);
+                      }
                     }
                     if (this._manufacturer.toLowerCase() === "aliyun") {
                       if (response == null ? void 0 : response.request_id) {
@@ -391,11 +438,12 @@ Current date: ${currentDate}`;
                         result.id = response.id;
                       }
                     }
-                    if (((_c2 = response.choices) == null ? void 0 : _c2.length) && ["openai", "azure"].indexOf(this._manufacturer.toLowerCase()) > -1) {
+                    if (((_j = response.choices) == null ? void 0 : _j.length) && ["openai", "azure", "tencent"].indexOf(this._manufacturer.toLowerCase()) > -1) {
                       const delta = response.choices[0].delta;
                       result.delta = delta.content;
                       if (delta == null ? void 0 : delta.content)
                         result.text += delta.content;
+                      result.role = "assistant";
                       if (delta.role) {
                         result.role = delta.role;
                       }
@@ -409,17 +457,17 @@ Current date: ${currentDate}`;
                       result.detail = response;
                       onProgress == null ? void 0 : onProgress(result);
                     } else if ((response == null ? void 0 : response.output) && this._manufacturer.toLowerCase() === "aliyun") {
-                      response.usage = Object.assign(response.usage, { prompt_tokens: (_d2 = response.usage) == null ? void 0 : _d2.input_tokens, completion_tokens: (_e2 = response.usage) == null ? void 0 : _e2.output_tokens, total_tokens: ((_f = response.usage) == null ? void 0 : _f.input_tokens) + ((_g = response.usage) == null ? void 0 : _g.output_tokens) });
+                      response.usage = Object.assign(response.usage, { prompt_tokens: (_k = response.usage) == null ? void 0 : _k.input_tokens, completion_tokens: (_l = response.usage) == null ? void 0 : _l.output_tokens, total_tokens: ((_m = response.usage) == null ? void 0 : _m.input_tokens) + ((_n = response.usage) == null ? void 0 : _n.output_tokens) });
                       result.delta = "";
-                      if ((_h = response == null ? void 0 : response.output) == null ? void 0 : _h.text)
-                        result.text = (_i = response == null ? void 0 : response.output) == null ? void 0 : _i.text;
+                      if ((_q = (_p = (_o = response == null ? void 0 : response.output) == null ? void 0 : _o.choices[0]) == null ? void 0 : _p.message) == null ? void 0 : _q.content)
+                        result.text = (_t = (_s = (_r = response == null ? void 0 : response.output) == null ? void 0 : _r.choices[0]) == null ? void 0 : _s.message) == null ? void 0 : _t.content;
                       result.role = "assistant";
                       result.detail = response;
                       onProgress == null ? void 0 : onProgress(result);
                     } else if ((response == null ? void 0 : response.data) && this._manufacturer.toLowerCase() === "zhipu") {
                       if (response.event === "finish") {
-                        if ((_j = response == null ? void 0 : response.meta) == null ? void 0 : _j.usage) {
-                          response.usage = (_k = response == null ? void 0 : response.meta) == null ? void 0 : _k.usage;
+                        if ((_u = response == null ? void 0 : response.meta) == null ? void 0 : _u.usage) {
+                          response.usage = (_v = response == null ? void 0 : response.meta) == null ? void 0 : _v.usage;
                         } else {
                           response.usage = { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 };
                         }
@@ -469,7 +517,7 @@ Current date: ${currentDate}`;
                   result.id = response.id;
                 }
               }
-              if (((_a = response == null ? void 0 : response.choices) == null ? void 0 : _a.length) && ["openai", "azure"].indexOf(this._manufacturer.toLowerCase()) > -1) {
+              if (((_a2 = response == null ? void 0 : response.choices) == null ? void 0 : _a2.length) && ["openai", "azure"].indexOf(this._manufacturer.toLowerCase()) > -1) {
                 const message2 = response.choices[0].message;
                 result.text = message2.content;
                 if (message2.role) {
@@ -498,13 +546,13 @@ Current date: ${currentDate}`;
         }
       }
     ).then(async (message2) => {
-      var _a, _b;
+      var _a2, _b;
       if (message2.detail && !message2.detail.usage) {
         try {
           const promptTokens = numTokens;
           let completionTokens = 0;
           if (["baidu", "aliyun"].indexOf(this._manufacturer.toLowerCase()) > -1) {
-            completionTokens = (_b = (_a = message2.detail) == null ? void 0 : _a.usage) == null ? void 0 : _b.total_tokens;
+            completionTokens = (_b = (_a2 = message2.detail) == null ? void 0 : _a2.usage) == null ? void 0 : _b.total_tokens;
           } else {
             completionTokens = await this._getTokenCount(message2.text);
           }
@@ -581,14 +629,14 @@ Current date: ${currentDate}`;
     const assistantLabel = ASSISTANT_LABEL_DEFAULT;
     const maxNumTokens = this._maxModelTokens - this._maxResponseTokens;
     let messages = [];
-    if (systemMessage && ["openai", "azure"].indexOf(this._manufacturer.toLowerCase()) > -1) {
+    if (systemMessage && ["openai", "azure", "aliyun"].indexOf(this._manufacturer.toLowerCase()) > -1) {
       messages.push({
         role: "system",
         content: systemMessage
       });
     }
     const systemMessageOffset = messages.length;
-    const userMessage = ["baidu", "azure", "zhipu", "xunfei"].indexOf(this._manufacturer.toLowerCase()) > -1 ? [{ role: "user", content: text }] : [{ role: "user", content: text, name: opts.name }];
+    const userMessage = ["baidu", "azure", "zhipu", "xunfei", "aliyun", "tencent"].indexOf(this._manufacturer.toLowerCase()) > -1 ? [{ role: "user", content: text }] : [{ role: "user", content: text, name: opts.name }];
     let nextMessages = text ? messages.concat(userMessage) : messages;
     let numTokens = 0;
     do {
@@ -623,7 +671,7 @@ ${message.content}`]);
         break;
       }
       const parentMessageRole = parentMessage.role || "user";
-      const parentMessageItem = ["baidu", "azure", "zhipu", "xunfei"].indexOf(this._manufacturer.toLowerCase()) > -1 ? {
+      const parentMessageItem = ["baidu", "azure", "zhipu", "xunfei", "aliyun", "tencent"].indexOf(this._manufacturer.toLowerCase()) > -1 ? {
         role: parentMessageRole,
         content: parentMessage.text
       } : {
@@ -647,28 +695,7 @@ ${message.content}`]);
       maxTokens = this._maxModelTokens;
       errorMessage = `${this._manufacturer}\uFF1A\u5F53\u524D\u63D0\u95EE\u4E0A\u4E0B\u6587\u5185\u5BB9\u957F\u5EA6${numTokens}tokns\u8D85\u957F\uFF0C\u8BE5\u6A21\u578B\u6700\u5927\u63D0\u95EE\u957F\u5EA6\u4E3A${this._maxModelTokens}tokens\uFF0C\u8BF7\u5207\u6362\u5176\u4ED6\u589E\u5F3AAI\u6A21\u578B\u6216\u51CF\u5C11\u5B57\u6570\u6216\u8005\u5173\u95ED\u4E0A\u4E0B\u6587\u5386\u53F2\u63D0\u9AD8\u5355\u6B21\u63D0\u95EE\u957F\u5EA6\uFF01`;
     }
-    let aliyunMessage;
-    if (this._manufacturer.toLowerCase() === "aliyun") {
-      aliyunMessage = { prompt: "", history: [] };
-      let onceMessage = { user: "", bot: "" };
-      messages.forEach((item, index) => {
-        if (index < messages.length - 1 && index > 1) {
-          if (!onceMessage.user || !onceMessage.bot) {
-            if (item.role === "user")
-              onceMessage.user = item.content;
-            if (item.role === "assistant")
-              onceMessage.bot = item.content;
-          }
-          if (onceMessage.user && onceMessage.bot) {
-            aliyunMessage.history.push(onceMessage);
-            onceMessage = { user: "", bot: "" };
-          }
-        } else {
-          aliyunMessage.prompt = item.content;
-        }
-      });
-    }
-    return { messages, aliyunMessage, maxTokens, numTokens, errorMessage };
+    return { messages, maxTokens, numTokens, errorMessage };
   }
   async _getTokenCount(text) {
     text = text.replace(/<\|endoftext\|>/g, "");
@@ -686,14 +713,6 @@ ${message.content}`]);
 // src/chatgpt-unofficial-proxy-api.ts
 import pTimeout2 from "p-timeout";
 import { v4 as uuidv42 } from "uuid";
-
-// src/utils.ts
-var uuidv4Re = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i;
-function isValidUUIDv4(str) {
-  return str && uuidv4Re.test(str);
-}
-
-// src/chatgpt-unofficial-proxy-api.ts
 var ChatGPTUnofficialProxyAPI = class {
   /**
    * @param fetch - Optional override for the `fetch` implementation to use. Defaults to the global `fetch` function.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@leikeduntech/leiai-js",
-  "version": "2.3.6",
+  "version": "2.5.0",
   "author": "liuhean",
   "repository": {
     "type": "git",
@@ -41,15 +41,16 @@
     "node": ">=14"
   },
   "dependencies": {
+    "@leikeduntech/spark-nodejs": "0.3.3",
     "cac": "^6.7.14",
     "conf": "^11.0.1",
+    "crypto-js": "^4.1.1",
     "eventsource-parser": "^1.0.0",
     "js-tiktoken": "^1.0.5",
     "keyv": "^4.5.2",
     "p-timeout": "^6.1.1",
     "quick-lru": "^6.1.1",
     "read-pkg-up": "^9.1.0",
-    "@leikeduntech/spark-nodejs": "0.3.3",
     "uuid": "^9.0.0"
   },
   "devDependencies": {