@leikeduntech/leiai-js 2.5.1 → 2.7.0

package/build/index.d.ts

@@ -1,6 +1,6 @@
 import Keyv from 'keyv';
 
-type Role = 'user' | 'assistant' | 'system';
+type Role = 'user' | 'assistant' | 'system' | 'function';
 type FetchFn = typeof fetch;
 type ChatGPTAPIOptions = {
     /** 模型厂商公司 **/
@@ -336,6 +336,7 @@ declare namespace openai {
          * @memberof CreateChatCompletionRequest
          */
         user?: string;
+        pluginList?: string | object;
     }
     /**
      * @type CreateChatCompletionRequestStop
@@ -473,6 +474,11 @@ declare class ChatGPTAPI {
      * @param fetch - Optional override for the `fetch` implementation to use. Defaults to the global `fetch` function.
      */
     constructor(opts: ChatGPTAPIOptions);
+    pluginListMap(pluginList: any): any;
+    DallE3Fun(): {
+        pluginList: string;
+        description: string;
+    };
     /**
      * Sends a message to the OpenAI chat completions endpoint, waits for the response
      * to resolve, and returns the response.

package/build/index.js

@@ -225,6 +225,16 @@ var ChatGPTAPI = class {
       throw new Error('Invalid "fetch" is not a function');
     }
   }
+  pluginListMap(pluginList) {
+    const list = {
+      "dall-e-3": "DallE3Fun"
+    };
+    return list[pluginList];
+  }
+  // DallE3文生图插件
+  DallE3Fun() {
+    return { pluginList: "dall-e-3", description: "\u4F7F\u7528DALL-E3\u6A21\u578B\u6839\u636E\u7528\u6237\u63CF\u8FF0\u63D0\u793A\u7ED8\u56FE\u753B\u753B\u4F5C\u56FE" };
+  }
   /**
    * Sends a message to the OpenAI chat completions endpoint, waits for the response
    * to resolve, and returns the response.
@@ -331,6 +341,11 @@ Current date: ${currentDate}`;
           url = "";
           delete headers.Authorization;
         }
+        let pluginList;
+        if (completionParams.pluginList) {
+          pluginList = completionParams.pluginList;
+          delete completionParams.pluginList;
+        }
         let body = {
           max_tokens: maxTokens,
           ...this._completionParams,
@@ -338,7 +353,33 @@ Current date: ${currentDate}`;
           messages,
           stream
         };
-        if (this._manufacturer.toLowerCase() === "aliyun") {
+        if (["openai"].indexOf(this._manufacturer.toLowerCase()) > -1) {
+          switch (pluginList) {
+            case "plugin1":
+              break;
+            case "dall-e-3":
+              body = Object.assign(body, { tools: [{
+                type: "function",
+                function: {
+                  "name": "DallE3Fun",
+                  "description": this.DallE3Fun().description,
+                  "parameters": {
+                    "type": "object",
+                    "properties": {
+                      "prompt": {
+                        "type": "string",
+                        "description": "\u7528\u6237\u7684\u63D0\u793A\u8BCD"
+                      }
+                    },
+                    "required": ["prompt"]
+                  }
+                }
+              }], tool_choice: "auto" });
+              break;
+            default:
+              break;
+          }
+        } else if (this._manufacturer.toLowerCase() === "aliyun") {
           body = Object.assign(body, { parameters: { result_format: "message" }, input: { messages } });
           delete body.messages;
         } else if (this._manufacturer.toLowerCase() === "zhipu") {
@@ -458,11 +499,19 @@ Current date: ${currentDate}`;
                     }
                     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;
+                      result.delta = "";
+                      if (response.choices[0].finish_reason === "tool_calls") {
+                        result.delta = text;
+                      } else if (delta.content) {
+                        result.delta = delta.content;
+                      }
                       if (delta == null ? void 0 : delta.content)
-                        result.text += delta.content;
+                        result.text += result.delta;
                       result.role = "assistant";
-                      if (delta.role) {
+                      if (response.choices[0].finish_reason === "tool_calls") {
+                        result.role = "function";
+                        result.name = this.pluginListMap(pluginList);
+                      } else if (delta.role) {
                         result.role = delta.role;
                       }
                       result.detail = response;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@leikeduntech/leiai-js",
-  "version": "2.5.1",
+  "version": "2.7.0",
   "author": "liuhean",
   "repository": {
     "type": "git",
@@ -41,7 +41,7 @@
     "node": ">=14"
   },
   "dependencies": {
-    "@leikeduntech/spark-nodejs": "0.3.3",
+    "@leikeduntech/spark-nodejs": "1.0.1",
     "cac": "^6.7.14",
     "conf": "^11.0.1",
     "crypto-js": "^4.1.1",