@shun-js/aibaiban-server 1.3.8 → 1.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.3.8",
+  "version": "1.4.0",
   "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": "e91cb46c02d591ce3bddd89e452abb089ca31dc5"
+  "gitHead": "c5efdb11f9e3d5fb520d76ec8e752eb6474effaf"
 }

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.images?.[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 - 通知前端执行线稿提取（处理在前端完成）