61 lines
1.8 KiB
TypeScript
61 lines
1.8 KiB
TypeScript
import AiService from "@/services/AiService";
|
|
|
|
export default class KaiService extends AiService {
|
|
getKaiChats = async (projectId: string): Promise<{ title: string, threadId: string }[]> => {
|
|
const r = await this.client.get(`/kai/${projectId}/chats`);
|
|
if (!r.ok) {
|
|
throw new Error('Failed to fetch chats');
|
|
}
|
|
const data = await r.json();
|
|
return data;
|
|
}
|
|
|
|
deleteKaiChat = async (projectId: string, threadId: string): Promise<boolean> => {
|
|
const r = await this.client.delete(`/kai/${projectId}/chats/${threadId}`);
|
|
if (!r.ok) {
|
|
throw new Error('Failed to delete chat');
|
|
}
|
|
return true;
|
|
}
|
|
|
|
getKaiChat = async (projectId: string, threadId: string): Promise<{ role: string, content: string, message_id: any, duration?: number }[]> => {
|
|
const r = await this.client.get(`/kai/${projectId}/chats/${threadId}`);
|
|
if (!r.ok) {
|
|
throw new Error('Failed to fetch chat');
|
|
}
|
|
const data = await r.json();
|
|
return data;
|
|
}
|
|
|
|
createKaiChat = async (projectId: string): Promise<number> => {
|
|
const r = await this.client.get(`/kai/${projectId}/chat/new`)
|
|
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) => {
|
|
const r = await this.client.post(`/kai/${projectId}/messages/feedback`, {
|
|
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) => {
|
|
const r = await this.client.post(`/kai/${projectId}/cancel/${threadId}`);
|
|
if (!r.ok) {
|
|
throw new Error('Failed to cancel generation');
|
|
}
|
|
|
|
const data = await r.json();
|
|
return data;
|
|
}
|
|
}
|