@shun-js/aibaiban-server 1.0.3 → 1.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -44,5 +44,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "6882b0a600e3ad421b67cee58134371ec478ce93"
+  "gitHead": "5eb0853768a4ff327c1ddda1eaa414b8645144b1"
 }

package/server/service/LLMService.js

@@ -7,7 +7,7 @@ const { chatFeishuMsg, errorFeishuMsg } = require('../util/feishu.js');
 
 /**
  * drawAgent - 流式 Agent 接口
- * 流程：router -> classify -> elaborate -> review -> generate
+ * 流程：router -> classify -> elaborate -> generate
  */
 exports.drawAgent = async (req, res) => {
   const methodName = 'drawAgent';
@@ -77,36 +77,7 @@ exports.drawAgent = async (req, res) => {
     chatFeishuMsg(req, `elaboration-${elaboration}`);
     res.streaming(`data: ${JSON.stringify({ step: 'elaborate', done: true, duration: elaborateTime })}\n\n`);
 
-    // 4. review - 质量检查
-    stepStart = Date.now();
-    res.streaming(`data: ${JSON.stringify({ step: 'review', status: 'start' })}\n\n`);
-    req.logger.info(methodName, 'step: review');
-    const reviewResult = await callLLMForJSON(
-      prompts.REVIEW_PROMPT.replace('{input}', input)
-        .replace('{diagramType}', diagramType)
-        .replace('{elaboration}', elaboration),
-      res,
-      'review',
-    );
-    const reviewTime = Date.now() - stepStart;
-    req.logger.info(methodName, 'reviewResult', reviewResult, `${reviewTime}ms`);
-    chatFeishuMsg(req, `reviewResult-${reviewResult}`);
-    res.streaming(`data: ${JSON.stringify({ step: 'review', result: reviewResult.result, duration: reviewTime })}\n\n`);
-
-    // 信息不足，追问用户
-    if (reviewResult.result === 'need_more_info') {
-      res.streaming(
-        `data: ${JSON.stringify({
-          step: 'review',
-          result: 'need_more_info',
-          questions: reviewResult.questions,
-        })}\n\n`,
-      );
-      res.streamingEnd();
-      return;
-    }
-
-    // 5. generate - 生成 Mermaid
+    // 4. generate - 生成 Mermaid
     stepStart = Date.now();
     res.streaming(`data: ${JSON.stringify({ step: 'generate', status: 'start' })}\n\n`);
     req.logger.info(methodName, 'step: generate');

package/server/util/prompt-agent.js

@@ -7,7 +7,7 @@ module.exports = {
   /**
    * 意图分类 - 判断是否与白板相关
    */
-  ROUTER_PROMPT: `你是一个意图分类器。判断用户输入是否与"画图/白板/图表/流程图/甘特图/思维导图/时序图/类图/状态图/ER图/饼图/Git分支图/用户旅程图"相关。
+  ROUTER_PROMPT: `你是一个意图分类器。判断用户输入是否与"画图/白板/图表/流程图/时序图/类图/ER图"相关。
 
 用户输入: {input}
 
@@ -19,16 +19,16 @@ module.exports = {
   CLASSIFY_PROMPT: `你是一个图表类型分析专家。根据用户需求，判断最适合的图表类型。
 
 可选类型:
-- flowchart: 流程图、步骤图、决策流程
-- sequence: 时序图、交互流程、调用链
-- classDiagram: 类图、对象关系、继承结构
-- stateDiagram: 状态图、状态机、状态流转
-- erDiagram: ER图、实体关系图、数据库设计
-- gantt: 甘特图、项目计划、时间线
-- pie: 饼图、占比图、比例分布
-- gitGraph: Git分支图、版本管理流程
-- journey: 用户旅程图、体验地图
-- mindmap: 思维导图、知识结构
+- flowchart: 流程图、步骤图、决策流程、状态图、思维导图等通用图表
+- sequence: 时序图、交互流程、调用链、消息传递
+- classDiagram: 类图、对象关系、继承结构、接口设计
+- erDiagram: ER图、实体关系图、数据库设计、数据模型
+
+注意:
+- 只能从以上4种类型中选择
+- 提到"ER图"、"实体关系"、"数据库"、"表结构"时必须选 erDiagram
+- 提到"类"、"继承"、"接口"、"对象"时选 classDiagram
+- 如果不确定，默认选 flowchart
 
 用户输入: {input}
 
@@ -47,32 +47,7 @@ module.exports = {
 - flowchart: {"nodes": [...], "edges": [...]}
 - sequence: {"participants": [...], "messages": [...]}
 - classDiagram: {"classes": [{"name": "", "attributes": [...], "methods": [...]}], "relations": [...]}
-- stateDiagram: {"states": [...], "transitions": [...]}
 - erDiagram: {"entities": [{"name": "", "attributes": [...]}], "relations": [...]}
-- gantt: {"tasks": [{"name": "", "start": "", "duration": ""}]}
-- pie: {"title": "", "slices": [{"label": "", "value": 0}]}
-- gitGraph: {"branches": [...], "commits": [...], "merges": [...]}
-- journey: {"title": "", "sections": [{"name": "", "tasks": [{"name": "", "score": 0}]}]}
-- mindmap: {"root": "", "children": [...]}
-
-只回复 JSON。`,
-
-  /**
-   * 质量检查
-   */
-  REVIEW_PROMPT: `你是一个质量检查专家。检查以下图表描述是否信息充足，能否生成完整的图表。
-
-用户原始输入: {input}
-图表类型: {diagramType}
-细化描述: {elaboration}
-
-判断标准:
-1. 节点/步骤是否足够（至少3个有意义的元素）
-2. 关系/连接是否清晰
-3. 是否有明显缺失的关键信息
-
-如果信息充足，回复: {"result": "pass"}
-如果信息不足，回复: {"result": "need_more_info", "questions": ["问题1", "问题2"]}
 
 只回复 JSON。`,
 
@@ -90,6 +65,12 @@ module.exports = {
 3. 节点文本和连接线文本中禁止使用括号()[]{}，这些是 Mermaid 的形状语法符号，会导致解析错误
 4. 连接线标签示例: -->|是| 、-->|否| 、-->|提交|，不要写 -->|否(注册)| 或 -->|是(通过)|
 5. 如需补充说明，用斜杠/顿号/破折号代替括号，如"否-注册"、"否/注册"
+6. erDiagram 属性格式严格为: type name PK/FK "注释"，每个属性只能有这4个部分
+   - 正确: string username PK "用户名"
+   - 正确: int age "年龄"
+   - 错误: VARCHAR(50) UNIQUE NOT NULL username "用户名"（不支持SQL语法）
+   - 错误: DATETIME DEFAULT CURRENT_TIMESTAMP created_at（不支持DEFAULT等修饰符）
+   - type只用简单类型: string, int, text, datetime, boolean, float
 
 要求:
 1. 生成合法的 Mermaid 语法
@@ -101,6 +82,5 @@ module.exports = {
   /**
    * 非白板请求的固定回复
    */
-  FIXED_REPLY:
-    '我是一个白板助手，专门帮你生成各种图表。支持：流程图、时序图、类图、状态图、ER图、甘特图、饼图、Git分支图、用户旅程图、思维导图。请告诉我你想画什么图吧！',
+  FIXED_REPLY: '我是一个白板助手，专门帮你生成各种图表。支持：流程图、时序图、类图、ER图。请告诉我你想画什么图吧！',
 };

package/views/index.html

@@ -109,7 +109,7 @@
     <script
       type="module"
       crossorigin
-      src="https://static-small.vincentqiao.com/aibaiban/static/index-CYIYkLbz.js"
+      src="https://static-small.vincentqiao.com/aibaiban/static/index-D3qgIOvl.js"
     ></script>
     <link
       rel="modulepreload"
@@ -134,7 +134,7 @@
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-WA-IihWf.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-BV1kaZIV.js"
     />
     <link
       rel="stylesheet"