@gientech/modual 1.4.7 → 1.5.2

package/src/modules/chat/ReferenceBar.tsx

@@ -0,0 +1,522 @@
+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;
+          }),
+        });
+        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/constants.tsx

@@ -1,30 +1,53 @@
 import { Icon } from '@mxmweb/zui';
+import { FileSymlink, Mic, Paperclip } from 'lucide-react';
 
 export const DefaultSenderConfig = {
   actions: [
     {
-      icon: <Icon type='rag/folder' size={18} />,
       name: 'history',
-      badgeCount: 0,
+      icon: <Paperclip color='#4E5969' size={20}/>,
+      // badgeCount: 0,
+      tip: '录音',
     },
+    {
+      name: 'upload',
+      icon: <Mic color='#4E5969' size={20}/>,
+      tip: '上传',
+    }
   ],
 
+  // {
+  //   name: 'reasoning',
+  //   label: '思维链',
+  //   type: 'switch' as const,
+  //   defaultValue: false,
+  //   enabled: true,
+  //   icon: <LoaderPinwheel size={16} />,
+  // },
+  // {
+  //   name: 'netSearch',
+  //   label: '网络搜索',
+  //   type: 'switch' as const,
+  //   defaultValue: true,
+  //   enabled: true,
+  //   icon: <Globe size={16} />,
+  // },
   switchs: [
     {
       label: '联网搜索',
       name: 'netSearch',
-      type: 'checkbox',
+      type: 'switch',
       defaultValue: false,
       enabled: true,
-      icon: <Icon type='rag/url' size={12} />,
+      icon: <Icon type='rag/url' size={16} />,
     },
     {
       label: '推理思考',
       name: 'reasoning',
-      type: 'checkbox',
+      type: 'switch',
       defaultValue: false,
       enabled: true,
-      icon: <Icon type='rag/think' size={12} />,
+      icon: <Icon type='rag/think' size={16} />,
     },
   ],
 };