@baishuyun/coze-provider 0.0.16 → 0.1.0

package/dist/index.mjs

@@ -12,8 +12,8 @@ import {
 var VERSION = "0.0.1";
 
 // src/coze-language-model.ts
-import { ServerSentEventTransformStream } from "parse-sse";
-import { TextDecoderStream } from "stream/web";
+import { ServerSentEventTransformStream as ServerSentEventTransformStream2 } from "parse-sse";
+import { TextDecoderStream as TextDecoderStream2 } from "stream/web";
 
 // src/utils/prompt-to-model-msg.ts
 import { RoleType } from "@coze/api";
@@ -107,6 +107,62 @@ var convertPromptToMessages = (messages) => {
   });
 };
 
+// src/utils/accumulate-sse-response.ts
+import { ServerSentEventTransformStream } from "parse-sse";
+import { TextDecoderStream } from "stream/web";
+async function accumulateSSEResponse(body) {
+  const sseStream = body.pipeThrough(
+    new TextDecoderStream()
+  ).pipeThrough(new ServerSentEventTransformStream());
+  let textContent = "";
+  let reasoningContent = "";
+  let finishReason = "stop";
+  let usage = { inputTokens: 0, outputTokens: 0, totalTokens: 0 };
+  const reader = sseStream.getReader();
+  try {
+    while (true) {
+      const { done, value: evt } = await reader.read();
+      if (done) break;
+      if (!evt) continue;
+      let evtData;
+      try {
+        evtData = JSON.parse(evt.data);
+      } catch {
+        continue;
+      }
+      const event = evt.type;
+      if (event === "conversation.message.delta") {
+        if (evtData.reasoning_content) {
+          reasoningContent += evtData.reasoning_content;
+        } else if (evtData.content) {
+          textContent += evtData.content;
+        }
+      } else if (event === "conversation.chat.completed" || event === "done") {
+        finishReason = "stop";
+        if (evtData.usage) {
+          usage = {
+            inputTokens: evtData.usage.input_count || 0,
+            outputTokens: evtData.usage.output_count || 0,
+            totalTokens: evtData.usage.total_count || 0
+          };
+        }
+      } else if (event === "error") {
+        finishReason = "error";
+      }
+    }
+  } finally {
+    reader.releaseLock();
+  }
+  const content = [];
+  if (reasoningContent) {
+    content.push({ type: "reasoning", text: reasoningContent });
+  }
+  if (textContent) {
+    content.push({ type: "text", text: textContent });
+  }
+  return { content, finishReason, usage };
+}
+
 // src/utils/create-error-stream.ts
 import { generateId } from "@ai-sdk/provider-utils";
 var createErrorStream = (error) => {
@@ -333,6 +389,31 @@ var LLMPartTrasnformStream = class extends TransformStream {
   }
 };
 
+// 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
+      }
+    ]
+  };
+};
+
 // src/coze-language-model.ts
 var CozeLanguageModel = class {
   specificationVersion = "v2";
@@ -346,8 +427,59 @@ var CozeLanguageModel = class {
     this.supportedUrls = config.supportedUrls?.() || {};
     this.config = config;
   }
-  doGenerate(options) {
-    throw new Error("Method not implemented.");
+  async doGenerate(options) {
+    const messages = convertPromptToMessages(options.prompt);
+    const warnings = [];
+    const url = this.config.url({ path: "/chat" });
+    const user_id = options.headers ? options.headers["x-user-id"] : "";
+    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: this.config.botId,
+      auto_save_history: false,
+      user_id: user_id || "7543191937730215936",
+      stream: true,
+      additional_messages: messages
+    };
+    if (customVar) {
+      body.custom_variables = customVar;
+    }
+    const headers = this.config.headers();
+    const abortSignal = options.abortSignal;
+    const response = await fetch(url, {
+      method: "POST",
+      headers,
+      credentials: "include",
+      body: JSON.stringify(body),
+      signal: abortSignal
+    });
+    if (!response.ok || !response.body) {
+      const rawError = await response.text();
+      let errorDetail;
+      try {
+        errorDetail = JSON.parse(rawError);
+      } catch {
+        errorDetail = rawError;
+      }
+      const errorMessage = typeof errorDetail === "object" && errorDetail?.msg ? errorDetail.msg : `Coze API error: ${response.status} ${response.statusText}`;
+      return BuildWarningGenerateResult(errorMessage);
+    }
+    const { content, finishReason, usage } = await accumulateSSEResponse(response.body);
+    return {
+      content,
+      finishReason,
+      usage,
+      request: { body },
+      response: { body: null },
+      warnings
+    };
   }
   isVisibleMsgPart(evtData) {
     if (evtData.type === "knowledge" && evtData.role === "assistant") {
@@ -361,7 +493,7 @@ var CozeLanguageModel = class {
     const url = this.config.url({
       path: "/chat"
     });
-    const user_id = (options.headers ? options.headers["x-user-id"] : "") || Date.now().toString();
+    const user_id = options.headers ? options.headers["x-user-id"] : "";
     const user_var = options.headers ? options.headers["x-user-var"] : "";
     let customVar;
     if (user_var) {
@@ -374,14 +506,12 @@ var CozeLanguageModel = class {
     const body = {
       bot_id: this.config.botId,
       auto_save_history: false,
-      user_id: "7543191937730215936",
-      // user_id,
+      user_id: user_id || "7543191937730215936",
       stream: true,
       additional_messages: messages
     };
-    console.log("Request body for Coze API:", JSON.stringify(body));
     if (customVar) {
-      body.custom_var = customVar;
+      body.custom_variables = customVar;
     }
     const headers = this.config.headers();
     const abortSignal = options.abortSignal;
@@ -413,8 +543,8 @@ var CozeLanguageModel = class {
     const isBuildStage = stage === "build";
     const isFillStage = stage === "fill";
     const result = response.body.pipeThrough(
-      new TextDecoderStream()
-    ).pipeThrough(new ServerSentEventTransformStream()).pipeThrough(new MCPChunkFilterTransformStream(isBuildStage || isFillStage)).pipeThrough(new LLMChunkInjectorTransformStream()).pipeThrough(new LLMPartTrasnformStream());
+      new TextDecoderStream2()
+    ).pipeThrough(new ServerSentEventTransformStream2()).pipeThrough(new MCPChunkFilterTransformStream(isBuildStage || isFillStage)).pipeThrough(new LLMChunkInjectorTransformStream()).pipeThrough(new LLMPartTrasnformStream());
     let finalResult = result;
     if (externalTransformers.length > 0) {
       finalResult = externalTransformers.reduce(
@@ -1659,6 +1789,9 @@ var RebuilderMap = {
   dept: rebuildDept,
   sn: rebuildSn,
   deptgroup: rebuildDeptGroup,
+  flowState: defaultBuilder,
+  flowstate: defaultBuilder,
+  formula: defaultBuilder,
   unknown: defaultBuilder
 };
 var rebuildFieldsJsonStructure = (truncatedJson, subItemsOnly = false) => {
@@ -1676,9 +1809,12 @@ var rebuildFieldsJsonStructure = (truncatedJson, subItemsOnly = false) => {
 // src/stream-trasnformer/fields-json-transform-stream.ts
 var createFieldsJsonTransformStream = (enableJsonParser) => {
   let parser;
+  let enqueueTextDelta;
   let id;
+  let isFirstChunk = true;
+  let passthroughMode = false;
   const chunkGuard = (chunk) => {
-    if (!enableJsonParser) {
+    if (!enableJsonParser || passthroughMode) {
       return false;
     }
     if (chunk.type !== "text-delta" || !("delta" in chunk)) {
@@ -1695,19 +1831,28 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
       }
     },
     transform: (chunk, controller) => {
-      if (!parser) {
+      if (!parser && !passthroughMode) {
         return;
       }
       try {
         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);
       }
     },
@@ -1715,7 +1860,7 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
       parser = new JSONParser({
         paths: ["$.*"]
       });
-      const enqueueTextDelta = enableJsonParser ? createTextInfoEnqueuer(controller) : (content) => {
+      enqueueTextDelta = enableJsonParser ? createTextInfoEnqueuer(controller) : (content) => {
       };
       parser.onValue = (parsedInfo) => {
         const value = parsedInfo.value;
@@ -1731,12 +1876,9 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
           id
         );
       };
-      let errorLogged = false;
       parser.onError = (err) => {
-        if (!errorLogged) {
-          console.error("JsonWidgetStream: JSON Parsing Error:", err);
-          errorLogged = true;
-        }
+        console.error("JsonWidgetStream: JSON Parsing Error:", err);
+        passthroughMode = true;
       };
       parser.onEnd = () => {
         enqueueTextDelta(