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

feat: refactor the request

Browse Source
This commit is contained in:
shiweisong.ssw 2023-07-05 13:46:12 +08:00
parent f4310a65db
commit a03bebde53
2 changed files with 62 additions and 103 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

127
datacenter/app/datastores/page.tsx
View File

@ -22,6 +22,7 @@ import {
Typography,
styled
} from '@/lib/mui'
import { sendSpaceGetRequest, sendSpacePostRequest, sendSpaceUploadPostRequest } from '@/utils/request';
const { Dragger } = Upload
@ -91,17 +92,7 @@ const Index = () => {
}
useEffect(() => {
async function fetchData() {
const res = await fetch(
`${process.env.API_BASE_URL}/knowledge/space/list`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({})
}
)
const data = await res.json()
const data = await sendSpacePostRequest('/knowledge/space/list')
if (data.success) {
setKnowledgeSpaceList(data.data)
}
@ -315,36 +306,19 @@ const Index = () => {
message.error('please input the name')
return
}
const res = await fetch(
`${process.env.API_BASE_URL}/knowledge/space/add`,
const data = await sendSpacePostRequest(
`/knowledge/space/add`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
name: knowledgeSpaceName,
vector_type: 'Chroma',
owner: 'keting',
desc: 'test1'
})
}
)
const data = await res.json()
if (data.success) {
message.success('success')
setActiveStep(1)
const res = await fetch(
`${process.env.API_BASE_URL}/knowledge/space/list`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({})
}
)
const data = await res.json()
const data = await sendSpacePostRequest('/knowledge/space/list')
if (data.success) {
setKnowledgeSpaceList(data.data)
}
@ -481,37 +455,18 @@ const Index = () => {
message.error('Please input the Web Page URL')
return
}
const res = await fetch(
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/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/${knowledgeSpaceName}/document/add`, {
doc_name: documentName,
content: webPageUrl,
doc_type: 'URL'
})
if (data.success) {
message.success('success')
setIsAddKnowledgeSpaceModalShow(false)
synchChecked &&
fetch(
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/sync`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_ids: [data.data]
})
}
)
sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/sync`, {
doc_ids: [data.data]
})
} else {
message.error(data.err_msg || 'failed')
}
@ -524,30 +479,15 @@ const Index = () => {
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/${knowledgeSpaceName}/document/upload`,
{
method: 'POST',
body: formData
}
)
const data = await res.json()
const data = await sendSpaceUploadPostRequest(`/knowledge/${knowledgeSpaceName}/document/upload`, formData);
if (data.success) {
message.success('success')
setIsAddKnowledgeSpaceModalShow(false)
synchChecked &&
fetch(
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/sync`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_ids: [data.data]
})
}
)
sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/sync`, {
doc_ids: [data.data]
});
} else {
message.error(data.err_msg || 'failed')
}
@ -556,38 +496,19 @@ const Index = () => {
message.error('Please input the text')
return
}
const res = await fetch(
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/add`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_name: documentName,
const data = await sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/add`, {
doc_name: documentName,
source: textSource,
content: text,
doc_type: 'TEXT'
})
}
)
const data = await res.json()
})
if (data.success) {
message.success('success')
setIsAddKnowledgeSpaceModalShow(false)
synchChecked &&
fetch(
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/sync`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_ids: [data.data]
})
}
)
sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/sync`, {
doc_ids: [data.data]
})
} else {
message.error(data.err_msg || 'failed')
}

38
datacenter/utils/request.ts
View File

@ -38,6 +38,24 @@ export const sendGetRequest = (url: string, qs?: { [key: string]: any }) => {
});
}
export const sendSpaceGetRequest = (url: string, qs?: { [key: string]: any }) => {
if (qs) {
const str = Object.keys(qs)
.filter(k => qs[k] !== undefined && qs[k] !== '')
.map(k => `${k}=${qs[k]}`)
.join('&');
if (str) {
url += `?${str}`;
}
}
return axios.get(url, {
headers: DEFAULT_HEADERS
}).then(res => res).catch(err => {
message.error(err);
Promise.reject(err);
});
}
export const sendPostRequest = (url: string, body?: any) => {
const reqBody = sanitizeBody(body);
return axios.post("/api" + url, {
@ -47,4 +65,24 @@ export const sendPostRequest = (url: string, body?: any) => {
message.error(err);
Promise.reject(err);
});
}
export const sendSpacePostRequest = (url: string, body?: any) => {
const reqBody = sanitizeBody(body);
return axios.post(url, {
body: reqBody,
headers: DEFAULT_HEADERS
}).then(res => res).catch(err => {
message.error(err);
Promise.reject(err);
});
}
export const sendSpaceUploadPostRequest = (url: string, body?: any) => {
return axios.post(url, {
body: body,
}).then(res => res).catch(err => {
message.error(err);
Promise.reject(err);
});
}