diff --git a/src/components/ai/AiChatDialog.vue b/src/components/ai/AiChatDialog.vue
index 955059a4a8..c5e65f8da5 100644
--- a/src/components/ai/AiChatDialog.vue
+++ b/src/components/ai/AiChatDialog.vue
@@ -238,6 +238,16 @@
         sendMessage()
     }
 
+    const setUserInput = (value: string): void => {
+        userInput.value = value
+        nextTick(() => {
+            autoResize()
+            textareaRef.value?.focus()
+        })
+    }
+
+    defineExpose({ setUserInput })
+
     const clearMessage = (): void => {
         abortController.value.abort()
         abortController.value = new AbortController()
diff --git a/src/components/layout/Search.vue b/src/components/layout/Search.vue
index 7c372d47e7..f554f7f7a4 100644
--- a/src/components/layout/Search.vue
+++ b/src/components/layout/Search.vue
@@ -233,6 +233,7 @@
                 <div class="modal-body row">
                     <Suspense>
                         <AiChatDialog
+                            ref="aiChatDialog"
                             :randomAiQuestions
                             @close="closeAiDialog"
                             @backToSearch="backToSearch"
@@ -322,9 +323,13 @@
                 this.selectedIndex = null
                 this.selectedItem = null
             },
-            focusSearchAi() {
-                document.querySelector("#ai-chat-input").value = ""
-                document.querySelector("#ai-chat-input").focus()
+            focusSearchAi(event) {
+                const openedFromSearch =
+                    event?.relatedTarget?.closest("#search-modal")
+                const prefill = openedFromSearch
+                    ? this.searchValue?.trim() || ""
+                    : ""
+                this.$refs.aiChatDialog?.setUserInput(prefill)
             },
             onHiddenAi() {},
             search(value) {