diff --git a/datacenter/app/datastores/documents/chunklist/page.tsx b/datacenter/app/datastores/documents/chunklist/page.tsx index 2664ed1bd..435290e3d 100644 --- a/datacenter/app/datastores/documents/chunklist/page.tsx +++ b/datacenter/app/datastores/documents/chunklist/page.tsx @@ -11,6 +11,7 @@ import { Link } from '@/lib/mui' import { Popover, Pagination } from 'antd' +import { sendSpaceGetRequest, sendSpacePostRequest, sendSpaceUploadPostRequest } from '@/utils/request'; const page_size = 20 const ChunkList = () => { @@ -23,21 +24,11 @@ const ChunkList = () => { const [chunkList, setChunkList] = useState([]) useEffect(() => { async function fetchChunks() { - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/chunk/list`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - document_id: documentId, - page: 1, - page_size - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/chunk/list`, { + document_id: documentId, + page: 1, + page_size + }) if (data.success) { setChunkList(data.data.data) setTotal(data.data.total) @@ -150,21 +141,11 @@ const ChunkList = () => { current={current} total={total} onChange={async (page) => { - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/chunk/list`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - document_id: documentId, - page, - page_size - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/chunk/list`, { + document_id: documentId, + page, + page_size + }) if (data.success) { setChunkList(data.data.data) setTotal(data.data.total) diff --git a/datacenter/app/datastores/documents/page.tsx b/datacenter/app/datastores/documents/page.tsx index 134fe5f4f..ff8ec0268 100644 --- a/datacenter/app/datastores/documents/page.tsx +++ b/datacenter/app/datastores/documents/page.tsx @@ -25,6 +25,7 @@ import CheckCircleOutlinedIcon from '@mui/icons-material/CheckCircleOutlined'; import CachedIcon from '@mui/icons-material/Cached'; import type { UploadProps } from 'antd' import { Upload, Pagination, Popover, message } from 'antd' +import { sendSpaceGetRequest, sendSpacePostRequest, sendSpaceUploadPostRequest } from '@/utils/request'; const { Dragger } = Upload const Item = styled(Sheet)(({ theme }) => ({ @@ -94,20 +95,10 @@ const Documents = () => { } useEffect(() => { async function fetchDocuments() { - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/list`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - page: 1, - page_size - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/document/list`, { + page: 1, + page_size + }) if (data.success) { setDocuments(data.data.data) setTotal(data.data.total) @@ -244,19 +235,9 @@ const Documents = () => { marginRight: '20px' }} onClick={async () => { - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/sync`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - doc_ids: [row.id] - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/document/sync`, { + doc_ids: [row.id] + }) if (data.success) { message.success('success') } else { @@ -297,20 +278,10 @@ const Documents = () => { current={current} total={total} onChange={async (page) => { - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/list`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - page, - page_size - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/document/list`, { + page, + page_size + }) if (data.success) { setDocuments(data.data.data) setTotal(data.data.total) @@ -484,52 +455,23 @@ const Documents = () => { message.error('Please input the Web Page URL') return } - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/add`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - doc_name: documentName, - content: webPageUrl, - doc_type: 'URL' - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/document/add`, { + doc_name: documentName, + content: webPageUrl, + doc_type: 'URL' + }) data.success && synchChecked && - fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/sync`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - doc_ids: [data.data] - }) - } - ) + sendSpacePostRequest(`/knowledge/${spaceName}/document/sync`, { + doc_ids: [data.data] + }) if (data.success) { message.success('success') setIsAddDocumentModalShow(false) - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/list`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - page: current, - page_size - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/document/list`, { + page: current, + page_size + }) if (data.success) { setDocuments(data.data.data) setTotal(data.data.total) @@ -547,45 +489,19 @@ const Documents = () => { formData.append('doc_name', documentName) formData.append('doc_file', originFileObj) formData.append('doc_type', 'DOCUMENT') - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/upload`, - { - method: 'POST', - body: formData - } - ) - const data = await res.json() + const data = await sendSpaceUploadPostRequest(`/knowledge/${spaceName}/document/upload`, formData); data.success && synchChecked && - fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/sync`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - doc_ids: [data.data] - }) - } - ) + sendSpacePostRequest(`/knowledge/${spaceName}/document/sync`, { + doc_ids: [data.data] + }) if (data.success) { message.success('success') setIsAddDocumentModalShow(false) - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/list`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - page: current, - page_size - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/document/list`, { + page: current, + page_size + }) if (data.success) { setDocuments(data.data.data) setTotal(data.data.total) @@ -599,53 +515,24 @@ const Documents = () => { message.error('Please input the text') return } - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/add`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - doc_name: documentName, - source: textSource, - content: text, - doc_type: 'TEXT' - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/document/add`, { + doc_name: documentName, + source: textSource, + content: text, + doc_type: 'TEXT' + }) data.success && synchChecked && - fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/sync`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - doc_ids: [data.data] - }) - } - ) + sendSpacePostRequest(`/knowledge/${spaceName}/document/sync`, { + doc_ids: [data.data] + }) if (data.success) { message.success('success') setIsAddDocumentModalShow(false) - const res = await fetch( - `${process.env.API_BASE_URL}/knowledge/${spaceName}/document/list`, - { - method: 'POST', - headers: { - 'Content-Type': 'application/json' - }, - body: JSON.stringify({ - page: current, - page_size - }) - } - ) - const data = await res.json() + const data = await sendSpacePostRequest(`/knowledge/${spaceName}/document/list`, { + page: current, + page_size + }); if (data.success) { setDocuments(data.data.data) setTotal(data.data.total)