feat: add ragflow web project & add pnpm workspace file

2025-11-09 11:18:58 +08:00
parent ed6e0ab282
commit b2053760be
1566 changed files with 218623 additions and 57 deletions
--- a/ragflow_web/src/components/message-item/index.tsx
+++ b/ragflow_web/src/components/message-item/index.tsx
@@ -0,0 +1,166 @@
+import { ReactComponent as AssistantIcon } from '@/assets/svg/assistant.svg';
+import { MessageType } from '@/constants/chat';
+import { IReference, IReferenceChunk } from '@/interfaces/database/chat';
+import classNames from 'classnames';
+import { memo, useCallback, useEffect, useMemo } from 'react';
+
+import {
+  useFetchDocumentInfosByIds,
+  useFetchDocumentThumbnailsByIds,
+} from '@/hooks/document-hooks';
+import { IRegenerateMessage, IRemoveMessageById } from '@/hooks/logic-hooks';
+import { cn } from '@/lib/utils';
+import { IMessage } from '@/pages/chat/interface';
+import MarkdownContent from '@/pages/chat/markdown-content';
+import { Avatar, Flex, Space } from 'antd';
+import { ReferenceDocumentList } from '../next-message-item/reference-document-list';
+import { InnerUploadedMessageFiles } from '../next-message-item/uploaded-message-files';
+import { useTheme } from '../theme-provider';
+import { AssistantGroupButton, UserGroupButton } from './group-button';
+import styles from './index.less';
+
+interface IProps extends Partial<IRemoveMessageById>, IRegenerateMessage {
+  item: IMessage;
+  reference: IReference;
+  loading?: boolean;
+  sendLoading?: boolean;
+  visibleAvatar?: boolean;
+  nickname?: string;
+  avatar?: string;
+  avatarDialog?: string | null;
+  clickDocumentButton?: (documentId: string, chunk: IReferenceChunk) => void;
+  index: number;
+  showLikeButton?: boolean;
+  showLoudspeaker?: boolean;
+}
+
+const MessageItem = ({
+  item,
+  reference,
+  loading = false,
+  avatar,
+  avatarDialog,
+  sendLoading = false,
+  clickDocumentButton,
+  index,
+  removeMessageById,
+  regenerateMessage,
+  showLikeButton = true,
+  showLoudspeaker = true,
+  visibleAvatar = true,
+}: IProps) => {
+  const { theme } = useTheme();
+  const isAssistant = item.role === MessageType.Assistant;
+  const isUser = item.role === MessageType.User;
+  const { data: documentList, setDocumentIds } = useFetchDocumentInfosByIds();
+  const { data: documentThumbnails, setDocumentIds: setIds } =
+    useFetchDocumentThumbnailsByIds();
+
+  const referenceDocumentList = useMemo(() => {
+    return reference?.doc_aggs ?? [];
+  }, [reference?.doc_aggs]);
+
+  const handleRegenerateMessage = useCallback(() => {
+    regenerateMessage?.(item);
+  }, [regenerateMessage, item]);
+
+  useEffect(() => {
+    const ids = item?.doc_ids ?? [];
+    if (ids.length) {
+      setDocumentIds(ids);
+      const documentIds = ids.filter((x) => !(x in documentThumbnails));
+      if (documentIds.length) {
+        setIds(documentIds);
+      }
+    }
+  }, [item.doc_ids, setDocumentIds, setIds, documentThumbnails]);
+
+  return (
+    <div
+      className={classNames(styles.messageItem, {
+        [styles.messageItemLeft]: item.role === MessageType.Assistant,
+        [styles.messageItemRight]: item.role === MessageType.User,
+      })}
+    >
+      <section
+        className={classNames(styles.messageItemSection, {
+          [styles.messageItemSectionLeft]: item.role === MessageType.Assistant,
+          [styles.messageItemSectionRight]: item.role === MessageType.User,
+        })}
+      >
+        <div
+          className={classNames(styles.messageItemContent, {
+            [styles.messageItemContentReverse]: item.role === MessageType.User,
+          })}
+        >
+          {visibleAvatar &&
+            (item.role === MessageType.User ? (
+              <Avatar size={40} src={avatar ?? '/logo.svg'} />
+            ) : avatarDialog ? (
+              <Avatar size={40} src={avatarDialog} />
+            ) : (
+              <AssistantIcon />
+            ))}
+
+          <Flex vertical gap={8} flex={1}>
+            <Space>
+              {isAssistant ? (
+                index !== 0 && (
+                  <AssistantGroupButton
+                    messageId={item.id}
+                    content={item.content}
+                    prompt={item.prompt}
+                    showLikeButton={showLikeButton}
+                    audioBinary={item.audio_binary}
+                    showLoudspeaker={showLoudspeaker}
+                  ></AssistantGroupButton>
+                )
+              ) : (
+                <UserGroupButton
+                  content={item.content}
+                  messageId={item.id}
+                  removeMessageById={removeMessageById}
+                  regenerateMessage={
+                    regenerateMessage && handleRegenerateMessage
+                  }
+                  sendLoading={sendLoading}
+                ></UserGroupButton>
+              )}
+
+              {/* <b>{isAssistant ? '' : nickname}</b> */}
+            </Space>
+            <div
+              className={cn(
+                isAssistant
+                  ? theme === 'dark'
+                    ? styles.messageTextDark
+                    : styles.messageText
+                  : styles.messageUserText,
+                { '!bg-bg-card': !isAssistant },
+              )}
+            >
+              <MarkdownContent
+                loading={loading}
+                content={item.content}
+                reference={reference}
+                clickDocumentButton={clickDocumentButton}
+              ></MarkdownContent>
+            </div>
+            {isAssistant && referenceDocumentList.length > 0 && (
+              <ReferenceDocumentList
+                list={referenceDocumentList}
+              ></ReferenceDocumentList>
+            )}
+            {isUser && documentList.length > 0 && (
+              <InnerUploadedMessageFiles
+                files={documentList}
+              ></InnerUploadedMessageFiles>
+            )}
+          </Flex>
+        </div>
+      </section>
+    </div>
+  );
+};
+
+export default memo(MessageItem);