@shun-js/aibaiban-server 1.0.4 → 1.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -44,5 +44,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "9e55cb650198206e7f85818beb61fe686d93dc2c"
+  "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

@@ -19,12 +19,16 @@ module.exports = {
   CLASSIFY_PROMPT: `你是一个图表类型分析专家。根据用户需求，判断最适合的图表类型。
 
 可选类型:
-- flowchart: 流程图、步骤图、决策流程、思维导图、甘特图等通用图表
-- sequence: 时序图、交互流程、调用链
-- classDiagram: 类图、对象关系、继承结构
-- erDiagram: ER图、实体关系图、数据库设计
+- flowchart: 流程图、步骤图、决策流程、状态图、思维导图等通用图表
+- sequence: 时序图、交互流程、调用链、消息传递
+- classDiagram: 类图、对象关系、继承结构、接口设计
+- erDiagram: ER图、实体关系图、数据库设计、数据模型
 
-注意: 只能从以上4种类型中选择。如果用户需求不完全匹配，选择最接近的类型（如思维导图用flowchart，状态图用flowchart）。
+注意:
+- 只能从以上4种类型中选择
+- 提到"ER图"、"实体关系"、"数据库"、"表结构"时必须选 erDiagram
+- 提到"类"、"继承"、"接口"、"对象"时选 classDiagram
+- 如果不确定，默认选 flowchart
 
 用户输入: {input}
 
@@ -45,25 +49,6 @@ module.exports = {
 - classDiagram: {"classes": [{"name": "", "attributes": [...], "methods": [...]}], "relations": [...]}
 - erDiagram: {"entities": [{"name": "", "attributes": [...]}], "relations": [...]}
 
-只回复 JSON。`,
-
-  /**
-   * 质量检查
-   */
-  REVIEW_PROMPT: `你是一个质量检查专家。检查以下图表描述是否信息充足，能否生成完整的图表。
-
-用户原始输入: {input}
-图表类型: {diagramType}
-细化描述: {elaboration}
-
-判断标准:
-1. 节点/步骤是否足够（至少3个有意义的元素）
-2. 关系/连接是否清晰
-3. 是否有明显缺失的关键信息
-
-如果信息充足，回复: {"result": "pass"}
-如果信息不足，回复: {"result": "need_more_info", "questions": ["问题1", "问题2"]}
-
 只回复 JSON。`,
 
   /**
@@ -80,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 语法

package/views/index.html

@@ -109,7 +109,7 @@
     <script
       type="module"
       crossorigin
-      src="https://static-small.vincentqiao.com/aibaiban/static/index-Nr2tpEOY.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-BE0oQTrv.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-BV1kaZIV.js"
     />
     <link
       rel="stylesheet"