@shun-js/aibaiban-server 0.7.7 → 0.8.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 qiaowenbin<uikoo9@qq.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/assets/bing.txt

@@ -0,0 +1 @@
+4cb288d7aef5469c92616c0f5b5aeb89

package/assets/robots.txt

@@ -0,0 +1,36 @@
+# AI白板 - robots.txt
+# 更新时间: 2026-01-26
+
+# 允许所有搜索引擎爬虫
+User-agent: *
+Allow: /
+
+# 禁止爬取静态资源文件（可选，减少服务器负载）
+# Disallow: /assets/
+# Disallow: /*.js$
+# Disallow: /*.css$
+
+# 禁止爬取 API 接口
+Disallow: /api/
+
+# 爬虫访问速率限制（可选，防止爬虫过度占用带宽1）
+# Crawl-delay: 1
+
+# 站点地图
+Sitemap: https://aibaiban.com/sitemap.xml
+
+# 特定爬虫配置
+# Google Bot
+User-agent: Googlebot
+Allow: /
+Crawl-delay: 0
+
+# Bing Bot
+User-agent: Bingbot
+Allow: /
+Crawl-delay: 0
+
+# 百度爬虫
+User-agent: Baiduspider
+Allow: /
+Crawl-delay: 1

package/assets/sitemap.xml

@@ -0,0 +1,16 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<urlset
+      xmlns="http://www.sitemaps.org/schemas/sitemap/0.9"
+      xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+      xsi:schemaLocation="http://www.sitemaps.org/schemas/sitemap/0.9
+            http://www.sitemaps.org/schemas/sitemap/0.9/sitemap.xsd">
+
+  <!-- 首页/落地页 - 最高优先级 -->
+  <url>
+    <loc>https://aibaiban.com/</loc>
+    <lastmod>2026-02-03</lastmod>
+    <changefreq>weekly</changefreq>
+    <priority>1.0</priority>
+  </url>
+
+</urlset>

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "0.7.7",
+  "version": "0.8.0",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -16,6 +16,9 @@
     "url": "https://github.com/uikoo9/shun-js/issues"
   },
   "files": [
+    "server",
+    "assets",
+    "views",
     "app.js"
   ],
   "scripts": {
@@ -40,5 +43,6 @@
   "publishConfig": {
     "access": "public",
     "registry": "https://registry.npmjs.org/"
-  }
+  },
+  "gitHead": "6d0b2408e4398e16f0f4c6e32d204e511957035a"
 }

package/server/controller/IndexController.js

@@ -0,0 +1,12 @@
+// service
+const service = require('../service/IndexService.js');
+
+/**
+ * controller
+ */
+module.exports = (app) => {
+  // index
+  app.get('/', (req, res) => {
+    service.index(req, res);
+  });
+};

package/server/controller/LLMController.js

@@ -0,0 +1,17 @@
+// service
+const service = require('../service/LLMService.js');
+
+/**
+ * controller
+ */
+module.exports = (app) => {
+  // intent
+  app.post('/intent', (req, res) => {
+    service.intent(req, res);
+  });
+
+  // drawWithTools (使用结构化 JSON 输出)
+  app.post('/drawWithTools', (req, res) => {
+    service.drawWithTools(req, res);
+  });
+};

package/server/controller/SEOController.js

@@ -0,0 +1,18 @@
+// service
+const service = require('../service/SEOService.js');
+
+/**
+ * controller
+ */
+module.exports = (app) => {
+  // seo
+  app.get('/robots.txt', (req, res) => {
+    service.robots(req, res);
+  });
+  app.get('/sitemap.xml', (req, res) => {
+    service.sitemap(req, res);
+  });
+  app.get('/4cb288d7aef5469c92616c0f5b5aeb89.txt', (req, res) => {
+    service.bingIndexNow(req, res);
+  });
+};

package/server/log-options.js

@@ -0,0 +1,34 @@
+/**
+ * log options
+ * @returns
+ */
+module.exports = () => {
+  // log options
+  const logLevel = 'debug';
+  const logPattern = 'yyyy-MM-dd-hh';
+  const logPath = require('path').resolve(__dirname, '../logs/qiao-z.log');
+
+  return {
+    pm2: true,
+    pm2InstanceVar: 'INSTANCE_ID',
+    appenders: {
+      stdout: {
+        type: 'stdout',
+      },
+      datefile: {
+        type: 'dateFile',
+        pattern: logPattern,
+        filename: logPath,
+        keepFileExt: true,
+        numBackups: 30,
+        compress: true,
+      },
+    },
+    categories: {
+      default: {
+        level: logLevel,
+        appenders: ['stdout', 'datefile'],
+      },
+    },
+  };
+};

package/server/service/IndexService.js

@@ -0,0 +1,16 @@
+/**
+ * index
+ * @param {*} req
+ * @param {*} res
+ */
+exports.index = async (req, res) => {
+  // const
+  const pagePath = './views/index.html';
+
+  // is static
+  const isStatic = await res.staticRender(pagePath);
+  if (isStatic) return;
+
+  // render
+  res.render(pagePath, {}, true);
+};

package/server/service/LLMService.js

@@ -0,0 +1,85 @@
+// llm
+const { llmParseIntent } = require('../util/llm-intent.js');
+
+// structured json output
+const { generateFlowchartWithTools } = require('../util/llm-toolcall.js');
+
+// util
+const { chatFeishuMsg, chatResFeishuMsg, errorFeishuMsg } = require('../util/feishu.js');
+
+/**
+ * intent
+ * @param {*} req
+ * @param {*} res
+ * @returns
+ */
+exports.intent = async (req, res) => {
+  const methodName = 'intent';
+
+  // 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
+  try {
+    const llmParseIntentRes = await llmParseIntent(userPrompt);
+    const llmParseIntentObj = JSON.parse(llmParseIntentRes);
+    req.logger.info(methodName, 'llmParseIntentObj', llmParseIntentObj);
+
+    // r
+    chatResFeishuMsg(req, JSON.stringify(llmParseIntentObj));
+    res.jsonSuccess('success', llmParseIntentObj);
+  } catch (error) {
+    const msg = 'parse intent error';
+    errorFeishuMsg(req, msg);
+    req.logger.error(methodName, msg, error);
+    res.jsonFail(msg);
+  }
+};
+
+/**
+ * drawWithTools - 使用结构化 JSON 输出生成流程图
+ * @param {*} req
+ * @param {*} res
+ * @returns
+ */
+exports.drawWithTools = async (req, res) => {
+  const methodName = 'drawWithTools';
+
+  // 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
+  try {
+    const diagram = await generateFlowchartWithTools(userPrompt);
+    req.logger.info(methodName, 'diagram', diagram);
+
+    // r
+    chatResFeishuMsg(req, JSON.stringify(diagram));
+    res.jsonSuccess('success', diagram);
+  } catch (error) {
+    const msg = 'draw with tools error';
+    errorFeishuMsg(req, msg);
+    req.logger.error(methodName, msg, error);
+    res.jsonFail(msg);
+  }
+};

package/server/service/SEOService.js

@@ -0,0 +1,38 @@
+// path
+const path = require('path');
+
+// qiao
+const { readFile } = require('qiao-file');
+
+/**
+ * robots
+ * @param {*} req
+ * @param {*} res
+ */
+exports.robots = async (req, res) => {
+  const txtPath = path.resolve(__dirname, '../../assets/robots.txt');
+  const txt = await readFile(txtPath);
+  res.send(txt);
+};
+
+/**
+ * sitemap
+ * @param {*} req
+ * @param {*} res
+ */
+exports.sitemap = async (req, res) => {
+  const sitemapPath = path.resolve(__dirname, '../../assets/sitemap.xml');
+  const sitemap = await readFile(sitemapPath);
+  res.send(sitemap);
+};
+
+/**
+ * bingIndexNow
+ * @param {*} req
+ * @param {*} res
+ */
+exports.bingIndexNow = async (req, res) => {
+  const txtPath = path.resolve(__dirname, '../../assets/bing.txt');
+  const txt = await readFile(txtPath);
+  res.send(txt);
+};

package/server/util/check.js

@@ -0,0 +1,23 @@
+// qiao
+const { userCheck } = require('qiao-z-service');
+
+/**
+ * checkUserAuth
+ * @param {*} req
+ * @param {*} res
+ * @returns
+ */
+exports.checkUserAuth = async function (req, res) {
+  const userCheckRes = await userCheck(global.QZ_CONFIG.user, {
+    userid: req.headers.userid,
+    usertoken: req.headers.usertoken,
+    paths: JSON.stringify(global.QZ_CONFIG.paths),
+    path: req.url.pathname,
+  });
+
+  // pass
+  if (userCheckRes && userCheckRes.type === 'success') return true;
+
+  // r
+  res.json(userCheckRes);
+};

package/server/util/feishu.js

@@ -0,0 +1,77 @@
+// services
+const { feishuBot } = require('@shun-js/shun-service');
+
+/**
+ * feishuMsg
+ * @param {*} msg
+ */
+exports.feishuMsg = (msg) => {
+  if (global.QZ_CONFIG.env !== 'production') return;
+
+  feishuBot({
+    url: global.QZ_CONFIG.feishu.url,
+    feishuUrl: global.QZ_CONFIG.feishu.feishuUrl,
+    feishuMsg: msg,
+  });
+};
+
+// is bot
+function isBot(req) {
+  const ua = req.useragent;
+  const hasOSName = ua && ua.os && ua.os.name;
+  const isGoogleBot = ua && ua.browser && ua.browser.name === 'Googlebot';
+  const isMetaBot = ua && ua.browser && ua.browser.name === 'meta-externalagent';
+  const isSafariBot = ua && ua.engine && ua.engine.name === 'WebKit' && ua.engine.version === '605.1.15';
+  return !hasOSName || isGoogleBot || isMetaBot || isSafariBot;
+}
+
+/**
+ * errorFeishuMsg
+ * @param {*} req
+ * @param {*} msg
+ * @returns
+ */
+exports.errorFeishuMsg = (req, msg) => {
+  // check
+  if (isBot(req)) return;
+
+  // msg
+  const uaJson = JSON.stringify(req.useragent || {});
+  exports.feishuMsg(`【通知】服务异常，${msg}，请查看日志，ua：${uaJson}`);
+};
+
+/**
+ * chatFeishuMsg
+ * @param {*} req
+ * @returns
+ */
+exports.chatFeishuMsg = (req) => {
+  // check
+  if (isBot(req)) return;
+
+  // msg
+  const uaJson = JSON.stringify(req.useragent || {});
+  const userid = req.headers.userid;
+  const prompt = decodeURIComponent(req.body.userPrompt);
+
+  const msg = `【通知】/chat被访问\nuserid:${userid}\nua:\n${uaJson}\nprompt:\n${prompt}`;
+  exports.feishuMsg(msg);
+};
+
+/**
+ * chatResFeishuMsg
+ * @param {*} req
+ * @returns
+ */
+exports.chatResFeishuMsg = (req, chatRes) => {
+  // check
+  if (isBot(req)) return;
+
+  // msg
+  const uaJson = JSON.stringify(req.useragent || {});
+  const userid = req.headers.userid;
+  const prompt = decodeURIComponent(req.body.userPrompt);
+
+  const msg = `【通知】/chat生成成功\nuserid:${userid}\nua:\n${uaJson}\nprompt:\n${prompt}\nres:${chatRes}`;
+  exports.feishuMsg(msg);
+};

package/server/util/llm-intent.js

@@ -0,0 +1,79 @@
+// 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');
+const gemini = GeminiVertex({
+  projectId: global.QZ_CONFIG.gemini.projectId,
+  location: global.QZ_CONFIG.gemini.location,
+  modelName: global.QZ_CONFIG.gemini.modelName,
+});
+
+// const
+const chatConfig = {
+  responseMimeType: 'application/json',
+  temperature: 0.2, // 保持较低温度，确保输出稳定
+  topP: 0.9,
+  topK: 40,
+  maxOutputTokens: 8192, // 足够大以支持复杂图表
+  thinkingConfig: {
+    thinkingBudget: 0,
+    includeThoughts: false,
+  },
+};
+const 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',
+  },
+];
+
+/**
+ * llmParseIntent
+ * @param {*} userPrompts
+ */
+let intentSystemPrompt = null;
+exports.llmParseIntent = async (userPrompts) => {
+  // intent system prompt - 支持版本切换
+  if (!intentSystemPrompt) {
+    const promptFile = './prompt-intent.md';
+    intentSystemPrompt = await readFile(path.resolve(__dirname, promptFile));
+  }
+
+  // chat config - 新增 confidence 字段支持
+  chatConfig.responseJsonSchema = zodToJsonSchema(
+    z.object({
+      intent: z.enum(['DRAW', 'REJECT']).describe('用户意图：DRAW（绘图）或 REJECT（非绘图）'),
+      reason: z.string().describe('判断理由（一句话说明）'),
+      confidence: z.number().min(0).max(1).optional().describe('置信度评分（0.0-1.0），可选字段'),
+    }),
+  );
+
+  // chat options
+  const chatOptions = {
+    contents: userPrompts,
+    systemInstruction: intentSystemPrompt,
+    config: chatConfig,
+    safetySettings: safetySettings,
+  };
+
+  // go
+  return await gemini.chat(chatOptions);
+};