feat: send message support ChatMessage

ant-design · Jun 11, 2024 · 5096c75 · 5096c75
1 parent 66b0a04
commit 5096c75
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/hooks/useChatList.ts b/src/hooks/useChatList.ts
@@ -196,9 +196,19 @@ export const useChatList = (props: ProChatUIUseListChatProps) => {
    * @param {string} message - The message to send.
    * @returns {Promise<void>} - A promise that resolves when the message is sent.
    */
-  const sendMessage = useRefFunction(async (message: string) => {
+  const sendMessage = useRefFunction(async (message: string | Partial<ChatMessage>) => {
     controller.current = new AbortController();
-    chatList.push(genMessageRecord({ content: message }, 'user'));
+    chatList.push(
+      genMessageRecord(
+        typeof message === 'string'
+          ? { content: message }
+          : {
+              ...message,
+              content: message.content,
+            },
+        'user',
+      ),
+    );
     setChatList([...chatList]);
     if (!props?.sendMessageRequest) return;
     setLoadingMessage(