openreplay/frontend/app/components/Kai/KaiService.ts

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;
  }
}