@gientech/modual 2.2.3-fix.5 → 2.2.3-fix.6

package/src/modules/chatCompare/components/UseChatInstanceHooks.ts

@@ -0,0 +1,406 @@
+import { uid } from 'uid';
+import { api_createDialogue } from './server';
+import { useRef } from 'react';
+import axios from 'axios';
+import { getUrlPrefix } from '../../chat/utils';
+import { mergeFiles } from '../../../utils/gientechCommon/utils/fileUtils';
+import { message } from 'antd';
+import { get } from 'animejs';
+export enum ChatSystemType {
+  Chat = 'KNO_INDEX', // 知识索引
+  Retrieval = 'INT_SESSION', // 智能问答
+  TEMPORARY = 'TEMPORARY', //临时
+}
+
+export default function UseChatInstanceHooks({
+  token,
+  sessionId,
+  api_startChat_re,
+  addEmptyMessage,
+  setCurrentChatData,
+  setAppStatus,
+  uploadController,
+  resetStopFlag,
+  setFileStatuses,
+  stoppedRef,
+  stopPoll,
+  pollFileStatusHook,
+  setFileManagerData,
+  senderSwitchValues,
+  tempKnowledgeBaseCheck,
+  setRecommandQuestions,
+  configId,
+  currentChatData,
+}: any) {
+  const isNew = useRef(true);
+  const createdialog = async (messageData: any) => {
+    const { content, assist_id } = messageData;
+    if (isNew.current) {
+      isNew.current = false;
+      const res = await api_createDialogue({
+        token,
+        params: {
+          label: content,
+          sessionId,
+          configId: assist_id,
+          sysType: ChatSystemType.TEMPORARY,
+        },
+      });
+      if (!res?.success || !res?.data) return;
+      return res.data;
+    } else {
+      return { configId: assist_id, name: '' };
+    }
+  };
+
+  const pollFileStatus = (
+    uids: string[],
+    onStatusUpdate: (serverStatuses: any[]) => void,
+    onComplete: (serverStatuses: any[]) => void,
+    onFail: (errorMsg: string) => void
+  ) => {
+    stopPoll = pollFileStatusHook(uids, {
+      onStatusUpdate,
+      onComplete,
+      onFail,
+    });
+  };
+
+  const fetchRecommendQuestions = async (queryId: string) => {
+    return;
+    if (!queryId) return;
+    try {
+      // 需要携带当前 configId
+      const res = await axios.get(
+        `${getUrlPrefix()}/qa/ai/recommendQuestion?queryId=${queryId}&configId=${currentChatData?.configId || configId || ''}`.replace(
+          /\?&/,
+          '?'
+        ),
+        {
+          headers: { Authorization: token },
+        }
+      );
+      setRecommandQuestions(res.data?.data || []);
+    } catch (e) {
+      setRecommandQuestions([]);
+    }
+  };
+
+  const sendMessageWithSessionId = (sessionId: string, data: any, clearFn: () => void) => {
+    const configIdToUse = data.assist_id || configId;
+    console.log('助手配置详情', configIdToUse);
+    const queryId = `chat-${uid(32)}`;
+    // === 数据流保障：明确保留内容格式，不进行任何 trim 或替换处理 ===
+    // 确保换行符、空格等格式完整保留，直接传递到后端
+    const rawContent = data.content || data.text || '';
+    const preservedContent = typeof rawContent === 'string' ? rawContent : String(rawContent);
+
+    const label = preservedContent ? preservedContent.slice(0, 20) : '未命名会话';
+    let localCurrentFiles = undefined;
+    if (data.files && data.files.length > 0) {
+      localCurrentFiles = data.files.map((fileItem: any) => ({
+        name: fileItem.file.name,
+        size: fileItem.file.size,
+        type: fileItem.file.type,
+        filePath: fileItem.filePath || '',
+        convertedFilePath: fileItem.convertedFilePath || '',
+        uid: fileItem.uid || '',
+        ...fileItem,
+      }));
+    }
+    console.log('本地文件上传', data.files);
+    addEmptyMessage({
+      content: preservedContent, // 使用保留格式的内容，确保换行和空格完整保留
+      queryId,
+      currentFiles: localCurrentFiles,
+      filePaths: localCurrentFiles ? JSON.stringify(localCurrentFiles) : undefined,
+    });
+    setCurrentChatData((c: any) => ({ ...c, label, isNew: false }));
+    // 2. 文件上传逻辑 ---
+    if (data.files && data.files.length > 0) {
+      setAppStatus((pre: any) => ({ ...pre, sender: 'uploading', display: 'uploading' }));
+      uploadController = new AbortController();
+      resetStopFlag(); // 使用 hook 提供的重置函数
+      const clientFileUids = data.files.map(
+        () => `rc-upload-${Date.now()}-${Math.random().toString(36).slice(2)}`
+      );
+      const initialFileStatuses = data.files.map((fileItem: any, index: number) => ({
+        ...fileItem,
+        uid: clientFileUids[index],
+        status: 'uploading',
+      }));
+      setFileStatuses(initialFileStatuses);
+      const formData = new FormData();
+      data.files.forEach((fileItem: any) => {
+        formData.append('files', fileItem.file);
+      });
+      const sessionIdStr = String(sessionId || '');
+      formData.append('sessionId', sessionIdStr);
+      formData.append('uids', clientFileUids.join(','));
+      axios
+        .post(`${getUrlPrefix()}/index/knowledgeBase/file/uploadFiles`, formData, {
+          headers: {
+            Authorization: token,
+            'Content-Type': 'multipart/form-data',
+          },
+          signal: uploadController.signal,
+        })
+        .then(response => {
+          console.log('文件上传的打印结果', response);
+          console.log('文件上传响应详情:', {
+            status: response.status,
+            success: response.data?.success,
+            dataType: Array.isArray(response.data?.data) ? 'array' : typeof response.data?.data,
+            dataLength: Array.isArray(response.data?.data) ? response.data.data.length : 'N/A',
+            errorMsg: response.data?.errorMsg,
+            errorCode: response.data?.errorCode,
+            fullResponse: response.data,
+          });
+
+          if (stoppedRef.current) return;
+
+          // 检查响应数据结构 - 严格按照 headlessChat 的逻辑：必须有有效数据才继续
+          const responseData = response.data?.data;
+          const isSuccess = response.data?.success === true;
+          const isValidData = Array.isArray(responseData) && responseData.length > 0;
+
+          if (isSuccess && isValidData) {
+            // 后端返回了有效数据，使用后端数据
+            const uploadedFilesData = responseData;
+            const serverUids = uploadedFilesData.map((f: any) => f.uid);
+
+            // === 合并后端返回字段到 fileStatuses ===
+            setFileStatuses((prev: any) => {
+              return prev.map((localFile: any) => {
+                const serverFile = responseData.find((f: any) => f.uid === localFile.uid);
+                return serverFile ? { ...localFile, ...serverFile } : localFile;
+              });
+            });
+
+            setAppStatus((pre: any) => ({ ...pre, sender: 'processing', display: 'analyzing' }));
+            setFileStatuses((prev: any) => prev.map((f: any) => ({ ...f, status: 'parsing' })));
+
+            pollFileStatus(
+              serverUids,
+              (serverStatuses: any[]) => {
+                // 添加空值检查，防止 serverStatuses 为 null 或 undefined
+                if (!serverStatuses || !Array.isArray(serverStatuses)) {
+                  console.warn('pollFileStatus 回调返回的 serverStatuses 无效:', serverStatuses);
+                  return;
+                }
+                setFileStatuses((currentLocalStatuses: any) => {
+                  return currentLocalStatuses.map((localFile: any) => {
+                    const correspondingServerFile = serverStatuses.find(
+                      (sf: any) => sf && sf.uid === localFile.uid
+                    );
+                    if (correspondingServerFile) {
+                      // 使用 fileParseStatus 判断文件解析状态：1:正常 2:解析中 3:解析成功 4:解析失败
+                      // 兼容旧数据，优先使用 fileParseStatus，如果没有则使用 status
+                      const parseStatus =
+                        correspondingServerFile.fileParseStatus ?? correspondingServerFile.status;
+                      switch (parseStatus) {
+                        case 3:
+                          return { ...localFile, status: 'done' };
+                        case 4:
+                          return { ...localFile, status: 'error' };
+                        case 1:
+                        case 2:
+                        default:
+                          return { ...localFile, status: 'parsing' };
+                      }
+                    }
+                    return localFile;
+                  });
+                });
+                // === 新增：全部完成时清空 sender 区 ===
+                // 使用 fileParseStatus 判断：所有文件解析成功（3）或解析失败（4）时停止轮询
+                const allDone =
+                  Array.isArray(serverStatuses) &&
+                  serverStatuses.length > 0 &&
+                  serverStatuses.every((file: any) => {
+                    if (!file) return false;
+                    const parseStatus = file.fileParseStatus ?? file.status;
+                    return parseStatus === 3 || parseStatus === 4; // 3:解析成功 4:解析失败
+                  });
+                if (allDone) {
+                  setFileStatuses([]);
+                  // 上传完成后合并新文件到 uploadedFiles，保证结构统一
+                  const map = new Map();
+                  (serverStatuses || []).forEach((item: any) => {
+                    map.set(item.id, item);
+                  });
+                  setFileManagerData((prev: any) => {
+                    const r = mergeFiles(prev.uploadedFiles, uploadedFilesData);
+
+                    return {
+                      uploadedFiles: r.map(item => {
+                        return {
+                          ...item,
+                          pdfPages: map.get(item.id)?.pdfPages || 0, // 保留后端返回的 pdfPages 字段
+                        };
+                      }),
+                    };
+                  });
+                }
+              },
+              // onComplete: 全部解析成功后才发起AI请求
+              (serverStatuses: any[]) => {
+                clearFn?.();
+                setAppStatus((pre: any) => ({
+                  ...pre,
+                  sender: 'processing',
+                  display: 'processing',
+                }));
+                // === 新增：回填 filePath 到 fileStatuses ===
+                setFileStatuses((prev: any) => {
+                  return prev.map((localFile: any) => {
+                    const serverFile = uploadedFilesData.find((f: any) => f.uid === localFile.uid);
+
+                    if (serverFile) {
+                      return {
+                        ...localFile,
+                        ...serverFile, // 合并后端所有字段
+                        url: serverFile.filePath, // 兼容 url 字段
+                        filePath: serverFile.filePath,
+                        name: serverFile.fileName || localFile.name,
+                        convertedFilePath:
+                          serverFile.convertedFilePath || localFile.convertedFilePath,
+                      };
+                    }
+                    return localFile;
+                  });
+                });
+                console.log('uploadedFilesData', uploadedFilesData);
+
+                // === 同步 currentFiles 字段到 chatData ===
+                setCurrentChatData((chatList: any) => {
+                  if (!chatList) return chatList;
+                  const { messages } = chatList;
+                  for (let i = messages.length - 1; i >= 0; i--) {
+                    if (messages[i].istype === 'user') {
+                      const oldFiles = messages[i].currentFiles || [];
+                      messages[i].currentFiles = oldFiles.map((localFile: any) => {
+                        const serverFile = uploadedFilesData.find(
+                          (f: any) => f.uid === localFile.uid
+                        );
+                        return serverFile ? { ...localFile, ...serverFile } : localFile;
+                      });
+                      break;
+                    }
+                  }
+                  return chatList;
+                });
+                // === 发起AI请求 ===
+                const now = Date.now();
+                api_startChat_re(
+                  {
+                    configId: configIdToUse,
+                    content: preservedContent, // 使用保留格式的内容，确保换行和空格完整传递到后端
+                    fileUids: serverUids,
+                    lastDate: now,
+                    name: label,
+                    queryId: queryId,
+                    sessionId: String(sessionId || ''),
+                    type: 'text',
+                    audioUrl: '',
+                    enableThinking: !!senderSwitchValues.reasoning,
+                    enableWebsearch: !!senderSwitchValues.netSearch,
+                    qaChannel: senderSwitchValues.qaChannel,
+                    tempKnowledgeBaseCheck: tempKnowledgeBaseCheck.current,
+                  } as any,
+                  (gmtModified: any) => {
+                    fetchRecommendQuestions(queryId);
+                  }
+                );
+              },
+              (errorMsg: string) => {
+                setAppStatus((pre: any) => ({ ...pre, display: 'ready', sender: 'ready' }));
+                setFileStatuses((prev: any) =>
+                  prev.map((f: any) => ({ ...f, status: 'error', message: errorMsg }))
+                );
+              }
+            );
+          } else {
+            // 详细的错误信息
+            let errorMsg = '文件上传失败';
+            if (response.data?.errorMsg) {
+              errorMsg = response.data.errorMsg;
+            } else if (!isSuccess) {
+              errorMsg = `上传失败: success=${response.data?.success}`;
+            } else if (!isValidData) {
+              if (!Array.isArray(responseData)) {
+                errorMsg = `上传失败: 返回数据格式错误，期望数组，实际为 ${typeof responseData}`;
+              } else if (responseData.length === 0) {
+                errorMsg = '上传失败: 返回数据为空';
+              }
+            }
+
+            console.error('文件上传失败详情:', {
+              errorMsg,
+              response: response.data,
+              isSuccess,
+              isValidData,
+            });
+
+            message.error(errorMsg);
+            setAppStatus((pre: any) => ({ ...pre, display: 'error', sender: 'ready' }));
+            setFileStatuses((prev: any) =>
+              prev.map((f: any) => ({ ...f, status: 'error', message: errorMsg }))
+            );
+          }
+        })
+        .catch(error => {
+          // 检查是否为文件大小超限错误 (413)
+          if (error?.response?.status === 413) {
+            if (error?.response?.statusText.includes('Request Entity Too Large')) {
+              message.error('文件超过最大可上传限制');
+            }
+          } else {
+            // 其他错误显示通用网络错误提示
+            message.error('网络错误');
+          }
+
+          setAppStatus((pre: any) => ({ ...pre, display: 'error', sender: 'ready' }));
+          setFileStatuses((prev: any) =>
+            prev.map((f: any) => ({ ...f, status: 'error', message: '网络错误' }))
+          );
+        });
+      return;
+    }
+    // 3. 无文件，直接发起AI请求
+    setAppStatus((pre: any) => ({ ...pre, display: 'processing', sender: 'processing' }));
+    const now = Date.now();
+    clearFn?.();
+    api_startChat_re(
+      {
+        configId: configIdToUse,
+        content: preservedContent, // 使用保留格式的内容，确保换行和空格完整传递到后端
+        fileUids: [],
+        lastDate: now + '',
+        name: label,
+        queryId: queryId,
+        sessionId: String(sessionId || ''),
+        type: 'text',
+        audioUrl: '',
+        enableThinking: !!senderSwitchValues.reasoning,
+        enableWebsearch: !!senderSwitchValues.netSearch,
+        qaChannel: senderSwitchValues.qaChannel,
+        tempKnowledgeBaseCheck: tempKnowledgeBaseCheck.current,
+      } as any,
+      (gmtModified: any) => {
+        fetchRecommendQuestions(queryId);
+      }
+    );
+  };
+
+  const handleSendMessage = async (messageData: any) => {
+    if (!messageData?.assist_id && !configId) {
+      return message.warning(`对话${messageData?.index + 1}请选择助手`);
+    }
+    console.log('Sending message:', messageData);
+    const { clearFn } = messageData;
+    await createdialog(messageData);
+    sendMessageWithSessionId(sessionId as string, messageData, clearFn);
+  };
+  return { handleSendMessage };
+}

package/src/modules/chatCompare/components/UseScrollHooks.ts

@@ -0,0 +1,36 @@
+import { useEffect, useRef } from 'react';
+
+export function UseScrollHooks({ scrollOld }: { scrollOld: boolean }) {
+  const handleScroll = () => {
+    chatContainerRef.current.addEventListener('wheel', (event: any) => {
+      event.preventDefault(); // 阻止默认滚动行为
+      const { deltaY } = event; // 获取滚动方向和速度
+      chatContainerRef.current.scrollTop -= deltaY; //  反转方向
+    });
+  };
+
+  const chatContainerRef: any = useRef(null);
+  const timer: any = useRef(null);
+
+  useEffect(() => {
+    if (scrollOld) return;
+    if (timer.current) return;
+    timer.current = setInterval(() => {
+      if (chatContainerRef?.current?.scrollTo) {
+        handleScroll();
+        clearInterval(timer.current);
+        timer.current = null;
+      }
+    }, 100);
+    return () => {
+      clearInterval(timer.current);
+      timer.current = null;
+      if (chatContainerRef?.current?.removeEventListener) {
+        chatContainerRef.current.removeEventListener('wheel', handleScroll);
+      }
+    };
+  }, []);
+  return { chatContainerRef };
+}
+
+export default UseScrollHooks;

package/src/modules/chatCompare/components/components/AllFiles.tsx

@@ -0,0 +1,156 @@
+import { useEffect, useRef, useState } from 'react';
+import { Tooltip } from 'antd';
+import axios from 'axios';
+import { getUrlPrefix } from '../../../../utils/gientechCommon/utils';
+import DrawerTitle from './DrawerTitle';
+import { renderIcon } from '../../../../utils';
+
+export const AllFiles = ({ fileData, token, onClose }: any) => {
+  const [fileList, setFileList] = useState<any[]>([]);
+  useEffect(() => {
+    const knowledgeBases = fileData?.knowledgeBases;
+    if (!knowledgeBases?.length) return;
+    const treeData = knowledgeBases[0]?.treeData || [];
+    if (!treeData?.length) return;
+    setFileList(treeData);
+  }, [fileData]);
+  const map: any = useRef(new Map());
+  const clickFileItem = async (item: any, key: number) => {
+    if (map.current.get(item?.fetchUrl)) {
+      return;
+    }
+    try {
+      const curFolder = fileList[key];
+      curFolder.loading = true;
+      const newFileList = [...fileList];
+      setFileList(newFileList);
+      const res = await axios.get(`${getUrlPrefix()}/${item.fetchUrl}`, {
+        headers: { Authorization: token },
+      });
+      const data = res?.data?.data?.records;
+      if (data) {
+        const curFolder = fileList[key];
+        curFolder.loading = false;
+        curFolder.children = data;
+        const newFileList = [...fileList];
+        newFileList[key] = curFolder;
+        setFileList(newFileList);
+        map.current.set(item?.fetchUrl, true);
+      } else {
+        const curFolder = fileList[key];
+        curFolder.loading = false;
+        const newFileList = [...fileList];
+        setFileList(newFileList);
+      }
+    } catch (error) {
+      const curFolder = fileList[key];
+      curFolder.loading = false;
+      const newFileList = [...fileList];
+      setFileList(newFileList);
+    }
+  };
+  return (
+    <>
+      <DrawerTitle
+        title="知识库文档目录"
+        setOpen={() => {
+          onClose();
+        }}
+      />
+      <div style={{ height: 'calc(100% - 60px)', overflowY: 'auto', padding: '16px' }}>
+        {fileList.map((item: any, index: number) => {
+          if (item.type === 'file') {
+            return (
+              <div className="flex" style={{ marginTop: '8px' }}>
+                {renderIcon(item)}
+                <Tooltip placement="top" title={item.name?.length > 23 ? item.name : ''}>
+                  <div
+                    style={{
+                      paddingLeft: '6px',
+                      width: '346px',
+                      overflow: 'hidden',
+                      textOverflow: 'ellipsis',
+                      whiteSpace: 'nowrap',
+                      cursor: 'default',
+                      fontSize: '14px',
+                    }}
+                  >
+                    {item.name}
+                  </div>
+                </Tooltip>
+              </div>
+            );
+          } else {
+            return (
+              <FolderCom item={item} key={index} index={index} clickFileItem={clickFileItem} />
+            );
+          }
+        })}
+      </div>
+    </>
+  );
+};
+
+const FolderCom = ({ item, index, clickFileItem }: any) => {
+  const [open, setOpen] = useState(false);
+
+  return (
+    <div style={{ marginTop: '8px' }}>
+      <div
+        className="flex cursor-pointer"
+        key={index}
+        onClick={() => {
+          setOpen(!open);
+          if (open) return;
+          clickFileItem(item, index);
+        }}
+      >
+        {renderIcon(item)}
+        <Tooltip placement="top" title={item.name?.length > 23 ? item.name : ''}>
+          <div
+            style={{
+              paddingLeft: '6px',
+              width: '346px',
+              overflow: 'hidden',
+              textOverflow: 'ellipsis',
+              whiteSpace: 'nowrap',
+              fontSize: '14px',
+            }}
+          >
+            {item.name} ({item.fileCount || 0})
+          </div>
+        </Tooltip>
+      </div>
+      <div
+        style={{
+          padding: '2px 0 2px 14px',
+          margin: '8px 0 0 7px',
+          borderLeft: '2px solid #E5E6EB',
+          display: open ? 'block' : 'none',
+        }}
+      >
+        {(item.children || []).map((child: any, index: number) => (
+          <div className="flex items-center" key={index}>
+            <div
+              style={{ width: '4px', height: '4px', borderRadius: '2px', background: 'black' }}
+            ></div>
+            <Tooltip placement="top" title={child.name?.length > 26 ? child.name : ''}>
+              <div
+                style={{
+                  paddingLeft: '6px',
+                  overflow: 'hidden',
+                  textOverflow: 'ellipsis',
+                  whiteSpace: 'nowrap',
+                  fontSize: '12px',
+                  width: '336px',
+                }}
+              >
+                {child.fileName}
+              </div>
+            </Tooltip>
+          </div>
+        ))}
+      </div>
+    </div>
+  );
+};

package/src/modules/chatCompare/components/components/DeleteModal.tsx

@@ -0,0 +1,75 @@
+import { Button, ConfigProvider, Form, Input, Modal } from "antd";
+import { useForm } from "antd/es/form/Form";
+import { useEffect } from "react";
+
+export const DeleteModal = ({
+  isModalOpen,
+  handleOk,
+  handleCancel,
+  data
+}: any) => {
+  const modalStyles = {
+    header: {
+      height: "60px",
+      borderBottom: "1px solid #F0F0F0",
+      display: "flex",
+      paddingLeft: "24px",
+      alignItems: "center",
+      marginBottom: "16px"
+    },
+    title: { fontSize: "16px" },
+    body: { padding: "0 24px" },
+    mask: {},
+    footer: { marginTop: "0px", padding: "0 24px 16px 24px" },
+    content: { padding: 0 }
+  };
+  const [rename_form] = useForm();
+  useEffect(() => {
+    if (!isModalOpen) {
+      rename_form.resetFields();
+      return;
+    }
+    rename_form.setFieldValue("new_name", data?.label);
+  }, [data, isModalOpen]);
+  return (
+    <ConfigProvider
+      modal={{
+        styles: modalStyles
+      }}
+    >
+      <Modal
+        title={<div className="text-[16px]">删除会话</div>}
+        closeIcon={null}
+        open={isModalOpen}
+        width={520}
+        centered
+        className="[&:[.ant-modal-content]]:bg-red"
+        footer={
+          <div className="mt-[16px] flex justify-end">
+            <Button
+              children="取消"
+              onClick={handleCancel}
+              className="mr-[8px]"
+            />
+            <Button
+              children="确认"
+              onClick={async () => {
+                try {
+                  const data = await rename_form.validateFields();
+                  handleOk("ok_remove", data);
+                } catch (e) {
+                  console.log(e);
+                  return;
+                }
+              }}
+              type="primary"
+              className="border-none"
+            />
+          </div>
+        }
+      >
+        <div>{`确定要删除会话「${data.label}」吗？删除后无法恢复。`}</div>
+      </Modal>
+    </ConfigProvider>
+  );
+};