chore: import upstream snapshot with attribution

libs/react-client/src/useChatInteract.ts

@@ -0,0 +1,224 @@
+import { useCallback, useContext } from 'react';
+import { useRecoilValue, useResetRecoilState, useSetRecoilState } from 'recoil';
+import {
+  actionState,
+  askUserState,
+  chatSettingsInputsState,
+  chatSettingsValueState,
+  currentThreadIdState,
+  elementState,
+  favoriteMessagesState,
+  firstUserInteraction,
+  loadingState,
+  messagesState,
+  sessionIdState,
+  sessionState,
+  sideViewState,
+  tasklistState,
+  threadIdToResumeState,
+  tokenCountState
+} from 'src/state';
+import { IFileRef, IStep } from 'src/types';
+import { addMessage } from 'src/utils/message';
+import { v4 as uuidv4 } from 'uuid';
+
+import { ChainlitContext } from './context';
+
+type PartialBy<T, K extends keyof T> = Omit<T, K> & Partial<Pick<T, K>>;
+
+const useChatInteract = () => {
+  const client = useContext(ChainlitContext);
+  const session = useRecoilValue(sessionState);
+  const askUser = useRecoilValue(askUserState);
+  const sessionId = useRecoilValue(sessionIdState);
+
+  const resetChatSettings = useResetRecoilState(chatSettingsInputsState);
+  const resetSessionId = useResetRecoilState(sessionIdState);
+  const resetChatSettingsValue = useResetRecoilState(chatSettingsValueState);
+
+  const setFirstUserInteraction = useSetRecoilState(firstUserInteraction);
+  const setLoading = useSetRecoilState(loadingState);
+  const setMessages = useSetRecoilState(messagesState);
+  const setElements = useSetRecoilState(elementState);
+  const setTasklists = useSetRecoilState(tasklistState);
+  const setActions = useSetRecoilState(actionState);
+  const setTokenCount = useSetRecoilState(tokenCountState);
+  const setIdToResume = useSetRecoilState(threadIdToResumeState);
+  const setSideView = useSetRecoilState(sideViewState);
+  const setCurrentThreadId = useSetRecoilState(currentThreadIdState);
+  const setFavoriteMessages = useSetRecoilState(favoriteMessagesState);
+
+  const clear = useCallback(() => {
+    session?.socket.emit('clear_session');
+    session?.socket.disconnect();
+    setIdToResume(undefined);
+    resetSessionId();
+    setFirstUserInteraction(undefined);
+    setMessages([]);
+    setElements([]);
+    setTasklists([]);
+    setActions([]);
+    setTokenCount(0);
+    resetChatSettings();
+    resetChatSettingsValue();
+    setSideView(undefined);
+    setCurrentThreadId(undefined);
+  }, [session]);
+
+  const sendMessage = useCallback(
+    (
+      message: PartialBy<IStep, 'createdAt' | 'id'>,
+      fileReferences: IFileRef[] = []
+    ) => {
+      if (!message.id) {
+        message.id = uuidv4();
+      }
+      if (!message.createdAt) {
+        message.createdAt = new Date().toISOString();
+      }
+      setMessages((oldMessages) => addMessage(oldMessages, message as IStep));
+
+      session?.socket.emit('client_message', { message, fileReferences });
+    },
+    [session?.socket]
+  );
+
+  const editMessage = useCallback(
+    (message: IStep) => {
+      session?.socket.emit('edit_message', { message });
+    },
+    [session?.socket]
+  );
+
+  const toggleMessageFavorite = useCallback(
+    (message: IStep) => {
+      const favorite = !(message.metadata?.favorite ?? false);
+      const updatedMetadata = {
+        ...(message.metadata || {}),
+        favorite
+      };
+
+      setMessages((oldMessages) =>
+        oldMessages.map((item) =>
+          item.id === message.id
+            ? { ...item, metadata: { ...(item.metadata || {}), favorite } }
+            : item
+        )
+      );
+
+      const nextMessage: IStep = {
+        ...message,
+        metadata: updatedMetadata
+      };
+
+      setFavoriteMessages((oldFavorites) => {
+        if (favorite) {
+          const filtered = oldFavorites.filter(
+            (step) => step.id !== message.id
+          );
+          return [nextMessage, ...filtered];
+        }
+        return oldFavorites.filter((step) => step.id !== message.id);
+      });
+
+      session?.socket.emit('message_favorite', { message: nextMessage });
+    },
+    [session?.socket, setFavoriteMessages, setMessages]
+  );
+
+  const windowMessage = useCallback(
+    (data: any) => {
+      session?.socket.emit('window_message', data);
+    },
+    [session?.socket]
+  );
+
+  const startAudioStream = useCallback(() => {
+    session?.socket.emit('audio_start');
+  }, [session?.socket]);
+
+  const sendAudioChunk = useCallback(
+    (
+      isStart: boolean,
+      mimeType: string,
+      elapsedTime: number,
+      data: Int16Array
+    ) => {
+      session?.socket.emit('audio_chunk', {
+        isStart,
+        mimeType,
+        elapsedTime,
+        data
+      });
+    },
+    [session?.socket]
+  );
+
+  const endAudioStream = useCallback(() => {
+    session?.socket.emit('audio_end');
+  }, [session?.socket]);
+
+  const replyMessage = useCallback(
+    (message: IStep) => {
+      if (askUser) {
+        if (askUser.parentId) message.parentId = askUser.parentId;
+        setMessages((oldMessages) => addMessage(oldMessages, message));
+        askUser.callback(message);
+      }
+    },
+    [askUser]
+  );
+
+  const updateChatSettings = useCallback(
+    (values: object) => {
+      session?.socket.emit('chat_settings_change', values);
+    },
+    [session?.socket]
+  );
+
+  const editChatSettings = useCallback(
+    (values: object) => {
+      session?.socket.emit('chat_settings_edit', values);
+    },
+    [session?.socket]
+  );
+
+  const stopTask = useCallback(() => {
+    setMessages((oldMessages) =>
+      oldMessages.map((m) => {
+        m.streaming = false;
+        return m;
+      })
+    );
+
+    setLoading(false);
+
+    session?.socket.emit('stop');
+  }, [session?.socket]);
+
+  const uploadFile = useCallback(
+    (file: File, onProgress: (progress: number) => void, parentId?: string) => {
+      return client.uploadFile(file, onProgress, sessionId, parentId);
+    },
+    [sessionId]
+  );
+
+  return {
+    uploadFile,
+    clear,
+    replyMessage,
+    sendMessage,
+    editMessage,
+    windowMessage,
+    startAudioStream,
+    sendAudioChunk,
+    endAudioStream,
+    stopTask,
+    setIdToResume,
+    updateChatSettings,
+    editChatSettings,
+    toggleMessageFavorite
+  };
+};
+
+export { useChatInteract };