github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-09 20:28:07 +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, Typography,
styled styled
} from '@/lib/mui' } from '@/lib/mui'
import { sendSpaceGetRequest, sendSpacePostRequest, sendSpaceUploadPostRequest } from '@/utils/request';
const { Dragger } = Upload const { Dragger } = Upload
@ -91,17 +92,7 @@ const Index = () => {
} }
useEffect(() => { useEffect(() => {
async function fetchData() { async function fetchData() {
const res = await fetch( const data = await sendSpacePostRequest('/knowledge/space/list')
`${process.env.API_BASE_URL}/knowledge/space/list`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({})
}
)
const data = await res.json()
if (data.success) { if (data.success) {
setKnowledgeSpaceList(data.data) setKnowledgeSpaceList(data.data)
} }
@ -315,36 +306,19 @@ const Index = () => {
message.error('please input the name') message.error('please input the name')
return return
} }
const res = await fetch( const data = await sendSpacePostRequest(
`${process.env.API_BASE_URL}/knowledge/space/add`, `/knowledge/space/add`,
{ {
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
name: knowledgeSpaceName, name: knowledgeSpaceName,
vector_type: 'Chroma', vector_type: 'Chroma',
owner: 'keting', owner: 'keting',
desc: 'test1' desc: 'test1'
})
} }
) )
const data = await res.json()
if (data.success) { if (data.success) {
message.success('success') message.success('success')
setActiveStep(1) setActiveStep(1)
const res = await fetch( const data = await sendSpacePostRequest('/knowledge/space/list')
`${process.env.API_BASE_URL}/knowledge/space/list`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({})
}
)
const data = await res.json()
if (data.success) { if (data.success) {
setKnowledgeSpaceList(data.data) setKnowledgeSpaceList(data.data)
} }
@ -481,37 +455,18 @@ const Index = () => {
message.error('Please input the Web Page URL') message.error('Please input the Web Page URL')
return return
} }
const res = await fetch( const data = await sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/add`, {
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/add`, doc_name: documentName,
{ content: webPageUrl,
method: 'POST', doc_type: 'URL'
headers: { })
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_name: documentName,
content: webPageUrl,
doc_type: 'URL'
})
}
)
const data = await res.json()
if (data.success) { if (data.success) {
message.success('success') message.success('success')
setIsAddKnowledgeSpaceModalShow(false) setIsAddKnowledgeSpaceModalShow(false)
synchChecked && synchChecked &&
fetch( sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/sync`, {
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/sync`, doc_ids: [data.data]
{ })
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_ids: [data.data]
})
}
)
} else { } else {
message.error(data.err_msg || 'failed') message.error(data.err_msg || 'failed')
} }
@ -524,30 +479,15 @@ const Index = () => {
formData.append('doc_name', documentName) formData.append('doc_name', documentName)
formData.append('doc_file', originFileObj) formData.append('doc_file', originFileObj)
formData.append('doc_type', 'DOCUMENT') formData.append('doc_type', 'DOCUMENT')
const res = await fetch(
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/upload`, const data = await sendSpaceUploadPostRequest(`/knowledge/${knowledgeSpaceName}/document/upload`, formData);
{
method: 'POST',
body: formData
}
)
const data = await res.json()
if (data.success) { if (data.success) {
message.success('success') message.success('success')
setIsAddKnowledgeSpaceModalShow(false) setIsAddKnowledgeSpaceModalShow(false)
synchChecked && synchChecked &&
fetch( sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/sync`, {
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/sync`, doc_ids: [data.data]
{ });
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_ids: [data.data]
})
}
)
} else { } else {
message.error(data.err_msg || 'failed') message.error(data.err_msg || 'failed')
} }
@ -556,38 +496,19 @@ const Index = () => {
message.error('Please input the text') message.error('Please input the text')
return return
} }
const res = await fetch( const data = await sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/add`, {
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/add`, doc_name: documentName,
{
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_name: documentName,
source: textSource, source: textSource,
content: text, content: text,
doc_type: 'TEXT' doc_type: 'TEXT'
}) })
}
)
const data = await res.json()
if (data.success) { if (data.success) {
message.success('success') message.success('success')
setIsAddKnowledgeSpaceModalShow(false) setIsAddKnowledgeSpaceModalShow(false)
synchChecked && synchChecked &&
fetch( sendSpacePostRequest(`/knowledge/${knowledgeSpaceName}/document/sync`, {
`${process.env.API_BASE_URL}/knowledge/${knowledgeSpaceName}/document/sync`, doc_ids: [data.data]
{ })
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
doc_ids: [data.data]
})
}
)
} else { } else {
message.error(data.err_msg || 'failed') 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) => { export const sendPostRequest = (url: string, body?: any) => {
const reqBody = sanitizeBody(body); const reqBody = sanitizeBody(body);
return axios.post("/api" + url, { return axios.post("/api" + url, {
@ -47,4 +65,24 @@ export const sendPostRequest = (url: string, body?: any) => {
message.error(err); message.error(err);
Promise.reject(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);
});
} }