From 91e5cd647cb38239680848f8d977927ae039f980 Mon Sep 17 00:00:00 2001
From: mortalYoung <yangwei1@outlook.com>
Date: Thu, 23 Jan 2025 17:42:17 +0800
Subject: [PATCH] fix(chat): fix alignment for chat's button

---
 src/chat/button/index.scss |  3 +++
 src/chat/demos/message.tsx | 26 +++++++++++++++++---------
 src/chat/message/index.tsx |  2 +-
 3 files changed, 21 insertions(+), 10 deletions(-)

diff --git a/src/chat/button/index.scss b/src/chat/button/index.scss
index a2616316c..7fb4a4369 100644
--- 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
index ad134baac..0c277550a 100644
--- 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
index dc4a00654..4eba6c8e9 100644
--- 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: {},
             };
         }