@baishuyun/coze-provider 0.0.5 → 0.0.7

package/.turbo/turbo-build.log

@@ -1,26 +1,23 @@
 
 
-> @baishuyun/coze-provider@0.0.1-alpha.1 build /Users/bbbottle/Code/bs-chat-sdk/packages/coze-provider
-> pnpm clean && tsup --tsconfig tsconfig.json
-
-
-> @baishuyun/coze-provider@0.0.1-alpha.1 clean /Users/bbbottle/Code/bs-chat-sdk/packages/coze-provider
-> del-cli dist *.tsbuildinfo
+> @baishuyun/coze-provider@0.0.6 build /Users/bbbottle/Code/bs-chat-sdk/packages/coze-provider
+> tsup --tsconfig tsconfig.json --clean
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.5.1
 CLI Using tsup config: /Users/bbbottle/Code/bs-chat-sdk/packages/coze-provider/tsup.config.ts
 CLI Target: es2022
+CLI Cleaning output folder
 CJS Build start
 ESM Build start
-CJS dist/index.js     45.32 KB
-CJS dist/index.js.map 83.63 KB
-CJS ⚡️ Build success in 20ms
-ESM dist/index.mjs     43.27 KB
-ESM dist/index.mjs.map 83.23 KB
-ESM ⚡️ Build success in 20ms
+ESM dist/index.mjs     44.72 KB
+ESM dist/index.mjs.map 85.63 KB
+ESM ⚡️ Build success in 21ms
+CJS dist/index.js     46.81 KB
+CJS dist/index.js.map 86.03 KB
+CJS ⚡️ Build success in 21ms
 DTS Build start
-DTS ⚡️ Build success in 754ms
-DTS dist/index.d.ts  1.54 KB
-DTS dist/index.d.mts 1.54 KB
+DTS ⚡️ Build success in 802ms
+DTS dist/index.d.ts  1.57 KB
+DTS dist/index.d.mts 1.57 KB

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @baishuyun/coze-provider
 
+## 0.0.7
+
+### Patch Changes
+
+- add ai filling
+
+## 0.0.6
+
+### Patch Changes
+
+- add ai filling
+
 ## 0.0.5
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1,7 +1,7 @@
 import { LanguageModelV2StreamPart, ProviderV2, LanguageModelV2, JSONValue } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 
-type CozeChatModelId = "chat";
+type CozeChatModelId = 'chat';
 
 interface CozeProviderSettings {
     apiKey?: string;
@@ -19,7 +19,7 @@ interface CozeProviderSettings {
   or to provide a custom fetch implementation for e.g. testing.
   */
     fetch?: FetchFunction;
-    extraStreamTransformers?: TransformStream<LanguageModelV2StreamPart, any>[];
+    extraStreamTransformers?: (() => TransformStream<LanguageModelV2StreamPart, any>)[];
 }
 interface CozeProvider extends ProviderV2 {
     (modelId: CozeChatModelId): LanguageModelV2;
@@ -37,6 +37,6 @@ declare class SuggestionTransformStream extends TransformStream<any, any> {
     constructor(enable: boolean);
 }
 
-declare const createTextInfoEnqueuer: (controller: TransformStreamDefaultController<LanguageModelV2StreamPart>) => (textInfo: string, metaPayload?: JSONValue, id?: string) => void;
+declare const createTextInfoEnqueuer: (controller: TransformStreamDefaultController<LanguageModelV2StreamPart>) => (textInfo: string, metaPayload?: JSONValue, id?: string, deltaOnly?: boolean) => void;
 
 export { SuggestionTransformStream, createCoze, createFieldsJsonTransformStream, createTextInfoEnqueuer };

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import { LanguageModelV2StreamPart, ProviderV2, LanguageModelV2, JSONValue } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 
-type CozeChatModelId = "chat";
+type CozeChatModelId = 'chat';
 
 interface CozeProviderSettings {
     apiKey?: string;
@@ -19,7 +19,7 @@ interface CozeProviderSettings {
   or to provide a custom fetch implementation for e.g. testing.
   */
     fetch?: FetchFunction;
-    extraStreamTransformers?: TransformStream<LanguageModelV2StreamPart, any>[];
+    extraStreamTransformers?: (() => TransformStream<LanguageModelV2StreamPart, any>)[];
 }
 interface CozeProvider extends ProviderV2 {
     (modelId: CozeChatModelId): LanguageModelV2;
@@ -37,6 +37,6 @@ declare class SuggestionTransformStream extends TransformStream<any, any> {
     constructor(enable: boolean);
 }
 
-declare const createTextInfoEnqueuer: (controller: TransformStreamDefaultController<LanguageModelV2StreamPart>) => (textInfo: string, metaPayload?: JSONValue, id?: string) => void;
+declare const createTextInfoEnqueuer: (controller: TransformStreamDefaultController<LanguageModelV2StreamPart>) => (textInfo: string, metaPayload?: JSONValue, id?: string, deltaOnly?: boolean) => void;
 
 export { SuggestionTransformStream, createCoze, createFieldsJsonTransformStream, createTextInfoEnqueuer };

package/dist/index.js

@@ -39,7 +39,7 @@ module.exports = __toCommonJS(index_exports);
 
 // src/coze-provider.ts
 var import_provider = require("@ai-sdk/provider");
-var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/const.ts
 var VERSION = "0.0.1";
@@ -105,7 +105,6 @@ var parseMessageContent = (message, isMultiModalMsg) => {
     const isExcel = part.mediaType === "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
     if (isImg) {
       const meta = part.providerOptions?.uploadMetadata;
-      console.log("img meta", meta);
       return meta?.parsedData;
     }
     if (isExcel) {
@@ -141,6 +140,33 @@ var convertPromptToMessages = (messages) => {
   });
 };
 
+// src/utils/create-error-stream.ts
+var import_provider_utils = require("@ai-sdk/provider-utils");
+var createErrorStream = (error) => {
+  return new ReadableStream({
+    start(controller) {
+      const id = (0, import_provider_utils.generateId)();
+      controller.enqueue({
+        id,
+        type: "text-start"
+      });
+      controller.enqueue({
+        id,
+        type: "text-delta",
+        providerMetadata: {
+          raw: { data: { type: "agent-error", error } }
+        },
+        delta: error
+      });
+      controller.enqueue({
+        type: "text-end",
+        id
+      });
+      controller.close();
+    }
+  });
+};
+
 // src/evt-sample.ts
 var EvtSample = class {
   _evtSampleMap = /* @__PURE__ */ new Map();
@@ -212,7 +238,6 @@ var MCPChunkFilterTransformStream = class extends TransformStream {
     super({
       transform: (evt, controller) => {
         if (!evt) return;
-        console.log("origin event", evt);
         const part = {
           event: evt.type,
           data: JSON.parse(evt.data) || {}
@@ -264,11 +289,17 @@ var mcp2llmChunk = (part) => {
           rawValue: part.data
         };
       }
+      if (part.status === "completed") {
+        return {
+          type: "text-end",
+          id: part.data.conversation_id,
+          // delta: part.data.content || part.data.reasoning_content || "",
+          providerMetadata: {}
+        };
+      }
       return {
-        type: "text-end",
-        id: part.data.conversation_id,
-        // delta: part.data.content || part.data.reasoning_content || "",
-        providerMetadata: {}
+        type: "raw",
+        rawValue: part.data
       };
       break;
     case import_api3.ChatEventType.CONVERSATION_CHAT_COMPLETED:
@@ -364,14 +395,27 @@ var CozeLanguageModel = class {
       path: "/chat"
     });
     const user_id = options.headers ? options.headers["x-user-id"] : "";
+    const user_var = options.headers ? options.headers["x-user-var"] : "";
+    const user_token = options.headers ? options.headers["x-user-token"] : "";
+    const idFromToken = user_token?.slice(-16);
+    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 || "123456",
+      user_id: user_id || idFromToken || "guest_" + Date.now(),
       stream: true,
       additional_messages: messages
     };
-    console.log("messages", messages);
+    if (customVar) {
+      body.custom_var = customVar;
+    }
     const headers = this.config.headers();
     const response = await fetch(url, {
       method: "POST",
@@ -387,19 +431,21 @@ var CozeLanguageModel = class {
       } catch {
         errorDetail = rawError;
       }
-      throw new Error(
-        `Coze API error: ${response.status} ${response.statusText}`
-      );
+      const errorMessage = typeof errorDetail === "object" && errorDetail?.msg ? errorDetail.msg : `Coze API error: ${response.status} ${response.statusText}`;
+      console.error(errorMessage);
+      return {
+        stream: createErrorStream("\u4F1A\u8BDD\u5F02\u5E38\uFF0C\u8BF7\u91CD\u8BD5")
+      };
     }
-    let externalTransformers = this.config.extraStreamTransformers || [];
+    let externalTransformers = (this.config.extraStreamTransformers || []).map(
+      (factory) => factory()
+    );
     const stage = options.headers ? options.headers["x-user-stage"] : "";
     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());
+    ).pipeThrough(new import_parse_sse.ServerSentEventTransformStream()).pipeThrough(new MCPChunkFilterTransformStream(isBuildStage || isFillStage)).pipeThrough(new LLMChunkInjectorTransformStream()).pipeThrough(new LLMPartTrasnformStream());
     let finalResult = result;
     if (externalTransformers.length > 0) {
       finalResult = externalTransformers.reduce(
@@ -418,12 +464,9 @@ var CozeLanguageModel = class {
 function createCoze(options = {
   botId: ""
 }) {
-  const baseURL = (0, import_provider_utils.withoutTrailingSlash)(options.baseURL);
-  const agentPartHeader = (0, import_provider_utils.withUserAgentSuffix)(
-    options.headers || {},
-    `ai-sdk/coze/${VERSION}`
-  );
-  const authorizationHeader = `Bearer ${(0, import_provider_utils.loadApiKey)({
+  const baseURL = (0, import_provider_utils2.withoutTrailingSlash)(options.baseURL);
+  const agentPartHeader = (0, import_provider_utils2.withUserAgentSuffix)(options.headers || {}, `ai-sdk/coze/${VERSION}`);
+  const authorizationHeader = `Bearer ${(0, import_provider_utils2.loadApiKey)({
     apiKey: options.apiKey,
     environmentVariableName: "COZE_API_KEY",
     description: ""
@@ -463,25 +506,32 @@ function createCoze(options = {
 var import_json = require("@streamparser/json");
 
 // src/utils/enqueue-text-info.ts
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
-var createTextInfoEnqueuer = (controller) => (textInfo, metaPayload, id = (0, import_provider_utils2.generateId)()) => {
-  controller.enqueue({
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
+var createTextInfoEnqueuer = (controller) => (textInfo, metaPayload, id = (0, import_provider_utils3.generateId)(), deltaOnly = false) => {
+  const i = id ?? (0, import_provider_utils3.generateId)();
+  const startPart = {
     type: "text-start",
-    id
-  });
-  const chunk = {
+    id: i
+  };
+  const deltaPart = {
     type: "text-delta",
-    id,
+    id: i,
     delta: textInfo,
     providerMetadata: metaPayload ? {
       raw: { data: metaPayload }
     } : void 0
   };
-  controller.enqueue(chunk);
-  controller.enqueue({
+  const endPart = {
     type: "text-end",
-    id
-  });
+    id: i
+  };
+  if (deltaOnly) {
+    controller.enqueue(deltaPart);
+    return;
+  }
+  controller.enqueue(startPart);
+  controller.enqueue(deltaPart);
+  controller.enqueue(endPart);
 };
 
 // src/utils/fields-json-rebuild/rebuild-text.ts
@@ -1677,7 +1727,6 @@ var createFieldsJsonTransformStream = (enableJsonParser) => {
     },
     transform: (chunk, controller) => {
       if (!parser) {
-        console.log("parser not init");
         return;
       }
       try {