@baishuyun/chat-backend 0.0.17 → 0.0.18

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @baishuyun/chat-backend
 
+## 0.0.18
+
+### Patch Changes
+
+- Updated dependencies [e5dd21a]
+- Updated dependencies [a934bc0]
+- Updated dependencies
+- Updated dependencies [4acb2cc]
+- Updated dependencies [35058a9]
+  - @baishuyun/coze-provider@0.1.0
+  - @baishuyun/agents@0.1.0
+  - @baishuyun/types@1.1.0
+
 ## 0.0.17
 
 ### Patch Changes

package/config/default.ts

@@ -19,7 +19,7 @@ export default {
 
     apiAuthKey: process.env.COZE_API_KEY,
 
-    deepseekApiKey: process.env.DS_API_KEY,
+    userId: process.env.BUILTIN_COZE_USER_ID,
 
     common: {
       baseUrl: `http://${process.env.AGENT_HOST}/v3/`,
@@ -31,6 +31,7 @@ export default {
         botId: process.env.BUILD_BOT_ID || '7579927677256073216',
         baseUrl: `http://${process.env.AGENT_HOST}/v3/`,
         apiKey: process.env.BOT_API_KEY, // load from env
+        intentBotId: process.env.FORM_BUILD_INTENT_BOT_ID || '7615799745042186240',
       },
 
       fill: {

package/dist/config/default.js

@@ -15,7 +15,7 @@ export default {
     agent: {
         host: process.env.AGENT_HOST || '47.99.202.157',
         apiAuthKey: process.env.COZE_API_KEY,
-        deepseekApiKey: process.env.DS_API_KEY,
+        userId: process.env.BUILTIN_COZE_USER_ID,
         common: {
             baseUrl: `http://${process.env.AGENT_HOST}/v3/`,
             apiKey: process.env.BOT_API_KEY, // load from env
@@ -25,6 +25,7 @@ export default {
                 botId: process.env.BUILD_BOT_ID || '7579927677256073216',
                 baseUrl: `http://${process.env.AGENT_HOST}/v3/`,
                 apiKey: process.env.BOT_API_KEY, // load from env
+                intentBotId: process.env.FORM_BUILD_INTENT_BOT_ID || '7615799745042186240',
             },
             fill: {
                 botId: process.env.FILL_BOT_ID || '7586483957357608960',

package/dist/src/app/main.js

@@ -3,6 +3,7 @@
  */
 // 应用实例
 import app from '../config/hono.config.js';
+import { createAgentRouter } from '../routes/agent/agent.route.js';
 import { createCommonRouter } from '../routes/common/common.route.js';
 // 子路由
 import { createFormRouter } from '../routes/form/form.route.js';
@@ -11,6 +12,7 @@ import { createReportRouter } from '../routes/report/report.route.js';
 app.route('web/api/form', createFormRouter());
 app.route('web/api/report', createReportRouter());
 app.route('web/api/common', createCommonRouter());
+app.route('web/api/agent', createAgentRouter());
 // 基础健康检查
 app.get('/web/api/health', (c) => {
     return c.json({

package/dist/src/config/cache.config.js

@@ -0,0 +1,6 @@
+import { LRUCache } from 'lru-cache';
+const options = {
+    max: 5000,
+    ttl: 1000 * 60 * 55, // 假设 Token 1 小时有效，我们存 55 分钟
+};
+export const tokenCache = new LRUCache(options);

package/dist/src/controllers/agent/bots.controller.js

@@ -0,0 +1,29 @@
+import config from 'config';
+import { logger } from '../../logger/index.js';
+import { safeJsonParser } from '../../utils/safeJsonParser.js';
+export const listBots = async (c) => {
+    const agentHost = config.get('agent.host');
+    const apiBots = `http://${agentHost}/v1/bots`;
+    const cozeInfo = c.get('X-Coze-Info');
+    logger.debug(`Fetching bots from ${apiBots} with cozeToken: ${cozeInfo.cozeToken}`);
+    const result = await fetch(apiBots, {
+        method: 'GET',
+        headers: {
+            Authorization: `Bearer ${cozeInfo.cozeToken}`,
+        },
+    });
+    const resJson = await result.json();
+    logger.debug(JSON.stringify(resJson, null, 2));
+    if (!resJson.bot_list || !Array.isArray(resJson.bot_list)) {
+        return c.json([]);
+    }
+    const minifyedBots = resJson.bot_list.map((bot) => ({
+        id: bot.bot_id,
+        name: bot.name,
+        description: bot.description,
+        icon: bot.icon_url,
+        lastEditTime: bot.update_time,
+        publisherConfig: safeJsonParser(bot.pre_publish_ext),
+    }));
+    return c.json(minifyedBots);
+};

package/dist/src/controllers/common/connect.controll.js

@@ -9,12 +9,17 @@ export const connectToAgent = async (c) => {
     catch (_) {
         return c.json({ error: 'Invalid JSON' }, 400);
     }
-    const botId = requestBody.botId;
+    const cozeInfo = c.get('X-Coze-Info');
     const messages = requestBody.messages;
+    const botId = c.req.header('X-Bot-Id') || '';
     const stream = streamText({
-        model: createBaseModel(botId),
+        model: createBaseModel(botId, cozeInfo.cozeToken),
         messages: convertToModelMessages(messages),
         includeRawChunks: true,
+        headers: {
+            'x-user-var': requestBody.userVar,
+            'x-user-id': cozeInfo.userId,
+        },
     });
     return stream.toUIMessageStreamResponse();
 };

package/dist/src/controllers/common/model.js

@@ -1,8 +1,8 @@
 import { createCoze } from '@baishuyun/coze-provider';
 import config from 'config';
-export const createBaseModel = (botId) => {
+export const createBaseModel = (botId, token) => {
     const coze = createCoze({
-        apiKey: config.get('agent.common.apiKey'),
+        apiKey: token || config.get('agent.common.apiKey'),
         baseURL: config.get('agent.common.baseUrl'),
         botId: botId,
     });

package/dist/src/controllers/form/build/build.controller.js

@@ -2,8 +2,6 @@ import { convertToModelMessages, streamText, generateText } from "ai";
 import {} from "hono";
 import { createModel } from "./model.js";
 import { createFieldsJsonTransformStream, SuggestionTransformStream, } from "@baishuyun/coze-provider";
-import { logger } from "../../../logger/index.js";
-import config from 'config';
 import { determineUserIntentByInput } from "./utils.js";
 /**
  * 搭建表单
@@ -26,30 +24,29 @@ export const buildForm = async (c) => {
         () => createFieldsJsonTransformStream(isBuildStage),
         () => new SuggestionTransformStream(isBuildStage),
     ]);
-    logger.debug("intent: " + intent);
-    const allMessages = [...requestBody.messages];
+    const allMsg = requestBody.messages || [];
+    const lastUserMsg = allMsg.length > 0 ? allMsg[allMsg.length - 1] : {
+        role: "user",
+        parts: [],
+    };
     if (isBuildStage && formName) {
-        allMessages.push({
-            role: "user",
-            parts: [
-                {
-                    type: "text",
-                    text: `确认搭建：${formName}`,
-                },
-            ],
+        lastUserMsg.parts = lastUserMsg.parts.map((p, index) => {
+            const isLastPart = index === lastUserMsg.parts.length - 1;
+            if (!isLastPart || p.type !== "text") {
+                return p;
+            }
+            return {
+                type: "text",
+                text: `【确认搭建：${formName}】 ${p.text}`
+            };
         });
     }
-    // clear empty text parts to avoid unnecessary streaming
-    allMessages.forEach((message) => {
-        message.parts = message.parts.filter((part) => part.type === "text" && part.text.trim() !== "");
-    });
     const stream = streamText({
         model,
-        messages: convertToModelMessages(allMessages),
+        messages: convertToModelMessages([lastUserMsg]),
         includeRawChunks: true,
         headers: {
-            "x-user-stage": requestBody.stage,
-            "x-user-id": Date.now().toString(),
+            "x-user-stage": intent,
             "x-user-token": c.req.header("authorization") || "",
         },
     });

package/dist/src/controllers/form/build/model.js

@@ -1,4 +1,3 @@
-import { createFormBuildIntentAgent } from "@baishuyun/agents";
 import { createCoze } from "@baishuyun/coze-provider";
 import config from "config";
 export const createModel = (extraStreamTransformers) => {
@@ -10,7 +9,11 @@ export const createModel = (extraStreamTransformers) => {
     });
     return coze.chat("chat");
 };
-export const createUserIntentAgent = () => {
-    const dsApiKey = config.get("agent.deepseekApiKey");
-    return createFormBuildIntentAgent(dsApiKey);
+export const createBuildIntentAgent = () => {
+    const coze = createCoze({
+        apiKey: config.get("agent.form.build.apiKey"),
+        baseURL: config.get("agent.form.build.baseUrl"),
+        botId: config.get("agent.form.build.intentBotId"),
+    });
+    return coze.chat("chat");
 };

package/dist/src/controllers/form/build/utils.js

@@ -1,4 +1,5 @@
-import { createUserIntentAgent } from "./model.js";
+import { logger } from "../../../logger/index.js";
+import { createBuildIntentAgent } from "./model.js";
 export const determineUserIntentByInput = async (input, userOriginIntent) => {
     if (userOriginIntent === "design") {
         return "design";
@@ -6,9 +7,9 @@ export const determineUserIntentByInput = async (input, userOriginIntent) => {
     if (!input || input.trim() === "") {
         return userOriginIntent;
     }
-    const agent = createUserIntentAgent();
-    const intent = await agent.generate({
-        messages: [{
+    const cozeAgent = createBuildIntentAgent();
+    const intent = await cozeAgent.doGenerate({
+        prompt: [{
                 role: "user",
                 content: [{
                         type: "text",
@@ -16,15 +17,12 @@ export const determineUserIntentByInput = async (input, userOriginIntent) => {
                     }]
             }]
     });
-    const result = intent.response.messages;
+    logger.debug("agent intent response: " + JSON.stringify(intent, null, 2));
+    const result = intent.content;
     if (!result || result.length === 0) {
         return userOriginIntent;
     }
-    const lastMessage = result[result.length - 1];
-    if (!lastMessage.content || lastMessage.content.length === 0) {
-        return userOriginIntent;
-    }
-    const lastContent = lastMessage.content[0];
+    const lastContent = result[result.length - 1];
     const intentText = lastContent.text === "build" ? "build" : lastContent.text === "design" ? "design" : userOriginIntent;
     return intentText;
 };

package/dist/src/controllers/report/query/query.controller.js

@@ -32,9 +32,6 @@ export const queryReport = async (c) => {
         model: createReportQueryModel(),
         messages: modelMessages,
         includeRawChunks: true,
-        headers: {
-            'x-user-id': Date.now().toString(), // uid,
-        },
     });
     return stream.toUIMessageStreamResponse({
         originalMessages: messages, // 建议添加，便于消息 ID 管理

package/dist/src/controllers/report/query/suggest.controller.js

@@ -10,7 +10,6 @@ export const queryReportSuggest = async (c) => {
     catch (_) {
         return c.json({ error: 'Invalid JSON' }, 400);
     }
-    const uid = c.req.header('X-User-Id') || '';
     const messages = requestBody.messages;
     const forms = requestBody.forms || [];
     const modelMessages = convertToModelMessages([
@@ -29,9 +28,6 @@ export const queryReportSuggest = async (c) => {
         model: createReportQueryModel(),
         messages: modelMessages,
         includeRawChunks: true,
-        headers: {
-            'x-user-id': uid,
-        },
     });
     return stream.toUIMessageStreamResponse({
         originalMessages: messages, // 建议添加，便于消息 ID 管理

package/dist/src/middleware/tokenExchange.js

@@ -0,0 +1,23 @@
+import { createMiddleware } from 'hono/factory';
+import { tokenCache } from '../config/cache.config.js';
+import { fetchCozeInfo } from '../services/fetchCozeInfo.js';
+import { logger } from '../logger/index.js';
+export const tokenExchange = () => createMiddleware(async (c, next) => {
+    const userAuth = c.req.header('X-Bs-Team-Id') || c.req.query('x-bs-team-id') || '';
+    if (!userAuth) {
+        return c.json({ error: 'Unauthorized' }, 401);
+    }
+    let cozeInfo = tokenCache.get(userAuth);
+    if (!cozeInfo) {
+        try {
+            cozeInfo = await fetchCozeInfo(userAuth);
+            tokenCache.set(userAuth, cozeInfo);
+        }
+        catch (error) {
+            logger.error(`Failed to exchange token for userAuth: ${userAuth}, error: ${error}`);
+            return c.json({ error: 'Failed to exchange token' }, 500);
+        }
+    }
+    c.set('X-Coze-Info', cozeInfo);
+    await next();
+});

package/dist/src/routes/agent/agent.route.js

@@ -0,0 +1,8 @@
+import { Hono } from 'hono';
+import { listBots } from '../../controllers/agent/bots.controller.js';
+import { tokenExchange } from '../../middleware/tokenExchange.js';
+export const createAgentRouter = () => {
+    const agentRouter = new Hono();
+    agentRouter.get('/bots', tokenExchange(), listBots);
+    return agentRouter;
+};

package/dist/src/routes/common/common.route.js

@@ -1,7 +1,8 @@
 import { Hono } from 'hono';
 import { connectToAgent } from '../../controllers/common/connect.controll.js';
+import { tokenExchange } from '../../middleware/tokenExchange.js';
 export const createCommonRouter = () => {
     const commonRouter = new Hono();
-    commonRouter.post('/connect', connectToAgent);
+    commonRouter.post('/connect', tokenExchange(), connectToAgent);
     return commonRouter;
 };

package/dist/src/services/fetchCozeInfo.js

@@ -0,0 +1,26 @@
+import config from 'config';
+import { logger } from '../logger/index.js';
+export const fetchCozeInfo = async (bsTeamId) => {
+    const agentHost = config.get('agent.host');
+    const apiUrl = `http://${agentHost}/api/by_teamid/tokens`;
+    const res = await fetch(apiUrl, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({ token: bsTeamId }),
+    });
+    const result = await res.json();
+    if (result.code !== 0) {
+        throw new Error('Failed to fetch Coze token');
+    }
+    const tokenList = result.data;
+    if (tokenList.length === 0) {
+        throw new Error('No Coze token returned');
+    }
+    const cozeToken = tokenList[0];
+    return {
+        cozeToken,
+        userId: `${result.team_uid}`,
+    };
+};

package/dist/src/types/coze.js

@@ -0,0 +1 @@
+export {};

package/dist/src/utils/safeJsonParser.js

@@ -0,0 +1,8 @@
+export const safeJsonParser = (str) => {
+    try {
+        return JSON.parse(str);
+    }
+    catch (error) {
+        return null; // or you can choose to return an empty object {} or any default value
+    }
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@baishuyun/chat-backend",
-  "version": "0.0.17",
+  "version": "0.0.18",
   "type": "module",
   "publishConfig": {
     "access": "public"
@@ -19,12 +19,13 @@
     "config": "^4.1.1",
     "dotenv": "^17.2.3",
     "hono": "^4.10.6",
+    "lru-cache": "^11.2.7",
     "parse-sse": "^0.1.0",
     "pino": "^10.1.0",
     "zod": "^4.1.13",
-    "@baishuyun/coze-provider": "0.0.17",
-    "@baishuyun/agents": "0.0.17",
-    "@baishuyun/types": "1.0.17"
+    "@baishuyun/coze-provider": "0.1.0",
+    "@baishuyun/types": "1.1.0",
+    "@baishuyun/agents": "0.1.0"
   },
   "devDependencies": {
     "@types/config": "^3.3.5",
@@ -35,7 +36,7 @@
     "pm2": "^6.0.14",
     "tsx": "^4.7.1",
     "typescript": "^5.8.3",
-    "@baishuyun/typescript-config": "0.0.17"
+    "@baishuyun/typescript-config": "0.1.0"
   },
   "scripts": {
     "dev": "cross-env NODE_ENV=development tsx watch src/index.ts",

package/src/app/main.ts

@@ -4,6 +4,7 @@
 
 // 应用实例
 import app from '../config/hono.config.js';
+import { createAgentRouter } from '../routes/agent/agent.route.js';
 import { createCommonRouter } from '../routes/common/common.route.js';
 
 // 子路由
@@ -14,6 +15,7 @@ import { createReportRouter } from '../routes/report/report.route.js';
 app.route('web/api/form', createFormRouter());
 app.route('web/api/report', createReportRouter());
 app.route('web/api/common', createCommonRouter());
+app.route('web/api/agent', createAgentRouter());
 
 // 基础健康检查
 app.get('/web/api/health', (c) => {

package/src/config/cache.config.ts

@@ -0,0 +1,9 @@
+import { LRUCache } from 'lru-cache';
+import type { ICozeInfoOfTokenExchange } from '../types/coze.js';
+
+const options = {
+  max: 5000,
+  ttl: 1000 * 60 * 55, // 假设 Token 1 小时有效，我们存 55 分钟
+};
+
+export const tokenCache = new LRUCache<string, ICozeInfoOfTokenExchange>(options);

package/src/controllers/agent/bots.controller.ts

@@ -0,0 +1,39 @@
+import type { Context } from 'hono';
+import config from 'config';
+import { logger } from '../../logger/index.js';
+import { safeJsonParser } from '../../utils/safeJsonParser.js';
+import type { ICozeInfoOfTokenExchange } from '../../types/coze.js';
+
+export const listBots = async (c: Context) => {
+  const agentHost = config.get<string>('agent.host');
+  const apiBots = `http://${agentHost}/v1/bots`;
+  const cozeInfo = c.get('X-Coze-Info') as ICozeInfoOfTokenExchange;
+
+  logger.debug(`Fetching bots from ${apiBots} with cozeToken: ${cozeInfo.cozeToken}`);
+
+  const result = await fetch(apiBots, {
+    method: 'GET',
+    headers: {
+      Authorization: `Bearer ${cozeInfo.cozeToken}`,
+    },
+  });
+
+  const resJson = await result.json();
+
+  logger.debug(JSON.stringify(resJson, null, 2));
+
+  if (!resJson.bot_list || !Array.isArray(resJson.bot_list)) {
+    return c.json([]);
+  }
+
+  const minifyedBots = resJson.bot_list.map((bot: any) => ({
+    id: bot.bot_id,
+    name: bot.name,
+    description: bot.description,
+    icon: bot.icon_url,
+    lastEditTime: bot.update_time,
+    publisherConfig: safeJsonParser(bot.pre_publish_ext),
+  }));
+
+  return c.json(minifyedBots);
+};

package/src/controllers/common/connect.controll.ts

@@ -1,6 +1,7 @@
 import { convertToModelMessages, streamText } from 'ai';
 import type { Context } from 'hono';
 import { createBaseModel } from './model.js';
+import type { ICozeInfoOfTokenExchange } from '../../types/coze.js';
 
 export const connectToAgent = async (c: Context) => {
   let requestBody;
@@ -11,13 +12,20 @@ export const connectToAgent = async (c: Context) => {
     return c.json({ error: 'Invalid JSON' }, 400);
   }
 
-  const botId = requestBody.botId;
+  const cozeInfo = c.get('X-Coze-Info') as ICozeInfoOfTokenExchange;
+
   const messages = requestBody.messages;
 
+  const botId = c.req.header('X-Bot-Id') || '';
+
   const stream = streamText({
-    model: createBaseModel(botId),
+    model: createBaseModel(botId, cozeInfo.cozeToken),
     messages: convertToModelMessages(messages),
     includeRawChunks: true,
+    headers: {
+      'x-user-var': requestBody.userVar,
+      'x-user-id': cozeInfo.userId,
+    },
   });
 
   return stream.toUIMessageStreamResponse();

package/src/controllers/common/model.ts

@@ -1,9 +1,9 @@
 import { createCoze } from '@baishuyun/coze-provider';
 import config from 'config';
 
-export const createBaseModel = (botId: string) => {
+export const createBaseModel = (botId: string, token?: string) => {
   const coze = createCoze({
-    apiKey: config.get<string>('agent.common.apiKey'),
+    apiKey: token || config.get<string>('agent.common.apiKey'),
     baseURL: config.get<string>('agent.common.baseUrl'),
     botId: botId,
   });

package/src/controllers/form/build/build.controller.ts

@@ -5,8 +5,6 @@ import {
   createFieldsJsonTransformStream,
   SuggestionTransformStream,
 } from "@baishuyun/coze-provider";
-import { logger } from "../../../logger/index.js";
-import config from 'config';
 import { determineUserIntentByInput } from "./utils.js";
 
 /**
@@ -33,35 +31,32 @@ export const buildForm = async (c: Context) => {
     () => new SuggestionTransformStream(isBuildStage),
   ]);
 
-  logger.debug("intent: " + intent);
+  const allMsg = requestBody.messages || [];
+  const lastUserMsg = allMsg.length > 0 ? allMsg[allMsg.length - 1] : {
+    role: "user",
+    parts: [],
+  };
 
-  const allMessages = [...requestBody.messages];
   if (isBuildStage && formName) {
-    allMessages.push({
-      role: "user",
-      parts: [
-        {
-          type: "text",
-          text: `确认搭建：${formName}`,
-        },
-      ],
+    lastUserMsg.parts = lastUserMsg.parts.map((p: any, index: number) => {
+      const isLastPart = index === lastUserMsg.parts.length - 1;
+      if (!isLastPart || p.type !== "text") {
+        return p;
+      }
+
+      return {
+        type: "text",
+        text: `【确认搭建：${formName}】 ${p.text}`
+      };
     });
   }
 
-  // clear empty text parts to avoid unnecessary streaming
-  allMessages.forEach((message) => {
-    message.parts = message.parts.filter(
-      (part: TextUIPart) => part.type === "text" && part.text.trim() !== ""
-    );
-  });
-
   const stream = streamText({
     model,
-    messages: convertToModelMessages(allMessages),
+    messages: convertToModelMessages([lastUserMsg]),
     includeRawChunks: true,
     headers: {
-      "x-user-stage": requestBody.stage,
-      "x-user-id": Date.now().toString(),
+      "x-user-stage": intent as string,
       "x-user-token": c.req.header("authorization") || "",
     },
   });

package/src/controllers/form/build/model.ts

@@ -1,5 +1,4 @@
 import type { LanguageModelV2StreamPart } from "@ai-sdk/provider";
-import { createFormBuildIntentAgent } from "@baishuyun/agents";
 import { createCoze } from "@baishuyun/coze-provider";
 import config from "config";
 
@@ -16,7 +15,12 @@ export const createModel = (
   return coze.chat("chat");
 };
 
-export const createUserIntentAgent = () => {
-  const dsApiKey = config.get<string>("agent.deepseekApiKey");
-  return createFormBuildIntentAgent(dsApiKey);
+export const createBuildIntentAgent = () => {
+  const coze = createCoze({
+    apiKey: config.get<string>("agent.form.build.apiKey"),
+    baseURL: config.get<string>("agent.form.build.baseUrl"),
+    botId: config.get<string>("agent.form.build.intentBotId"),
+  });
+
+  return coze.chat("chat");
 }

package/src/controllers/form/build/utils.ts

@@ -1,4 +1,5 @@
-import { createUserIntentAgent } from "./model.js"
+import { logger } from "../../../logger/index.js";
+import { createBuildIntentAgent } from "./model.js"
 
 export const determineUserIntentByInput = async (input: string, userOriginIntent: "build" | "design" | null): Promise<"build" | "design" | null> => {
     if (userOriginIntent === "design") {
@@ -9,30 +10,25 @@ export const determineUserIntentByInput = async (input: string, userOriginIntent
         return userOriginIntent;
     }
 
-  const agent = createUserIntentAgent();
-
-  const intent = await agent.generate({
-    messages: [{
+  const cozeAgent = createBuildIntentAgent();
+  const intent = await cozeAgent.doGenerate({
+    prompt: [{
       role: "user",
       content: [{
         type: "text",
         text: input,
       }]
     }]
-  })
-
-  const result = intent.response.messages;
-  if (!result || result.length === 0) {
-    return userOriginIntent;
-  }
+  });
 
-  const lastMessage = result[result.length - 1];
+  logger.debug("agent intent response: " + JSON.stringify(intent, null, 2));
 
-  if (!lastMessage.content || lastMessage.content.length === 0) {
+  const result = intent.content;
+  if (!result || result.length === 0) {
     return userOriginIntent;
   }
 
-  const lastContent = lastMessage.content[0] as {
+  const lastContent = result[result.length - 1] as {
     type: "text";
     text: string;
   };

package/src/controllers/report/query/query.controller.ts

@@ -39,9 +39,6 @@ export const queryReport = async (c: Context) => {
     model: createReportQueryModel(),
     messages: modelMessages,
     includeRawChunks: true,
-    headers: {
-      'x-user-id': Date.now().toString(), // uid,
-    },
   });
 
   return stream.toUIMessageStreamResponse({

package/src/controllers/report/query/suggest.controller.ts

@@ -12,7 +12,6 @@ export const queryReportSuggest = async (c: Context) => {
     return c.json({ error: 'Invalid JSON' }, 400);
   }
 
-  const uid = c.req.header('X-User-Id') || '';
   const messages: UIMessage[] = requestBody.messages;
 
   const forms = requestBody.forms || [];
@@ -35,9 +34,6 @@ export const queryReportSuggest = async (c: Context) => {
     model: createReportQueryModel(),
     messages: modelMessages,
     includeRawChunks: true,
-    headers: {
-      'x-user-id': uid,
-    },
   });
 
   return stream.toUIMessageStreamResponse({

package/src/middleware/tokenExchange.ts

@@ -0,0 +1,26 @@
+import { createMiddleware } from 'hono/factory';
+import { tokenCache } from '../config/cache.config.js';
+import { fetchCozeInfo } from '../services/fetchCozeInfo.js';
+import { logger } from '../logger/index.js';
+
+export const tokenExchange = () =>
+  createMiddleware(async (c, next) => {
+    const userAuth = c.req.header('X-Bs-Team-Id') || c.req.query('x-bs-team-id') || '';
+    if (!userAuth) {
+      return c.json({ error: 'Unauthorized' }, 401);
+    }
+
+    let cozeInfo = tokenCache.get(userAuth);
+    if (!cozeInfo) {
+      try {
+        cozeInfo = await fetchCozeInfo(userAuth);
+        tokenCache.set(userAuth, cozeInfo);
+      } catch (error) {
+        logger.error(`Failed to exchange token for userAuth: ${userAuth}, error: ${error}`);
+        return c.json({ error: 'Failed to exchange token' }, 500);
+      }
+    }
+
+    c.set('X-Coze-Info', cozeInfo);
+    await next();
+  });

package/src/routes/agent/agent.route.ts

@@ -0,0 +1,11 @@
+import { Hono } from 'hono';
+import { listBots } from '../../controllers/agent/bots.controller.js';
+import { tokenExchange } from '../../middleware/tokenExchange.js';
+
+export const createAgentRouter = () => {
+  const agentRouter = new Hono();
+
+  agentRouter.get('/bots', tokenExchange(), listBots);
+
+  return agentRouter;
+};

package/src/routes/common/common.route.ts

@@ -1,10 +1,11 @@
 import { Hono } from 'hono';
 import { connectToAgent } from '../../controllers/common/connect.controll.js';
+import { tokenExchange } from '../../middleware/tokenExchange.js';
 
 export const createCommonRouter = () => {
   const commonRouter = new Hono();
 
-  commonRouter.post('/connect', connectToAgent);
+  commonRouter.post('/connect', tokenExchange(), connectToAgent);
 
   return commonRouter;
 };

package/src/services/fetchCozeInfo.ts

@@ -0,0 +1,33 @@
+import config from 'config';
+import type { ICozeInfoOfTokenExchange, OpenSpaceData } from '../types/coze.js';
+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 res = await fetch(apiUrl, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+    },
+    body: JSON.stringify({ token: bsTeamId }),
+  });
+
+  const result = await res.json();
+  if (result.code !== 0) {
+    throw new Error('Failed to fetch Coze token');
+  }
+
+  const tokenList = result.data as Array<string>;
+  if (tokenList.length === 0) {
+    throw new Error('No Coze token returned');
+  }
+
+  const cozeToken = tokenList[0];
+
+  return {
+    cozeToken,
+    userId: `${result.team_uid}`,
+  };
+};

package/src/types/coze.ts

@@ -0,0 +1,13 @@
+export interface ICozeInfoOfTokenExchange {
+  cozeToken: string;
+  userId: string;
+}
+
+export type OpenSpaceData = Array<IOpenSpace>;
+
+export interface IOpenSpace {
+  id: string;
+  name: string;
+  icon_url: string;
+  owner_uid: string;
+}

package/src/utils/safeJsonParser.ts

@@ -0,0 +1,7 @@
+export const safeJsonParser = (str: string) => {
+  try {
+    return JSON.parse(str);
+  } catch (error) {
+    return null; // or you can choose to return an empty object {} or any default value
+  }
+};