52 lines
1.6 KiB
TypeScript
52 lines
1.6 KiB
TypeScript
import { useState, useCallback } from 'react'
|
|
import Button from '@/components/ui/Button'
|
|
import {
|
|
Dialog,
|
|
DialogContent,
|
|
DialogDescription,
|
|
DialogHeader,
|
|
DialogTitle,
|
|
DialogTrigger
|
|
} from '@/components/ui/Dialog'
|
|
import { toast } from 'sonner'
|
|
import { errorMessage } from '@/lib/utils'
|
|
import { clearDocuments } from '@/api/lightrag'
|
|
|
|
import { EraserIcon } from 'lucide-react'
|
|
|
|
export default function ClearDocumentsDialog() {
|
|
const [open, setOpen] = useState(false) // 添加状态控制
|
|
|
|
const handleClear = useCallback(async () => {
|
|
try {
|
|
const result = await clearDocuments()
|
|
if (result.status === 'success') {
|
|
toast.success('Documents cleared successfully')
|
|
setOpen(false)
|
|
} else {
|
|
toast.error(`Clear Documents Failed:\n${result.message}`)
|
|
}
|
|
} catch (err) {
|
|
toast.error('Clear Documents Failed:\n' + errorMessage(err))
|
|
}
|
|
}, [setOpen])
|
|
|
|
return (
|
|
<Dialog open={open} onOpenChange={setOpen}>
|
|
<DialogTrigger asChild>
|
|
<Button variant="outline" tooltip="Clear documents" side="bottom" size="icon">
|
|
<EraserIcon />
|
|
</Button>
|
|
</DialogTrigger>
|
|
<DialogContent className="sm:max-w-xl" onCloseAutoFocus={(e) => e.preventDefault()}>
|
|
<DialogHeader>
|
|
<DialogTitle>Clear documents</DialogTitle>
|
|
<DialogDescription>Do you really want to clear all documents?</DialogDescription>
|
|
</DialogHeader>
|
|
<Button variant="destructive" onClick={handleClear}>
|
|
YES
|
|
</Button>
|
|
</DialogContent>
|
|
</Dialog>
|
|
)
|
|
}
|