Function VectorStoreQuery

packages/ui/src/views/docstore/VectorStoreQuery.jsx:56–498

()

}))

const VectorStoreQuery = () => {
    const customization = useSelector((state) => state.customization)
    const navigate = useNavigate()
    const theme = useTheme()
    const dispatch = useDispatch()
    const inputRef = useRef(null)
    const { hasAssignedWorkspace } = useAuth()

    useNotifier()

    const enqueueSnackbar = (...args) => dispatch(enqueueSnackbarAction(...args))
    const closeSnackbar = (...args) => dispatch(closeSnackbarAction(...args))

    const { storeId } = useParams()

    const [documentChunks, setDocumentChunks] = useState([])
    const [loading, setLoading] = useState(false)
    const [showExpandedChunkDialog, setShowExpandedChunkDialog] = useState(false)
    const [expandedChunkDialogProps, setExpandedChunkDialogProps] = useState({})
    const [documentStore, setDocumentStore] = useState({})
    const [query, setQuery] = useState('')

    const [timeTaken, setTimeTaken] = useState(-1)
    const [retrievalError, setRetrievalError] = useState(undefined)

    const getSpecificDocumentStoreApi = useApi(documentsApi.getSpecificDocumentStore)
    const queryVectorStoreApi = useApi(documentsApi.queryVectorStore)

    const getVectorStoreNodeDetailsApi = useApi(nodesApi.getSpecificNode)
    const [selectedVectorStoreProvider, setSelectedVectorStoreProvider] = useState({})

    const handleVectorStoreProviderDataChange = ({ inputParam, newValue }) => {
        setSelectedVectorStoreProvider((prevData) => {
            const updatedData = { ...prevData }
            updatedData.inputs[inputParam.name] = newValue
            updatedData.inputParams = showHideInputParams(updatedData)
            return updatedData
        })
    }

    const chunkSelected = (chunkId, selectedChunkNumber) => {
        const selectedChunk = documentChunks.find((chunk) => chunk.id === chunkId)
        const dialogProps = {
            data: {
                selectedChunk,
                selectedChunkNumber
            }
        }
        setExpandedChunkDialogProps(dialogProps)
        setShowExpandedChunkDialog(true)
    }

    const handleEnter = (e) => {
        // Check if IME composition is in progress
        const isIMEComposition = e.isComposing || e.keyCode === 229
        if (e.key === 'Enter' && query && !isIMEComposition) {
            if (!e.shiftKey && query) {
                if (inputRef.current) {