@gientech/modual 1.5.0 → 1.5.2

package/src/main.tsx.backup

@@ -1,5 +1,5 @@
-import React from 'react';
-import ReactDOM from 'react-dom/client';
-import Chat from './examples/chat/index';
-
-ReactDOM.createRoot(document.getElementById('root')!).render(<Chat />);
+import React from 'react';
+import ReactDOM from 'react-dom/client';
+import Chat from './examples/chat/index';
+
+ReactDOM.createRoot(document.getElementById('root')!).render(<Chat />);

package/src/modules/chat/ReferenceBar.tsx

@@ -1,13 +1,522 @@
-export default function ReferenceBar(data: any) {
-  return (
-    <div>
-      <h1>ReferenceBar</h1>
-      <div onClick={() => {
-        const closeEvent = new CustomEvent('aichat:right_set', {
-          detail: { collapsed: true },
+import { ConfigProvider, Drawer, Tooltip } from 'antd';
+import SingleQa from '../../assets/img/singleQa.png';
+import DownLoad from '../../assets/img/downLoad.png';
+import GraphIcon from '../../assets/img/graphIcon.png';
+import WebSearch from '../../assets/img/webSearch.png';
+import Database from '../../assets/img/database.png';
+import { useEffect, useMemo, useState } from 'react';
+import './style.css';
+import DrawerGraphPreview from './referenceCom/DrawerGraphPreview';
+import { DrawerContent } from './referenceCom/DrawerContent';
+import DrawerPreview from './referenceCom/DrawerPreview';
+import DrawerDatabase from './referenceCom/DrawerDatabase';
+import DrawerTitle from './referenceCom/DrawerTitle';
+import TagCom from './referenceCom/TagCom';
+import {
+  downloadByStream,
+  getFileIcon,
+  getFileName,
+  getMarkExcludeType,
+  TempBaseUrl,
+} from './utils';
+import { DrawerType } from '../../lib_enter';
+
+enum ListName {
+  KNOWLEDGE = 'knowledgeList',
+  SEARCH = 'searchList',
+  DATABASE = 'databaseList',
+  GRAPH = 'graphList',
+}
+
+export const bgColor: any = {
+  [ListName.KNOWLEDGE]: '#EAF4FD',
+  [ListName.SEARCH]: '#E8F7F0',
+  [ListName.DATABASE]: '#FFF4E5',
+  [ListName.GRAPH]: '#E7F8FB',
+};
+
+export const textColor: any = {
+  [ListName.KNOWLEDGE]: '#2072F0',
+  [ListName.SEARCH]: '#39B83F',
+  [ListName.DATABASE]: '#F48C14',
+  [ListName.GRAPH]: '#10BBD9',
+};
+
+const BTNS = [
+  {
+    key: ListName.KNOWLEDGE,
+    name: '知识库文档',
+  },
+  {
+    key: ListName.SEARCH,
+    name: '联网搜索',
+    hideBtns: true,
+  },
+  {
+    key: ListName.DATABASE,
+    name: '数据库',
+  },
+  {
+    key: ListName.GRAPH,
+    name: '知识图谱',
+    hideBtns: true,
+  },
+];
+
+export const ReferenceBar = ({
+  type,
+  data,
+  token,
+  eventsEmit,
+  is_download,
+}: {
+  type: string;
+  data: any;
+  token: string;
+  eventsEmit?: (eventName: string, data: any) => void;
+  is_download: boolean;
+}) => {
+  const [btns, setBtns] = useState(BTNS);
+  const [drawerType, setDrawerType] = useState<string>('mark');
+  const [drawerOpen, setDrawerOpen] = useState<boolean>(false);
+  const [curFileInfo, setCurFileInfo] = useState<any>({});
+  const [activeBtn, setActiveBtn] = useState<any>(ListName.KNOWLEDGE);
+  const [lists, seLists] = useState<any>({
+    [ListName.KNOWLEDGE]: [],
+    [ListName.SEARCH]: [],
+    [ListName.DATABASE]: [],
+    [ListName.GRAPH]: [],
+  });
+
+  const getEmptyBoxs = useMemo(() => {
+    const n = 4 - btns.length;
+    const arr = Array(n)
+      .fill(null)
+      .map((v, i) => {
+        return {
+          name: '',
+          key: 100 + i,
+        };
+      });
+
+    const r = [...btns, ...arr];
+    console.log('rrrrrr', r);
+
+    return r;
+  }, [btns]);
+
+  const filterScore = (score: any) => {
+    if (isNaN(score)) {
+      return '';
+    } else {
+      return (+score).toFixed(2);
+    }
+  };
+
+  const getIcon = (btn: string, filtType?: string) => {
+    switch (btn) {
+      case ListName.KNOWLEDGE:
+        return getFileIcon(filtType as any);
+      case ListName.SEARCH:
+        return WebSearch;
+      case ListName.DATABASE:
+        return Database;
+      case ListName.GRAPH:
+        return GraphIcon;
+    }
+  };
+
+  const dealData = (type: string, data: any[]) => {
+    if (type === ListName.GRAPH) {
+      return data.map(item => {
+        return {
+          ...item,
+          name: item.graphName,
+        };
+      });
+    }
+
+    if (type === ListName.KNOWLEDGE) {
+      return data.map(item => {
+        return {
+          ...item,
+          name: item.file_name,
+          score: item.rrf_score,
+        };
+      });
+    }
+
+    if (type === ListName.SEARCH) {
+      return data.map(item => {
+        return {
+          ...item,
+          name: item.file_name,
+        };
+      });
+    }
+
+    return data;
+  };
+
+  const clickTag = (file: any, tag: any, btn: any) => {
+    switch (btn.key) {
+      case ListName.KNOWLEDGE:
+        console.log('click knowledge tag', file, tag);
+        const curFile = file;
+        const { pdfPages, parsedFilePath, order_num, metadata } = curFile;
+        const id = tag.id || order_num || 'Unknown ID';
+        // 解析 source 字段
+        const sourceParts = metadata.source.split('/');
+        const fileName = sourceParts[sourceParts.length - 1] || '';
+        const fileParts = fileName.split('.');
+        const label = fileParts.slice(0, -1).join('.') || 'Unknown Label';
+        const file_type = fileParts[fileParts.length - 1] || 'Unknown Type';
+
+        setCurFileInfo({
+          id,
+          title: label,
+          file_name: fileName,
+          file_type: file_type,
+          file_source: getMarkExcludeType(file_type)
+            ? `${TempBaseUrl}${metadata.source}`
+            : parsedFilePath,
+          metadata: metadata,
+          pdfPages: pdfPages,
+        });
+        setDrawerType('mark');
+        setDrawerOpen(true);
+        break;
+      case ListName.SEARCH:
+        console.log('click search tag', file, tag);
+        break;
+      case ListName.DATABASE:
+        console.log('click database tag', file, tag);
+        break;
+      case ListName.GRAPH:
+        console.log('click graph tag', file, tag);
+        setDrawerType('graphPreview');
+        setCurFileInfo({
+          ...file,
+          traceNode: (data?.traceNode || []).find((item: any) => {
+            return item.id === tag.id;
+          }),
         });
-        window.dispatchEvent(closeEvent);
-      }}>关闭</div>
-    </div>
+        setDrawerOpen(true);
+        break;
+    }
+  };
+
+  useEffect(() => {
+    if (type !== DrawerType.REFERENCELIST) return;
+    console.log('data in reference drawer', data);
+    if (!data) return;
+    const { graphRefs, knowledgeRefs, webRefs } = data;
+    if (graphRefs?.length) {
+      lists[ListName.GRAPH] = dealData(ListName.GRAPH, graphRefs);
+    }
+    if (knowledgeRefs?.length) {
+      lists[ListName.KNOWLEDGE] = dealData(ListName.KNOWLEDGE, knowledgeRefs);
+    }
+    if (webRefs?.length) {
+      lists[ListName.SEARCH] = dealData(ListName.SEARCH, webRefs);
+    }
+
+    const curBtns = BTNS.filter((item: any) => {
+      return lists[item.key]?.length;
+    });
+
+    setBtns(curBtns);
+
+    if (curBtns?.length) {
+      setActiveBtn(curBtns[0].key);
+    }
+
+    seLists({ ...lists });
+  }, [data]);
+
+  return (
+    <>
+      <div className="listWrap w-full h-full">
+        <DrawerTitle
+          title="知识溯源"
+          setOpen={() => {
+            const closeEvent = new CustomEvent('aichat:right_set', {
+              detail: { collapsed: true },
+            });
+            window.dispatchEvent(closeEvent);
+          }}
+        />
+        <div style={{ height: 'calc(100% - 60px)' }}>
+          <div
+            className="tabs"
+            style={{ display: 'flex', justifyContent: 'space-between', padding: '12px 20px' }}
+          >
+            {getEmptyBoxs.map(item => (
+              <div
+                key={item.key}
+                className="tab-item"
+                style={{
+                  display: 'inline-block',
+                  padding: '5.5px 12px',
+                  backgroundColor: !item.name
+                    ? 'transparent'
+                    : activeBtn === item.key
+                      ? '#4E6EF2'
+                      : '#F5F6F9',
+                  cursor: 'pointer',
+                  borderRadius: '4px',
+                  color: activeBtn === item.key ? '#FFFFFF' : '#4E5969',
+                  fontSize: '14px',
+                  minWidth: '80px',
+                }}
+                onClick={() => {
+                  if (item.name) {
+                    setActiveBtn(item.key);
+                  }
+                }}
+              >
+                {item.name}
+              </div>
+            ))}
+          </div>
+          <div
+            className="listBox"
+            style={{
+              height: 'calc(100% - 66px)',
+              overflowY: 'auto',
+              display: 'flex',
+              flexDirection: 'column',
+              alignItems: 'center',
+            }}
+          >
+            {btns.map((item: any) => {
+              let list: any[] = lists[item.key] || [];
+              if (activeBtn !== item.key) return null;
+              return (
+                <div key={item.key}>
+                  {list.map(
+                    (i: {
+                      name: string;
+                      id: string;
+                      score: number | string;
+                      tags?: any[];
+                      hideBtns?: boolean;
+                      file_type?: string;
+                      link?: string;
+                      metadata: any;
+                      file_name: any;
+                      parsedFilePath: any;
+                    }) => {
+                      return (
+                        <div
+                          key={i.id}
+                          style={{
+                            width: '360px',
+                            padding: '4px 8px',
+                            borderRadius: '4px',
+                            margin: '6px 0',
+                            cursor: 'pointer',
+                          }}
+                          className="listItemBox"
+                          onClick={() => {
+                            if (item.key === ListName.SEARCH) {
+                              window.open(i.link, '_blank');
+                            }
+                          }}
+                        >
+                          <div
+                            style={{
+                              display: 'flex',
+                              justifyContent: 'space-between',
+                              alignItems: 'center',
+                            }}
+                          >
+                            <div
+                              style={{
+                                width: item.hideBtns ? '339px' : '288px',
+                                fontSize: '14px',
+                                color: '#1D2129',
+                                fontWeight: '700',
+                                display: 'flex',
+                                alignItems: 'center',
+                              }}
+                            >
+                              <img
+                                src={getIcon(item.key, i.file_type)}
+                                style={{ width: '16px', height: '16px', marginRight: '2px' }}
+                              />
+                              <Tooltip
+                                placement="topLeft"
+                                title={i?.name?.length > 16 ? i.name : ''}
+                              >
+                                <div
+                                  style={{
+                                    whiteSpace: 'nowrap',
+                                    overflow: 'hidden',
+                                    textOverflow: 'ellipsis',
+                                    width: item.hideBtns ? '322px' : '258px',
+                                  }}
+                                >
+                                  {i.name}
+                                </div>
+                              </Tooltip>
+                            </div>
+                            <div
+                              style={{
+                                width: '52px',
+                                justifyContent: 'space-between',
+                                alignItems: 'center',
+                                marginRight: '8px',
+                                display: item.hideBtns ? 'none' : 'flex',
+                              }}
+                            >
+                              <img
+                                src={SingleQa}
+                                style={{ width: '16px', height: '16px', cursor: 'pointer' }}
+                                onClick={() => {
+                                  eventsEmit?.('referenceFile_filechat', {
+                                    data: {
+                                      ...i,
+                                      file_type: i.file_type,
+                                      file_name: i.file_name,
+                                      metadata: i.metadata,
+                                      parsedFilePath: i.parsedFilePath,
+                                    },
+                                  });
+                                }}
+                              />
+                              <div
+                                style={{ borderRight: '1px solid #E5E6EB', height: '14px' }}
+                              ></div>
+                              <img
+                                src={DownLoad}
+                                style={{
+                                  width: '16px',
+                                  height: '16px',
+                                  cursor: is_download ? 'pointer' : 'not-allowed',
+                                }}
+                                onClick={() => {
+                                  if (!i?.metadata || !is_download) return;
+                                  downloadByStream({
+                                    filePath: `${TempBaseUrl}${i.metadata.source}`,
+                                    fileName: getFileName(i),
+                                    token,
+                                  });
+                                }}
+                              />
+                            </div>
+                          </div>
+                          {(typeof i.score === 'string' ||
+                            typeof i.score === 'number' ||
+                            (i?.tags?.length && i.tags.length > 0 && i.tags.length <= 10)) && (
+                            <div
+                              style={{
+                                display: 'flex',
+                                justifyContent: 'space-between',
+                                height: '18px',
+                                marginTop: '4px',
+                              }}
+                            >
+                              {(typeof i.score === 'string' || typeof i.score === 'number') && (
+                                <div
+                                  style={{
+                                    backgroundColor: '#F2F2FB',
+                                    borderRadius: '4px',
+                                    padding: '2px 3px',
+                                    display: 'flex',
+                                    justifyContent: 'center',
+                                    alignItems: 'center',
+                                    fontSize: '12px',
+                                  }}
+                                >
+                                  <span style={{ color: '#4E5969', marginRight: '6px' }}>
+                                    分数:
+                                  </span>
+                                  <span style={{ color: '#1D2129' }}>{filterScore(i.score)}</span>
+                                </div>
+                              )}
+                              <div style={{ flex: 1, display: 'flex', justifyContent: 'flex-end' }}>
+                                {(i?.tags?.length && i.tags.length > 0 && i.tags.length <= 10
+                                  ? i.tags
+                                  : []
+                                ).map((j: any) => {
+                                  return (
+                                    <TagCom
+                                      key={j.id}
+                                      i={i}
+                                      item={item}
+                                      j={j}
+                                      clickTag={clickTag}
+                                    />
+                                  );
+                                })}
+                              </div>
+                            </div>
+                          )}
+                          <div
+                            style={{
+                              flex: 1,
+                              display: 'flex',
+                              justifyContent: 'flex-start',
+                              marginTop: '4px',
+                            }}
+                          >
+                            {(i?.tags?.length && i.tags.length > 10 ? i.tags : []).map((j: any) => {
+                              return (
+                                <TagCom key={j.id} i={i} item={item} j={j} clickTag={clickTag} />
+                              );
+                            })}
+                          </div>
+                        </div>
+                      );
+                    }
+                  )}
+                </div>
+              );
+            })}
+          </div>
+        </div>
+      </div>
+      <ConfigProvider
+        drawer={{
+          styles: {
+            body: { padding: 0 },
+            header: { display: 'none' },
+          },
+        }}
+      >
+        <Drawer
+          autoFocus={false}
+          onClose={() => setDrawerOpen(false)}
+          open={drawerOpen}
+          width={720}
+          className="overflow-x-hidden"
+          destroyOnClose
+          getContainer={false}
+          afterOpenChange={open => {
+            if (!open) {
+              setCurFileInfo(null);
+            }
+          }}
+        >
+          {drawerType === 'graphPreview' && (
+            <DrawerGraphPreview
+              curFileInfo={curFileInfo as any}
+              onClose={setDrawerOpen}
+              token={token}
+            />
+          )}
+          {drawerType === 'mark' && (
+            <DrawerContent content={curFileInfo as any} onClose={setDrawerOpen} token={token} />
+          )}
+          {drawerType === 'preview' && (
+            <DrawerPreview curFileInfo={curFileInfo} onClose={setDrawerOpen} token={token} />
+          )}
+          {drawerType === 'database' && (
+            <DrawerDatabase curFileInfo={curFileInfo} onClose={setDrawerOpen} />
+          )}
+        </Drawer>
+      </ConfigProvider>
+    </>
   );
-}
+};

package/src/modules/chat/index.tsx

@@ -9,9 +9,10 @@ import '@mxmweb/rtext/style.css';
 import '@mxmweb/aichat/style.css';
 import '@mxmweb/zui/style.css';
 import { defaultTheme, deepMergeTheme, type Styles } from '@mxmweb/zui';
-import ReferenceBar from './ReferenceBar';
 import { DefaultSenderConfig } from './constants';
 import GientechConversationPanel from './Conversations';
+import { registerPDFWorker } from '@mxmweb/fviewer';
+registerPDFWorker('/worker/pdf.worker1.min.js');
 import {
   convertQueryReplyPairListToMessages,
   getFileTypeByName,
@@ -38,8 +39,7 @@ import DisplayError from '../../utils/gientechCommon/components/DisplayError';
 import { maxPollCount, maxPollInterval } from '../../utils/gientechCommon/configs/commonConfig';
 import { RenameModal } from '../../utils/gientechCommon/components/RenameModal';
 import { DeleteModal } from '../../utils/gientechCommon/components/DeleteModal';
-import useAichatEvent from '../../utils/gientechCommon/components/hooks/useAichatEvent';
-import { ReferenceListDrawer } from '../../utils/gientechCommon/components/ReferenceListDrawer';
+import { ReferenceBar } from './ReferenceBar';
 
 export enum DrawerType {
   REFERENCELIST = 'referencelist',
@@ -118,11 +118,6 @@ export default function withGientechChatAdopter(WrappedComponent = AiChat) {
     scrollOld,
     ...rest
   }: GientechChatAdopterProps) {
-    const [drawerConfig, setDrawerConfig] = useState<{ type: string; data?: any }>({
-      type: 'string',
-    });
-    const [open, setOpen] = useState(false);
-    const {} = useAichatEvent();
     // 统一归一化样式：使用 UI 库的 deepMergeTheme
     const normalizedStyles: Styles = useMemo(() => {
       const baseStyles: Styles = { theme: defaultTheme, mode: 'light' };
@@ -1482,6 +1477,7 @@ export default function withGientechChatAdopter(WrappedComponent = AiChat) {
         console.log('error', error);
       }
     };
+
     /**
      * 统一事件分发器，连接业务逻辑与所有子组件的事件
      * @param {string} eventName - 事件名称
@@ -1676,21 +1672,24 @@ export default function withGientechChatAdopter(WrappedComponent = AiChat) {
           eventsEmit?.('referenceFile_download', data);
           break;
         case 'references:click': {
-          setDrawerConfig({
-            type: DrawerType.REFERENCELIST,
-            data,
-          });
-          setOpen(true);
           // 使用 AiChat 内置的右侧栏事件
           const Content = () => {
-            const safe = (() => {
-              try {
-                return typeof data === 'string' ? data : JSON.stringify(data, null, 2);
-              } catch (e) {
-                return String(data);
-              }
-            })();
-            return <ReferenceBar data={data} />;
+            // const safe = (() => {
+            //   try {
+            //     return typeof data === 'string' ? data : JSON.stringify(data, null, 2);
+            //   } catch (e) {
+            //     return String(data);
+            //   }
+            // })();
+            return (
+              <ReferenceBar
+                type={DrawerType.REFERENCELIST}
+                data={data}
+                token={token}
+                eventsEmit={eventsEmit}
+                is_download={is_download}
+              />
+            );
           };
           const setContentEvent = new CustomEvent('aichat:right_set_content', {
             detail: { content: <Content /> },
@@ -2041,6 +2040,7 @@ export default function withGientechChatAdopter(WrappedComponent = AiChat) {
           CustomComponents={mergedCustomComponents}
           activeSessionId={activeSessionId}
           scrollOld={scrollOld}
+          rightbarWidth={'400px'}
         />
         <FeedBackModal
           open={openFeed}
@@ -2060,14 +2060,6 @@ export default function withGientechChatAdopter(WrappedComponent = AiChat) {
           handleCancel={() => handleEvent('close_remove')}
           data={curEditData?.data || {}}
         />
-        <ReferenceListDrawer
-          open={open}
-          setOpen={setOpen}
-          type={drawerConfig.type}
-          data={drawerConfig.data}
-          token={token}
-          eventsEmit={eventsEmit}
-        />
       </div>
     );
   };

package/src/{utils/gientechCommon/components → modules/chat}/referenceCom/DrawerTitle.tsx

@@ -1,4 +1,4 @@
-import Close from '../../../../assets/img/close.png';
+import Close from '../../../assets/img/close.png';
 const DrawerTitle = ({ title, setOpen }: { title: string; setOpen: (arg: boolean) => void }) => {
   return (
     <div

package/src/{utils/gientechCommon/components → modules/chat}/referenceCom/TagCom.tsx

@@ -1,4 +1,4 @@
-import { bgColor, textColor } from '../ReferenceListDrawer';
+import { bgColor, textColor } from '../ReferenceBar';
 
 const TagCom = ({ i, item, j, clickTag }: any) => {
   console.log('j in tag', j);

package/src/{utils/gientechCommon/components → modules/chat}/utils/index.ts

@@ -1,9 +1,9 @@
-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 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';