110 lines
3.6 KiB
TypeScript
110 lines
3.6 KiB
TypeScript
import AiService from "@/services/AiService";
|
|
|
|
export default class KaiService extends AiService {
|
|
getKaiChats = async (userId: string, projectId: string): Promise<{ title: string, threadId: string }[]> => {
|
|
// const r = await this.client.get('/kai/PROJECT_ID/chats');
|
|
const jwt = window.env.KAI_TESTING // this.client.getJwt()
|
|
const r = await fetch(`http://localhost:8700/kai/${projectId}/chats?user_id=${userId}`, {
|
|
headers: new Headers({
|
|
Accept: 'application/json',
|
|
'Content-Type': 'application/json',
|
|
'Authorization': `Bearer ${jwt}`
|
|
}),
|
|
});
|
|
if (!r.ok) {
|
|
throw new Error('Failed to fetch chats');
|
|
}
|
|
const data = await r.json();
|
|
return data;
|
|
}
|
|
|
|
deleteKaiChat = async (projectId: string, userId: string, threadId: string): Promise<boolean> => {
|
|
const jwt = window.env.KAI_TESTING // this.client.getJwt()
|
|
const r = await fetch(`http://localhost:8700/kai/${projectId}/chats/${threadId}?user_id=${userId}`, {
|
|
method: 'DELETE',
|
|
headers: new Headers({
|
|
Accept: 'application/json',
|
|
'Content-Type': 'application/json',
|
|
'Authorization': `Bearer ${jwt}`
|
|
}),
|
|
});
|
|
if (!r.ok) {
|
|
throw new Error('Failed to delete chat');
|
|
}
|
|
return true;
|
|
}
|
|
|
|
getKaiChat = async (projectId: string, userId: string, threadId: string): Promise<{ role: string, content: string, message_id: any }[]> => {
|
|
const jwt = window.env.KAI_TESTING // this.client.getJwt()
|
|
const r = await fetch(`http://localhost:8700/kai/${projectId}/chats/${threadId}?user_id=${userId}`, {
|
|
method: 'GET',
|
|
headers: new Headers({
|
|
Accept: 'application/json',
|
|
'Content-Type': 'application/json',
|
|
'Authorization': `Bearer ${jwt}`
|
|
}),
|
|
});
|
|
if (!r.ok) {
|
|
throw new Error('Failed to fetch chat');
|
|
}
|
|
const data = await r.json();
|
|
return data;
|
|
}
|
|
|
|
createKaiChat = async (projectId: string, userId: string): Promise<number> => {
|
|
const jwt = window.env.KAI_TESTING // this.client.getJwt()
|
|
const r = await fetch(`http://localhost:8700/kai/${projectId}/chat/new?user_id=${userId}`, {
|
|
method: 'GET',
|
|
headers: new Headers({
|
|
Accept: 'application/json',
|
|
'Content-Type': 'application/json',
|
|
'Authorization': `Bearer ${jwt}`
|
|
}),
|
|
})
|
|
if (!r.ok) {
|
|
throw new Error('Failed to create chat');
|
|
}
|
|
const data = await r.json();
|
|
return data;
|
|
}
|
|
|
|
feedback = async (positive: boolean | null, messageId: string, projectId: string, userId: string) => {
|
|
const jwt = window.env.KAI_TESTING // this.client.getJwt()
|
|
const r = await fetch(`http://localhost:8700/kai/${projectId}/messages/feedback`, {
|
|
method: 'POST',
|
|
headers: new Headers({
|
|
Accept: 'application/json',
|
|
'Content-Type': 'application/json',
|
|
'Authorization': `Bearer ${jwt}`
|
|
}),
|
|
body: JSON.stringify({
|
|
message_id: messageId,
|
|
value: positive,
|
|
user_id: userId,
|
|
}),
|
|
});
|
|
if (!r.ok) {
|
|
throw new Error('Failed to send feedback');
|
|
}
|
|
|
|
return await r.json()
|
|
}
|
|
|
|
cancelGeneration = async (projectId: string, threadId: string, userId: string) => {
|
|
const jwt = window.env.KAI_TESTING // this.client.getJwt()
|
|
const r = await fetch(`http://localhost:8700/kai/${projectId}/cancel/${threadId}?user_id=${userId}`, {
|
|
method: 'POST',
|
|
headers: new Headers({
|
|
Accept: 'application/json',
|
|
'Content-Type': 'application/json',
|
|
'Authorization': `Bearer ${jwt}`
|
|
}),
|
|
});
|
|
if (!r.ok) {
|
|
throw new Error('Failed to cancel generation');
|
|
}
|
|
|
|
const data = await r.json();
|
|
return data;
|
|
}
|
|
}
|