@shun-js/aibaiban-server 1.3.7 → 1.3.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.3.7",
+  "version": "1.3.9",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -37,7 +37,7 @@
     "qiao-z-nuser": "^6.0.0",
     "qiao-z-service": "^6.0.0",
     "qiao-z-sms": "^6.0.0",
-    "viho-llm": "^1.0.9",
+    "viho-llm": "^1.1.0",
     "zod": "^4.3.6",
     "zod-to-json-schema": "^3.25.1"
   },
@@ -45,5 +45,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "d630d50c88faa610d607a2293cb315861910bd14"
+  "gitHead": "7288be3407eb8a311adbe7e6026ce0ed80ec4b83"
 }

package/server/service/LLMService.js

@@ -1,5 +1,5 @@
 // llm
-const { OpenAIAPI, runAgents } = require('viho-llm');
+const { OpenAIAPI, LibLibAPI, runAgents } = require('viho-llm');
 const prompts = require('../util/prompt-agent.js');
 
 // util
@@ -12,8 +12,9 @@ const llm = OpenAIAPI(finalLLMConfig);
 const modelName = finalLLMConfig.modelName;
 const thinking = finalLLMConfig.thinking;
 
-// 图片生成配置
+// 图片生成配置 (LibLib)
 const imageGenConfig = global.QZ_CONFIG.imageGen;
+const liblib = LibLibAPI(imageGenConfig);
 
 /**
  * convertToolCallToSkeleton - 将单个 tool call 转为 ExcalidrawElementSkeleton
@@ -335,39 +336,35 @@ exports.drawAgent = async (req, res) => {
         res.streaming(`data: ${JSON.stringify({ type: 'message', content: canvasResponse.content })}\n\n`);
       }
     } else if (decision.action === 'image') {
-      // 4. Image - 调用图片生成 API
+      // 4. Image - 调用 LibLib 图片生成 API
       res.streaming(`data: ${JSON.stringify({ type: 'status', step: 'image' })}\n\n`);
       req.logger.info(methodName, 'step: image generation', decision.prompt);
       chatFeishuMsg(req, `image-${decision.prompt}`);
 
-      const imageResponse = await fetch(`${imageGenConfig.baseURL}/images/generations`, {
-        method: 'POST',
-        headers: {
-          Authorization: `Bearer ${imageGenConfig.apiKey}`,
-          'Content-Type': 'application/json',
-        },
-        body: JSON.stringify({
-          model: imageGenConfig.modelName,
-          prompt: decision.prompt,
-        }),
-      });
+      const submitRes = await liblib.text2img(decision.prompt, { aspectRatio: 'square' });
+      const generateUuid = submitRes.data?.generateUuid;
 
-      const imageResult = await imageResponse.json();
-      const imageUrl = imageResult.data?.[0]?.url;
+      if (!generateUuid) {
+        req.logger.error(methodName, 'liblib submit failed', JSON.stringify(submitRes));
+        res.streaming(`data: ${JSON.stringify({ type: 'message', content: '图片生成失败，请稍后重试。' })}\n\n`);
+      } else {
+        const result = await liblib.waitForResult(generateUuid);
+        const imageUrl = result.imagesInfo?.[0]?.imageUrl;
 
-      if (imageUrl) {
-        // 下载图片转 base64（火山引擎 TOS URL 不支持浏览器跨域 fetch）
-        const imgResponse = await fetch(imageUrl);
-        const imgBuffer = Buffer.from(await imgResponse.arrayBuffer());
-        const contentType = imgResponse.headers.get('content-type') || 'image/jpeg';
-        const dataURL = `data:${contentType};base64,${imgBuffer.toString('base64')}`;
+        if (imageUrl) {
+          // 下载图片转 base64
+          const imgResponse = await fetch(imageUrl);
+          const imgBuffer = Buffer.from(await imgResponse.arrayBuffer());
+          const contentType = imgResponse.headers.get('content-type') || 'image/jpeg';
+          const dataURL = `data:${contentType};base64,${imgBuffer.toString('base64')}`;
 
-        const duration = Date.now() - startTime;
-        req.logger.info(methodName, 'image generated', `${duration}ms`);
-        res.streaming(`data: ${JSON.stringify({ type: 'image', dataURL, duration })}\n\n`);
-      } else {
-        req.logger.error(methodName, 'image generation failed', JSON.stringify(imageResult));
-        res.streaming(`data: ${JSON.stringify({ type: 'message', content: '图片生成失败，请稍后重试。' })}\n\n`);
+          const duration = Date.now() - startTime;
+          req.logger.info(methodName, 'image generated', `${duration}ms`);
+          res.streaming(`data: ${JSON.stringify({ type: 'image', dataURL, duration })}\n\n`);
+        } else {
+          req.logger.error(methodName, 'liblib no image url', JSON.stringify(result));
+          res.streaming(`data: ${JSON.stringify({ type: 'message', content: '图片生成失败，请稍后重试。' })}\n\n`);
+        }
       }
     } else if (decision.action === 'sketch') {
       // 5. Sketch - 通知前端执行线稿提取（处理在前端完成）

package/views/index.html

@@ -105,7 +105,7 @@
     <script
       type="module"
       crossorigin
-      src="https://static-small.vincentqiao.com/aibaiban/static/index-BHX1lvXv.js"
+      src="https://static-small.vincentqiao.com/aibaiban/static/index-DhOf2fPx.js"
     ></script>
     <link
       rel="modulepreload"
@@ -130,7 +130,7 @@
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-W79wllRE.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-CaLPmsCY.js"
     />
     <link
       rel="stylesheet"