From 72699ac06fd629612a8f7a3ed23cd14c73755ea6 Mon Sep 17 00:00:00 2001
From: chaos <1585767724@qq.com>
Date: Wed, 18 Sep 2024 23:06:30 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0setInputArea=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/ProChat/index.tsx          |  3 +++
 src/components/ProChatInputArea/index.tsx | 14 +++++++++++---
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/components/ProChat/index.tsx b/src/components/ProChat/index.tsx
index 092dce7..b6a7e1a 100644
--- a/src/components/ProChat/index.tsx
+++ b/src/components/ProChat/index.tsx
@@ -84,6 +84,8 @@ export interface ProChatInstance<T = Record<string, any>> {
     message: Partial<ChatMessage<T>>,
     userType: 'assistant' | 'user',
   ) => ChatMessage<T>;
+
+  setInputAreaValue: (value: string) => void;
 }
 
 export type ProChatChatReference<T = Record<string, any>> = MutableRefObject<
@@ -529,6 +531,7 @@ export function ProChat<
           inputAreaRender={inputAreaRender}
           inputRender={inputRender}
           inputAreaProps={inputAreaProps}
+          chatRef={chatRef}
           actionStyle={styles?.chatInputAction}
           sendAreaStyle={styles?.chatSendAreaStyle}
           areaStyle={styles?.chatInputArea}
diff --git a/src/components/ProChatInputArea/index.tsx b/src/components/ProChatInputArea/index.tsx
index 02f7658..17dfd73 100644
--- a/src/components/ProChatInputArea/index.tsx
+++ b/src/components/ProChatInputArea/index.tsx
@@ -1,10 +1,10 @@
 import { useRefFunction } from '@/hooks/useRefFunction';
-import { ChatMessage } from '@/index';
+import { ChatMessage, ProChatChatReference } from '@/index';
 import { ProChatLocale } from '@/locale';
 import { SendOutlined } from '@ant-design/icons';
 import { Button, ButtonProps, ConfigProvider, Flex } from 'antd';
 import cx from 'classnames';
-import { useContext, useMemo, useRef, useState } from 'react';
+import { useContext, useEffect, useMemo, useRef, useState } from 'react';
 import AnimationItem from '../Animation';
 import { ProChatActionBar, ProChatActionBarProps } from './ActionBar';
 import { MentionsTextArea, MentionsTextAreaProps } from './AutoCompleteTextArea';
@@ -75,6 +75,8 @@ export type ChatInputAreaProps = {
    * Locale configuration for the ProChat component.
    */
   locale?: ProChatLocale;
+
+  chatRef?: ProChatChatReference;
   /**
    * Callback function to clear the message input.
    */
@@ -133,6 +135,7 @@ export const ChatInputArea = (props: ChatInputAreaProps) => {
     onMessageSend,
     actionStyle,
     locale,
+    chatRef,
     mentionRequest,
     actionsRender,
   } = props || {};
@@ -144,7 +147,12 @@ export const ChatInputArea = (props: ChatInputAreaProps) => {
   const prefixClass = getPrefixCls('pro-chat-input-area');
 
   const { wrapSSR, hashId } = useStyle(prefixClass);
-
+  useEffect(() => {
+    if (!chatRef || !chatRef.current) return;
+    chatRef.current.setInputAreaValue = (value) => {
+      setMessage(value);
+    };
+  }, [chatRef]);
   const send = useRefFunction(async () => {
     if (onSend && message) {
       const success = await onSend(message);