@shun-js/aibaiban-server 0.8.3 → 0.8.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "0.8.3",
+  "version": "0.8.5",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -44,5 +44,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "331bc8dc9ccfa3de0ea2c3c7326301f660cab528"
+  "gitHead": "f3fa85784a8d086877a5057a29964d862126791c"
 }

package/server/service/LLMService.js

@@ -116,6 +116,8 @@ exports.drawAgent = async (req, res) => {
     const intentResult = await callLLMForJSON(prompts.ROUTER_PROMPT.replace('{input}', input), res, 'router');
     const intent = intentResult.intent;
     req.logger.info(methodName, 'intent', intent);
+    // 发送结果
+    res.streaming(`data: ${JSON.stringify({ step: 'router', intent })}\n\n`);
 
     // 非白板请求
     if (intent === 'irrelevant') {
@@ -132,6 +134,8 @@ exports.drawAgent = async (req, res) => {
     const classifyResult = await callLLMForJSON(prompts.CLASSIFY_PROMPT.replace('{input}', input), res, 'classify');
     const diagramType = classifyResult.diagramType;
     req.logger.info(methodName, 'diagramType', diagramType);
+    // 发送结果
+    res.streaming(`data: ${JSON.stringify({ step: 'classify', diagramType })}\n\n`);
 
     // 3. elaborate - 细化内容
     res.streaming(`data: ${JSON.stringify({ step: 'elaborate', status: 'start' })}\n\n`);
@@ -142,6 +146,8 @@ exports.drawAgent = async (req, res) => {
       'elaborate',
     );
     req.logger.info(methodName, 'elaboration', elaboration.slice(0, 100) + '...');
+    // 发送结果（不显示内容，只标记完成）
+    res.streaming(`data: ${JSON.stringify({ step: 'elaborate', done: true })}\n\n`);
 
     // 4. review - 质量检查
     res.streaming(`data: ${JSON.stringify({ step: 'review', status: 'start' })}\n\n`);
@@ -154,6 +160,8 @@ exports.drawAgent = async (req, res) => {
       'review',
     );
     req.logger.info(methodName, 'reviewResult', reviewResult);
+    // 发送结果
+    res.streaming(`data: ${JSON.stringify({ step: 'review', result: reviewResult.result })}\n\n`);
 
     // 信息不足，追问用户
     if (reviewResult.result === 'need_more_info') {

package/server/util/llm-agent.js

@@ -7,7 +7,7 @@ const { OpenAIAPI } = require('viho-llm');
 const llmConfig = {
   apiKey: global.QZ_CONFIG.kimi.apiKey,
   baseURL: global.QZ_CONFIG.kimi.baseURL,
-  modelName: global.QZ_CONFIG.kimi.modelID,
+  modelName: global.QZ_CONFIG.kimi.modelName,
 };
 
 const llm = OpenAIAPI(llmConfig);
@@ -43,9 +43,9 @@ function extractJSON(text) {
 
 /**
  * 流式调用 LLM 并解析 JSON
- * 使用非流式 API，但过程流式输出给客户端
+ * 不发送中间 delta，只返回解析后的结果
  */
-exports.callLLMForJSON = async (prompt, res, step) => {
+exports.callLLMForJSON = async (prompt) => {
   // 非流式调用，等待完整响应
   const response = await llm.chat({
     model: llmConfig.modelName,
@@ -53,16 +53,15 @@ exports.callLLMForJSON = async (prompt, res, step) => {
   });
 
   const fullContent = response.content || '';
-  // 流式输出给客户端
-  res.streaming(`data: ${JSON.stringify({ step, delta: fullContent })}\n\n`);
-
+  // 不发送 delta，直接返回结果让调用方处理
   return extractJSON(fullContent);
 };
 
 /**
  * 流式调用 LLM（普通文本）
+ * 不发送中间 delta，只返回完整内容
  */
-exports.callLLM = async (prompt, res, step) => {
+exports.callLLM = async (prompt) => {
   let fullContent = '';
 
   await llm.chatWithStreaming(
@@ -70,8 +69,7 @@ exports.callLLM = async (prompt, res, step) => {
     {
       contentCallback: (chunk) => {
         fullContent += chunk;
-        // 直接流式返回给客户端
-        res.streaming(`data: ${JSON.stringify({ step, delta: chunk })}\n\n`);
+        // 不发送中间 delta，避免显示问题
       },
     },
   );