github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-09 04:08:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/new-page-framework' into dev_ty_06_end

Browse Source
This commit is contained in:
aries_ckt 2023-07-05 14:05:54 +08:00
commit 8e01229fb9
2 changed files with 56 additions and 188 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
datacenter/app/datastores/documents/chunklist/page.tsx
View File

@ -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<any>([])
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)

203
datacenter/app/datastores/documents/page.tsx
View File

@ -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)