@mxmweb/aichat 1.1.1 → 1.1.8

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

@@ -0,0 +1,308 @@
+import { EventSourceMessage, fetchEventSource } from '@microsoft/fetch-event-source';
+
+import { uid } from 'uid';
+import { AppStatusManager } from '../../../components/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;
+
+const AichatUseController = ({
+  baseUrl,
+  token,
+  setChatData,
+  activeSessionId,
+  setAppStatus,
+  setConversationList,
+}: {
+  baseUrl: string;
+  token: string;
+  setChatData: (arg: any[] | ((prev: any[]) => any[])) => void;
+  activeSessionId: string | undefined;
+  setAppStatus: React.Dispatch<React.SetStateAction<AppStatusManager>>;
+  setConversationList: React.Dispatch<React.SetStateAction<any[]>>;
+}) => {
+  const reset_chat = ({
+    content,
+    status,
+    respTime,
+    filePaths,
+    reference,
+    llmType,
+    resultType,
+    queryId,
+    tokens,
+  }: {
+    content: string;
+    status: StatusType;
+    respTime: number;
+    filePaths: any[];
+    reference: string;
+    llmType: number;
+    resultType: number;
+    queryId?: string;
+    tokens: number;
+    [key: string]: any;
+  }) => {
+    setChatData((chatList: any) => {
+      if (!chatList[0]) return chatList;
+      chatList[0].isNew = false;
+
+      const { messages } = chatList[0];
+      const lastMessage = messages[messages.length - 1];
+      if (!lastMessage && lastMessage.istype !== 'ai') return;
+
+      if (filePaths?.length) {
+        const lastSecondMessage = messages[messages.length - 2];
+        if (lastSecondMessage && lastSecondMessage.istype === 'user') {
+          lastSecondMessage.filePaths = filePaths;
+        }
+      }
+
+      lastMessage.content = lastMessage.content + content;
+      lastMessage.status = status;
+      lastMessage.respTime = respTime;
+      lastMessage.filePaths = filePaths;
+      lastMessage.reference = reference;
+      lastMessage.resultType = resultType;
+      lastMessage.llmType = llmType;
+      lastMessage.tokens = tokens;
+      lastMessage.queryId = queryId || lastMessage.queryId || uid(32);
+      lastMessage.clientSideId = lastMessage.clientSideId || lastMessage.queryId + '-ai';
+      return [...chatList];
+    });
+  };
+
+  const onErrorCallback = (err: any) => {
+    if (err instanceof FatalError) {
+      setAppStatus(pre => ({ ...pre, display: 'error', sender: 'ready' }));
+      if (currentStreamController) {
+        currentStreamController.abort();
+        currentStreamController = null;
+      }
+      throw err;
+    }
+    // 其他错误可选处理
+  };
+
+  const api_startChat_re = (data: Api_Chat, onSuccess?: (gmtModified: number) => void) => {
+    if (currentStreamController) {
+      currentStreamController.abort();
+    }
+    currentStreamController = new AbortController();
+    const eventSourceUrl = `${baseUrl}/qa/ai/question_re`;
+    console.log('data', data);
+    const params = {
+      sessionId: data.sessionId,
+      queryId: data.queryId,
+      name: data.name,
+      lastDate: data.lastDate,
+      type: data.type,
+      content: data.content,
+      configId: data.configId,
+      audioUrl: data.audioUrl,
+      fileUids: data.fileUids,
+      sysType: data.sysType,
+    };
+    fetchEventSource(eventSourceUrl, {
+      method: 'post',
+      signal: currentStreamController.signal,
+      headers: {
+        'Content-Type': 'application/json',
+        Accept: 'text/event-stream',
+        Authorization: token || '',
+      },
+      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);
+        if (activeSessionId !== curData.sessionId) return;
+        const { reply, resultType, respTime, reference, filePaths, status, docs, llmType } =
+          curData;
+        const _filePaths = filePaths ? JSON.parse(filePaths) : [];
+        //console.log('curData```````````````````````',curData,curData.tokens);
+        if (status === StatusType.Done) {
+          reset_chat({
+            content: reply.content || '',
+            status: resultType === 2 ? StatusType.Process : StatusType.Done,
+            respTime: respTime,
+            filePaths: _filePaths,
+            reference: reference || docs || '',
+            llmType,
+            resultType,
+            queryId: curData.queryId,
+            tokens: curData.tokens,
+          });
+          setAppStatus(pre => {
+            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 || '',
+            llmType,
+            resultType,
+            queryId: curData.queryId,
+            tokens: curData.tokens,
+          });
+        }
+      },
+      onerror: onErrorCallback,
+    });
+  };
+
+  const setLastEmptyMessage = ({ now }: { now: number }) => {
+    setChatData((chatList: any[]) => {
+      if (!chatList[0]) return chatList;
+      const { messages } = chatList[0];
+      if (!messages?.length) return chatList;
+      const lastIndex = messages.length - 1;
+      if (lastIndex > -1) {
+        const lastMessage = messages[lastIndex];
+        if (lastMessage.istype !== 'ai') return chatList;
+        messages[lastIndex] = {
+          istype: 'ai',
+          content: '', // AI回复流式填充
+          time: now,
+          streaming: true,
+          status: StatusType.Pending,
+        };
+
+        return [...chatList];
+      } else {
+        return chatList;
+      }
+    });
+  };
+
+  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;
+    }
+    setAppStatus(pre => ({ ...pre, display: 'ready', sender: 'ready' }));
+  };
+
+  const updateUserFilesMessage = ({ currentFiles }: { currentFiles: any[] }) => {
+    setChatData((chatList: any[]) => {
+      if (!chatList[0]) return chatList;
+      const { messages } = chatList[0];
+      if (!messages?.length) return chatList;
+      const lastIndex = messages.length - 2;
+      if (lastIndex > -1) {
+        const lastMessage = messages[lastIndex];
+        if (lastMessage.istype !== 'user') return chatList;
+        messages[lastIndex] = {
+          ...messages[lastIndex],
+          currentFiles,
+        };
+
+        return [...chatList];
+      } else {
+        return chatList;
+      }
+    });
+  };
+
+  return {
+    updateUserFilesMessage,
+    api_startChat_re,
+    addEmptyMessage,
+    setLastEmptyMessage,
+    stopStream,
+  };
+};
+
+export default AichatUseController;