@zero-library/chat-agent 2.1.10 → 2.1.11

package/dist/index.esm.js

@@ -1,13 +1,13 @@
-import { useRefState, useDebounce, useSyncInput, RenderWrapper, shouldRender, isFunction, useDeepEffect, useWebSocket, getWebSocketUrl, getToken, isLocalhost, downloadFile, FilePreview, MarkdownEditor, useCreateValtioContext, isNumber, getFileSuffixName, isObject, isNullOrUnDef, isBoolean, UserAvatar, request, deepCopy, getCurrentUser, deepMerge, RenderMarkdown, copyText, FileIcon, isExternal } from '@zero-library/common';
+import { useRefState, useDebounce, useSyncInput, RenderWrapper, shouldRender, isFunction, useDeepEffect, useWebSocket, getWebSocketUrl, getToken, isLocalhost, downloadFile, FilePreview, MarkdownEditor, useCreateValtioContext, isNumber, getFileSuffixName, isObject, isNullOrUnDef, isBoolean, UserAvatar, deepCopy, getCurrentUser, deepMerge, createRequest, RenderMarkdown, copyText, FileIcon, isExternal } from '@zero-library/common';
 import { App, Badge, Button, Flex, Typography, Spin, Splitter, Tag, Popover, List, Avatar, Space, message, Empty, Collapse, Modal, Row, Col, Drawer } from 'antd';
 import dayjs from 'dayjs';
 import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { forwardRef, useRef, useEffect, useImperativeHandle, useMemo, useState } from 'react';
+import { useSnapshot, proxy } from 'valtio';
 import { CloudUploadOutlined, PaperClipOutlined, EnterOutlined, UserSwitchOutlined, CloseOutlined, PlusOutlined, CommentOutlined, OpenAIOutlined, CopyOutlined, LikeOutlined, DislikeOutlined, DeleteOutlined, RedoOutlined, PlayCircleOutlined, CaretRightOutlined } from '@ant-design/icons';
 import { Attachments, Sender, Suggestion, XProvider, Welcome, Prompts, Bubble, Conversations } from '@ant-design/x';
 export * from '@ant-design/x';
 import classNames8 from 'classnames';
-import { useSnapshot, proxy } from 'valtio';
 import InfiniteScroll from 'react-infinite-scroll-component';
 
 var __defProp = Object.defineProperty;
@@ -19,19 +19,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 request.get(`/uc/doc?${paramsStr}`);
-    };
-    fileCreate = async (fileContent, fileName, targetFormat = "docx") => {
-      const blob = await request.post(`/uc/doc/convert`, { fileContent, fileName, targetFormat }, { responseType: "blob" });
-      const url = URL.createObjectURL(blob);
-      downloadFile(url, fileName);
-    };
-  }
-});
 function transform(source, fieldMap) {
   const result = {};
   for (const [targetKey, mapping] of Object.entries(fieldMap)) {
@@ -147,14 +134,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 = useSnapshot(chatStore.config);
       const [content, setContent] = useState("");
       const [loading, setLoading] = 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);
@@ -301,52 +290,95 @@ var init_QuoteList = __esm({
   }
 });
 
-// src/ui/layouts/index.tsx
-init_file();
-
 // src/stores/index.ts
 init_utils();
-var conversationsQuery = (params) => {
-  return request.get("/lolr/conversation", params);
-};
-var conversationCreate = (params) => {
-  return request.post("/lolr/conversation", params);
-};
-var conversationDelete = (conversationId) => {
-  return request.delete("/lolr/conversation", { conversationId });
-};
-var conversationRecentQuery = (params) => {
-  return request.get("/lolr/conversation/recent", params);
-};
-var conversationMessagesQuery = (params) => {
-  return request.get("/lolr/conversation/message", params);
-};
-var feedbackUpdate = (params) => {
-  return request.put("/lolr/conversation/message/feedback", params);
-};
-var messageSuggestedQuery = (msgId, maxCount = 3) => {
-  return request.get("/lolr/conversation/suggest", { msgId, maxCount });
-};
-var conversationMemberQuery = (conversationId) => {
-  return request.get("/lolr/conversation/member", { conversationId });
-};
-var conversationMessageSend = (params) => {
-  return request.post("/lolr/conversation/send", params);
-};
-var conversationStop = (conversationId) => {
-  return request.post("/lolr/conversation/stop", { conversationId });
-};
-var agentInfoQuery = (agentId) => {
-  return request.get("/lolr/agent/detail", { agentId });
-};
-var agentCharacterQuery = (agentId) => {
-  return request.get("/lolr/character", { agentId });
-};
-var agentCharacterSelect = (agentId, characterId) => {
-  return 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 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);
+    downloadFile(url, fileName);
+  };
+  return {
+    docQuery,
+    fileCreate
+  };
 };
 
 // src/stores/index.ts
@@ -390,6 +422,9 @@ var defaultExpertLayout = {
 };
 function createChatStore() {
   const config = proxy({
+    services: {
+      websocketBaseUrls: []
+    },
     hooks: {},
     layout: {},
     preview: {
@@ -402,6 +437,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 = createRequest(baseUrl);
+    config.services.request = { ...createChatService(request), ...createFileService(request) };
+  };
   const setPreview = (file = {}, isEdit = false) => {
     if (shouldRender(config.layout.preview)) {
       config.preview = {
@@ -457,7 +500,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",
@@ -508,7 +551,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);
@@ -529,7 +572,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 {
@@ -552,7 +595,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,
@@ -579,7 +622,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);
       message.success("\u5220\u9664\u6210\u529F");
       conversations.list.items = conversations.list.items.filter((item) => item.key !== conversationId);
       delete conversation.messages[conversationId];
@@ -630,7 +673,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;
@@ -638,7 +681,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 });
       message.success("\u611F\u8C22\u60A8\u7684\u53CD\u9988");
       conversation.messages[conversationId].message[index].msgFeedback = msgFeedback;
     } finally {
@@ -659,7 +702,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,
@@ -667,7 +710,7 @@ function createChatStore() {
       });
       return data;
     } else {
-      const { data } = await conversationCreate({
+      const { data } = await config.services.request.conversationCreate({
         receiverId,
         receiverType: 3,
         businessId: config.params.businessId,
@@ -682,7 +725,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) {
@@ -742,7 +785,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
@@ -802,7 +845,7 @@ function createChatStore() {
       setHeaderOpen(false);
     }
     try {
-      await conversationMessageSend(sendParams);
+      await config.services.request.conversationMessageSend(sendParams);
       config.hooks?.onAfterSend?.();
     } finally {
       conversation.messages[conversationId].loading = false;
@@ -810,7 +853,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);
@@ -880,6 +923,7 @@ function createChatStore() {
   };
   return {
     config,
+    setServices,
     setPreview,
     setLayout,
     setHooks,
@@ -1433,7 +1477,7 @@ var styles_module_default3 = {
   chatAttachments: "styles_module_chatAttachments",
   chatSender: "styles_module_chatSender"
 };
-var Attachments_default = forwardRef(({ fileUploadConfig = [], fileList = [], onChange, extraParams }, ref) => {
+var Attachments_default = forwardRef(({ fileUpload, fileUploadConfig = [], fileList = [], onChange, extraParams }, ref) => {
   const { message: message3 } = App.useApp();
   const fileListRef = useRef([]);
   const [attachedFiles, setAttachedFiles, getAttachedFiles] = useRefState([]);
@@ -1598,11 +1642,11 @@ var ChatSender_default = forwardRef(
     extraFooter,
     extraFooterBelow,
     sendBtnProps,
-    fileUpload: fileUpload2
+    fileUpload
   }, ref) => {
     const senderRef = useRef(null);
     const { inputValue, setInputValue } = useSyncInput(content || "", onContentChange);
-    const attachmentsNode = (fileUpload2?.config?.length ?? 0 > 0) && /* @__PURE__ */ jsx(Badge, { dot: !!fileList?.length && !headerOpen, children: /* @__PURE__ */ jsx(Button, { type: "text", icon: /* @__PURE__ */ jsx(PaperClipOutlined, {}), onClick: () => onHeaderOpenChange(!headerOpen) }) });
+    const attachmentsNode = (fileUpload?.config?.length ?? 0 > 0) && /* @__PURE__ */ jsx(Badge, { dot: !!fileList?.length && !headerOpen, children: /* @__PURE__ */ jsx(Button, { type: "text", icon: /* @__PURE__ */ jsx(PaperClipOutlined, {}), onClick: () => onHeaderOpenChange(!headerOpen) }) });
     const attachmentsRef = useRef();
     const senderHeader = /* @__PURE__ */ jsxs(Fragment, { children: [
       /* @__PURE__ */ jsx(Sender.Header, { title: "\u9009\u62E9\u6587\u4EF6", open: headerOpen, onOpenChange: onHeaderOpenChange, children: /* @__PURE__ */ jsx(
@@ -1610,9 +1654,10 @@ var ChatSender_default = forwardRef(
         {
           ref: attachmentsRef,
           fileList,
+          fileUpload: fileUpload.request,
           onChange: onFileListChange,
-          fileUploadConfig: fileUpload2.config,
-          extraParams: fileUpload2.params
+          fileUploadConfig: fileUpload.config,
+          extraParams: fileUpload.params
         }
       ) }),
       extraHeader
@@ -1708,6 +1753,7 @@ var SenderPromptsItems_default = () => {
 var ChatSender_default2 = forwardRef(
   ({ placeholder, extraBtn = false, referencesBtn = false, sendBtnProps, footerBelow = false, prompts = true }, ref) => {
     const chatStore = useChatStore();
+    const configState = useSnapshot(chatStore.config);
     const receiverState = useSnapshot(chatStore.receiver);
     const conversationState = useSnapshot(chatStore.conversation);
     const chatMessage = useMemo(
@@ -1739,6 +1785,7 @@ var ChatSender_default2 = forwardRef(
           onCancel: chatStore.cancelReceive,
           onFocus: chatStore.config.hooks?.onSenderFocus,
           fileUpload: {
+            request: configState.services.request?.fileUpload,
             config: receiverState.active.config?.fileUpload
           },
           sendBtnProps: isFunction(sendBtnProps) ? sendBtnProps() : {},
@@ -1843,9 +1890,12 @@ var styles_module_default4 = {
   nsChatBody: "styles_module_nsChatBody",
   nsBodyWidth: "styles_module_nsBodyWidth"
 };
-var layouts_default = forwardRef(({ theme, params, userInfo, hooks, layout, config }, ref) => {
+var layouts_default = forwardRef(({ theme, params, userInfo, hooks, layout, config, services }, ref) => {
   const chatStore = useMemo(() => createChatStore(), []);
   const senderRef = useRef();
+  useDeepEffect(() => {
+    chatStore.setServices(services);
+  }, [services]);
   useDeepEffect(() => {
     if (config?.conversationId) {
       chatStore.switchConversation(config.conversationId);
@@ -1879,6 +1929,7 @@ var layouts_default = forwardRef(({ theme, params, userInfo, hooks, layout, conf
       setReferences: chatStore.setReferences,
       setMessage: chatStore.setContent,
       setFiles: chatStore.setFileList,
+      setServices: chatStore.setServices,
       setParams: chatStore.setParams,
       senderFocus: (options) => {
         senderRef.current?.focus(options);
@@ -1901,7 +1952,19 @@ var layouts_default = forwardRef(({ theme, params, userInfo, hooks, layout, conf
     }
   }, [hasPreView]);
   useWebSocket({
-    url: getWebSocketUrl(`/api/lolr/conversation/ws/subscribe?NS-TOKEN=${getToken()}`, isLocalhost() ? "192.168.6.23" : ""),
+    url: configState.services.websocketBaseUrls?.[0] && getWebSocketUrl(
+      `${configState.services.websocketBaseUrls?.[0]}/lolr/conversation/ws/subscribe?NS-TOKEN=${getToken()}`,
+      isLocalhost() ? "192.168.6.23" : ""
+    ),
+    onMessage: chatStore.acceptMessage,
+    clientHeartbeat: false,
+    reconnectInterval: 1e4
+  });
+  useWebSocket({
+    url: configState.services.websocketBaseUrls?.[1] && getWebSocketUrl(
+      `${configState.services.websocketBaseUrls?.[1]}/lolr/conversation/ws/subscribe?NS-TOKEN=${getToken()}`,
+      isLocalhost() ? "192.168.6.23" : ""
+    ),
     onMessage: chatStore.acceptMessage,
     clientHeartbeat: false,
     reconnectInterval: 1e4
@@ -1958,7 +2021,7 @@ var layouts_default = forwardRef(({ theme, params, userInfo, hooks, layout, conf
                   }
                 });
               },
-              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__ */ jsx(Fragment, { children: /* @__PURE__ */ jsx(Tag, { bordered: false, color: "default", className: "cursor-pointer", onClick: () => chatStore.setPreview(), children: /* @__PURE__ */ jsx(CloseOutlined, {}) }) })
             }
           )