From c61dde50854eb098a5167049974926dfba88b368 Mon Sep 17 00:00:00 2001
From: ousugo <dkzyxh@gmail.com>
Date: Sun, 23 Feb 2025 23:22:55 +0800
Subject: [PATCH] feat: Enhance knowledge search popup input focus behavior

---
 .../src/pages/knowledge/components/KnowledgeSearchPopup.tsx  | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/renderer/src/pages/knowledge/components/KnowledgeSearchPopup.tsx b/src/renderer/src/pages/knowledge/components/KnowledgeSearchPopup.tsx
index ac04ddac..e1ffa3cd 100644
--- a/src/renderer/src/pages/knowledge/components/KnowledgeSearchPopup.tsx
+++ b/src/renderer/src/pages/knowledge/components/KnowledgeSearchPopup.tsx
@@ -4,7 +4,7 @@ import { DEFAULT_KNOWLEDGE_THRESHOLD } from '@renderer/config/constant'
 import { getFileFromUrl, getKnowledgeBaseParams } from '@renderer/services/KnowledgeService'
 import { FileType, KnowledgeBase } from '@renderer/types'
 import { Input, List, Modal, Spin, Typography } from 'antd'
-import { useState } from 'react'
+import { useRef, useState } from 'react'
 import { useTranslation } from 'react-i18next'
 import styled from 'styled-components'
 
@@ -25,6 +25,7 @@ const PopupContainer: React.FC<Props> = ({ base, resolve }) => {
   const [results, setResults] = useState<Array<ExtractChunkData & { file: FileType | null }>>([])
   const [searchKeyword, setSearchKeyword] = useState('')
   const { t } = useTranslation()
+  const searchInputRef = useRef<any>(null)
 
   const handleSearch = async (value: string) => {
     if (!value.trim()) {
@@ -91,6 +92,7 @@ const PopupContainer: React.FC<Props> = ({ base, resolve }) => {
       onOk={onOk}
       onCancel={onCancel}
       afterClose={onClose}
+      afterOpenChange={(visible) => visible && searchInputRef.current?.focus()}
       width={800}
       footer={null}
       centered
@@ -102,6 +104,7 @@ const PopupContainer: React.FC<Props> = ({ base, resolve }) => {
           enterButton
           size="large"
           onSearch={handleSearch}
+          ref={searchInputRef}
         />
         <ResultsContainer>
           {loading ? (