Add File
This commit is contained in:
336
frontend/src/api/chat.ts
Normal file
336
frontend/src/api/chat.ts
Normal file
@@ -0,0 +1,336 @@
|
||||
import { request } from '@/utils/request'
|
||||
import { getDate } from '@/utils/utils.ts'
|
||||
|
||||
export const questionApi = {
|
||||
pager: (pageNumber: number, pageSize: number) =>
|
||||
request.get(`/chat/question/pager/${pageNumber}/${pageSize}`),
|
||||
/* add: (data: any) => new Promise((resolve, reject) => {
|
||||
request.post('/chat/question', data, { responseType: 'stream', timeout: 0, onDownloadProgress: p => {
|
||||
resolve(p)
|
||||
}}).catch(e => reject(e))
|
||||
}), */
|
||||
// add: (data: any) => request.post('/chat/question', data),
|
||||
add: (data: any, controller?: AbortController) =>
|
||||
request.fetchStream('/chat/question', data, controller),
|
||||
edit: (data: any) => request.put('/chat/question', data),
|
||||
delete: (id: number) => request.delete(`/chat/question/${id}`),
|
||||
query: (id: number) => request.get(`/chat/question/${id}`),
|
||||
}
|
||||
|
||||
export interface ChatMessage {
|
||||
role: 'user' | 'assistant'
|
||||
create_time?: Date | string
|
||||
content?: string | number
|
||||
record?: ChatRecord
|
||||
isTyping?: boolean
|
||||
first_chat?: boolean
|
||||
recommended_question?: string
|
||||
index: number
|
||||
}
|
||||
|
||||
export class ChatRecord {
|
||||
id?: number
|
||||
chat_id?: number
|
||||
create_time?: Date | string
|
||||
finish_time?: Date | string
|
||||
question?: string
|
||||
sql_answer?: string
|
||||
sql?: string
|
||||
data?: string | any
|
||||
chart_answer?: string
|
||||
chart?: string
|
||||
analysis?: string
|
||||
analysis_thinking?: string
|
||||
predict?: string
|
||||
predict_content?: string
|
||||
predict_data?: string | any
|
||||
finish?: boolean = false
|
||||
error?: string
|
||||
run_time: number = 0
|
||||
first_chat: boolean = false
|
||||
recommended_question?: string
|
||||
analysis_record_id?: number
|
||||
predict_record_id?: number
|
||||
|
||||
constructor()
|
||||
constructor(
|
||||
id: number,
|
||||
chat_id: number,
|
||||
create_time: Date | string,
|
||||
finish_time: Date | string | undefined,
|
||||
question: string,
|
||||
sql_answer: string | undefined,
|
||||
sql: string | undefined,
|
||||
data: string | any | undefined,
|
||||
chart_answer: string | undefined,
|
||||
chart: string | undefined,
|
||||
analysis: string | undefined,
|
||||
analysis_thinking: string | undefined,
|
||||
predict: string | undefined,
|
||||
predict_content: string | undefined,
|
||||
predict_data: string | any | undefined,
|
||||
finish: boolean,
|
||||
error: string | undefined,
|
||||
run_time: number,
|
||||
first_chat: boolean,
|
||||
recommended_question: string | undefined,
|
||||
analysis_record_id: number | undefined,
|
||||
predict_record_id: number | undefined
|
||||
)
|
||||
constructor(
|
||||
id?: number,
|
||||
chat_id?: number,
|
||||
create_time?: Date | string,
|
||||
finish_time?: Date | string,
|
||||
question?: string,
|
||||
sql_answer?: string,
|
||||
sql?: string,
|
||||
data?: string | any,
|
||||
chart_answer?: string,
|
||||
chart?: string,
|
||||
analysis?: string,
|
||||
analysis_thinking?: string,
|
||||
predict?: string,
|
||||
predict_content?: string,
|
||||
predict_data?: string | any,
|
||||
finish?: boolean,
|
||||
error?: string,
|
||||
run_time?: number,
|
||||
first_chat?: boolean,
|
||||
recommended_question?: string,
|
||||
analysis_record_id?: number,
|
||||
predict_record_id?: number
|
||||
) {
|
||||
this.id = id
|
||||
this.chat_id = chat_id
|
||||
this.create_time = getDate(create_time)
|
||||
this.finish_time = getDate(finish_time)
|
||||
this.question = question
|
||||
this.sql_answer = sql_answer
|
||||
this.sql = sql
|
||||
this.data = data
|
||||
this.chart_answer = chart_answer
|
||||
this.chart = chart
|
||||
this.analysis = analysis
|
||||
this.analysis_thinking = analysis_thinking
|
||||
this.predict = predict
|
||||
this.predict_content = predict_content
|
||||
this.predict_data = predict_data
|
||||
this.finish = !!finish
|
||||
this.error = error
|
||||
this.run_time = run_time ?? 0
|
||||
this.first_chat = !!first_chat
|
||||
this.recommended_question = recommended_question
|
||||
this.analysis_record_id = analysis_record_id
|
||||
this.predict_record_id = predict_record_id
|
||||
}
|
||||
}
|
||||
|
||||
export class Chat {
|
||||
id?: number
|
||||
create_time?: Date | string
|
||||
create_by?: number
|
||||
brief?: string
|
||||
chat_type?: string
|
||||
datasource?: number
|
||||
engine_type?: string
|
||||
ds_type?: string
|
||||
|
||||
constructor()
|
||||
constructor(
|
||||
id: number,
|
||||
create_time: Date | string,
|
||||
create_by: number,
|
||||
brief: string,
|
||||
chat_type: string,
|
||||
datasource: number,
|
||||
engine_type: string
|
||||
)
|
||||
constructor(
|
||||
id?: number,
|
||||
create_time?: Date | string,
|
||||
create_by?: number,
|
||||
brief?: string,
|
||||
chat_type?: string,
|
||||
datasource?: number,
|
||||
engine_type?: string
|
||||
) {
|
||||
this.id = id
|
||||
this.create_time = getDate(create_time)
|
||||
this.create_by = create_by
|
||||
this.brief = brief
|
||||
this.chat_type = chat_type
|
||||
this.datasource = datasource
|
||||
this.engine_type = engine_type
|
||||
}
|
||||
}
|
||||
|
||||
export class ChatInfo extends Chat {
|
||||
datasource_name?: string
|
||||
datasource_exists: boolean = true
|
||||
records: Array<ChatRecord> = []
|
||||
|
||||
constructor()
|
||||
constructor(chat: Chat)
|
||||
constructor(
|
||||
id: number,
|
||||
create_time: Date | string,
|
||||
create_by: number,
|
||||
brief: string,
|
||||
chat_type: string,
|
||||
datasource: number,
|
||||
engine_type: string,
|
||||
ds_type: string,
|
||||
datasource_name: string,
|
||||
datasource_exists: boolean,
|
||||
records: Array<ChatRecord>
|
||||
)
|
||||
constructor(
|
||||
param1?: number | Chat,
|
||||
create_time?: Date | string,
|
||||
create_by?: number,
|
||||
brief?: string,
|
||||
chat_type?: string,
|
||||
datasource?: number,
|
||||
engine_type?: string,
|
||||
ds_type?: string,
|
||||
datasource_name?: string,
|
||||
datasource_exists: boolean = true,
|
||||
records: Array<ChatRecord> = []
|
||||
) {
|
||||
super()
|
||||
if (param1 !== undefined) {
|
||||
if (param1 instanceof Chat) {
|
||||
this.id = param1.id
|
||||
this.create_time = getDate(param1.create_time)
|
||||
this.create_by = param1.create_by
|
||||
this.brief = param1.brief
|
||||
this.chat_type = param1.chat_type
|
||||
this.datasource = param1.datasource
|
||||
this.engine_type = param1.engine_type
|
||||
this.ds_type = param1.ds_type
|
||||
} else {
|
||||
this.id = param1
|
||||
this.create_time = getDate(create_time)
|
||||
this.create_by = create_by
|
||||
this.brief = brief
|
||||
this.chat_type = chat_type
|
||||
this.datasource = datasource
|
||||
this.engine_type = engine_type
|
||||
this.ds_type = ds_type
|
||||
}
|
||||
}
|
||||
this.datasource_name = datasource_name
|
||||
this.datasource_exists = datasource_exists
|
||||
this.records = records
|
||||
}
|
||||
}
|
||||
|
||||
const toChatRecord = (data?: any): ChatRecord | undefined => {
|
||||
if (!data) {
|
||||
return undefined
|
||||
}
|
||||
return new ChatRecord(
|
||||
data.id,
|
||||
data.chat_id,
|
||||
data.create_time,
|
||||
data.finish_time,
|
||||
data.question,
|
||||
data.sql_answer,
|
||||
data.sql,
|
||||
data.data,
|
||||
data.chart_answer,
|
||||
data.chart,
|
||||
data.analysis,
|
||||
data.analysis_thinking,
|
||||
data.predict,
|
||||
data.predict_content,
|
||||
data.predict_data,
|
||||
data.finish,
|
||||
data.error,
|
||||
data.run_time,
|
||||
data.first_chat,
|
||||
data.recommended_question,
|
||||
data.analysis_record_id,
|
||||
data.predict_record_id
|
||||
)
|
||||
}
|
||||
const toChatRecordList = (list: any = []): ChatRecord[] => {
|
||||
const records: Array<ChatRecord> = []
|
||||
for (let i = 0; i < list.length; i++) {
|
||||
const record = toChatRecord(list[i])
|
||||
if (record) {
|
||||
records.push(record)
|
||||
}
|
||||
}
|
||||
return records
|
||||
}
|
||||
|
||||
export const chatApi = {
|
||||
toChatInfo: (data?: any): ChatInfo | undefined => {
|
||||
if (!data) {
|
||||
return undefined
|
||||
}
|
||||
return new ChatInfo(
|
||||
data.id,
|
||||
data.create_time,
|
||||
data.create_by,
|
||||
data.brief,
|
||||
data.chat_type,
|
||||
data.datasource,
|
||||
data.engine_type,
|
||||
data.ds_type,
|
||||
data.datasource_name,
|
||||
data.datasource_exists,
|
||||
toChatRecordList(data.records)
|
||||
)
|
||||
},
|
||||
toChatInfoList: (list: any[] = []): ChatInfo[] => {
|
||||
const infos: Array<ChatInfo> = []
|
||||
for (let i = 0; i < list.length; i++) {
|
||||
const chatInfo = chatApi.toChatInfo(list[i])
|
||||
if (chatInfo) {
|
||||
infos.push(chatInfo)
|
||||
}
|
||||
}
|
||||
return infos
|
||||
},
|
||||
list: (): Promise<Array<ChatInfo>> => {
|
||||
return request.get('/chat/list')
|
||||
},
|
||||
get: (id: number): Promise<ChatInfo> => {
|
||||
return request.get(`/chat/get/${id}`)
|
||||
},
|
||||
get_with_Data: (id: number): Promise<ChatInfo> => {
|
||||
return request.get(`/chat/get/with_data/${id}`)
|
||||
},
|
||||
get_chart_data: (record_id?: number): Promise<any> => {
|
||||
return request.get(`/chat/record/get/${record_id}/data`)
|
||||
},
|
||||
get_chart_predict_data: (record_id?: number): Promise<any> => {
|
||||
return request.get(`/chat/record/get/${record_id}/predict_data`)
|
||||
},
|
||||
startChat: (data: any): Promise<ChatInfo> => {
|
||||
return request.post('/chat/start', data)
|
||||
},
|
||||
startAssistantChat: (): Promise<ChatInfo> => {
|
||||
return request.post('/chat/assistant/start')
|
||||
},
|
||||
renameChat: (chat_id: number | undefined, brief: string): Promise<string> => {
|
||||
return request.post('/chat/rename', { id: chat_id, brief: brief })
|
||||
},
|
||||
deleteChat: (id: number | undefined): Promise<string> => {
|
||||
return request.get(`/chat/delete/${id}`)
|
||||
},
|
||||
analysis: (record_id: number | undefined, controller?: AbortController) => {
|
||||
return request.fetchStream(`/chat/record/${record_id}/analysis`, {}, controller)
|
||||
},
|
||||
predict: (record_id: number | undefined, controller?: AbortController) => {
|
||||
return request.fetchStream(`/chat/record/${record_id}/predict`, {}, controller)
|
||||
},
|
||||
recommendQuestions: (record_id: number | undefined, controller?: AbortController) => {
|
||||
return request.fetchStream(`/chat/recommend_questions/${record_id}`, {}, controller)
|
||||
},
|
||||
checkLLMModel: () => request.get('/system/aimodel/default', { requestOptions: { silent: true } }),
|
||||
export2Excel: (data: any) => request.post('/chat/excel/export', data, { responseType: 'blob' }),
|
||||
}
|
||||
Reference in New Issue
Block a user