🐛 fix: fix messageItemExtraRender item no set extra error (#18)

ant-design · Dec 6, 2023 · ba006e1 · ba006e1
1 parent 44f293d
commit ba006e1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ProChat/components/ChatList/Extras/Assistant.tsx b/src/ProChat/components/ChatList/Extras/Assistant.tsx
@@ -17,7 +17,7 @@ export const AssistantMessageExtra: RenderMessageExtra = memo(({ extra, ...rest
 
   const showExtra = showModelTag || hasTranslate;
 
-  const dom = messageItemExtraRender?.(rest, 'assistant');
+  const dom = messageItemExtraRender?.({ extra, ...rest }, 'assistant');
   if (!showExtra && !dom) return;
 
   return (

diff --git a/src/ProChat/components/ChatList/Extras/User.tsx b/src/ProChat/components/ChatList/Extras/User.tsx
@@ -16,7 +16,7 @@ export const UserMessageExtra: RenderMessageExtra = memo(({ extra, ...rest }) =>
           <Divider style={{ margin: '12px 0' }} />
         </div>
       )}
-      {messageItemExtraRender?.(rest, 'user')}
+      {messageItemExtraRender?.({ extra, ...rest }, 'user')}
     </Flexbox>
   );
 });