feat: add initialInput prop to AIChatContainer (#278)

Author: farazcsk

src/components/AIChat/AIChatContainer.tsx

@@ -25,6 +25,7 @@ export interface AIChatContainerProps
   components?: Partial<AIChatComponents>
   toolCallApproval?: ToolCallApprovalProps
   children?: ReactNode
+  initialInput?: string
 }
 
 interface AIChatComponents {
@@ -41,6 +42,7 @@ export function AIChatContainer({
   components,
   toolCallApproval,
   children,
+  initialInput,
 }: AIChatContainerProps) {
   return (
     <AIChatContext.Provider
@@ -52,6 +54,7 @@ export function AIChatContainer({
         onModelChange: modelSelector?.onModelChange,
         availableModels: modelSelector?.availableModels,
         toolCallApproval,
+        initialInput,
       }}
     >
       <div className={cn('flex h-full min-h-0 max-w-3xl flex-col', className)}>

src/components/AIChat/AIChatMessageComposer.tsx

@@ -72,10 +72,16 @@ export function AIChatMessageComposer({
   className,
   components,
 }: AIChatMessageComposerProps) {
-  const [message, setMessage] = useState('')
+  const {
+    onSendMessage,
+    isLoading,
+    model,
+    onModelChange,
+    availableModels,
+    initialInput,
+  } = useAIChat()
+  const [message, setMessage] = useState(initialInput || '')
   const [isFocused, setIsFocused] = useState(false)
-  const { onSendMessage, isLoading, model, onModelChange, availableModels } =
-    useAIChat()
   const textareaRef = useRef<HTMLTextAreaElement>(null)
 
   const adjustHeight = () => {
@@ -90,6 +96,13 @@ export function AIChatMessageComposer({
     adjustHeight()
   }, [message])
 
+  // Set initial input only once when component mounts with an initialInput
+  useEffect(() => {
+    if (initialInput) {
+      setMessage(initialInput)
+    }
+  }, [])
+
   const handleSubmit = (e?: React.FormEvent) => {
     e?.preventDefault()
     if (message.trim() && onSendMessage && !isLoading) {

src/components/AIChat/types.ts

@@ -78,6 +78,7 @@ export interface AIChatContextValue {
   onModelChange?: (model: string) => void
   availableModels?: { label: string; value: string }[]
   toolCallApproval?: ToolCallApprovalProps
+  initialInput?: string
 }
 
 export interface BasePartProps {