@flexem/chat-box 1.0.0

package/miniprogram_dist/utils/api.js

@@ -0,0 +1,474 @@
+/**
+ * API 封装模块
+ * 与 AiChatWeb 后端接口对接
+ *
+ * 配置说明:
+ * - idsServiceUrl: IDS 认证服务地址，用于获取 serviceUrls
+ * - aiChatUrl: 主要 AI 聊天 API 地址（从 serviceUrls 获取）
+ * - ossServiceUrl: OSS/七牛云服务地址（从 serviceUrls 获取）
+ * - blobStorageDownloadServiceUrl: 文件下载服务地址（从 serviceUrls 获取）
+ */
+
+// 缓存的服务配置
+let cachedServiceUrls = null;
+
+// 生成唯一消息ID（GUID格式）
+function generateMsgId() {
+  return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, function(c) {
+    const r = Math.random() * 16 | 0;
+    const v = c === 'x' ? r : (r & 0x3 | 0x8);
+    return v.toString(16);
+  });
+}
+
+// 请求封装
+function request(options) {
+  const { url, method = 'GET', data, header = {}, token } = options;
+
+  return new Promise((resolve, reject) => {
+    wx.request({
+      url,
+      method,
+      data,
+      header: {
+        'Content-Type': 'application/json',
+        'X-Access-Token': token ? `Bearer ${token}` : '',
+        ...header
+      },
+      success: (res) => {
+        if (res.statusCode === 401) {
+          reject({ code: 401, message: '请登录后使用' });
+          return;
+        }
+        if (res.statusCode >= 200 && res.statusCode < 300) {
+          // 处理新旧API响应格式
+          const result = res.data;
+          if (result && typeof result === 'object' && 'code' in result) {
+            if (result.code === 0 || result.code === 200) {
+              resolve(result.data);
+            } else if (result.code === 401) {
+              reject({ code: 401, message: '请登录后使用' });
+            } else {
+              reject({ code: result.code, message: result.message || '请求失败' });
+            }
+          } else {
+            resolve(result);
+          }
+        } else {
+          reject({ code: res.statusCode, message: res.data?.message || '请求失败' });
+        }
+      },
+      fail: (err) => {
+        reject({ code: -1, message: err.errMsg || '网络错误' });
+      }
+    });
+  });
+}
+
+/**
+ * 获取对话列表
+ */
+function getChatSessions(baseUrl, token, page = 1, pageSize = 20) {
+  return request({
+    url: `${baseUrl}/app/sessions/paging?current_page=${page}&page_size=${pageSize}`,
+    method: 'GET',
+    token
+  });
+}
+
+/**
+ * 创建新对话
+ */
+function createChatSession(baseUrl, token, title = '新对话', configs = {}) {
+  return request({
+    url: `${baseUrl}/app/sessions`,
+    method: 'POST',
+    data: {
+      title: title.substring(0, 30),
+      configs: {
+        defaultDeepThinking: configs.deepThinking || false,
+        defaultWebSearch: configs.webSearch || false
+      }
+    },
+    token
+  });
+}
+
+/**
+ * 重命名对话
+ */
+function renameChatSession(baseUrl, token, sessionId, newTitle) {
+  return request({
+    url: `${baseUrl}/app/sessions/modify-title`,
+    method: 'POST',
+    data: {
+      id: sessionId,
+      title: newTitle.substring(0, 30)
+    },
+    token
+  });
+}
+
+/**
+ * 删除对话
+ */
+function deleteChatSession(baseUrl, token, sessionId, userId) {
+  return request({
+    url: `${baseUrl}/app/sessions/${sessionId}`,
+    method: 'DELETE',
+    data: {
+      session_id: sessionId,
+      current_user_id: userId
+    },
+    token
+  });
+}
+
+/**
+ * 获取聊天历史记录
+ */
+function getChatHistories(baseUrl, token, sessionId, page = 1, pageSize = 20) {
+  return request({
+    url: `${baseUrl}/app/chat-histories/mine?session_id=${sessionId}&current_page=${page}&page_size=${pageSize}`,
+    method: 'GET',
+    token
+  });
+}
+
+/**
+ * 删除指定时间后的聊天记录（用于重新生成）
+ */
+function deleteChatHistoriesByTime(baseUrl, token, sessionId, chatHistoryTime) {
+  return request({
+    url: `${baseUrl}/app/chat-histories/delete_by_time`,
+    method: 'POST',
+    data: {
+      session_id: sessionId,
+      chat_history_time: chatHistoryTime
+    },
+    token
+  });
+}
+
+/**
+ * 发送聊天消息（SSE 流式传输）
+ * 微信小程序不支持原生SSE，使用 requestTask 模拟
+ */
+function sendChatMessage(options) {
+  const {
+    baseUrl,
+    token,
+    sessionId,
+    messages,
+    settings = {},
+    onMessage,
+    onComplete,
+    onError
+  } = options;
+
+  const msgId = generateMsgId();
+
+  // 构建请求数据
+  const requestData = {
+    messages: messages.map(msg => {
+      const messageObj = {
+        content: msg.content,
+        role: msg.role
+      };
+      // 只有当有附件时才添加 attachments 字段
+      if (msg.attachments && msg.attachments.length > 0) {
+        messageObj.attachments = msg.attachments;
+      }
+      return messageObj;
+    }),
+    session_id: sessionId,
+    msg_id: msgId,
+    chat_setting: {
+      deep_thinking: settings.deepThinking || false,
+      online_search: settings.webSearch || false
+    }
+  };
+
+  let fullContent = '';
+  let thinkingText = '';
+  let hasError = false;  // 标记是否已发生错误，防止 onComplete 被调用
+
+  const requestTask = wx.request({
+    url: `${baseUrl}/app/api/chatstream`,
+    method: 'POST',
+    data: requestData,
+    header: {
+      'Content-Type': 'application/json',
+      'Accept': 'text/event-stream',
+      'Cache-Control': 'no-cache',
+      'X-Access-Token': `Bearer ${token}`
+    },
+    enableChunked: true,
+    success: () => {
+      // 如果已经发生错误，不再调用 onComplete
+      if (hasError) {
+        return;
+      }
+      if (onComplete) {
+        onComplete({
+          content: fullContent,
+          thinkingText: thinkingText,
+          msgId: msgId
+        });
+      }
+    },
+    fail: (err) => {
+      // 检查是否是用户主动中止的请求
+      const isAborted = err.errMsg && (
+        err.errMsg.includes('abort') ||
+        err.errMsg.includes('cancel')
+      );
+
+      if (onError && !isAborted) {
+        hasError = true;
+        onError({ code: -1, message: err.errMsg || '发送失败' });
+      }
+    }
+  });
+
+  // 监听数据块
+  requestTask.onChunkReceived((res) => {
+    try {
+      const text = arrayBufferToString(res.data);
+      const lines = text.split('\n');
+
+      lines.forEach(line => {
+        if (line.startsWith('data: ')) {
+          const data = line.substring(6);
+          if (data === '[DONE]') {
+            return;
+          }
+          try {
+            const json = JSON.parse(data);
+            if (json.error) {
+              hasError = true;  // 标记错误，防止 onComplete 被调用
+              if (onError) {
+                onError({ code: -1, message: json.error });
+              }
+            } else if (json.ai) {
+              fullContent += json.ai;
+              if (onMessage) {
+                onMessage({
+                  content: fullContent,
+                  delta: json.ai,
+                  isStreaming: true
+                });
+              }
+            } else if (json.thinking) {
+              thinkingText += json.thinking;
+            }
+          } catch (e) {
+            // 忽略解析错误
+          }
+        }
+      });
+    } catch (e) {
+      console.error('解析响应数据失败', e);
+    }
+  });
+
+  return {
+    abort: () => requestTask.abort(),
+    msgId: msgId
+  };
+}
+
+/**
+ * ArrayBuffer 转字符串
+ */
+function arrayBufferToString(buffer) {
+  const uint8Array = new Uint8Array(buffer);
+  let str = '';
+  for (let i = 0; i < uint8Array.length; i++) {
+    str += String.fromCharCode(uint8Array[i]);
+  }
+  try {
+    return decodeURIComponent(escape(str));
+  } catch (e) {
+    return str;
+  }
+}
+
+/**
+ * 获取服务 URLs 配置
+ * @param {string} idsServiceUrl - IDS 服务地址
+ * @param {string} token - 用户 Token
+ * @returns {Promise<Object>} - 包含 aiChatUrl, ossServiceUrl, blobStorageDownloadServiceUrl 等
+ */
+async function getServiceUrls(idsServiceUrl, token) {
+  if (cachedServiceUrls) {
+    return cachedServiceUrls;
+  }
+
+  const result = await request({
+    url: `${idsServiceUrl}/api/aiChatWeb/serviceUrls`,
+    method: 'GET',
+    token
+  });
+
+  cachedServiceUrls = result;
+  return result;
+}
+
+/**
+ * 清除服务配置缓存
+ */
+function clearServiceUrlsCache() {
+  cachedServiceUrls = null;
+}
+
+/**
+ * 获取七牛上传 Token
+ * @param {string} ossServiceUrl - OSS 服务地址
+ * @param {string} token - 用户 Token
+ * @returns {Promise<string>} - 七牛上传 token 字符串
+ */
+function getQiniuToken(ossServiceUrl, token) {
+  return new Promise((resolve, reject) => {
+    wx.request({
+      url: `${ossServiceUrl}/api/app/externalStorage/tickets`,
+      method: 'POST',
+      header: {
+        'Content-Type': 'application/json',
+        'X-Access-Token': token ? `Bearer ${token}` : ''
+      },
+      success: (res) => {
+        if (res.statusCode === 401) {
+          reject({ code: 401, message: '请登录后使用' });
+          return;
+        }
+        if (res.statusCode >= 200 && res.statusCode < 300) {
+          // API 直接返回 token 字符串
+          resolve(res.data);
+        } else {
+          reject({ code: res.statusCode, message: res.data?.message || '获取上传凭证失败' });
+        }
+      },
+      fail: (err) => {
+        reject({ code: -1, message: err.errMsg || '网络错误' });
+      }
+    });
+  });
+}
+
+/**
+ * 上传文件到七牛云
+ */
+function uploadFile(options) {
+  const { filePath, uploadToken, key, onProgress, onSuccess, onError } = options;
+
+  const uploadTask = wx.uploadFile({
+    url: 'https://up-z2.qiniup.com',
+    filePath: filePath,
+    name: 'file',
+    formData: {
+      token: uploadToken,
+      key: key
+    },
+    success: (res) => {
+      if (res.statusCode === 200) {
+        try {
+          const data = JSON.parse(res.data);
+          if (onSuccess) {
+            onSuccess(data);
+          }
+        } catch (e) {
+          if (onError) {
+            onError({ message: '解析上传结果失败' });
+          }
+        }
+      } else {
+        if (onError) {
+          onError({ message: '上传失败' });
+        }
+      }
+    },
+    fail: (err) => {
+      if (onError) {
+        onError(err);
+      }
+    }
+  });
+
+  uploadTask.onProgressUpdate((res) => {
+    if (onProgress) {
+      onProgress(res.progress);
+    }
+  });
+
+  return uploadTask;
+}
+
+/**
+ * 解析文件内容
+ */
+function parseFileContent(baseUrl, token, fileUrl) {
+  return request({
+    url: `${baseUrl}/app/tempFile/parse`,
+    method: 'POST',
+    data: { url: fileUrl },
+    token
+  });
+}
+
+/**
+ * 获取聊天设置
+ */
+function getChatSetting(baseUrl, token) {
+  return request({
+    url: `${baseUrl}/app/chat-setting`,
+    method: 'GET',
+    token
+  });
+}
+
+/**
+ * 自动生成会话标题
+ */
+function updateSessionTitle(baseUrl, token, msgId, userQuestion, aiAnswer) {
+  return request({
+    url: `${baseUrl}/app/sessions/title`,
+    method: 'POST',
+    data: {
+      id: msgId,
+      user_question: userQuestion,
+      ai_answer: aiAnswer
+    },
+    token
+  });
+}
+
+/**
+ * 搜索对话和聊天记录
+ */
+function searchChats(baseUrl, token, keyword) {
+  return request({
+    url: `${baseUrl}/app/chat-session/search?content=${encodeURIComponent(keyword)}`,
+    method: 'GET',
+    token
+  });
+}
+
+module.exports = {
+  generateMsgId,
+  getServiceUrls,
+  clearServiceUrlsCache,
+  getChatSessions,
+  createChatSession,
+  renameChatSession,
+  deleteChatSession,
+  getChatHistories,
+  deleteChatHistoriesByTime,
+  sendChatMessage,
+  getQiniuToken,
+  uploadFile,
+  parseFileContent,
+  getChatSetting,
+  updateSessionTitle,
+  searchChats
+};