@baishuyun/coze-provider 0.0.16 → 0.1.0

package/.turbo/turbo-build.log

@@ -1,6 +1,6 @@
 
 
-> @baishuyun/coze-provider@0.0.8 build /Users/bbbottle/Code/bs-chat-sdk/packages/coze-provider
+> @baishuyun/coze-provider@0.0.17 build /Users/bbbottle/Code/bs-chat-sdk/packages/coze-provider
 > tsup --tsconfig tsconfig.json --clean
 
 CLI Building entry: src/index.ts
@@ -11,13 +11,13 @@
 CLI Cleaning output folder
 CJS Build start
 ESM Build start
-CJS dist/index.js     46.81 KB
-CJS dist/index.js.map 86.03 KB
-CJS ⚡️ Build success in 32ms
-ESM dist/index.mjs     44.72 KB
-ESM dist/index.mjs.map 85.63 KB
-ESM ⚡️ Build success in 32ms
+CJS dist/index.js     50.91 KB
+CJS dist/index.js.map 94.87 KB
+CJS ⚡️ Build success in 30ms
+ESM dist/index.mjs     48.87 KB
+ESM dist/index.mjs.map 94.55 KB
+ESM ⚡️ Build success in 31ms
 DTS Build start
-DTS ⚡️ Build success in 1097ms
+DTS ⚡️ Build success in 860ms
 DTS dist/index.d.ts  1.57 KB
 DTS dist/index.d.mts 1.57 KB

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # @baishuyun/coze-provider
 
+## 0.1.0
+
+### Minor Changes
+
+- widget agent
+- 4acb2cc: widget agent
+- 35058a9: widget agent
+
+### Patch Changes
+
+- e5dd21a: widget agent
+- a934bc0: add widget agent
+
+## 0.0.17
+
+### Patch Changes
+
+- update style and bug fix
+
 ## 0.0.16
 
 ### Patch Changes

package/dist/index.js

@@ -45,8 +45,8 @@ var import_provider_utils2 = require("@ai-sdk/provider-utils");
 var VERSION = "0.0.1";
 
 // src/coze-language-model.ts
-var import_parse_sse = require("parse-sse");
-var import_web = require("stream/web");
+var import_parse_sse2 = require("parse-sse");
+var import_web2 = require("stream/web");
 
 // src/utils/prompt-to-model-msg.ts
 var import_api = require("@coze/api");
@@ -140,6 +140,62 @@ var convertPromptToMessages = (messages) => {
   });
 };
 
+// src/utils/accumulate-sse-response.ts
+var import_parse_sse = require("parse-sse");
+var import_web = require("stream/web");
+async function accumulateSSEResponse(body) {
+  const sseStream = body.pipeThrough(
+    new import_web.TextDecoderStream()
+  ).pipeThrough(new import_parse_sse.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
 var import_provider_utils = require("@ai-sdk/provider-utils");
 var createErrorStream = (error) => {
@@ -366,6 +422,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";
@@ -379,8 +460,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") {
@@ -394,7 +526,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) {
@@ -407,14 +539,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;
@@ -446,8 +576,8 @@ var CozeLanguageModel = class {
     const isBuildStage = stage === "build";
     const isFillStage = stage === "fill";
     const result = response.body.pipeThrough(
-      new import_web.TextDecoderStream()
-    ).pipeThrough(new import_parse_sse.ServerSentEventTransformStream()).pipeThrough(new MCPChunkFilterTransformStream(isBuildStage || isFillStage)).pipeThrough(new LLMChunkInjectorTransformStream()).pipeThrough(new LLMPartTrasnformStream());
+      new import_web2.TextDecoderStream()
+    ).pipeThrough(new import_parse_sse2.ServerSentEventTransformStream()).pipeThrough(new MCPChunkFilterTransformStream(isBuildStage || isFillStage)).pipeThrough(new LLMChunkInjectorTransformStream()).pipeThrough(new LLMPartTrasnformStream());
     let finalResult = result;
     if (externalTransformers.length > 0) {
       finalResult = externalTransformers.reduce(
@@ -1692,6 +1822,9 @@ var RebuilderMap = {
   dept: rebuildDept,
   sn: rebuildSn,
   deptgroup: rebuildDeptGroup,
+  flowState: defaultBuilder,
+  flowstate: defaultBuilder,
+  formula: defaultBuilder,
   unknown: defaultBuilder
 };
 var rebuildFieldsJsonStructure = (truncatedJson, subItemsOnly = false) => {
@@ -1709,9 +1842,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)) {
@@ -1728,19 +1864,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);
       }
     },
@@ -1748,7 +1893,7 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
       parser = new import_json.JSONParser({
         paths: ["$.*"]
       });
-      const enqueueTextDelta = enableJsonParser ? createTextInfoEnqueuer(controller) : (content) => {
+      enqueueTextDelta = enableJsonParser ? createTextInfoEnqueuer(controller) : (content) => {
       };
       parser.onValue = (parsedInfo) => {
         const value = parsedInfo.value;
@@ -1764,12 +1909,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(