80 lines
1.9 KiB
TypeScript
80 lines
1.9 KiB
TypeScript
import AiService from '@/services/AiService';
|
|
|
|
export default class KaiService extends AiService {
|
|
getKaiChats = async (
|
|
projectId: string,
|
|
): Promise<{ title: string; thread_id: 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;
|
|
feedback: boolean | null;
|
|
}[]
|
|
> => {
|
|
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,
|
|
});
|
|
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;
|
|
};
|
|
}
|