paul-ai-assistant 1.0.5 → 1.0.7

package/es/hooks/useAutoSend.js

@@ -0,0 +1,95 @@
+import { useCallback, useRef, useEffect } from 'react';
+import { assistantChat } from "../service/conversation";
+import { getAssistantWsStatus, reconnectAssistantWs } from "../utils/apushAssistant";
+import { REQUEST_SUCCESS_CODE } from "../constants/common";
+/**
+ * 内置自动发送消息逻辑的 Hook
+ * 如果用户传入了 onSend，直接返回用户的 onSend
+ * 否则返回内置的发送逻辑
+ */
+export function useAutoSend(props) {
+  var bizId = props.bizId,
+    _props$bizDomain = props.bizDomain,
+    bizDomain = _props$bizDomain === void 0 ? 'CHAT' : _props$bizDomain,
+    userOnSend = props.onSend,
+    onSendSuccess = props.onSendSuccess,
+    onSendError = props.onSendError;
+  var pendingSendRef = useRef(null);
+
+  // 监听 WebSocket 连接状态，处理待发送消息
+  useEffect(function () {
+    if (pendingSendRef.current) {
+      var _getAssistantWsStatus;
+      var status = (getAssistantWsStatus === null || getAssistantWsStatus === void 0 || (_getAssistantWsStatus = getAssistantWsStatus()) === null || _getAssistantWsStatus === void 0 ? void 0 : _getAssistantWsStatus.status) || 'disconnected';
+      if (status === 'connected') {
+        var pendingSend = pendingSendRef.current;
+        pendingSendRef.current = null;
+        pendingSend();
+      }
+    }
+  }, [bizId]);
+
+  // 内置 onSend 逻辑
+  var internalOnSend = useCallback(function (data, onSuccessCallback) {
+    var _getAssistantWsStatus2;
+    var _ref = (data === null || data === void 0 ? void 0 : data.params) || {},
+      query = _ref.query;
+    if (!bizId) {
+      console.warn('useAutoSend: bizId is required for sending message');
+      onSendError === null || onSendError === void 0 || onSendError(new Error('bizId is required'));
+      return;
+    }
+    var doSend = function doSend() {
+      var _data$params;
+      assistantChat({
+        subjectId: String(bizId),
+        content: query,
+        featureConfig: (data === null || data === void 0 ? void 0 : data.featureConfig) || {},
+        attachments: (data === null || data === void 0 || (_data$params = data.params) === null || _data$params === void 0 ? void 0 : _data$params.attachments) || []
+      }).then(function (res) {
+        if ((res === null || res === void 0 ? void 0 : res.code) === REQUEST_SUCCESS_CODE) {
+          onSuccessCallback === null || onSuccessCallback === void 0 || onSuccessCallback(res);
+          onSendSuccess === null || onSendSuccess === void 0 || onSendSuccess(res);
+        } else {
+          onSendError === null || onSendError === void 0 || onSendError(res);
+        }
+      }).catch(function (error) {
+        console.error('useAutoSend: send message failed', error);
+        onSendError === null || onSendError === void 0 || onSendError(error);
+      });
+    };
+
+    // 检查 WebSocket 连接状态
+    var status = (getAssistantWsStatus === null || getAssistantWsStatus === void 0 || (_getAssistantWsStatus2 = getAssistantWsStatus()) === null || _getAssistantWsStatus2 === void 0 ? void 0 : _getAssistantWsStatus2.status) || 'disconnected';
+    if (status === 'connected') {
+      doSend();
+    } else {
+      // 保存待发送消息
+      pendingSendRef.current = doSend;
+
+      // 如果连接断开，触发重连
+      if (status === 'disconnected') {
+        reconnectAssistantWs === null || reconnectAssistantWs === void 0 || reconnectAssistantWs({
+          bizId: String(bizId),
+          bizDomain: bizDomain
+        });
+      }
+
+      // 设置超时，避免无限等待
+      setTimeout(function () {
+        if (pendingSendRef.current === doSend) {
+          pendingSendRef.current = null;
+          // 超时后直接尝试发送
+          doSend();
+        }
+      }, 5000);
+    }
+  }, [bizId, bizDomain, onSendSuccess, onSendError]);
+
+  // 如果用户传入了 onSend，直接返回用户的
+  if (userOnSend) {
+    return userOnSend;
+  }
+  return internalOnSend;
+}
+export default useAutoSend;

package/es/index.js

@@ -25,6 +25,9 @@ import { ASSIST_CHAT_EVENT } from "./constants/assistChatEvent";
 import { BIZ_DOMAIN_TYPE } from "./constants/common";
 import ErrorBoundary from "./ErrorBoundary";
 import { parseLlmComponentUrl } from "./utils/parseUrl";
+// 开箱即用相关导出
+import { getConversationConnection, assistantChat } from "./service/conversation";
+import { useAutoSend } from "./hooks/useAutoSend";
 import "./index.scss";
 
 // 导出大模型卡片组件
@@ -60,4 +63,8 @@ registerAssistChatForm,
 // 表单组件
 PaulButtonGroupProps, SelectBlurry, ArrayInput,
 // utils
-parseLlmComponentUrl, ShareContent };
+parseLlmComponentUrl, ShareContent,
+// 开箱即用相关
+getConversationConnection, assistantChat, useAutoSend };
+
+// 导出开箱即用相关类型

package/es/service/conversation.js

@@ -0,0 +1,74 @@
+import _regeneratorRuntime from "@babel/runtime/helpers/esm/regeneratorRuntime";
+import _asyncToGenerator from "@babel/runtime/helpers/esm/asyncToGenerator";
+import { customRequest } from "../utils/request";
+
+/**
+ * 会话连接信息
+ */
+
+/**
+ * 获取会话连接 API 响应
+ */
+
+/**
+ * 获取会话连接信息
+ * 用于自动创建或获取现有会话
+ * @param params 请求参数
+ * @returns Promise<IConversationConnectionResponse>
+ */
+export function getConversationConnection(_x) {
+  return _getConversationConnection.apply(this, arguments);
+}
+
+/**
+ * 发送聊天消息
+ * @param params 消息参数
+ * @returns Promise<any>
+ */
+function _getConversationConnection() {
+  _getConversationConnection = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(params) {
+    return _regeneratorRuntime().wrap(function _callee$(_context) {
+      while (1) switch (_context.prev = _context.next) {
+        case 0:
+          return _context.abrupt("return", customRequest({
+            method: 'GET',
+            url: '/cognipilot/api/conversation/connection',
+            query: {
+              bizDomain: params.bizDomain || 'CHAT',
+              subjectId: params.subjectId
+            }
+          }));
+        case 1:
+        case "end":
+          return _context.stop();
+      }
+    }, _callee);
+  }));
+  return _getConversationConnection.apply(this, arguments);
+}
+export function assistantChat(_x2) {
+  return _assistantChat.apply(this, arguments);
+}
+function _assistantChat() {
+  _assistantChat = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(params) {
+    return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+      while (1) switch (_context2.prev = _context2.next) {
+        case 0:
+          return _context2.abrupt("return", customRequest({
+            method: 'POST',
+            url: '/cognipilot/api/assistant/chat',
+            query: {
+              subjectId: params.subjectId,
+              content: params.content,
+              featureConfig: params.featureConfig || {},
+              attachments: params.attachments || []
+            }
+          }));
+        case 1:
+        case "end":
+          return _context2.stop();
+      }
+    }, _callee2);
+  }));
+  return _assistantChat.apply(this, arguments);
+}

package/es/service/index.js

@@ -0,0 +1 @@
+export { getConversationConnection, assistantChat } from "./conversation";

package/es/utils/apushAssistant.js

@@ -4,6 +4,7 @@ import { customRequest } from "./request";
 import { JsonParse } from "./formatData";
 import { AesCustomLogEventEnum } from "./log";
 import { logger } from "./websocketLogger";
+import IM from "./im";
 
 // 扩展 Window 接口，声明全局属性
 
@@ -50,29 +51,38 @@ var msgCallback = function msgCallback(v) {
     return;
   }
 
-  // 检查 ticketId 是否匹配
+  // 检查 ticketId 或 subjectId 是否匹配
+  // STREAM 消息使用 subjectId，其他消息使用 ticketId
   var ticketId = argJson === null || argJson === void 0 ? void 0 : argJson.ticketId;
-  var isMatch = ticketId === bizId;
+  var subjectId = argJson === null || argJson === void 0 ? void 0 : argJson.subjectId;
+  var isMatch = ticketId === bizId || subjectId === bizId;
   if (isMatch) {
     event.emit(msgType, argJson);
     logger.debug('消息已分发', {
       msgType: msgType,
-      ticketId: ticketId
+      ticketId: ticketId,
+      subjectId: subjectId
     });
   } else {
     // 消息被过滤，记录调试日志
-    logger.debug('消息 ticketId 不匹配，已过滤', {
+    logger.debug('消息 ticketId/subjectId 不匹配，已过滤', {
       expected: bizId,
-      received: ticketId,
+      receivedTicketId: ticketId,
+      receivedSubjectId: subjectId,
       msgType: msgType
     });
   }
 };
-var callback = function callback(msgBody, bizId) {
+var callback = function callback(message, bizId) {
   try {
+    // message 是完整的 WebSocket 消息对象，真正的消息体在 message.msgBody 中
+    var msgBody = (message === null || message === void 0 ? void 0 : message.msgBody) || message;
     var _ref3 = msgBody || {},
-      version = _ref3.version,
-      msgDetail = _ref3.msgDetail;
+      msgVersion = _ref3.msgVersion,
+      msgDetail = _ref3.msgDetail,
+      msgType = _ref3.msgType;
+    // 兼容旧版本的 version 字段
+    var version = msgVersion || (msgBody === null || msgBody === void 0 ? void 0 : msgBody.version);
     if (version === '1.0.0') {
       var _msgDetail$, _msgDetail$2;
       if (msgDetail !== null && msgDetail !== void 0 && (_msgDetail$ = msgDetail[0]) !== null && _msgDetail$ !== void 0 && _msgDetail$.arg && msgDetail !== null && msgDetail !== void 0 && (_msgDetail$2 = msgDetail[0]) !== null && _msgDetail$2 !== void 0 && _msgDetail$2.app) {
@@ -93,8 +103,8 @@ var callback = function callback(msgBody, bizId) {
     } else if (version === '2.0.0' || !version) {
       // version 不传默认为 2.0.0
       msgCallback({
-        msgType: msgBody === null || msgBody === void 0 ? void 0 : msgBody.msgType,
-        arg: msgBody === null || msgBody === void 0 ? void 0 : msgBody.msgDetail,
+        msgType: msgType,
+        arg: msgDetail,
         bizId: bizId
       });
     }
@@ -191,15 +201,16 @@ export var startAssistantWs = function startAssistantWs(wsProps) {
   // const url = window.location.host.includes('aliyun.com') ? Landlord.paulAiAliyunCom : 'paul-ai.aliyun.work';
 
   var serverDomain = function serverDomain() {
-    if (isThirdEmbed) {
-      return 'paul.aliyun.com';
-    }
-    if (window.location.host === 'pre.gts.work') {
-      return 'pre-paul-ai.aliyun.work';
-    } else if (window.location.host === 'gts.work') {
-      return 'paul-ai.aliyun.work';
-    }
-    return window.location.host;
+    return 'ai-space.work';
+    // if (isThirdEmbed) {
+    //   return 'paul.aliyun.com';
+    // }
+    // if (window.location.host === 'pre.gts.work') {
+    //   return 'pre-paul-ai.aliyun.work';
+    // } else if (window.location.host === 'gts.work') {
+    //   return 'paul-ai.aliyun.work';
+    // }
+    // return window.location.host;
   };
   var url = "wss://".concat(serverDomain(), "/paulWebPush/ws");
   var webPushTokenUrl = '/sso/webpush/getToken';
@@ -239,11 +250,11 @@ export var startAssistantWs = function startAssistantWs(wsProps) {
       });
     }
   };
+  console.log(appId, userId, '=====');
   var im = new IM(_objectSpread({
     url: url,
     appKey: "".concat(appId, ",").concat(userId),
     // appKey: `${appId},1000`,
-
     commonParams: {
       appId: appId,
       userId: userId

package/es/utils/im.js

@@ -0,0 +1,395 @@
+import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
+import _regeneratorRuntime from "@babel/runtime/helpers/esm/regeneratorRuntime";
+import _asyncToGenerator from "@babel/runtime/helpers/esm/asyncToGenerator";
+import _classCallCheck from "@babel/runtime/helpers/esm/classCallCheck";
+import _createClass from "@babel/runtime/helpers/esm/createClass";
+import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
+/**
+ * @file WebSocket IM 客户端
+ * 本地实现，替代 @alife/magic-chat-im
+ */
+
+import { logger } from "./websocketLogger";
+
+/**
+ * IM 配置选项
+ */
+
+/**
+ * 事件类型
+ */
+
+/**
+ * 事件回调类型
+ */
+
+/**
+ * 重连配置
+ */
+var RECONNECT_INTERVAL_BASE = 1000; // 基础重连间隔 1 秒
+var RECONNECT_INTERVAL_MAX = 30000; // 最大重连间隔 30 秒
+var HEARTBEAT_INTERVAL = 30000; // 心跳间隔 30 秒
+
+/**
+ * IM WebSocket 客户端
+ */
+var IM = /*#__PURE__*/function () {
+  function IM(options) {
+    _classCallCheck(this, IM);
+    _defineProperty(this, "url", void 0);
+    _defineProperty(this, "appKey", void 0);
+    _defineProperty(this, "commonParams", void 0);
+    _defineProperty(this, "connectCallback", void 0);
+    _defineProperty(this, "token", void 0);
+    _defineProperty(this, "getToken", void 0);
+    _defineProperty(this, "ws", null);
+    _defineProperty(this, "eventListeners", new Map());
+    _defineProperty(this, "reconnectAttempts", 0);
+    _defineProperty(this, "reconnectTimer", null);
+    _defineProperty(this, "heartbeatTimer", null);
+    _defineProperty(this, "isManualClose", false);
+    _defineProperty(this, "isConnecting", false);
+    this.url = options.url;
+    this.appKey = options.appKey;
+    this.commonParams = options.commonParams;
+    this.connectCallback = options.connectCallback;
+    this.token = options.token;
+    this.getToken = options.getToken;
+  }
+
+  /**
+   * 注册事件监听器
+   */
+  _createClass(IM, [{
+    key: "on",
+    value: function on(event, callback) {
+      if (!this.eventListeners.has(event)) {
+        this.eventListeners.set(event, []);
+      }
+      this.eventListeners.get(event).push(callback);
+    }
+
+    /**
+     * 触发事件
+     */
+  }, {
+    key: "emit",
+    value: function emit(event, data) {
+      var listeners = this.eventListeners.get(event);
+      if (listeners) {
+        listeners.forEach(function (callback) {
+          try {
+            callback(data);
+          } catch (e) {
+            logger.error("\u4E8B\u4EF6 ".concat(event, " \u56DE\u8C03\u6267\u884C\u5931\u8D25"), e);
+          }
+        });
+      }
+    }
+
+    /**
+     * 建立连接
+     */
+  }, {
+    key: "connect",
+    value: (function () {
+      var _connect = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
+        var _this = this;
+        var token, wsUrl;
+        return _regeneratorRuntime().wrap(function _callee$(_context) {
+          while (1) switch (_context.prev = _context.next) {
+            case 0:
+              if (!this.isConnecting) {
+                _context.next = 3;
+                break;
+              }
+              logger.warn('正在连接中，跳过重复连接请求');
+              return _context.abrupt("return");
+            case 3:
+              this.isConnecting = true;
+              this.isManualClose = false;
+              _context.prev = 5;
+              // 获取 token
+              token = this.token;
+              if (!(!token && this.getToken)) {
+                _context.next = 11;
+                break;
+              }
+              _context.next = 10;
+              return this.getToken();
+            case 10:
+              token = _context.sent;
+            case 11:
+              // 构建 WebSocket URL
+              wsUrl = this.buildUrl(token);
+              logger.info('开始建立 WebSocket 连接', {
+                url: this.url
+              });
+
+              // 创建 WebSocket 连接
+              this.ws = new WebSocket(wsUrl);
+              this.ws.onopen = function () {
+                var _this$connectCallback;
+                logger.info('WebSocket 连接已建立');
+                _this.isConnecting = false;
+                _this.reconnectAttempts = 0;
+
+                // 发送鉴权消息
+                _this.sendAuth(token);
+
+                // 启动心跳
+                _this.startHeartbeat();
+
+                // 触发连接回调
+                (_this$connectCallback = _this.connectCallback) === null || _this$connectCallback === void 0 || _this$connectCallback.call(_this);
+              };
+              this.ws.onmessage = function (event) {
+                try {
+                  var _data = JSON.parse(event.data);
+
+                  // 处理心跳响应
+                  if (_data.type === 'pong' || _data.msgType === 'pong') {
+                    logger.debug('收到心跳响应');
+                    return;
+                  }
+
+                  // 触发消息事件
+                  _this.emit('message', _data);
+                } catch (e) {
+                  logger.error('消息解析失败', e);
+                }
+              };
+              this.ws.onclose = function (event) {
+                logger.info('WebSocket 连接已关闭', {
+                  code: event.code,
+                  reason: event.reason,
+                  wasClean: event.wasClean
+                });
+                _this.isConnecting = false;
+                _this.stopHeartbeat();
+                var closeInfo = {
+                  code: event.wasClean ? 'CONNECTION_CLOSED_CLEANLY' : event.code,
+                  reason: event.reason
+                };
+                _this.emit('close', closeInfo);
+
+                // 非手动关闭时尝试重连
+                if (!_this.isManualClose) {
+                  _this.scheduleReconnect();
+                }
+              };
+              this.ws.onerror = function (error) {
+                logger.error('WebSocket 连接错误', error);
+                _this.isConnecting = false;
+                _this.emit('error', error);
+              };
+              _context.next = 26;
+              break;
+            case 20:
+              _context.prev = 20;
+              _context.t0 = _context["catch"](5);
+              this.isConnecting = false;
+              logger.error('建立连接失败', _context.t0);
+              this.emit('error', _context.t0);
+              this.scheduleReconnect();
+            case 26:
+            case "end":
+              return _context.stop();
+          }
+        }, _callee, this, [[5, 20]]);
+      }));
+      function connect() {
+        return _connect.apply(this, arguments);
+      }
+      return connect;
+    }()
+    /**
+     * 解析 JWT token 的 payload
+     */
+    )
+  }, {
+    key: "parseJwtPayload",
+    value: function parseJwtPayload(token) {
+      try {
+        var parts = token.split('.');
+        if (parts.length !== 3) {
+          return null;
+        }
+        // Base64Url 解码
+        var payload = parts[1].replace(/-/g, '+').replace(/_/g, '/');
+        var decoded = atob(payload);
+        return JSON.parse(decoded);
+      } catch (e) {
+        logger.error('JWT 解析失败', e);
+        return null;
+      }
+    }
+
+    /**
+     * 构建 WebSocket URL
+     * 格式：wss://domain/paulWebPush/ws?param={bizDomain},{userId},{token}
+     * 其中 userId 从 token 的 payload 中获取
+     */
+  }, {
+    key: "buildUrl",
+    value: function buildUrl(token) {
+      var _this$commonParams;
+      var bizDomain = ((_this$commonParams = this.commonParams) === null || _this$commonParams === void 0 ? void 0 : _this$commonParams.appId) || '';
+
+      // 从 token 中解析 userId
+      var userId = '';
+      if (token) {
+        var payload = this.parseJwtPayload(token);
+        userId = (payload === null || payload === void 0 ? void 0 : payload.userId) || '';
+      }
+      var param = "".concat(bizDomain, ",").concat(userId, ",").concat(token || '');
+      return "".concat(this.url, "?param=").concat(param);
+    }
+
+    /**
+     * 发送鉴权消息
+     */
+  }, {
+    key: "sendAuth",
+    value: function sendAuth(token) {
+      if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+        return;
+      }
+      var authMessage = _objectSpread({
+        type: 'auth',
+        appKey: this.appKey,
+        token: token
+      }, this.commonParams);
+      this.ws.send(JSON.stringify(authMessage));
+      logger.debug('已发送鉴权消息');
+    }
+
+    /**
+     * 启动心跳
+     */
+  }, {
+    key: "startHeartbeat",
+    value: function startHeartbeat() {
+      var _this2 = this;
+      this.stopHeartbeat();
+      this.heartbeatTimer = setInterval(function () {
+        if (_this2.ws && _this2.ws.readyState === WebSocket.OPEN) {
+          _this2.ws.send(JSON.stringify({
+            type: 'ping'
+          }));
+          logger.debug('发送心跳');
+        }
+      }, HEARTBEAT_INTERVAL);
+    }
+
+    /**
+     * 停止心跳
+     */
+  }, {
+    key: "stopHeartbeat",
+    value: function stopHeartbeat() {
+      if (this.heartbeatTimer) {
+        clearInterval(this.heartbeatTimer);
+        this.heartbeatTimer = null;
+      }
+    }
+
+    /**
+     * 安排重连
+     */
+  }, {
+    key: "scheduleReconnect",
+    value: function scheduleReconnect() {
+      var _this3 = this;
+      if (this.isManualClose) {
+        return;
+      }
+
+      // 计算重连间隔（指数退避）
+      var interval = Math.min(RECONNECT_INTERVAL_BASE * Math.pow(2, this.reconnectAttempts), RECONNECT_INTERVAL_MAX);
+      this.reconnectAttempts++;
+      logger.info("\u5C06\u5728 ".concat(interval, "ms \u540E\u5C1D\u8BD5\u7B2C ").concat(this.reconnectAttempts, " \u6B21\u91CD\u8FDE"));
+
+      // 触发重连事件
+      this.emit('reconnect', {
+        code: 'RECONNECTING',
+        err: null,
+        attempt: this.reconnectAttempts
+      });
+      this.reconnectTimer = setTimeout(function () {
+        _this3.connect();
+      }, interval);
+    }
+
+    /**
+     * 关闭连接
+     */
+  }, {
+    key: "close",
+    value: function close() {
+      this.isManualClose = true;
+
+      // 清除重连定时器
+      if (this.reconnectTimer) {
+        clearTimeout(this.reconnectTimer);
+        this.reconnectTimer = null;
+      }
+
+      // 停止心跳
+      this.stopHeartbeat();
+
+      // 关闭 WebSocket
+      if (this.ws) {
+        try {
+          this.ws.close(1000, 'Manual close');
+        } catch (e) {
+          logger.error('关闭 WebSocket 失败', e);
+        }
+        this.ws = null;
+      }
+      logger.info('WebSocket 连接已手动关闭');
+    }
+
+    /**
+     * 发送消息
+     */
+  }, {
+    key: "send",
+    value: function send(data) {
+      if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+        logger.warn('WebSocket 未连接，无法发送消息');
+        return false;
+      }
+      try {
+        var message = typeof data === 'string' ? data : JSON.stringify(data);
+        this.ws.send(message);
+        return true;
+      } catch (e) {
+        logger.error('发送消息失败', e);
+        return false;
+      }
+    }
+
+    /**
+     * 获取连接状态
+     */
+  }, {
+    key: "getReadyState",
+    value: function getReadyState() {
+      var _this$ws$readyState, _this$ws;
+      return (_this$ws$readyState = (_this$ws = this.ws) === null || _this$ws === void 0 ? void 0 : _this$ws.readyState) !== null && _this$ws$readyState !== void 0 ? _this$ws$readyState : WebSocket.CLOSED;
+    }
+
+    /**
+     * 是否已连接
+     */
+  }, {
+    key: "isConnected",
+    value: function isConnected() {
+      var _this$ws2;
+      return ((_this$ws2 = this.ws) === null || _this$ws2 === void 0 ? void 0 : _this$ws2.readyState) === WebSocket.OPEN;
+    }
+  }]);
+  return IM;
+}();
+export default IM;