chore：替换了ai-service.js

common/js/ai-service.js

@@ -60,55 +60,107 @@ export default {
   /**
    * 流式消息入口（自动适配平台）
    */
-  async sendStreamMessage(message, onChunk, onComplete) {
+ async sendStreamMessage(message, onChunk, onComplete) {
-    // #ifdef MP-WEIXIN
+   // #ifdef MP-WEIXIN
-    // 微信小程序：降级为普通请求 + 前端打字模拟
+   return new Promise((resolve, reject) => {
-    try {
+     const socketTask = wx.connectSocket({
-      const result = await this.sendMessage(message);
+       url: 'wss://dev.aigc-quickapp.com/ws/aikefu',
-      const content = result.content || '';
+       header: {}
-      const conversationId = result.conversationId || '';
+     });
-
+ 
-      // 保存会话ID（确保连续对话）
+     let content = '';
-      if (conversationId) {
+     let conversationId = '';
-        this.setConversationId(conversationId);
+     let isAuthenticated = false;
-      }
+ 
-
+     socketTask.onOpen(() => {
-      // 模拟打字效果
+       console.log('WebSocket 连接成功，开始认证...');
-      let index = 0;
+       socketTask.send({
-      const chunkSize = 2; // 每次显示2个字符
+         data: JSON.stringify({
-      return new Promise((resolve) => {
+           action: 'auth',
-        const timer = setInterval(() => {
+           uniacid: store.state.uniacid || '1',
-          if (index < content.length) {
+           token: store.state.token || 'test_token',
-            const chunk = content.substring(index, index + chunkSize);
+           user_id: store.state.memberInfo?.id || 'anonymous'
-            index += chunkSize;
+         })
-            if (onChunk) onChunk(chunk);
+       });
-          } else {
+     });
-            clearInterval(timer);
+ 
-            if (onComplete) {
+     socketTask.onMessage((res) => {
-              onComplete({
+       try {
-                content: content,
+         const data = JSON.parse(res.data);
-                conversation_id: conversationId
+         console.log('收到 WebSocket 消息:', data);
-              });
+ 
-            }
+         if (data.type === 'auth_success') {
-            resolve({ content, conversation_id: conversationId });
+           console.log('认证成功，发送聊天消息...');
-          }
+           isAuthenticated = true;
-        }, 80); // 打字速度：80ms/次
+           socketTask.send({
-      });
+             data: JSON.stringify({
-    } catch (error) {
+               action: 'chat',
-      console.error('小程序流式消息降级失败:', error);
+               uniacid: store.state.uniacid || '1',
-      if (onComplete) {
+               query: message,
-        onComplete({ error: error.message || '发送失败' });
+               user_id: store.state.memberInfo?.id || 'anonymous',
-      }
+               conversation_id: this.getConversationId() || ''
-      throw error;
+             })
-    }
+           });
+         } else if (data.type === 'auth_failed') {
+           const errorMsg = '认证失败，请重新登录';
+           console.error(errorMsg, data);
+           reject(new Error(errorMsg));
+           if (onComplete) onComplete({ error: errorMsg });
+           socketTask.close();
+         }
+ 
+         // 处理流式消息块
+         else if (data.type === 'message' || data.event === 'message') {
+           const text = data.answer || data.content || data.text || '';
+           content += text;
+           if (onChunk) onChunk(text);
+         }
+ 
+         // 处理流结束
+         else if (data.event === 'message_end' || data.type === 'message_end') {
+           conversationId = data.conversation_id || '';
+           if (conversationId) {
+             this.setConversationId(conversationId);
+           }
+           if (onComplete) {
+             onComplete({ content, conversation_id: conversationId });
+           }
+           resolve({ content, conversation_id: conversationId });
+           socketTask.close();
+         }
+ 
+         // 可选：处理 done
+         else if (data.type === 'done') {
+           console.log('对话完成:', data);
+         }
+ 
+       } catch (e) {
+         console.error('WebSocket 消息解析失败:', e, '原始数据:', res.data);
+       }
+     });
+ 
+     socketTask.onError((err) => {
+       const errorMsg = 'WebSocket 连接失败';
+       console.error(errorMsg, err);
+       reject(new Error(errorMsg));
+       if (onComplete) onComplete({ error: errorMsg });
+     });
+ 
+     socketTask.onClose(() => {
+       console.log('WebSocket 连接已关闭');
+     });
+ 
+     const timeout = setTimeout(() => {
+       if (!isAuthenticated || content === '') {
+         console.warn('WebSocket 超时，强制关闭');
+         socketTask.close();
+         reject(new Error('AI服务响应超时'));
+         if (onComplete) onComplete({ error: 'AI服务响应超时' });
+       }
+     }, 10000);
+   });
     // #endif
-
+},
-    // #ifdef H5
-    // H5：使用真实流式（EventSource / Fetch）
-    return this.sendHttpStream(message, onChunk, onComplete);
-    // #endif
-  },
-
   /**
    * HTTP 流式请求（仅 H5 使用）
    */
@@ -131,64 +183,75 @@ export default {
           conversation_id: this.getConversationId() || ''
         })
       });
-
+  
-      if (!response.ok) {
+      if (!response.ok || !response.body) {
-        throw new Error(`HTTP ${response.status}`);
+        throw new Error('无效响应');
       }
-      if (!response.body) {
+  
-        throw new Error('响应体不可用');
-      }
-
       const reader = response.body.getReader();
       const decoder = new TextDecoder('utf-8');
       let buffer = '';
       let content = '';
       let conversationId = '';
-
+  
-      function processBuffer(buf, callback) {
+      while (true) {
-        const lines = buf.split('\n');
+        const { done, value } = await reader.read();
-        buf = lines.pop() || '';
+        if (done) break;
+  
+        buffer += decoder.decode(value, { stream: true });
+  
+        // 按行分割，保留不完整的最后一行
+        const lines = buffer.split('\n');
+        buffer = lines.pop() || ''; // 未完成的行留到下次
+  
         for (const line of lines) {
           const trimmed = line.trim();
           if (trimmed.startsWith('data:')) {
-            const jsonStr = trimmed.slice(5).trim();
+            try {
-            if (jsonStr) {
+              const jsonStr = trimmed.slice(5).trim();
-              try {
+              if (jsonStr && jsonStr !== '[DONE]') {
                 const data = JSON.parse(jsonStr);
                 if (data.event === 'message') {
                   const text = data.answer || data.text || '';
                   content += text;
-                  callback(text);
+                  if (onChunk) onChunk(text);
                 }
                 if (data.conversation_id) {
                   conversationId = data.conversation_id;
                 }
                 if (data.event === 'message_end') {
-                  // 可选：提前完成
+                  // 可提前结束
                 }
-              } catch (e) {
-                console.warn('解析流数据失败:', e);
               }
+            } catch (e) {
+              console.warn('解析失败:', e, line);
             }
           }
         }
-        return buf;
       }
-
+  
-      while (true) {
+      // 处理最后残留的 buffer（如果有）
-        const { done, value } = await reader.read();
+      if (buffer.trim().startsWith('data:')) {
-        if (done) break;
+        try {
-        buffer += decoder.decode(value, { stream: true });
+          const jsonStr = buffer.trim().slice(5);
-        buffer = processBuffer(buffer, (chunk) => {
+          if (jsonStr) {
-          if (onChunk) onChunk(chunk);
+            const data = JSON.parse(jsonStr);
-        });
+            if (data.event === 'message') {
+              const text = data.answer || '';
+              content += text;
+              if (onChunk) onChunk(text);
+            }
+            if (data.conversation_id) {
+              conversationId = data.conversation_id;
+            }
+          }
+        } catch (e) {
+          console.warn('最后 buffer 解析失败:', e);
+        }
       }
-
+  
       if (onComplete) {
-        onComplete({
+        onComplete({ content, conversation_id: conversationId });
-          content,
-          conversation_id: conversationId
-        });
       }
       return { content, conversation_id: conversationId };
     } catch (error) {
@@ -196,11 +259,6 @@ export default {
       throw error;
     }
     // #endif
-
-    // #ifdef MP-WEIXIN
-    // 理论上不会执行到这里，但防止 fallback
-    return this.sendStreamMessage(message, onChunk, onComplete);
-    // #endif
   },
 
   /**