@mxmweb/aichat 1.1.8 → 1.2.0

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

@@ -1,308 +0,0 @@
-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;