@baishuyun/coze-provider 0.0.15 → 0.0.17

package/dist/index.mjs

@@ -333,6 +333,118 @@ var LLMPartTrasnformStream = class extends TransformStream {
   }
 };
 
+// src/utils/create-coze-non-streaming-chat.ts
+var createCozeNonStreamingChat = async (botId, options, chatConfig) => {
+  const messages = convertPromptToMessages(options.prompt);
+  const url = chatConfig.url({
+    path: "/chat"
+  });
+  const user_id = (options.headers ? options.headers["x-user-id"] : "") || Date.now().toString();
+  const user_var = options.headers ? options.headers["x-user-var"] : "";
+  let customVar;
+  if (user_var) {
+    try {
+      customVar = JSON.parse(user_var);
+    } catch (e) {
+      console.warn("x-user-var parse error", e);
+    }
+  }
+  const body = {
+    bot_id: botId,
+    auto_save_history: true,
+    user_id: user_id || "7543191937730215936",
+    additional_messages: messages,
+    stream: false
+  };
+  if (customVar) {
+    body.custom_var = customVar;
+  }
+  const headers = chatConfig.headers();
+  const abortSignal = options.abortSignal;
+  const response = await fetch(url, {
+    method: "POST",
+    headers,
+    credentials: "include",
+    body: JSON.stringify(body),
+    signal: abortSignal
+  });
+  return response;
+};
+
+// src/utils/get-coze-chat-msg-detail.ts
+var getCozeChatMsgDetail = async (conversationId, chatId, chatConfig) => {
+  const url = chatConfig.url({
+    path: `/chat/message/list?conversation_id=${conversationId}&chat_id=${chatId}`
+  });
+  const response = await fetch(url, {
+    method: "POST",
+    headers: chatConfig.headers(),
+    credentials: "include"
+  });
+  return response;
+};
+
+// src/utils/build-generate-result-by-chat-res.ts
+var BuildWarningGenerateResult = (message) => {
+  return {
+    content: [],
+    finishReason: "error",
+    request: {
+      body: null
+    },
+    usage: {
+      inputTokens: 0,
+      outputTokens: 0,
+      totalTokens: 0
+    },
+    response: {
+      body: null
+    },
+    warnings: [
+      {
+        type: "other",
+        message
+      }
+    ]
+  };
+};
+var BuildGenerateResultByChatRes = async (response, chatConfig, reqBody) => {
+  if (!response.ok) {
+    return Promise.reject(
+      BuildWarningGenerateResult(`Coze API error: ${response.status} ${response.statusText}`)
+    );
+  }
+  const res = await response.json();
+  if (res.code !== 0) {
+    return Promise.reject(BuildWarningGenerateResult(`Coze API error: ${res.code} ${res.msg}`));
+  }
+  const msgRes = await getCozeChatMsgDetail(res.data.conversation_id, res.data.id, chatConfig);
+  if (!msgRes.ok) {
+    return Promise.reject(
+      BuildWarningGenerateResult(
+        `Failed to get message detail: ${msgRes.status} ${msgRes.statusText}`
+      )
+    );
+  }
+  const msgDetail = await msgRes.json();
+  return {
+    content: msgDetail.data.message.content,
+    finishReason: msgDetail.data.message.status === "finished" ? "stop" : "error",
+    request: {
+      body: reqBody
+    },
+    usage: {
+      inputTokens: 0,
+      outputTokens: 0,
+      totalTokens: 0
+    },
+    response: {
+      body: msgDetail
+    },
+    warnings: []
+  };
+};
+
 // src/coze-language-model.ts
 var CozeLanguageModel = class {
   specificationVersion = "v2";
@@ -346,8 +458,23 @@ var CozeLanguageModel = class {
     this.supportedUrls = config.supportedUrls?.() || {};
     this.config = config;
   }
-  doGenerate(options) {
-    throw new Error("Method not implemented.");
+  // @not-support
+  // coze 目前不支持非流式接口，为了兼容接口规范，仍然实现了 doGenerate 方法
+  async doGenerate(options) {
+    return BuildWarningGenerateResult(
+      "Coze provider does not support non-streaming calls, please use streaming interface"
+    );
+    const botIdFromProvider = options.providerOptions?.bot?.botId;
+    if (!botIdFromProvider) {
+      return BuildWarningGenerateResult("bot id missing in provider options");
+    }
+    const response = await createCozeNonStreamingChat(
+      botIdFromProvider,
+      options,
+      this.config
+    );
+    const result = await BuildGenerateResultByChatRes(response, this.config);
+    return result;
   }
   isVisibleMsgPart(evtData) {
     if (evtData.type === "knowledge" && evtData.role === "assistant") {
@@ -374,20 +501,22 @@ var CozeLanguageModel = class {
     const body = {
       bot_id: this.config.botId,
       auto_save_history: false,
-      user_id,
+      user_id: "7543191937730215936",
+      // user_id,
       stream: true,
       additional_messages: messages
     };
-    console.log("Request body for Coze API:", JSON.stringify(body));
     if (customVar) {
       body.custom_var = customVar;
     }
     const headers = this.config.headers();
+    const abortSignal = options.abortSignal;
     const response = await fetch(url, {
       method: "POST",
       headers,
       credentials: "include",
-      body: JSON.stringify(body)
+      body: JSON.stringify(body),
+      signal: abortSignal
     });
     if (!response.ok || !response.body) {
       const rawError = await response.text();
@@ -1656,6 +1785,9 @@ var RebuilderMap = {
   dept: rebuildDept,
   sn: rebuildSn,
   deptgroup: rebuildDeptGroup,
+  flowState: defaultBuilder,
+  flowstate: defaultBuilder,
+  formula: defaultBuilder,
   unknown: defaultBuilder
 };
 var rebuildFieldsJsonStructure = (truncatedJson, subItemsOnly = false) => {
@@ -1674,6 +1806,7 @@ var rebuildFieldsJsonStructure = (truncatedJson, subItemsOnly = false) => {
 var createFieldsJsonTransformStream = (enableJsonParser) => {
   let parser;
   let id;
+  let isFirstChunk = true;
   const chunkGuard = (chunk) => {
     if (!enableJsonParser) {
       return false;
@@ -1699,12 +1832,21 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
         if (chunkGuard(chunk)) {
           if ("id" in chunk && "delta" in chunk) {
             id = chunk.id;
-            parser.write(chunk.delta);
+            let delta = chunk.delta;
+            if (isFirstChunk) {
+              delta = delta.replace(/^\uFEFF/, "");
+              isFirstChunk = false;
+            }
+            parser.write(delta);
+          } else {
+            console.warn("Unexpected chunk format for JSON parsing:", chunk);
+            controller.enqueue(chunk);
           }
         } else {
           controller.enqueue(chunk);
         }
       } catch (e) {
+        console.log("write or enqueue error", e);
         controller.error(e);
       }
     },
@@ -1731,6 +1873,7 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
       let errorLogged = false;
       parser.onError = (err) => {
         if (!errorLogged) {
+          console.log("is enabled", enableJsonParser);
           console.error("JsonWidgetStream: JSON Parsing Error:", err);
           errorLogged = true;
         }