@baishuyun/coze-provider 0.0.17 → 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,57 +389,6 @@ 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 {
@@ -408,42 +413,6 @@ var BuildWarningGenerateResult = (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 {
@@ -458,23 +427,59 @@ var CozeLanguageModel = class {
     this.supportedUrls = config.supportedUrls?.() || {};
     this.config = config;
   }
-  // @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 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 response = await createCozeNonStreamingChat(
-      botIdFromProvider,
-      options,
-      this.config
-    );
-    const result = await BuildGenerateResultByChatRes(response, this.config);
-    return result;
+    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") {
@@ -488,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) {
@@ -501,13 +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
     };
     if (customVar) {
-      body.custom_var = customVar;
+      body.custom_variables = customVar;
     }
     const headers = this.config.headers();
     const abortSignal = options.abortSignal;
@@ -539,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(
@@ -1805,10 +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)) {
@@ -1825,7 +1831,7 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
       }
     },
     transform: (chunk, controller) => {
-      if (!parser) {
+      if (!parser && !passthroughMode) {
         return;
       }
       try {
@@ -1854,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;
@@ -1870,13 +1876,9 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
           id
         );
       };
-      let errorLogged = false;
       parser.onError = (err) => {
-        if (!errorLogged) {
-          console.log("is enabled", enableJsonParser);
-          console.error("JsonWidgetStream: JSON Parsing Error:", err);
-          errorLogged = true;
-        }
+        console.error("JsonWidgetStream: JSON Parsing Error:", err);
+        passthroughMode = true;
       };
       parser.onEnd = () => {
         enqueueTextDelta(