This commit is contained in:
rebecca554owen 2025-02-17 17:44:31 +08:00 committed by 亢奋猫
parent 83ddb8e161
commit 25a3cd5073

View File

@ -50,7 +50,7 @@ const Inputbar: FC = () => {
const [spaceClickCount, setSpaceClickCount] = useState(0) const [spaceClickCount, setSpaceClickCount] = useState(0)
const spaceClickTimer = useRef<NodeJS.Timeout>() const spaceClickTimer = useRef<NodeJS.Timeout>()
const [isTranslating, setIsTranslating] = useState(false) const [isTranslating, setIsTranslating] = useState(false)
const [selectedKnowledgeBase, setSelectedKnowledgeBase] = useState<KnowledgeBase>() const [selectedKnowledgeBase, setSelectedKnowledgeBase] = useState<KnowledgeBase | undefined>()
const isVision = useMemo(() => isVisionModel(model), [model]) const isVision = useMemo(() => isVisionModel(model), [model])
const supportExts = useMemo(() => [...textExts, ...documentExts, ...(isVision ? imageExts : [])], [isVision]) const supportExts = useMemo(() => [...textExts, ...documentExts, ...(isVision ? imageExts : [])], [isVision])
@ -274,8 +274,8 @@ const Inputbar: FC = () => {
} }
}, []) }, [])
const handleKnowledgeBaseSelect = (base?: KnowledgeBase) => { const handleKnowledgeBaseSelect = (bases: KnowledgeBase[]) => {
setSelectedKnowledgeBase(base) setSelectedKnowledgeBase(bases?.[0])
} }
return ( return (
@ -317,7 +317,7 @@ const Inputbar: FC = () => {
</Popconfirm> </Popconfirm>
</Tooltip> </Tooltip>
<KnowledgeBaseButton <KnowledgeBaseButton
selectedBase={selectedKnowledgeBase} selectedBases={selectedKnowledgeBase ? [selectedKnowledgeBase] : []}
onSelect={handleKnowledgeBaseSelect} onSelect={handleKnowledgeBaseSelect}
ToolbarButton={ToolbarButton} ToolbarButton={ToolbarButton}
disabled={files.length > 0} disabled={files.length > 0}