@baishuyun/chat-backend 0.0.10 → 0.0.12

package/.env.example

@@ -18,3 +18,6 @@ FILL_BOT_ID=xxxxx
 
 # query bot id
 QUERY_BOT_ID=xxxxx
+
+# query suggest bot id
+QUERY_SUGGEST_BOT_ID=xxxx

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # @baishuyun/chat-backend
 
+## 0.0.12
+
+### Patch Changes
+
+- Updated dependencies
+  - @baishuyun/coze-provider@0.0.12
+  - @baishuyun/types@1.0.12
+
+## 0.0.11
+
+### Patch Changes
+
+- Updated dependencies
+  - @baishuyun/coze-provider@0.0.11
+  - @baishuyun/types@1.0.11
+
 ## 0.0.10
 
 ### Patch Changes

package/ecosystem.config.cjs

@@ -7,6 +7,10 @@ module.exports = {
       autorestart: true, // 异常时自动重启（核心）
       watch: false, // 生产环境关闭文件监听（避免容器内文件变动触发重启）
       max_memory_restart: '2G', // 内存超过 1G 自动重启（防止内存泄漏）
+
+      listen_timeout: 10000,
+      kill_timeout: 5000,
+
       env: {
         // 开发环境变量
         NODE_ENV: 'development',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@baishuyun/chat-backend",
-  "version": "0.0.10",
+  "version": "0.0.12",
   "type": "module",
   "publishConfig": {
     "access": "public"
@@ -22,8 +22,8 @@
     "parse-sse": "^0.1.0",
     "pino": "^10.1.0",
     "zod": "^4.1.13",
-    "@baishuyun/coze-provider": "0.0.10",
-    "@baishuyun/types": "1.0.10"
+    "@baishuyun/coze-provider": "0.0.12",
+    "@baishuyun/types": "1.0.12"
   },
   "devDependencies": {
     "@types/config": "^3.3.5",
@@ -34,7 +34,7 @@
     "pm2": "^6.0.14",
     "tsx": "^4.7.1",
     "typescript": "^5.8.3",
-    "@baishuyun/typescript-config": "0.0.10"
+    "@baishuyun/typescript-config": "0.0.12"
   },
   "scripts": {
     "dev": "cross-env NODE_ENV=development tsx watch src/index.ts",

package/src/app/main.ts

@@ -13,7 +13,12 @@ import { createReportRouter } from '../routes/report/report.route.js';
 app.route('web/api/form', createFormRouter());
 app.route('web/api/report', createReportRouter());
 
-// add test route
-app.get('web/api/test', (c) => {
-  return c.json({ message: 'Test route is working!' });
+// 基础健康检查
+app.get('/web/api/health', (c) => {
+  return c.json({
+    status: 'ok',
+    timestamp: new Date().toISOString(),
+    uptime: process.uptime(),
+    service: 'hono-app',
+  });
 });

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

@@ -0,0 +1,54 @@
+import { type LanguageModelV2StreamPart } from '@ai-sdk/provider';
+import { JSONParser } from '@streamparser/json';
+import { type IQueryResult } from '@baishuyun/types';
+import {
+  createJsonStreamTransformer,
+  type IParserCtx,
+} from '../../../utils/createJsonStreamTransformer.js';
+import { logger } from '../../../logger/index.js';
+
+export interface ISuggestion {
+  t: string;
+  v: string;
+}
+
+type suggestions = Array<ISuggestion>;
+
+export const createQuerySuggestionTransformer = () => {
+  return createJsonStreamTransformer<suggestions>({
+    createJSONParser: createJSONParser,
+    onParseValue: handleParsedValue,
+    onParseError(error) {
+      logger.error('JSON parsing error in query transformer: ', error);
+      return '数据解析异常，请重试';
+    },
+    onErrorChunk(chunk) {
+      logger.error('Received error chunk in query transformer');
+      return '';
+    },
+  });
+};
+
+function createJSONParser(): JSONParser {
+  return new JSONParser({
+    stringBufferSize: undefined,
+    numberBufferSize: undefined,
+    separator: '',
+    paths: ['$.*'],
+    keepStack: true,
+  });
+}
+
+function handleParsedValue(ctx: IParserCtx<suggestions>): void {
+  const { parsedInfo, getResult, currentChunkId, deltaChunkEnqueuer: enqueueTextDelta, ctrl } = ctx;
+  const { value } = parsedInfo;
+
+  logger.debug('Parsed JSON value: ' + JSON.stringify(value));
+
+  enqueueTextDelta(
+    `${JSON.stringify(value)},`,
+    { type: 'query-suggestion-parsed-info', result: JSON.stringify(getResult()) },
+    currentChunkId + 2,
+    true
+  );
+}

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

@@ -2,6 +2,7 @@ import { createCoze } from '@baishuyun/coze-provider';
 
 import config from 'config';
 import { createQueryResultTransformer } from './createQueryTransformStream.js';
+import { createQuerySuggestionTransformer } from './createQuerySuggestionTransStream.js';
 
 export const createReportQueryModel = () => {
   const reportQuery = createCoze({
@@ -15,3 +16,16 @@ export const createReportQueryModel = () => {
 
   return reportQueryModel;
 };
+
+export const createQuerySuggestionModel = () => {
+  const reportQuery = createCoze({
+    apiKey: config.get<string>('agent.report.query.apiKey'),
+    baseURL: config.get<string>('agent.report.query.baseUrl'),
+    botId: config.get<string>('agent.report.query.botId'),
+    extraStreamTransformers: [createQuerySuggestionTransformer],
+  });
+
+  const reportQueryModel = reportQuery.chat('chat');
+
+  return reportQueryModel;
+};

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

@@ -0,0 +1,46 @@
+import { type Context } from 'hono';
+import { convertToModelMessages, streamText, type UIMessage, generateId } from 'ai';
+import { createReportQueryModel } from './model.js';
+
+export const queryReportSuggest = async (c: Context) => {
+  let requestBody;
+
+  try {
+    const json = await c.req.json();
+    requestBody = json;
+  } catch (_) {
+    return c.json({ error: 'Invalid JSON' }, 400);
+  }
+
+  const uid = c.req.header('X-User-Id') || '';
+  const messages: UIMessage[] = requestBody.messages;
+
+  const forms = requestBody.forms || [];
+
+  const modelMessages = convertToModelMessages([
+    {
+      role: 'user',
+      parts: [
+        {
+          type: 'text',
+          text: JSON.stringify(forms),
+        },
+      ],
+    },
+  ]);
+
+  console.log('Final messages for query suggestion: ', JSON.stringify(modelMessages, null, 2));
+
+  const stream = streamText({
+    model: createReportQueryModel(),
+    messages: modelMessages,
+    includeRawChunks: true,
+    headers: {
+      'x-user-id': uid,
+    },
+  });
+
+  return stream.toUIMessageStreamResponse({
+    originalMessages: messages, // 建议添加，便于消息 ID 管理
+  });
+};

package/src/routes/report/report.route.ts

@@ -1,5 +1,6 @@
 import { Hono } from 'hono';
 import { queryReport } from '../../controllers/report/query/query.controller.js';
+import { queryReportSuggest } from '../../controllers/report/query/suggest.controller.js';
 
 export const createReportRouter = () => {
   const reportRouter = new Hono();
@@ -7,5 +8,7 @@ export const createReportRouter = () => {
   // 智能问数
   reportRouter.post('/query', queryReport);
 
+  reportRouter.post('/query/suggest', queryReportSuggest);
+
   return reportRouter;
 };