@baishuyun/chat-backend 0.0.18 → 0.0.20

package/src/services/fetchCozeInfo.ts

@@ -5,7 +5,7 @@ import { logger } from '../logger/index.js';
 export const fetchCozeInfo = async (bsTeamId: string): Promise<ICozeInfoOfTokenExchange> => {
   const agentHost = config.get<string>('agent.host');
 
-  const apiUrl = `http://${agentHost}/api/by_teamid/tokens`;
+  const apiUrl = `https://${agentHost}/api/by_teamid/tokens`;
   const res = await fetch(apiUrl, {
     method: 'POST',
     headers: {

package/src/utils/createFakeUIMessageStreamResponse.ts

@@ -0,0 +1,27 @@
+import { createUIMessageStream, createUIMessageStreamResponse, generateId } from 'ai';
+import { createSpecialPartMeta } from './createSpecialPartMeta.js';
+
+/**
+ * 创建一个与 `streamText().toUIMessageStreamResponse()` 格式一致的假流式响应，
+ * 用于在无法调用真实模型时向客户端推送一条指定的文本消息。
+ */
+export const createFakeUIMessageStreamResponse = (text: string): Response => {
+  const stream = createUIMessageStream({
+    execute: ({ writer }) => {
+      const messageId = generateId();
+      const textId = generateId();
+      writer.write({ type: 'start', messageId });
+      writer.write({ type: 'text-start', id: textId });
+      writer.write({
+        type: 'text-delta',
+        id: textId,
+        delta: text,
+        providerMetadata: createSpecialPartMeta('error-bot-id-required'),
+      });
+      writer.write({ type: 'text-end', id: textId });
+      writer.write({ type: 'finish', finishReason: 'stop' });
+    },
+  });
+
+  return createUIMessageStreamResponse({ stream });
+};

package/src/utils/createJsonStreamTransformer.ts

@@ -31,6 +31,8 @@ export interface JsonStreamTransformerOptions<T> {
   onErrorChunk?: (chunk: LanguageModelV2StreamPart) => void;
   /** JSON 解析出错时的回调，返回要展示的错误消息（可选） */
   onParseError?: (error: any) => string;
+
+  bypassParseError?: boolean; // 是否完全绕过解析错误处理，直接透传原始 chunk（不推荐，除非特殊场景）
 }
 
 /**
@@ -75,6 +77,7 @@ class JsonStreamProcessor<T> {
   private parseCompleted: Promise<void> | null = null;
   private resolveParseCompleted: (() => void) | null = null;
   private terminated = false;
+  private passthroughMode = false; // 当启用且遇到非 JSON chunk 时，切换到透传模式
 
   constructor(options: JsonStreamTransformerOptions<T>) {
     this.options = options;
@@ -96,6 +99,11 @@ class JsonStreamProcessor<T> {
       return;
     }
 
+    if (this.passthroughMode) {
+      controller.enqueue(chunk);
+      return;
+    }
+
     try {
       // 1) error chunk → 通知下游后终止
       if (chunk.type === 'error') {
@@ -117,6 +125,11 @@ class JsonStreamProcessor<T> {
         this.parser.write(stripCodeFence(chunk.delta));
       }
     } catch (error) {
+      if (this.options.bypassParseError) {
+        this.passthroughMode = true;
+        controller.enqueue(chunk);
+        return;
+      }
       this.enqueueError(controller, error);
     }
   }
@@ -169,6 +182,11 @@ class JsonStreamProcessor<T> {
     };
 
     this.parser.onError = (err: any) => {
+      if (this.options.bypassParseError) {
+        this.passthroughMode = true;
+        return;
+      }
+
       if (!this.terminated) {
         const msg = this.options.onParseError
           ? this.options.onParseError(err)
@@ -180,8 +198,9 @@ class JsonStreamProcessor<T> {
     };
 
     this.parser.onEnd = () => {
-      logger.debug('Parsing completed');
-      enqueue(' ', JSON.stringify({ appendConfirm: 'save-fields' }));
+      if (!this.options.bypassParseError) {
+        enqueue(' ', JSON.stringify({ appendConfirm: 'save-fields' }));
+      }
       this.completeParsing();
     };
   }

package/src/utils/createSpecialPartMeta.ts

@@ -0,0 +1,7 @@
+import type { SpecialPart } from '@baishuyun/types';
+
+export const createSpecialPartMeta = (type: SpecialPart, content?: string) => {
+  return {
+    raw: { data: { specialPart: true, type: 'error-bot-id-required' } },
+  };
+};