@shun-js/aibaiban-server 0.4.8 → 0.4.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "0.4.8",
+  "version": "0.4.9",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -44,5 +44,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "1521c2b05c17217a24c716881c05f35b66480fed"
+  "gitHead": "822aa69649e8feaff1bed2c0e02c2b527d432568"
 }

package/server/controller/IndexController.js

@@ -11,6 +11,9 @@ module.exports = (app) => {
   });
 
   // chat
+  app.post('/chat', (req, res) => {
+    service.chat(req, res);
+  });
   app.post('/chat-streaming', (req, res) => {
     service.chatWithStreaming(req, res);
   });

package/server/service/IndexService.js

@@ -1,8 +1,8 @@
 // llm
-const { chatWithStreaming } = require('../util/llm-intent.js');
+const { llmParseIntent } = require('../util/llm');
 
 // util
-const { chatFeishuMsg, errorFeishuMsg } = require('../util/feishu.js');
+const { chatFeishuMsg, chatResFeishuMsg, errorFeishuMsg } = require('../util/feishu.js');
 
 /**
  * index
@@ -22,13 +22,13 @@ exports.index = async (req, res) => {
 };
 
 /**
- * chatWithStreaming
+ * chat
  * @param {*} req
  * @param {*} res
  * @returns
  */
-exports.chatWithStreaming = async (req, res) => {
-  const methodName = 'chatWithStreaming';
+exports.chat = async (req, res) => {
+  const methodName = 'chat';
 
   // check
   if (!req.body.userPrompt) {
@@ -44,28 +44,63 @@ exports.chatWithStreaming = async (req, res) => {
   chatFeishuMsg(req);
 
   // go
-  const userPrompts = [{ text: userPrompt }];
-  chatWithStreaming(userPrompts, {
-    beginCallback: () => {
-      req.logger.info(methodName, 'beginCallback');
-      res.streamingStart();
-    },
-    firstContentCallback: () => {
-      req.logger.info(methodName, 'firstContentCallback');
-      // res.streaming('First chunk received!');
-    },
-    contentCallback: (content) => {
-      req.logger.info(methodName, 'contentCallback', content);
-      res.streaming(content);
-    },
-    endCallback: () => {
-      req.logger.info(methodName, 'endCallback');
-      res.streamingEnd();
-    },
-    errorCallback: (error) => {
-      errorFeishuMsg(req, 'errorCallback');
-      req.logger.info(methodName, 'errorCallback', error);
-      res.streamingEnd();
-    },
-  });
+  try {
+    const llmParseIntentRes = await llmParseIntent(userPrompt);
+    req.logger.info(methodName, 'llmParseIntentRes', llmParseIntentRes);
+
+    // r
+    chatResFeishuMsg(req, JSON.stringify(llmParseIntentRes));
+    res.jsonSuccess('success', llmParseIntentRes);
+  } catch (error) {
+    const msg = 'chat error';
+    errorFeishuMsg(req, msg);
+    req.logger.error(methodName, msg, error);
+    res.jsonFail(msg);
+  }
+};
+
+/**
+ * chatWithStreaming
+ * @param {*} req
+ * @param {*} res
+ * @returns
+ */
+exports.chatWithStreaming = async () => {
+  // const methodName = 'chatWithStreaming';
+  // // check
+  // if (!req.body.userPrompt) {
+  //   const msg = 'need userPrompt';
+  //   req.logger.error(methodName, msg);
+  //   res.jsonFail(msg);
+  //   return;
+  // }
+  // // const
+  // const userPrompt = decodeURIComponent(req.body.userPrompt);
+  // req.logger.info(methodName, 'userPrompt', userPrompt);
+  // chatFeishuMsg(req);
+  // // go
+  // const userPrompts = [{ text: userPrompt }];
+  // chatWithStreaming(userPrompts, {
+  //   beginCallback: () => {
+  //     req.logger.info(methodName, 'beginCallback');
+  //     res.streamingStart();
+  //   },
+  //   firstContentCallback: () => {
+  //     req.logger.info(methodName, 'firstContentCallback');
+  //     // res.streaming('First chunk received!');
+  //   },
+  //   contentCallback: (content) => {
+  //     req.logger.info(methodName, 'contentCallback', content);
+  //     res.streaming(content);
+  //   },
+  //   endCallback: () => {
+  //     req.logger.info(methodName, 'endCallback');
+  //     res.streamingEnd();
+  //   },
+  //   errorCallback: (error) => {
+  //     errorFeishuMsg(req, 'errorCallback');
+  //     req.logger.info(methodName, 'errorCallback', error);
+  //     res.streamingEnd();
+  //   },
+  // });
 };

package/server/util/llm.js

@@ -8,172 +8,38 @@ const { zodToJsonSchema } = require('zod-to-json-schema');
 
 // llm
 const { GeminiVertex } = require('viho-llm');
-
-// gemini
 const gemini = GeminiVertex({
   projectId: global.QZ_CONFIG.gemini.projectId,
   location: global.QZ_CONFIG.gemini.location,
   modelName: global.QZ_CONFIG.gemini.modelName,
 });
 
-// ==================== 缓存（懒加载） ====================
-
+// const
 let cachedSystemPrompt = null;
-let cachedCompiledSchema = null;
-
-// ==================== JSON Schema 定义（避免联合类型）====================
-
 const jsonSchema = (() => {
-  const DiagramNodeSchema = z.object({
-    id: z.string().describe('节点唯一标识符，使用驼峰命名法，如 userService, mysqlDB'),
-    label: z.string().describe('节点显示文本，支持 \\n 换行'),
-    type: z
-      .enum(['rectangle', 'ellipse', 'diamond', 'hexagon', 'cylinder', 'cloud'])
-      .optional()
-      .describe('形状类型，默认 rectangle'),
-    color: z
-      .enum(['blue', 'green', 'purple', 'orange', 'red', 'gray', 'yellow', 'pink', 'black'])
-      .optional()
-      .describe('节点颜色，默认 blue'),
-    x: z.number().optional().describe('X 坐标，不指定则自动布局'),
-    y: z.number().optional().describe('Y 坐标，不指定则自动布局'),
-    width: z.number().optional().describe('宽度，不指定则使用默认值'),
-    height: z.number().optional().describe('高度，不指定则使用默认值'),
-  });
-
-  const DiagramConnectionSchema = z.object({
-    from: z.string().describe('起始节点 id'),
-    to: z.string().describe('目标节点 id'),
-    label: z.string().optional().describe('连接线上的文字标签'),
-    type: z.enum(['arrow', 'line']).optional().describe('连接类型：arrow=箭头线，line=直线，默认 arrow'),
-    style: z
-      .enum(['solid', 'dashed', 'dotted'])
-      .optional()
-      .describe('线条样式：solid=实线，dashed=虚线，dotted=点线，默认 solid'),
-  });
-
-  const TextAnnotationSchema = z.object({
-    id: z.string().describe('标注唯一标识符'),
-    text: z.string().describe('标注文本内容'),
-    x: z.number().optional().describe('X 坐标'),
-    y: z.number().optional().describe('Y 坐标'),
-    fontSize: z.number().optional().describe('字体大小'),
-    color: z.string().optional().describe('文字颜色'),
-  });
-
-  const FrameSchema = z.object({
-    id: z.string().describe('框架唯一标识符'),
-    label: z.string().optional().describe('框架标题'),
-    children: z.array(z.string()).describe('包含的节点 id 列表'),
-    color: z.string().optional().describe('框架颜色'),
-    x: z.number().optional().describe('X 坐标'),
-    y: z.number().optional().describe('Y 坐标'),
-    width: z.number().optional().describe('宽度'),
-    height: z.number().optional().describe('高度'),
-  });
-
-  const ImageElementSchema = z.object({
-    id: z.string().describe('图片唯一标识符'),
-    imageUrl: z.string().url().describe('图片 URL'),
-    alt: z.string().optional().describe('图片描述'),
-    x: z.number().optional().describe('X 坐标'),
-    y: z.number().optional().describe('Y 坐标'),
-    width: z.number().optional().describe('宽度'),
-    height: z.number().optional().describe('高度'),
-  });
-
-  // 注意：freedraws 使用简化格式，避免 tuple 导致的兼容性问题
-  const FreedrawElementSchema = z.object({
-    id: z.string().describe('手绘元素唯一标识符'),
-    points: z.array(z.number()).describe('路径点坐标数组，格式：[x1,y1,x2,y2,...]'),
-    color: z.string().optional().describe('线条颜色'),
-    strokeWidth: z.number().optional().describe('线条宽度'),
-  });
-
-  // 统一的响应 Schema（包含图表和错误两种情况）
-  // 通过 type 字段区分：'architecture'|'flowchart'|'sequence'|'custom' = 图表，'error' = 错误
-  return z.object({
-    // 通用字段：type 用于区分响应类型
-    type: z
-      .enum(['architecture', 'flowchart', 'sequence', 'custom', 'error'])
-      .describe('响应类型：architecture/flowchart/sequence/custom=图表类型，error=错误响应'),
-
-    // ========== 图表相关字段（当 type != error 时使用）==========
-    title: z.string().optional().describe('图表标题'),
-    nodes: z.array(DiagramNodeSchema).optional().describe('节点列表'),
-    connections: z.array(DiagramConnectionSchema).optional().describe('连接关系列表'),
-    annotations: z.array(TextAnnotationSchema).optional().describe('独立文本标注列表（可选）'),
-    frames: z.array(FrameSchema).optional().describe('分组框架列表（可选）'),
-    images: z.array(ImageElementSchema).optional().describe('图片元素列表（可选）'),
-    freedraws: z.array(FreedrawElementSchema).optional().describe('手绘元素列表（可选）'),
-
-    // ========== 错误相关字段（当 type = error 时使用）==========
-    error: z
-      .enum(['NON_DRAWING_REQUEST', 'INVALID_REQUEST', 'UNKNOWN_ERROR'])
-      .optional()
-      .describe('错误类型（仅当 type=error 时需要）'),
-    message: z.string().optional().describe('错误提示消息（仅当 type=error 时需要）'),
-  });
+  return zodToJsonSchema(
+    z.object({
+      intent: z.enum(['DRAW', 'REJECT']).describe('用户意图：DRAW（绘图）或 REJECT（非绘图）'),
+      reason: z.string().describe('判断理由（一句话说明）'),
+    }),
+  );
 })();
 
-// ==================== 懒加载函数 ====================
-
-/**
- * 获取系统提示词（懒加载+缓存）
- */
-async function getSystemPrompt() {
-  if (!cachedSystemPrompt) {
-    const promptPath = path.resolve(__dirname, './prompt.md');
-    cachedSystemPrompt = await readFile(promptPath);
-  }
-  return cachedSystemPrompt;
-}
-
-/**
- * 获取编译后的 JSON Schema（懒加载+缓存）
- */
-function getCompiledSchema() {
-  if (!cachedCompiledSchema) {
-    const fullSchema = zodToJsonSchema(jsonSchema, {
-      name: 'AIResponse',
-      $refStrategy: 'none',
-    });
-
-    // 关键修复：Gemini 不支持顶层 $ref，直接使用 definitions 中的定义
-    if (fullSchema.definitions && fullSchema.definitions.AIResponse) {
-      cachedCompiledSchema = fullSchema.definitions.AIResponse;
-    } else if (fullSchema.$ref) {
-      // 如果顶层是 $ref，解析它
-      const refName = fullSchema.$ref.split('/').pop();
-      cachedCompiledSchema = fullSchema.definitions[refName];
-    } else {
-      cachedCompiledSchema = fullSchema;
-    }
-  }
-  return cachedCompiledSchema;
-}
-
-// ==================== 主要功能 ====================
-
 /**
- * chatWithStreaming
+ * llmParseIntent
  * @param {*} userPrompts
- * @param {*} callbackOptions
  */
-exports.chatWithStreaming = async (userPrompts, callbackOptions) => {
-  // 懒加载：第一次使用时加载并缓存
-  const systemPrompt = await getSystemPrompt();
-  const compiledSchema = getCompiledSchema();
-  console.log('compiledSchema');
-  console.log(JSON.stringify(compiledSchema));
+exports.llmParseIntent = async (userPrompts) => {
+  // cache
+  if (!cachedSystemPrompt) cachedSystemPrompt = await readFile(path.resolve(__dirname, './prompt-intent.md'));
 
   // chat options
   const chatOptions = {
     contents: userPrompts,
-    systemInstruction: systemPrompt,
+    systemInstruction: cachedSystemPrompt,
     config: {
       responseMimeType: 'application/json',
-      responseJsonSchema: compiledSchema,
+      responseJsonSchema: jsonSchema,
       temperature: 0.2,
       topP: 0.9,
       topK: 40,
@@ -203,6 +69,6 @@ exports.chatWithStreaming = async (userPrompts, callbackOptions) => {
     },
   };
 
-  // Send a chat message with streaming
-  await gemini.chatWithStreaming(chatOptions, callbackOptions);
+  // go
+  return await gemini.chat(chatOptions);
 };

package/server/util/llm1.js

@@ -0,0 +1,208 @@
+// path
+const path = require('path');
+const { readFile } = require('qiao-file');
+
+// z
+const { z } = require('zod');
+const { zodToJsonSchema } = require('zod-to-json-schema');
+
+// llm
+const { GeminiVertex } = require('viho-llm');
+
+// gemini
+const gemini = GeminiVertex({
+  projectId: global.QZ_CONFIG.gemini.projectId,
+  location: global.QZ_CONFIG.gemini.location,
+  modelName: global.QZ_CONFIG.gemini.modelName,
+});
+
+// ==================== 缓存（懒加载） ====================
+
+let cachedSystemPrompt = null;
+let cachedCompiledSchema = null;
+
+// ==================== JSON Schema 定义（避免联合类型）====================
+
+const jsonSchema = (() => {
+  const DiagramNodeSchema = z.object({
+    id: z.string().describe('节点唯一标识符，使用驼峰命名法，如 userService, mysqlDB'),
+    label: z.string().describe('节点显示文本，支持 \\n 换行'),
+    type: z
+      .enum(['rectangle', 'ellipse', 'diamond', 'hexagon', 'cylinder', 'cloud'])
+      .optional()
+      .describe('形状类型，默认 rectangle'),
+    color: z
+      .enum(['blue', 'green', 'purple', 'orange', 'red', 'gray', 'yellow', 'pink', 'black'])
+      .optional()
+      .describe('节点颜色，默认 blue'),
+    x: z.number().optional().describe('X 坐标，不指定则自动布局'),
+    y: z.number().optional().describe('Y 坐标，不指定则自动布局'),
+    width: z.number().optional().describe('宽度，不指定则使用默认值'),
+    height: z.number().optional().describe('高度，不指定则使用默认值'),
+  });
+
+  const DiagramConnectionSchema = z.object({
+    from: z.string().describe('起始节点 id'),
+    to: z.string().describe('目标节点 id'),
+    label: z.string().optional().describe('连接线上的文字标签'),
+    type: z.enum(['arrow', 'line']).optional().describe('连接类型：arrow=箭头线，line=直线，默认 arrow'),
+    style: z
+      .enum(['solid', 'dashed', 'dotted'])
+      .optional()
+      .describe('线条样式：solid=实线，dashed=虚线，dotted=点线，默认 solid'),
+  });
+
+  const TextAnnotationSchema = z.object({
+    id: z.string().describe('标注唯一标识符'),
+    text: z.string().describe('标注文本内容'),
+    x: z.number().optional().describe('X 坐标'),
+    y: z.number().optional().describe('Y 坐标'),
+    fontSize: z.number().optional().describe('字体大小'),
+    color: z.string().optional().describe('文字颜色'),
+  });
+
+  const FrameSchema = z.object({
+    id: z.string().describe('框架唯一标识符'),
+    label: z.string().optional().describe('框架标题'),
+    children: z.array(z.string()).describe('包含的节点 id 列表'),
+    color: z.string().optional().describe('框架颜色'),
+    x: z.number().optional().describe('X 坐标'),
+    y: z.number().optional().describe('Y 坐标'),
+    width: z.number().optional().describe('宽度'),
+    height: z.number().optional().describe('高度'),
+  });
+
+  const ImageElementSchema = z.object({
+    id: z.string().describe('图片唯一标识符'),
+    imageUrl: z.string().url().describe('图片 URL'),
+    alt: z.string().optional().describe('图片描述'),
+    x: z.number().optional().describe('X 坐标'),
+    y: z.number().optional().describe('Y 坐标'),
+    width: z.number().optional().describe('宽度'),
+    height: z.number().optional().describe('高度'),
+  });
+
+  // 注意：freedraws 使用简化格式，避免 tuple 导致的兼容性问题
+  const FreedrawElementSchema = z.object({
+    id: z.string().describe('手绘元素唯一标识符'),
+    points: z.array(z.number()).describe('路径点坐标数组，格式：[x1,y1,x2,y2,...]'),
+    color: z.string().optional().describe('线条颜色'),
+    strokeWidth: z.number().optional().describe('线条宽度'),
+  });
+
+  // 统一的响应 Schema（包含图表和错误两种情况）
+  // 通过 type 字段区分：'architecture'|'flowchart'|'sequence'|'custom' = 图表，'error' = 错误
+  return z.object({
+    // 通用字段：type 用于区分响应类型
+    type: z
+      .enum(['architecture', 'flowchart', 'sequence', 'custom', 'error'])
+      .describe('响应类型：architecture/flowchart/sequence/custom=图表类型，error=错误响应'),
+
+    // ========== 图表相关字段（当 type != error 时使用）==========
+    title: z.string().optional().describe('图表标题'),
+    nodes: z.array(DiagramNodeSchema).optional().describe('节点列表'),
+    connections: z.array(DiagramConnectionSchema).optional().describe('连接关系列表'),
+    annotations: z.array(TextAnnotationSchema).optional().describe('独立文本标注列表（可选）'),
+    frames: z.array(FrameSchema).optional().describe('分组框架列表（可选）'),
+    images: z.array(ImageElementSchema).optional().describe('图片元素列表（可选）'),
+    freedraws: z.array(FreedrawElementSchema).optional().describe('手绘元素列表（可选）'),
+
+    // ========== 错误相关字段（当 type = error 时使用）==========
+    error: z
+      .enum(['NON_DRAWING_REQUEST', 'INVALID_REQUEST', 'UNKNOWN_ERROR'])
+      .optional()
+      .describe('错误类型（仅当 type=error 时需要）'),
+    message: z.string().optional().describe('错误提示消息（仅当 type=error 时需要）'),
+  });
+})();
+
+// ==================== 懒加载函数 ====================
+
+/**
+ * 获取系统提示词（懒加载+缓存）
+ */
+async function getSystemPrompt() {
+  if (!cachedSystemPrompt) {
+    const promptPath = path.resolve(__dirname, './prompt.md');
+    cachedSystemPrompt = await readFile(promptPath);
+  }
+  return cachedSystemPrompt;
+}
+
+/**
+ * 获取编译后的 JSON Schema（懒加载+缓存）
+ */
+function getCompiledSchema() {
+  if (!cachedCompiledSchema) {
+    const fullSchema = zodToJsonSchema(jsonSchema, {
+      name: 'AIResponse',
+      $refStrategy: 'none',
+    });
+
+    // 关键修复：Gemini 不支持顶层 $ref，直接使用 definitions 中的定义
+    if (fullSchema.definitions && fullSchema.definitions.AIResponse) {
+      cachedCompiledSchema = fullSchema.definitions.AIResponse;
+    } else if (fullSchema.$ref) {
+      // 如果顶层是 $ref，解析它
+      const refName = fullSchema.$ref.split('/').pop();
+      cachedCompiledSchema = fullSchema.definitions[refName];
+    } else {
+      cachedCompiledSchema = fullSchema;
+    }
+  }
+  return cachedCompiledSchema;
+}
+
+// ==================== 主要功能 ====================
+
+/**
+ * chatWithStreaming
+ * @param {*} userPrompts
+ * @param {*} callbackOptions
+ */
+exports.chatWithStreaming = async (userPrompts, callbackOptions) => {
+  // 懒加载：第一次使用时加载并缓存
+  const systemPrompt = await getSystemPrompt();
+  const compiledSchema = getCompiledSchema();
+  console.log('compiledSchema');
+  console.log(JSON.stringify(compiledSchema));
+
+  // chat options
+  const chatOptions = {
+    contents: userPrompts,
+    systemInstruction: systemPrompt,
+    config: {
+      responseMimeType: 'application/json',
+      responseJsonSchema: compiledSchema,
+      temperature: 0.2,
+      topP: 0.9,
+      topK: 40,
+      maxOutputTokens: 8192,
+      thinkingConfig: {
+        thinkingBudget: 0,
+        includeThoughts: false,
+      },
+      safetySettings: [
+        {
+          category: 'HARM_CATEGORY_HATE_SPEECH',
+          threshold: 'BLOCK_ONLY_HIGH',
+        },
+        {
+          category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
+          threshold: 'BLOCK_ONLY_HIGH',
+        },
+        {
+          category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
+          threshold: 'BLOCK_ONLY_HIGH',
+        },
+        {
+          category: 'HARM_CATEGORY_HARASSMENT',
+          threshold: 'BLOCK_ONLY_HIGH',
+        },
+      ],
+    },
+  };
+
+  // Send a chat message with streaming
+  await gemini.chatWithStreaming(chatOptions, callbackOptions);
+};

package/server/util/prompt-intent.md

@@ -1,217 +1,107 @@
-# AI 白板助手 - 意图识别与设计说明（流式输出）
+# 意图识别：判断用户是否要绘制图表
 
-你是一个 AI 白板绘图助手，负责理解用户意图并提供设计说明。
+分析用户输入，判断是否为绘图请求。
 
-## 核心职责
+## 输出要求
 
-1. **判断用户意图**：识别用户是否在请求绘制图表
-2. **流式输出设计说明**：如果是绘图请求，实时输出设计思路
-3. **友好拒绝**：如果不是绘图请求，礼貌说明能力范围
-
-## 响应格式要求
-
-你的回复必须以特殊标记开始，用于后续处理：
-
-### 非绘图请求格式
-
-```
-[INTENT:REJECT]
-
-抱歉，我是 AI 白板的绘图助手，我只能帮你绘制图表、流程图、架构图等。
-
-我可以帮你：
-• 绘制流程图和架构图
-• 设计系统结构
-• 创建各种图形和连接
-
-如果你想画图，可以试试这样说：
-「画一个登录流程图」
-「设计一个微服务架构」
-「画三个矩形连接起来」
-```
-
-### 绘图请求格式
+以 JSON 格式输出：
 
+```json
+{
+  "intent": "DRAW" 或 "REJECT",
+  "reason": "判断理由（一句话说明）"
+}
 ```
-[INTENT:DRAW]
-
-好的，我来帮你设计[图表类型]！
-
-设计思路：
-[流式输出你的设计思路，包括：]
-1. 会使用哪些元素（矩形、圆形、箭头等）
-2. 元素之间的关系和布局
-3. 颜色和样式的选择理由
-4. 整体的视觉效果说明
-
-正在生成图表数据...
-```
-
-## 判断规则
-
-### 属于绘图请求的情况：
-
-- 明确要求绘制图表：「画一个流程图」「设计架构图」
-- 描述需要可视化的内容：「用户登录的流程」「微服务架构」
-- 要求创建图形：「画三个矩形」「连接这些模块」
-- 使用图表相关词汇：流程图、架构图、组织架构、思维导图、ER图、时序图等
-
-### 不属于绘图请求的情况：
-
-- 闲聊和问候：「你好」「今天天气怎么样」
-- 知识问答：「什么是微服务」「React 和 Vue 的区别」
-- 代码相关：「帮我写一段代码」「这段代码有什么问题」
-- 其他任务：「帮我写文章」「翻译这段话」「计算数学题」
-
-## 设计说明输出指南
 
-当确认是绘图请求后，你需要流式输出设计说明，包括：
+## 判断标准
 
-1. **欢迎语**：简短确认用户需求
-2. **设计分析**：
-   - 识别出的关键元素和实体
-   - 元素之间的关系（顺序、包含、依赖等）
-   - 推荐的图表类型（流程图、架构图、关系图等）
+### DRAW（绘图请求）
 
-3. **布局说明**：
-   - 元素的排列方式（横向、纵向、层次化）
-   - 分组和分层的逻辑
-   - 视觉重点的突出方式
+- 包含动作词：画、绘制、设计、生成、创建、制作
+- 包含图表词：图、流程图、架构图、思维导图、关系图、结构图、示意图
+- 描述可视化需求：「用户登录流程」「微服务架构」「数据库关系」
+- 描述图形元素：「三个矩形」「圆形连接」
 
-4. **样式选择**：
-   - 颜色方案的选择理由
-   - 形状类型的使用场景
-   - 箭头和连接线的样式
+### REJECT（非绘图请求）
 
-5. **结束语**：提示正在生成图表数据
+- 问候寒暄：你好、早上好、谢谢
+- 知识问答：什么是、如何、为什么、介绍、解释
+- 代码任务：写代码、debug、实现功能
+- 文字任务：翻译、写文章、总结
+- 其他任务：计算、推荐、建议
 
 ## 示例
 
-### 示例 1：非绘图请求
+输入：画一个用户登录流程图
+输出：
 
+```json
+{
+  "intent": "DRAW",
+  "reason": "包含动作词'画'和图表词'流程图'"
+}
 ```
-用户：你好，今天天气怎么样？
 
-助手：
-[INTENT:REJECT]
-
-抱歉，我是 AI 白板的绘图助手，我只能帮你绘制图表、流程图、架构图等。
-
-我可以帮你：
-• 绘制流程图和架构图
-• 设计系统结构
-• 创建各种图形和连接
-
-如果你想画图，可以试试这样说：
-「画一个登录流程图」
-「设计一个微服务架构」
-「画三个矩形连接起来」
-```
-
-### 示例 2：简单绘图请求
+输入：设计微服务架构
+输出：
 
+```json
+{
+  "intent": "DRAW",
+  "reason": "包含动作词'设计'和可视化需求'架构'"
+}
 ```
-用户：画三个矩形连接起来
 
-助手：
-[INTENT:DRAW]
+输入：三个矩形连接起来
+输出：
 
-好的，我来帮你设计一个简单的连接图！
-
-设计思路：
-1. 创建三个矩形节点，使用蓝色作为主色调
-2. 从左到右水平排列，保持均匀间距
-3. 使用箭头连接，表示从左到右的流动关系
-4. 矩形大小统一，简洁清晰
-
-正在生成图表数据...
+```json
+{
+  "intent": "DRAW",
+  "reason": "描述了具体的图形元素和连接关系"
+}
 ```
 
-### 示例 3：复杂绘图请求
+输入：简单介绍一下上海市
+输出：
 
+```json
+{
+  "intent": "REJECT",
+  "reason": "这是知识问答请求，不是绘图需求"
+}
 ```
-用户：设计一个用户登录的流程图
-
-助手：
-[INTENT:DRAW]
-
-好的，我来帮你设计用户登录流程图！
-
-设计思路：
 
-1. **关键节点识别**：
-   - 开始节点（椭圆形）
-   - 输入用户名密码（矩形）
-   - 验证凭证（菱形决策节点）
-   - 成功/失败处理（矩形）
-   - 结束节点（椭圆形）
+输入：什么是微服务架构？
+输出：
 
-2. **流程关系**：
-   - 线性主流程：开始 → 输入 → 验证 → 成功 → 结束
-   - 异常分支：验证失败 → 错误提示 → 返回输入
-
-3. **布局设计**：
-   - 采用纵向布局，从上到下展示流程
-   - 决策节点居中，两侧分出成功/失败分支
-   - 失败分支使用红色标注，成功路径使用绿色
-
-4. **视觉效果**：
-   - 起止节点：绿色椭圆，突出流程边界
-   - 处理节点：蓝色矩形，表示操作步骤
-   - 决策节点：橙色菱形，强调判断点
-   - 错误提示：红色矩形，警示异常情况
-
-正在生成图表数据...
+```json
+{
+  "intent": "REJECT",
+  "reason": "这是知识问答，用户在询问概念而非要求绘图"
+}
 ```
 
-### 示例 4：架构图请求
+输入：你好，今天天气怎么样
+输出：
 
+```json
+{
+  "intent": "REJECT",
+  "reason": "这是闲聊问候，与绘图无关"
+}
 ```
-用户：画一个微服务架构图
-
-助手：
-[INTENT:DRAW]
-
-好的，我来帮你设计微服务架构图！
-
-设计思路：
-
-1. **架构层次**：
-   - 前端层：客户端和 Web 应用
-   - 网关层：API Gateway
-   - 服务层：多个微服务（用户服务、订单服务、支付服务等）
-   - 数据层：各服务独立数据库
-
-2. **元素选择**：
-   - 服务节点：使用圆角矩形（cylinder 样式）表示微服务
-   - 数据库：使用圆柱体（cylinder）图标
-   - 网关：使用六边形突出其特殊地位
-   - 客户端：使用矩形
-
-3. **连接关系**：
-   - 客户端到网关：使用粗箭头表示请求流向
-   - 网关到服务：使用细箭头表示分发
-   - 服务到数据库：使用虚线表示数据访问
-   - 服务间调用：使用双向箭头
-
-4. **颜色方案**：
-   - 前端层：蓝色系，表示用户界面
-   - 网关层：紫色，突出中间层地位
-   - 服务层：绿色系，表示业务逻辑
-   - 数据层：灰色，表示存储
 
-5. **布局安排**：
-   - 分层垂直布局，从上到下展示请求流向
-   - 同层服务水平排列，体现并行关系
-   - 使用 Frame 框将同层元素分组
+输入：帮我写一段 Python 代码
+输出：
 
-正在生成图表数据...
+```json
+{
+  "intent": "REJECT",
+  "reason": "这是代码编写任务，不是绘图请求"
+}
 ```
 
-## 注意事项
+## 用户输入
 
-1. **标记必须准确**：必须以 `[INTENT:REJECT]` 或 `[INTENT:DRAW]` 开头
-2. **流式友好**：设计说明要分段输出，适合流式显示
-3. **简洁清晰**：避免过度冗长，重点突出设计思路
-4. **专业但友好**：使用专业术语但保持易懂
-5. **匹配用户需求**：设计说明要紧密对应用户的具体请求
+{USER_INPUT}

package/server/util/llm-intent.js

@@ -1,82 +0,0 @@
-// path
-const path = require('path');
-const { readFile } = require('qiao-file');
-
-// z
-// const { z } = require('zod');
-// const { zodToJsonSchema } = require('zod-to-json-schema');
-
-// llm
-const { GeminiVertex } = require('viho-llm');
-
-// gemini
-const gemini = GeminiVertex({
-  projectId: global.QZ_CONFIG.gemini.projectId,
-  location: global.QZ_CONFIG.gemini.location,
-  modelName: global.QZ_CONFIG.gemini.modelName,
-});
-
-// ==================== 缓存（懒加载） ====================
-
-let cachedSystemPrompt = null;
-
-// ==================== 懒加载函数 ====================
-
-/**
- * 获取系统提示词（懒加载+缓存）
- */
-async function getSystemPrompt(promptPath) {
-  if (!cachedSystemPrompt) cachedSystemPrompt = await readFile(path.resolve(__dirname, promptPath));
-  return cachedSystemPrompt;
-}
-
-// ==================== 主要功能 ====================
-
-/**
- * chatWithStreaming
- * @param {*} userPrompts
- * @param {*} callbackOptions
- */
-exports.chatWithStreaming = async (userPrompts, callbackOptions) => {
-  // 懒加载：第一次使用时加载并缓存
-  const systemPrompt = await getSystemPrompt('./prompt-intent.md');
-
-  // chat options
-  const chatOptions = {
-    contents: userPrompts,
-    systemInstruction: systemPrompt,
-    config: {
-      // responseMimeType: 'application/json',
-      // responseJsonSchema: compiledSchema,
-      temperature: 0.2,
-      topP: 0.9,
-      topK: 40,
-      maxOutputTokens: 8192,
-      thinkingConfig: {
-        thinkingBudget: 0,
-        includeThoughts: false,
-      },
-      safetySettings: [
-        {
-          category: 'HARM_CATEGORY_HATE_SPEECH',
-          threshold: 'BLOCK_ONLY_HIGH',
-        },
-        {
-          category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
-          threshold: 'BLOCK_ONLY_HIGH',
-        },
-        {
-          category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
-          threshold: 'BLOCK_ONLY_HIGH',
-        },
-        {
-          category: 'HARM_CATEGORY_HARASSMENT',
-          threshold: 'BLOCK_ONLY_HIGH',
-        },
-      ],
-    },
-  };
-
-  // Send a chat message with streaming
-  await gemini.chatWithStreaming(chatOptions, callbackOptions);
-};