@gientech/modual 1.4.7 → 1.5.2

package/src/modules/chat/referenceCom/DrawerTitle.tsx

@@ -0,0 +1,26 @@
+import Close from '../../../assets/img/close.png';
+const DrawerTitle = ({ title, setOpen }: { title: string; setOpen: (arg: boolean) => void }) => {
+  return (
+    <div
+      style={{
+        width: '100%',
+        height: '60px',
+        borderBottom: '1px solid #EBEBEB',
+        padding: '20px',
+        display: 'flex',
+        justifyContent: 'space-between',
+        alignItems: 'center',
+        fontSize: '16px',
+        fontWeight: 600,
+      }}
+    >
+      <div>{title}</div>
+      <img
+        src={Close}
+        style={{ width: '16px', height: '16px', cursor: 'pointer' }}
+        onClick={() => setOpen(false)}
+      />
+    </div>
+  );
+};
+export default DrawerTitle;

package/src/modules/chat/referenceCom/RenameModal.tsx

@@ -0,0 +1,86 @@
+import { BorderBottomOutlined } from "@ant-design/icons";
+import { Button, ConfigProvider, Form, Input, Modal } from "antd";
+import { useForm } from "antd/es/form/Form";
+import { useEffect } from "react";
+
+export const RenameModal = ({
+  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_rename", data);
+                } catch (e) {
+                  console.log(e);
+                  return;
+                }
+              }}
+              type="primary"
+              className="border-none"
+            />
+          </div>
+        }
+      >
+        <Form form={rename_form}>
+          <Form.Item
+            name="new_name"
+            rules={[
+              { required: true, message: "请输入" },
+              { max: 100, message: "最大字数限制100" }
+            ]}
+          >
+            <Input />
+          </Form.Item>
+        </Form>
+      </Modal>
+    </ConfigProvider>
+  );
+};

package/src/modules/chat/referenceCom/TagCom.tsx

@@ -0,0 +1,30 @@
+import { bgColor, textColor } from '../ReferenceBar';
+
+const TagCom = ({ i, item, j, clickTag }: any) => {
+  console.log('j in tag', j);
+  console.log('i in tag', i);
+  console.log('item in tag', item);
+
+  return (
+    <div
+      key={j.id}
+      style={{
+        minWidth: '16px',
+        height: '16px',
+        backgroundColor: bgColor[item.key],
+        margin: '0 4px',
+        color: textColor[item.key],
+        display: 'flex',
+        justifyContent: 'center',
+        alignItems: 'center',
+        borderRadius: '2px',
+        fontSize: '12px',
+        cursor: 'pointer',
+      }}
+      onClick={() => clickTag && clickTag(i, j, item)}
+    >
+      {j.index}
+    </div>
+  );
+};
+export default TagCom;

package/src/modules/chat/style.css

@@ -0,0 +1,3 @@
+.listWrap .listBox .listItemBox:hover {
+  background-color: #eaf4fd;
+}

package/src/modules/chat/utils/index.ts

@@ -0,0 +1,326 @@
+import xls from '../../../assets/img/excel.png';
+import txt from '../../../assets/img/txt.png';
+import doc from '../../../assets/img/word.png';
+import pdf from '../../../assets/img/pdf.png';
+import ppt from '../../../assets/img/ppt.png';
+import img from '../../../assets/img/img.png';
+import dayjs from 'dayjs';
+import axios from 'axios';
+
+export const dealMinioUrl = (url: string) => {
+  // | undefined | string[]
+  return url || '';
+  // if (!url) return "";
+  // if (Array.isArray(url)) {
+  //   const urlArr = url.map((item) => {
+  //     const arr = item.split("/").slice(3);
+  //     return `${getMinioUrlPrefix()}/` + arr.join("/");
+  //   });
+  //   return urlArr;
+  // }
+
+  // const arr = url.split("/").slice(3);
+
+  // return `${getMinioUrlPrefix()}/` + arr.join("/");
+};
+
+export const uniqArr2By = (array1: any[], array2: any[], key: string) => {
+  return array2.filter(item2 => !array1.some(item1 => item1[key] === item2[key]));
+};
+
+export const fileViewTypes = [
+  //"txt",
+  'pdf',
+  // "xlsx",
+  // "xls",
+  // "csv",
+  'md',
+  'jpeg',
+  'jpg',
+  'png',
+];
+
+export const getMarkExcludeType = (file_type: string) => {
+  return [''].includes(file_type); //"xls", "xlsx", "csv"
+};
+
+export const isImage = (type: string) => {
+  return ['png', 'jpg', 'jpeg'].includes(type);
+};
+
+export const TempBaseUrl = `http://${location.hostname}:9000/`;
+
+export const getFileHost = (convertedPath: string) => {
+  if (convertedPath.includes('http')) {
+    const url = new URL(convertedPath);
+    return `${url.origin}/`;
+  }
+  return TempBaseUrl;
+};
+export const FILEIMG: any = {
+  docx: doc,
+  doc: doc,
+  pdf: pdf,
+  xls: xls,
+  xlsx: xls,
+  csv: xls,
+  ppt: ppt,
+  pptx: ppt,
+  pub: ppt,
+  txt: txt,
+  png: img,
+  jpg: img,
+  jpge: img,
+};
+
+export const getFileIcon = (fileType: string) => {
+  return FILEIMG[fileType] || FILEIMG['txt'];
+};
+
+export const smoothScroll = (container: any, distance?: number) => {
+  const d = distance || 20;
+  const { scrollTop, scrollHeight, clientHeight } = container;
+  const isBottom = scrollTop + clientHeight >= scrollHeight;
+  requestAnimationFrame(() => {
+    if (!isBottom) {
+      if (scrollTop + clientHeight + d < scrollHeight) {
+        container.scrollBy({
+          top: d,
+          behavior: 'smooth',
+        });
+        smoothScroll(container, d);
+      } else {
+        container.scrollTo({
+          top: scrollHeight,
+          behavior: 'smooth',
+        });
+      }
+    }
+  });
+};
+
+export const judgeXlS = (item: { type?: string; fileName: string }) => {
+  const validTypes = ['xlsx', 'xls', 'csv', 'png', 'jpg', 'jpeg'];
+  const fileExtension = item.fileName.split('.').pop()?.toLowerCase();
+  return validTypes.includes(item.type || '') || validTypes.includes(fileExtension || '');
+};
+
+export const FileModalSize = {
+  width: 960,
+  height: 742,
+};
+
+export const formatTime = (val: string) => {
+  return val ? dayjs(val).format('YY-MM-DD HH:mm') : '';
+};
+
+export const getUrlPrefix = () => {
+  return import.meta.env.VITE_RAG_PREFIX;
+};
+
+export const getMinioUrlPrefix = () => {
+  return import.meta.env.VITE_MINIO_PREFIX;
+};
+
+export const getCecUrlPrefix = () => {
+  return import.meta.env.VITE_CEC_PREFIX;
+};
+
+export const handleDownLoad = async (url: string) => {
+  if (!url) return;
+
+  try {
+    const response = await fetch(encodeURI(url)); // 处理特殊字符
+    const blob = await response.blob();
+    const blobUrl = URL.createObjectURL(blob);
+
+    const link = document.createElement('a');
+    link.href = blobUrl;
+    link.download = url.split('/').pop() || 'file.pdf'; // 提取文件名
+    link.style.display = 'none';
+
+    document.body.appendChild(link);
+    link.click();
+
+    // 清理资源
+    setTimeout(() => {
+      URL.revokeObjectURL(blobUrl);
+      document.body.removeChild(link);
+    }, 100);
+  } catch (error) {
+    console.error('下载失败:', error);
+  }
+};
+
+/**
+ * 获取文档预览组件数据
+ */
+
+export const getFilePath = (content: {
+  file_type?: string;
+  file_source?: string;
+  metadata: { source?: string };
+}) => {
+  let fileType = content?.file_type;
+  const source = content?.metadata?.source;
+  if (source) {
+    fileType = source.split('.').slice(-1)[0];
+  } else {
+    return content?.file_source || '';
+  }
+  if (fileViewTypes.includes(fileType)) {
+    if (source.includes('http')) {
+      return source;
+    } else {
+      return `${TempBaseUrl}${source}`;
+    }
+  } else {
+    return content?.file_source || '';
+  }
+};
+
+export const getFileName = (content: {
+  file_name?: string;
+  file_source?: string;
+  metadata: { source?: string };
+}) => {
+  if (!content) return '';
+  const { file_name } = content;
+  if (file_name) return file_name;
+  const source = content?.metadata?.source;
+  if (source) {
+    const arr = source.split('/');
+    if (!arr.length) return '';
+    return arr[arr.length - 1];
+  } else {
+    return '';
+  }
+};
+
+export const getFileType = (content: { metadata: { source?: string } }, filePath: string) => {
+  if (!content) return '';
+  const source = content?.metadata?.source;
+  if (source) {
+    const arr = source.split('.');
+    if (!arr.length) return '';
+    const type = arr[arr.length - 1];
+
+    const _arr = (filePath || '').split('.');
+    if (!_arr.length) return '';
+    const _type = _arr[_arr.length - 1];
+    switch (type.toLocaleLowerCase()) {
+      case 'ppt':
+      case 'pub':
+      case 'pptx':
+        return 'pdf_slides';
+      case 'xls':
+      case 'xlsx':
+      case 'csv':
+      case 'md':
+        if (_type === 'pdf') return 'pdf';
+        return 'markdown_table';
+      case 'png':
+      case 'jpg':
+      case 'jpeg':
+        return 'image';
+      case 'html':
+      case 'htm':
+        return 'html';
+      case 'txt':
+        return 'text';
+      default:
+        if (_type === 'ppt' || _type === 'pub' || _type === 'pptx') return 'pdf_slides';
+        return 'pdf';
+    }
+  } else {
+    return '';
+  }
+};
+
+export const downloadByStream = async ({
+  filePath,
+  fileName,
+  token,
+}: {
+  filePath: string;
+  fileName: string;
+  token: string;
+}) => {
+  if (filePath && fileName) {
+    try {
+      const response: any = await axios.post(
+        '/proxy/index/knowledgeBase/file/stream',
+        { convertedFilePath: filePath },
+        {
+          headers: {
+            'Content-Type': 'application/json',
+            Authorization: token,
+          },
+          responseType: 'arraybuffer',
+        }
+      );
+      //const response = await fetch(filePath);
+      console.log('response', response);
+      const blob = new Blob([response.data]);
+      const url = window.URL.createObjectURL(blob);
+      const _link = document.createElement('a');
+      const _body = document.body;
+
+      _link.href = url;
+      _link.download = fileName;
+      _link.style.display = 'none';
+      _body.appendChild(_link);
+
+      const event = new MouseEvent('click', {
+        view: window,
+        bubbles: true,
+        cancelable: true,
+      });
+      _link.dispatchEvent(event);
+
+      setTimeout(() => {
+        _body.removeChild(_link);
+        window.URL.revokeObjectURL(url);
+      }, 100);
+    } catch (error) {
+      console.error('下载文件失败：', error);
+    }
+  } else {
+    console.error('缺少文件名或文件地址：');
+  }
+};
+
+export const getInitialZoom = (boxWidth: number, curContent: any) => {
+  try {
+    if (!curContent?.metadata?.page_size?.length) {
+      const filePath = dealMinioUrl(getFilePath(curContent));
+      const fileType = getFileType(curContent, filePath);
+      let file_type = fileType;
+      const s = curContent?.file_name || curContent?.metadata?.source;
+      if (s) {
+        const sarr = s.split('.');
+        file_type = sarr?.[sarr.length - 1];
+      }
+      return judgePPT(fileType) || judgePPT(file_type) ? 0.6 : 1;
+    } else {
+      let pageSize = curContent?.metadata?.page_size || [[0, 0]];
+      if (!Array.isArray(pageSize)) {
+        pageSize = JSON.parse(pageSize);
+      }
+      const pageWidth = pageSize[0]?.[0] || 0;
+      const pageHeight = pageSize[0]?.[1] || 0;
+      if (pageHeight > pageWidth) return 1;
+      if (!boxWidth || !pageWidth) return 1;
+      if (pageWidth <= boxWidth) return 1;
+      return +(boxWidth / pageWidth).toFixed(4);
+    }
+  } catch (error) {
+    console.log('error', error);
+    return 1;
+  }
+};
+
+export const judgePPT = (type: string) => {
+  const types = ['ppt', 'pptx', 'pub'];
+  return types.includes(type.toLocaleLowerCase());
+};