@gientech/modual 2.0.4 → 2.0.5

package/src/utils/gientechCommon/components/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/utils/gientechCommon/components/style.less

@@ -0,0 +1,11 @@
+.selfTree {
+  .ant-tree .ant-tree-treenode {
+    padding: 4px 0 4px 0;
+  }
+  .ant-tree-switcher {
+    display: none;
+  }
+  .ant-tree-indent {
+    display: none;
+  }
+}

package/src/utils/gientechCommon/configs/commonConfig.ts

@@ -0,0 +1,2 @@
+export const maxPollCount = 1000;
+export const maxPollInterval = 3000;

package/src/utils/gientechCommon/configs/stylesConfig.ts

@@ -0,0 +1,142 @@
+// 主题类型定义
+export type SizeType = 'sm' | 'md' | 'lg'
+export type FontWeightType = 'normal' | 'bold' |'thin'
+export type FontFamilyType = 'default' | 'serif' | 'mono'
+
+export interface AiChatStyles {
+  theme?: AppTheme
+  mode?: 'light' | 'dark'
+  [key: string]: any
+}
+
+export interface ThemeColors {
+  primary: string
+  secondary: string
+  success: string
+  warning: string
+  error: string
+  info: string
+  background: string
+  text: string
+  border: string
+  disabled: string
+  disabledBackground: string
+  disabledText: string
+  shadow: string
+  appBackground: string
+}
+
+export interface ThemeOthers {
+
+}
+
+export interface ThemeSpace {
+  sidebar: string
+  size: string
+  radius: string
+  padding: string
+  margin: string
+  shadow: string
+  lineHeight: string
+}
+
+export interface ThemeIcons {
+  aiAvatar: string
+  userAvatar: string
+  pdf: string
+  word: string
+  image: string
+  video: string
+  audio: string
+  excel: string
+  ppt: string
+}
+
+export interface AppTheme {
+  colors: ThemeColors
+
+  space: ThemeSpace
+  icons: ThemeIcons
+  others: ThemeOthers
+}
+
+// 默认主题实现
+export const defaultTheme: AppTheme = {
+  colors: {
+    primary: '#007bff',
+    secondary: '#6c757d',
+    success: '#008000',
+    warning: '#FFA500',
+    error: '#FF0000',
+    info: '#0000FF',
+    background: '#f8f9fa',
+    text: '#343a40',
+    border: '#dee2e6',
+    disabled: '#808080',
+    disabledBackground: '#F5F5F5',
+    disabledText: '#808080',
+    shadow: '#000000',
+    appBackground: '#fff',
+  },
+  space: {
+    sidebar: 'md',
+    size: 'md',
+    radius: 'md',
+    padding: 'md',
+    margin: 'md',
+    shadow: 'md',
+    lineHeight: 'md',
+  },
+  icons: {
+    aiAvatar: 'http://10.15.12.13:9000/dev-rag-data/header_pic/JIABINBIN00/%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20231212161213.jpg',
+    userAvatar: 'http://10.15.12.13:9005/assets/nav-logo-BKaOCDqi.png',
+    pdf: 'https://ai-chat-1326151111.cos.ap-shanghai.myqcloud.com/pdf-icon.png',
+    word: 'https://ai-chat-1326151111.cos.ap-shanghai.myqcloud.com/word-icon.png',
+    image: 'https://ai-chat-1326151111.cos.ap-shanghai.myqcloud.com/image-icon.png',
+    video: 'https://ai-chat-1326151111.cos.ap-shanghai.myqcloud.com/video-icon.png',
+    audio: 'https://ai-chat-1326151111.cos.ap-shanghai.myqcloud.com/audio-icon.png',
+    excel: 'https://ai-chat-1326151111.cos.c zap-shanghai.myqcloud.com/excel-icon.png',
+    ppt: 'https://ai-chat-1326151111.cos.ap-shanghai.myqcloud.com/ppt-icon.png',
+  },
+  others: {
+    baseFontWeight: 'normal',
+  },
+}
+
+// SizeType 映射表
+export const sidebarMap = {
+  sm: '240px',
+  md: '260px',
+  lg: '280px',
+};
+export const sizeMap = {
+  sm: '12px',
+  md: '16px',
+  lg: '20px',
+};
+export const radiusMap = {
+  sm: '2px',
+  md: '4px',
+  lg: '8px',
+};
+export const paddingMap = {
+  sm: '4px',
+  md: '8px',
+  lg: '16px',
+};
+export const marginMap = {
+  sm: '4px',
+  md: '8px',
+  lg: '16px',
+};
+export const shadowMap = {
+  sm: '0 1px 2px #eee',
+  md: '0 2px 8px #ddd',
+  lg: '0 4px 16px #ccc',
+};
+export const lineHeightMap = {
+  sm: '1.2',
+  md: '1.5',
+  lg: '2',
+};
+

package/src/utils/gientechCommon/hooks/AichatUseController.tsx

@@ -0,0 +1,417 @@
+import { EventSourceMessage, fetchEventSource } from '@microsoft/fetch-event-source';
+
+import { uid } from 'uid';
+import { AppStatusManager } from '@mxmweb/aichat';
+
+interface Api_Chat {
+  sessionId: string;
+  queryId: string;
+  name: string;
+  lastDate: string | number;
+  type: string;
+  content: string;
+  configId: string;
+  audioUrl: string;
+  fileUids: any[];
+  signal?: AbortSignal;
+  sysType?: 'KNO_INDEX' | 'INT_SESSION' | 'TEMPORARY';
+}
+
+export enum StatusType {
+  Pending = 0,
+  Process,
+  Done,
+  Error,
+}
+
+class FatalError extends Error {}
+
+let currentStreamController: AbortController | null = null;
+// 记录当前流式请求的 sessionId，用于在切换会话时判断是否需要取消请求
+let currentStreamSessionId: string | null = null;
+
+const AichatUseController = ({
+  baseUrl,
+  token,
+  CSRFToken,
+  setChatData,
+  activeSessionId,
+  setAppStatus,
+  setConversationList,
+}: {
+  baseUrl: string;
+  token: string;
+  CSRFToken?: string;
+  setChatData: (arg: any[] | ((prev: any[]) => any[])) => void;
+  activeSessionId: string | undefined;
+  setAppStatus: React.Dispatch<React.SetStateAction<AppStatusManager>>;
+  setConversationList: React.Dispatch<React.SetStateAction<any[]>>;
+}) => {
+  // 判断 graphReference 是否有有效值
+  const isValidGraphReference = (ref: any): boolean => {
+    if (!ref) return false;
+    if (typeof ref === 'string') {
+      const trimmed = ref.trim();
+      return trimmed !== '' && trimmed !== '[]' && trimmed !== 'null';
+    }
+    return true;
+  };
+
+  const reset_chat = ({
+    content,
+    status,
+    respTime,
+    filePaths,
+    reference,
+    webReference,
+    graphReference,
+    databaseReference,
+    llmType,
+    resultType,
+    queryId,
+    tokens,
+  }: {
+    content: string;
+    status: StatusType;
+    respTime: number;
+    filePaths: any[];
+    reference: string;
+    webReference: string;
+    graphReference?: string;
+    databaseReference?: any;
+    llmType: number;
+    resultType: number;
+    queryId?: string;
+    tokens: number;
+    [key: string]: any;
+  }) => {
+    setChatData((chatList: any) => {
+      if (!chatList[0]) return chatList;
+      const current = chatList[0];
+      const messages = current.messages || [];
+      const lastIndex = messages.length - 1;
+      if (lastIndex < 0) return chatList;
+      const lastMessage = messages[lastIndex];
+      if (lastMessage.istype !== 'ai') return chatList;
+
+      // 更新上一条 user 的 filePaths（不可变）
+      let newMessages = messages.slice();
+      if (filePaths?.length) {
+        const lastSecondIndex = messages.length - 2;
+        if (lastSecondIndex >= 0 && messages[lastSecondIndex]?.istype === 'user') {
+          const lastSecond = { ...messages[lastSecondIndex], filePaths };
+          newMessages[lastSecondIndex] = lastSecond;
+        }
+      }
+
+      // 只有当 graphReference 有有效值时才更新，否则保留之前的值
+      // 如果传入的 graphReference 是 undefined，说明本次没有更新，保留之前的值
+      // 如果传入的 graphReference 是 null 或无效值，也保留之前的值
+      const finalGraphReference = graphReference !== undefined
+        ? (isValidGraphReference(graphReference) ? graphReference : lastMessage.graphReference)
+        : lastMessage.graphReference;
+      // 调试：检查 reset_chat 中的 graphReference 处理
+      if (status === StatusType.Done && graphReference !== undefined) {
+        console.log('[AichatUseController] reset_chat status=2 时的 graphReference:', {
+          graphReference,
+          isValid: isValidGraphReference(graphReference),
+          lastMessageGraphReference: lastMessage.graphReference,
+          finalGraphReference,
+        });
+      }
+
+      const updatedLast = {
+        ...lastMessage,
+        content: (lastMessage.content || '') + content,
+        status,
+        respTime,
+        filePaths,
+        reference,
+        webReference,
+        graphReference: finalGraphReference,
+        // 流式阶段若后端带来了 databaseReference，写入当前 AI 消息
+        databaseReference: databaseReference ?? lastMessage.databaseReference,
+        resultType,
+        llmType,
+        tokens,
+        queryId: queryId || lastMessage.queryId || uid(32),
+      } as any;
+      updatedLast.clientSideId = updatedLast.clientSideId || updatedLast.queryId + '-ai';
+      newMessages[lastIndex] = updatedLast;
+
+      const newChat0 = { ...current, isNew: false, messages: newMessages };
+      return [newChat0, ...chatList.slice(1)];
+    });
+  };
+
+  const onErrorCallback = (err: any) => {
+    if (err instanceof FatalError) {
+      setAppStatus((pre:any) => ({ ...pre, display: 'error', sender: 'ready' }));
+      if (currentStreamController) {
+        currentStreamController.abort();
+        currentStreamController = null;
+      }
+      currentStreamSessionId = null;
+      throw err;
+    }
+    // 其他错误可选处理
+  };
+
+  const api_startChat_re = (data: Api_Chat, onSuccess?: (gmtModified: number) => void) => {
+    if (currentStreamController) {
+      currentStreamController.abort();
+    }
+    currentStreamController = new AbortController();
+    // 记录当前流式请求的 sessionId
+    currentStreamSessionId = data.sessionId;
+    const eventSourceUrl = `${baseUrl}/qa/ai/question_re`;
+    console.log('data', data);
+    const params = { ...data };
+    fetchEventSource(eventSourceUrl, {
+      method: 'post',
+      signal: currentStreamController.signal,
+      headers: {
+        'Content-Type': 'application/json',
+        Accept: 'text/event-stream',
+        Authorization: token || '',
+        'X-CSRF-TOKEN-IN': CSRFToken || '',
+      },
+      openWhenHidden: true,
+      body: JSON.stringify(params),
+      onopen: async response => {
+        if (response.status === 500) {
+          throw new FatalError('Internal Server Error');
+        }
+        if (!response.ok) {
+          throw new FatalError(`HTTP error: ${response.status}`);
+        }
+        // 其他状态可根据需要处理
+      },
+      onmessage: event => {
+        const curData = JSON.parse(event.data);
+        // 检查当前请求的 sessionId 是否仍然匹配，如果不匹配则忽略所有后续数据
+        if (currentStreamSessionId !== curData.sessionId) {
+          console.log('[AichatUseController] 流式请求的 sessionId 已不匹配，忽略数据', {
+            currentStreamSessionId,
+            receivedSessionId: curData.sessionId,
+            activeSessionId,
+          });
+          return;
+        }
+        if (activeSessionId !== curData.sessionId) return;
+        const {
+          reply,
+          resultType,
+          respTime,
+          reference,
+          webReference,
+          graphReference,
+          filePaths,
+          status,
+          docs,
+          llmType,
+          databaseReference,
+        } = curData;
+        const _filePaths = filePaths ? JSON.parse(filePaths) : [];
+        // === 新增：流式错误兜底 ===
+        if (
+          status === StatusType.Done &&
+          reply &&
+          reply.content &&
+          (reply.content.includes('模型暂无法返回') ||
+            reply.content.includes('失败') ||
+            reply.content.includes('错误') ||
+            reply.content.includes('请稍后再试'))
+        ) {
+          setAppStatus((pre:any) => ({ ...pre, display: 'ready', sender: 'ready' }));
+        }
+        // 处理 graphReference：只有当有有效值时才传递，否则传递 undefined 让 reset_chat 保留之前的值
+        const finalGraphReference = isValidGraphReference(graphReference) ? graphReference : undefined;
+        // 调试：检查最后一条消息的 graphReference
+        if (status === StatusType.Done) {
+          console.log('[AichatUseController] status=2 时的 graphReference:', {
+            graphReference,
+            type: typeof graphReference,
+            isValid: isValidGraphReference(graphReference),
+            finalGraphReference,
+          });
+        }
+
+        if (status === StatusType.Done) {
+          reset_chat({
+            content: reply.content || '',
+            status: StatusType.Done, // 流式完成时，消息状态应该是 Done
+            respTime: respTime,
+            filePaths: _filePaths,
+            reference: reference || docs || '',
+            webReference: webReference || '',
+            graphReference: finalGraphReference,
+            databaseReference,
+            llmType,
+            resultType,
+            queryId: curData.queryId,
+            tokens: curData.tokens,
+          });
+          setAppStatus((pre:any) => {
+            return {
+              ...pre,
+              display: 'ready',
+              sender: 'ready',
+            };
+          });
+          if (onSuccess) {
+            const gmtModified = respTime || Date.now();
+            console.log('[AichatUseController] 消息发送成功，gmtModified:', gmtModified);
+            onSuccess(gmtModified);
+          }
+        } else {
+          reset_chat({
+            content: reply.content || '',
+            status: status,
+            respTime: respTime,
+            filePaths: _filePaths,
+            reference: reference || docs || '',
+            webReference: webReference || '',
+            graphReference: finalGraphReference, // 流式过程中保持之前的graphReference值
+            databaseReference,
+            llmType,
+            resultType,
+            queryId: curData.queryId,
+            tokens: curData.tokens,
+          });
+        }
+      },
+      onerror: onErrorCallback,
+      onclose: () => {
+        // 流式请求结束时，清理 sessionId 记录
+        if (currentStreamSessionId === data.sessionId) {
+          currentStreamSessionId = null;
+        }
+      },
+    });
+  };
+
+  const setLastEmptyMessage = ({ now }: { now: number }) => {
+    setChatData((chatList: any[]) => {
+      if (!chatList[0]) return chatList;
+      const current = chatList[0];
+      const messages = current.messages || [];
+      if (!messages.length) return chatList;
+      const lastIndex = messages.length - 1;
+      if (lastIndex < 0) return chatList;
+      const lastMessage = messages[lastIndex];
+      if (lastMessage.istype !== 'ai') return chatList;
+      const newMessages = messages.slice();
+      newMessages[lastIndex] = {
+        ...lastMessage,
+        istype: 'ai',
+        content: '',
+        time: now,
+        streaming: true,
+        status: StatusType.Pending,
+      } as any;
+      const newChat0 = { ...current, messages: newMessages };
+      return [newChat0, ...chatList.slice(1)];
+    });
+  };
+
+  const addEmptyMessage = ({
+    queryId,
+    content,
+    filePaths,
+    currentFiles,
+  }: {
+    queryId: string;
+    content: string;
+    filePaths?: string;
+    currentFiles?: any[];
+  }) => {
+    setChatData((chatList: any[]) => {
+      const now = new Date().toISOString();
+      const _queryId = queryId || uid(32); // 保证只生成一次
+      const userMessage = {
+        istype: 'user',
+        content: content,
+        time: now,
+        queryId: _queryId,
+        clientSideId: _queryId + '-user',
+        filePaths: filePaths || '',
+        currentFiles: currentFiles || [],
+      };
+      const aiMessage = {
+        istype: 'ai',
+        content: '', // AI回复流式填充
+        time: new Date(Date.now() + 1).toISOString(),
+        streaming: true,
+        status: StatusType.Pending,
+        queryId: _queryId,
+        clientSideId: _queryId + '-ai',
+      };
+
+      // 找到当前激活的对话
+      const activeChat = chatList.find(chat => chat.id === activeSessionId);
+
+      if (activeChat) {
+        // 如果激活的对话存在，则向其添加新消息
+        activeChat.messages.push(userMessage, aiMessage);
+      } else {
+        // 如果没有激活的对话（例如首次加载），则需要特殊处理
+        // 这里的逻辑可能需要根据产品需求调整，是创建新对话还是报错
+        // 假设我们总是有一个激活的对话
+        console.warn('addEmptyMessage called without an active session.');
+      }
+      return [...chatList];
+    });
+  };
+
+  // 主动终止当前 stream
+  const stopStream = () => {
+    if (currentStreamController) {
+      currentStreamController.abort();
+      currentStreamController = null;
+    }
+    currentStreamSessionId = null;
+    setAppStatus((pre:any) => ({ ...pre, display: 'ready', sender: 'ready' }));
+  };
+
+  // 根据 sessionId 取消流式请求（用于切换会话时）
+  const cancelStreamBySessionId = (sessionId: string) => {
+    if (currentStreamController && currentStreamSessionId === sessionId) {
+      console.log('[AichatUseController] 切换会话，取消正在进行的流式请求', sessionId);
+      currentStreamController.abort();
+      currentStreamController = null;
+      currentStreamSessionId = null;
+      setAppStatus((pre:any) => ({ ...pre, display: 'ready', sender: 'ready' }));
+      return true;
+    }
+    return false;
+  };
+
+  const updateUserFilesMessage = ({ currentFiles }: { currentFiles: any[] }) => {
+    setChatData((chatList: any[]) => {
+      if (!chatList[0]) return chatList;
+      const current = chatList[0];
+      const messages = current.messages || [];
+      if (!messages.length) return chatList;
+      const lastIndex = messages.length - 2;
+      if (lastIndex < 0) return chatList;
+      const lastMessage = messages[lastIndex];
+      if (lastMessage.istype !== 'user') return chatList;
+      const newMessages = messages.slice();
+      newMessages[lastIndex] = { ...messages[lastIndex], currentFiles } as any;
+      const newChat0 = { ...current, messages: newMessages };
+      return [newChat0, ...chatList.slice(1)];
+    });
+  };
+
+  return {
+    updateUserFilesMessage,
+    api_startChat_re,
+    addEmptyMessage,
+    setLastEmptyMessage,
+    stopStream,
+    cancelStreamBySessionId,
+  };
+};
+
+export default AichatUseController;