Function Chat

apps/web/lib/fe/components/chat.tsx:152–469

({
  chat,
  chatMessages,
  documents,
  citations,
  onJumpToPage,
}: {
  chat: ChatResponse;
  chatMessages: ChatMessageResponse[];
  documents?: DocumentsWithIndexingStatus[];
  citations?: CitationResponse[];
  onJumpToPage?: (docId: string, pageIndex: number) => void;
})

};

export function Chat({
  chat,
  chatMessages,
  documents,
  citations,
  onJumpToPage,
}: {
  chat: ChatResponse;
  chatMessages: ChatMessageResponse[];
  documents?: DocumentsWithIndexingStatus[];
  citations?: CitationResponse[];
  onJumpToPage?: (docId: string, pageIndex: number) => void;
}) {
  const chatId = Id.from<ChatResponse>(chat.id);

  const [title, setTitle] = useState<string | undefined>(DEFAULT_CHAT_TITLE);
  const [isTitleGenerating, setIsTitleGenerating] = useState<boolean>(false);
  const [isProcessingDocuments, setIsProcessingDocuments] =
    useState<boolean>(false);
  const [numberOfProcessedDocuments, setNumberOfProcessedDocuments] =
    useState(0);
  const [processingDocumentName, setProcessingDocumentName] = useState("");
  const [processingDocumentStatus, setProcessingDocumentStatus] = useState("");
  const [processingDocumentsError, setProcessingDocumentsError] = useState("");
  const [messageCitations, setMessageCitations] = useState<
    Map<string /* messageId */, CitationResponse[]>
  >(toCitationMap(citations));
  const formRef = useRef(null);
  const messagesRef = useRef<Message[]>([]);
  const {
    messages,
    setMessages,
    input,
    handleInputChange,
    handleSubmit,
    append,
    isLoading,
  } = useChat({
    id: chat.id,
    api: postChatMessagesGenerateApiPath(chatId),
    onFinish: async (message) => {
      if (message.role !== "assistant") {
        return;
      }

      // Fetch latest message from backend to get the correct message id
      const { response: chatMessageResponses } = await get<
        ChatMessageResponse[]
      >(
        getChatMessagesApiPath({
          chatId: chatId,
          ordering: {
            orderBy: "createdAt",
            order: "desc",
          },
          pagination: {
            page: 1,
            pageSize: 1,