@zero-library/chat-agent 2.1.10 → 2.1.11

package/dist/index.cjs.js

@@ -5,10 +5,10 @@ var antd = require('antd');
 var dayjs = require('dayjs');
 var jsxRuntime = require('react/jsx-runtime');
 var react = require('react');
+var valtio = require('valtio');
 var icons = require('@ant-design/icons');
 var x = require('@ant-design/x');
 var classNames8 = require('classnames');
-var valtio = require('valtio');
 var InfiniteScroll = require('react-infinite-scroll-component');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
@@ -26,19 +26,6 @@ var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
 };
-var docQuery, fileCreate;
-var init_file = __esm({
-  "src/services/file.ts"() {
-    docQuery = (paramsStr) => {
-      return common.request.get(`/uc/doc?${paramsStr}`);
-    };
-    fileCreate = async (fileContent, fileName, targetFormat = "docx") => {
-      const blob = await common.request.post(`/uc/doc/convert`, { fileContent, fileName, targetFormat }, { responseType: "blob" });
-      const url = URL.createObjectURL(blob);
-      common.downloadFile(url, fileName);
-    };
-  }
-});
 function transform(source, fieldMap) {
   const result = {};
   for (const [targetKey, mapping] of Object.entries(fieldMap)) {
@@ -154,14 +141,16 @@ var init_FileView = __esm({
 var DocDrawer_default;
 var init_DocDrawer = __esm({
   "src/ui/common/markdownAlert/components/DocDrawer.tsx"() {
-    init_file();
+    init_Context();
     DocDrawer_default = ({ title, open, onClose, fileUrl }) => {
+      const chatStore = useChatStore();
+      const configState = valtio.useSnapshot(chatStore.config);
       const [content, setContent] = react.useState("");
       const [loading, setLoading] = react.useState(false);
       const getContent = async () => {
         try {
           setLoading(true);
-          const res = await docQuery(fileUrl);
+          const res = await configState.services.request.docQuery(fileUrl);
           setContent(res.data?.content);
         } finally {
           setLoading(false);
@@ -308,52 +297,95 @@ var init_QuoteList = __esm({
   }
 });
 
-// src/ui/layouts/index.tsx
-init_file();
-
 // src/stores/index.ts
 init_utils();
-var conversationsQuery = (params) => {
-  return common.request.get("/lolr/conversation", params);
-};
-var conversationCreate = (params) => {
-  return common.request.post("/lolr/conversation", params);
-};
-var conversationDelete = (conversationId) => {
-  return common.request.delete("/lolr/conversation", { conversationId });
-};
-var conversationRecentQuery = (params) => {
-  return common.request.get("/lolr/conversation/recent", params);
-};
-var conversationMessagesQuery = (params) => {
-  return common.request.get("/lolr/conversation/message", params);
-};
-var feedbackUpdate = (params) => {
-  return common.request.put("/lolr/conversation/message/feedback", params);
-};
-var messageSuggestedQuery = (msgId, maxCount = 3) => {
-  return common.request.get("/lolr/conversation/suggest", { msgId, maxCount });
-};
-var conversationMemberQuery = (conversationId) => {
-  return common.request.get("/lolr/conversation/member", { conversationId });
-};
-var conversationMessageSend = (params) => {
-  return common.request.post("/lolr/conversation/send", params);
-};
-var conversationStop = (conversationId) => {
-  return common.request.post("/lolr/conversation/stop", { conversationId });
-};
-var agentInfoQuery = (agentId) => {
-  return common.request.get("/lolr/agent/detail", { agentId });
-};
-var agentCharacterQuery = (agentId) => {
-  return common.request.get("/lolr/character", { agentId });
-};
-var agentCharacterSelect = (agentId, characterId) => {
-  return common.request.post("/lolr/character/user", { agentId, characterId });
+
+// src/services/index.ts
+var createChatService = (request) => {
+  const conversationsQuery = (params) => {
+    return request.get("/lolr/conversation", params);
+  };
+  const conversationCreate = (params) => {
+    return request.post("/lolr/conversation", params);
+  };
+  const conversationDelete = (conversationId) => {
+    return request.delete("/lolr/conversation", { conversationId });
+  };
+  const conversationRecentUpdate = (params) => {
+    return request.put("/lolr/conversation/recent", params);
+  };
+  const conversationRecentQuery = (params) => {
+    return request.get("/lolr/conversation/recent", params);
+  };
+  const conversationMessagesQuery = (params) => {
+    return request.get("/lolr/conversation/message", params);
+  };
+  const conversationMessageRead = (conversationId) => {
+    return request.put("/lolr/conversation/message/read", { conversationId });
+  };
+  const feedbackUpdate = (params) => {
+    return request.put("/lolr/conversation/message/feedback", params);
+  };
+  const messageSuggestedQuery = (msgId, maxCount = 3) => {
+    return request.get("/lolr/conversation/suggest", { msgId, maxCount });
+  };
+  const conversationMemberQuery = (conversationId) => {
+    return request.get("/lolr/conversation/member", { conversationId });
+  };
+  const conversationStateUpdate = (conversationId, state) => {
+    return request.put("/lolr/conversation/state", { conversationId, state });
+  };
+  const conversationMessageSend = (params) => {
+    return request.post("/lolr/conversation/send", params);
+  };
+  const conversationStop = (conversationId) => {
+    return request.post("/lolr/conversation/stop", { conversationId });
+  };
+  const agentInfoQuery = (agentId) => {
+    return request.get("/lolr/agent/detail", { agentId });
+  };
+  const agentCharacterQuery = (agentId) => {
+    return request.get("/lolr/character", { agentId });
+  };
+  const agentCharacterSelect = (agentId, characterId) => {
+    return request.post("/lolr/character/user", { agentId, characterId });
+  };
+  const fileUpload = (formData) => {
+    return request.post("/lolr/storage/upload", formData, { timeout: 5 * 60 * 1e3 });
+  };
+  return {
+    conversationsQuery,
+    conversationCreate,
+    conversationDelete,
+    conversationRecentUpdate,
+    conversationRecentQuery,
+    conversationMessagesQuery,
+    conversationMessageRead,
+    feedbackUpdate,
+    messageSuggestedQuery,
+    conversationMemberQuery,
+    conversationStateUpdate,
+    conversationMessageSend,
+    conversationStop,
+    agentInfoQuery,
+    agentCharacterQuery,
+    agentCharacterSelect,
+    fileUpload
+  };
 };
-var fileUpload = (formData) => {
-  return common.request.post("/lolr/storage/upload", formData, { timeout: 5 * 60 * 1e3 });
+var createFileService = (request) => {
+  const docQuery = (paramsStr) => {
+    return request.get(`/uc/doc?${paramsStr}`);
+  };
+  const fileCreate = async (fileContent, fileName, targetFormat = "docx") => {
+    const blob = await request.post(`/uc/doc/convert`, { fileContent, fileName, targetFormat }, { responseType: "blob" });
+    const url = URL.createObjectURL(blob);
+    common.downloadFile(url, fileName);
+  };
+  return {
+    docQuery,
+    fileCreate
+  };
 };
 
 // src/stores/index.ts
@@ -397,6 +429,9 @@ var defaultExpertLayout = {
 };
 function createChatStore() {
   const config = valtio.proxy({
+    services: {
+      websocketBaseUrls: []
+    },
     hooks: {},
     layout: {},
     preview: {
@@ -409,6 +444,14 @@ function createChatStore() {
     userInfo: {},
     params: {}
   });
+  const setServices = ({ baseUrl = "/api" } = {}) => {
+    config.services.baseUrl = baseUrl;
+    if (!config.services.websocketBaseUrls?.includes(baseUrl)) {
+      config.services.websocketBaseUrls = [...config.services.websocketBaseUrls, baseUrl];
+    }
+    const request = common.createRequest(baseUrl);
+    config.services.request = { ...createChatService(request), ...createFileService(request) };
+  };
   const setPreview = (file = {}, isEdit = false) => {
     if (common.shouldRender(config.layout.preview)) {
       config.preview = {
@@ -464,7 +507,7 @@ function createChatStore() {
     }
     try {
       receiver.loading = true;
-      const { data } = await agentInfoQuery(agentId);
+      const { data } = await config.services.request.agentInfoQuery(agentId);
       receiver.active = transform(data, {
         name: "agentName",
         logo: "logo",
@@ -515,7 +558,7 @@ function createChatStore() {
   const getCharacters = async (agentId) => {
     try {
       character.loading = true;
-      const { data } = await agentCharacterQuery(agentId);
+      const { data } = await config.services.request.agentCharacterQuery(agentId);
       if (receiver.active.id !== agentId) return;
       character.list = data;
       const active = data.find((item) => item.selected);
@@ -536,7 +579,7 @@ function createChatStore() {
     if (!c.id) return;
     try {
       character.switchLoading = true;
-      await agentCharacterSelect(receiver.active.id, c.id);
+      await config.services.request.agentCharacterSelect(receiver.active.id, c.id);
       character.active = { ...c };
       closeCharacterList();
     } finally {
@@ -559,7 +602,7 @@ function createChatStore() {
   const getConversations = async (targetId, targetType) => {
     try {
       conversations.loading = true;
-      const { data } = await conversationsQuery({
+      const { data } = await config.services.request.conversationsQuery({
         targetId,
         targetType,
         businessId: config.params.businessId,
@@ -586,7 +629,7 @@ function createChatStore() {
       const canProceed = await config.hooks?.onBeforeDelConversation?.(conversationId);
       if (canProceed === false) return;
       conversations.delLoading = true;
-      await conversationDelete(conversationId);
+      await config.services.request.conversationDelete(conversationId);
       antd.message.success("\u5220\u9664\u6210\u529F");
       conversations.list.items = conversations.list.items.filter((item) => item.key !== conversationId);
       delete conversation.messages[conversationId];
@@ -637,7 +680,7 @@ function createChatStore() {
     if (!messages?.length) return;
     let lastMessage = messages[messages.length - 1];
     if (!lastMessage || lastMessage.sender.type !== 3 || lastMessage.id !== messageId) return;
-    const { data } = await messageSuggestedQuery(messageId);
+    const { data } = await config.services.request.messageSuggestedQuery(messageId);
     lastMessage = messages[messages.length - 1];
     if (lastMessage.id !== messageId) return;
     conversation.messages[conversationId].questionList = data;
@@ -645,7 +688,7 @@ function createChatStore() {
   const feedback = async (conversationId, messageId, msgFeedback, index) => {
     try {
       conversation.feedback.loading = true;
-      await feedbackUpdate({ conversationId, id: messageId, msgFeedback });
+      await config.services.request.feedbackUpdate({ conversationId, id: messageId, msgFeedback });
       antd.message.success("\u611F\u8C22\u60A8\u7684\u53CD\u9988");
       conversation.messages[conversationId].message[index].msgFeedback = msgFeedback;
     } finally {
@@ -666,7 +709,7 @@ function createChatStore() {
   };
   const resolveConversationId = async (receiverId, strategy = 2) => {
     if (strategy === 1) {
-      const { data } = await conversationRecentQuery({
+      const { data } = await config.services.request.conversationRecentQuery({
         receiverId,
         receiverType: 3,
         businessId: config.params.businessId,
@@ -674,7 +717,7 @@ function createChatStore() {
       });
       return data;
     } else {
-      const { data } = await conversationCreate({
+      const { data } = await config.services.request.conversationCreate({
         receiverId,
         receiverType: 3,
         businessId: config.params.businessId,
@@ -689,7 +732,7 @@ function createChatStore() {
   };
   const initConversation = async (conversationId) => {
     try {
-      const { data } = await conversationMemberQuery(conversationId);
+      const { data } = await config.services.request.conversationMemberQuery(conversationId);
       conversation.active = { id: conversationId, member: {} };
       data.forEach((member) => {
         if (member.memberType === 3) {
@@ -749,7 +792,7 @@ function createChatStore() {
       setInitMessage(conversationId);
     }
     if (conversation.messages[conversationId].message.length === 0) {
-      const { data } = await conversationMessagesQuery({
+      const { data } = await config.services.request.conversationMessagesQuery({
         pageNo: 1,
         pageSize: 1e3,
         conversationId
@@ -809,7 +852,7 @@ function createChatStore() {
       setHeaderOpen(false);
     }
     try {
-      await conversationMessageSend(sendParams);
+      await config.services.request.conversationMessageSend(sendParams);
       config.hooks?.onAfterSend?.();
     } finally {
       conversation.messages[conversationId].loading = false;
@@ -817,7 +860,7 @@ function createChatStore() {
   };
   const cancelReceive = async () => {
     console.log("\u53D6\u6D88\u63A5\u6536");
-    await conversationStop(conversation.active.id);
+    await config.services.request.conversationStop(conversation.active.id);
   };
   const updateConversationTitle = (conversationId, title) => {
     const idx = conversations.list.items.findIndex((item) => item.id === conversationId);
@@ -887,6 +930,7 @@ function createChatStore() {
   };
   return {
     config,
+    setServices,
     setPreview,
     setLayout,
     setHooks,
@@ -1440,7 +1484,7 @@ var styles_module_default3 = {
   chatAttachments: "styles_module_chatAttachments",
   chatSender: "styles_module_chatSender"
 };
-var Attachments_default = react.forwardRef(({ fileUploadConfig = [], fileList = [], onChange, extraParams }, ref) => {
+var Attachments_default = react.forwardRef(({ fileUpload, fileUploadConfig = [], fileList = [], onChange, extraParams }, ref) => {
   const { message: message3 } = antd.App.useApp();
   const fileListRef = react.useRef([]);
   const [attachedFiles, setAttachedFiles, getAttachedFiles] = common.useRefState([]);
@@ -1605,11 +1649,11 @@ var ChatSender_default = react.forwardRef(
     extraFooter,
     extraFooterBelow,
     sendBtnProps,
-    fileUpload: fileUpload2
+    fileUpload
   }, ref) => {
     const senderRef = react.useRef(null);
     const { inputValue, setInputValue } = common.useSyncInput(content || "", onContentChange);
-    const attachmentsNode = (fileUpload2?.config?.length ?? 0 > 0) && /* @__PURE__ */ jsxRuntime.jsx(antd.Badge, { dot: !!fileList?.length && !headerOpen, children: /* @__PURE__ */ jsxRuntime.jsx(antd.Button, { type: "text", icon: /* @__PURE__ */ jsxRuntime.jsx(icons.PaperClipOutlined, {}), onClick: () => onHeaderOpenChange(!headerOpen) }) });
+    const attachmentsNode = (fileUpload?.config?.length ?? 0 > 0) && /* @__PURE__ */ jsxRuntime.jsx(antd.Badge, { dot: !!fileList?.length && !headerOpen, children: /* @__PURE__ */ jsxRuntime.jsx(antd.Button, { type: "text", icon: /* @__PURE__ */ jsxRuntime.jsx(icons.PaperClipOutlined, {}), onClick: () => onHeaderOpenChange(!headerOpen) }) });
     const attachmentsRef = react.useRef();
     const senderHeader = /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
       /* @__PURE__ */ jsxRuntime.jsx(x.Sender.Header, { title: "\u9009\u62E9\u6587\u4EF6", open: headerOpen, onOpenChange: onHeaderOpenChange, children: /* @__PURE__ */ jsxRuntime.jsx(
@@ -1617,9 +1661,10 @@ var ChatSender_default = react.forwardRef(
         {
           ref: attachmentsRef,
           fileList,
+          fileUpload: fileUpload.request,
           onChange: onFileListChange,
-          fileUploadConfig: fileUpload2.config,
-          extraParams: fileUpload2.params
+          fileUploadConfig: fileUpload.config,
+          extraParams: fileUpload.params
         }
       ) }),
       extraHeader
@@ -1715,6 +1760,7 @@ var SenderPromptsItems_default = () => {
 var ChatSender_default2 = react.forwardRef(
   ({ placeholder, extraBtn = false, referencesBtn = false, sendBtnProps, footerBelow = false, prompts = true }, ref) => {
     const chatStore = useChatStore();
+    const configState = valtio.useSnapshot(chatStore.config);
     const receiverState = valtio.useSnapshot(chatStore.receiver);
     const conversationState = valtio.useSnapshot(chatStore.conversation);
     const chatMessage = react.useMemo(
@@ -1746,6 +1792,7 @@ var ChatSender_default2 = react.forwardRef(
           onCancel: chatStore.cancelReceive,
           onFocus: chatStore.config.hooks?.onSenderFocus,
           fileUpload: {
+            request: configState.services.request?.fileUpload,
             config: receiverState.active.config?.fileUpload
           },
           sendBtnProps: common.isFunction(sendBtnProps) ? sendBtnProps() : {},
@@ -1850,9 +1897,12 @@ var styles_module_default4 = {
   nsChatBody: "styles_module_nsChatBody",
   nsBodyWidth: "styles_module_nsBodyWidth"
 };
-var layouts_default = react.forwardRef(({ theme, params, userInfo, hooks, layout, config }, ref) => {
+var layouts_default = react.forwardRef(({ theme, params, userInfo, hooks, layout, config, services }, ref) => {
   const chatStore = react.useMemo(() => createChatStore(), []);
   const senderRef = react.useRef();
+  common.useDeepEffect(() => {
+    chatStore.setServices(services);
+  }, [services]);
   common.useDeepEffect(() => {
     if (config?.conversationId) {
       chatStore.switchConversation(config.conversationId);
@@ -1886,6 +1936,7 @@ var layouts_default = react.forwardRef(({ theme, params, userInfo, hooks, layout
       setReferences: chatStore.setReferences,
       setMessage: chatStore.setContent,
       setFiles: chatStore.setFileList,
+      setServices: chatStore.setServices,
       setParams: chatStore.setParams,
       senderFocus: (options) => {
         senderRef.current?.focus(options);
@@ -1908,7 +1959,19 @@ var layouts_default = react.forwardRef(({ theme, params, userInfo, hooks, layout
     }
   }, [hasPreView]);
   common.useWebSocket({
-    url: common.getWebSocketUrl(`/api/lolr/conversation/ws/subscribe?NS-TOKEN=${common.getToken()}`, common.isLocalhost() ? "192.168.6.23" : ""),
+    url: configState.services.websocketBaseUrls?.[0] && common.getWebSocketUrl(
+      `${configState.services.websocketBaseUrls?.[0]}/lolr/conversation/ws/subscribe?NS-TOKEN=${common.getToken()}`,
+      common.isLocalhost() ? "192.168.6.23" : ""
+    ),
+    onMessage: chatStore.acceptMessage,
+    clientHeartbeat: false,
+    reconnectInterval: 1e4
+  });
+  common.useWebSocket({
+    url: configState.services.websocketBaseUrls?.[1] && common.getWebSocketUrl(
+      `${configState.services.websocketBaseUrls?.[1]}/lolr/conversation/ws/subscribe?NS-TOKEN=${common.getToken()}`,
+      common.isLocalhost() ? "192.168.6.23" : ""
+    ),
     onMessage: chatStore.acceptMessage,
     clientHeartbeat: false,
     reconnectInterval: 1e4
@@ -1965,7 +2028,7 @@ var layouts_default = react.forwardRef(({ theme, params, userInfo, hooks, layout
                   }
                 });
               },
-              onDownloadFile: (content, type) => fileCreate(content, configState.preview.file.fileName, type),
+              onDownloadFile: (content, type) => configState.services.request.fileCreate(content, configState.preview.file.fileName, type),
               extraNav: /* @__PURE__ */ jsxRuntime.jsx(jsxRuntime.Fragment, { children: /* @__PURE__ */ jsxRuntime.jsx(antd.Tag, { bordered: false, color: "default", className: "cursor-pointer", onClick: () => chatStore.setPreview(), children: /* @__PURE__ */ jsxRuntime.jsx(icons.CloseOutlined, {}) }) })
             }
           )