fix(chat): fix alignment for chat's button

DTStack · Jan 23, 2025 · 91e5cd6 · 91e5cd6
1 parent f28dbcb
commit 91e5cd6
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 10 deletions.
diff --git a/src/chat/button/index.scss b/src/chat/button/index.scss
@@ -4,6 +4,9 @@ $primaryGradientHover: #08C4FF 0%, #4892FF 50%, #8A61FF 100%;
 .ant-btn.dtc__aigc__button {
     border: none;
     border-radius: 4px;
+    font-size: 12px;
+    display: inline-flex;
+    align-items: center;
     &.ant-btn-sm {
         border-radius: 2px;
     }

diff --git a/src/chat/demos/message.tsx b/src/chat/demos/message.tsx
@@ -1,19 +1,26 @@
 import React, { useMemo, useState } from 'react';
 import { Button, Space } from 'antd';
 import { Chat } from 'dt-react-component';
-import { MessageStatus } from 'dt-react-component/chat/entity';
+import { Message, MessageStatus, Prompt } from 'dt-react-component/chat/entity';
+
+class BasicPrompt extends Prompt {}
+class BasicMessage extends Message {}
 
 export default function () {
     const [status, setStatus] = useState<MessageStatus>(MessageStatus.DONE);
 
     const data = useMemo(() => {
-        return [
-            {
-                id: new Date().valueOf().toString(),
-                content: status === MessageStatus.DONE ? '输出完成' : '正在打字中...',
-                status,
-            } as any,
-        ];
+        return new BasicPrompt({
+            id: `prompt_${new Date().valueOf().toString()}`,
+            title: '这是一个标题',
+            messages: [
+                new BasicMessage({
+                    id: 'message_${new Date().valueOf().toString()}',
+                    content: status === MessageStatus.DONE ? '输出完成' : '正在打字中...',
+                    status,
+                }),
+            ],
+        });
     }, [status]);
 
     return (
@@ -30,7 +37,8 @@ export default function () {
                 </Button>
             </Space>
             <Chat.Message
-                data={data}
+                prompt={data}
+                data={data.messages}
                 regenerate
                 onStop={() => setStatus(MessageStatus.STOPPED)}
                 onRegenerate={() => console.log('regenerate')}

diff --git a/src/chat/message/index.tsx b/src/chat/message/index.tsx
@@ -96,7 +96,7 @@ export default function Message({
     const copyInfo = useMemo<{ disabled: boolean; options: CopyOptions }>(() => {
         if (typeof copy === 'boolean') {
             return {
-                disabled: copy,
+                disabled: !copy,
                 options: {},
             };
         }