buster/web/src/api/buster_socket/chats/chatRequests.ts

import type { BusterSocketRequestBase } from '../base_interfaces';

export type ChatCreateNewChat = BusterSocketRequestBase<
  '/chats/post',
  {
    dataset_id: string | null;
    prompt?: string; //send if we are starting a new chat
    chat_id?: string | null; //send if we are following up on a chat
    suggestion_id?: string | null; //send if we clicked on a suggestion
    message_id?: string; //send if we want to REPLACE current message
    metric_id?: string; //send if we want to create a chat from a metric
    draft_session_id?: string; //TODO: do we need this?
  }
>;

export type ChatGetChat = BusterSocketRequestBase<'/chats/get', { id: string }>;

export type ChatUnsubscribeFromChat = BusterSocketRequestBase<'/chats/unsubscribe', { id: string }>;

export type ChatListEmitPayload = BusterSocketRequestBase<
  '/chats/list',
  {
    page_token: number;
    page_size: number;
    admin_view: boolean;
    filters?: {};
  }
>;

export type ChatDeleteChat = BusterSocketRequestBase<'/chats/delete', { id: string }>;

export type ChatUpdateChat = BusterSocketRequestBase<
  '/chats/update',
  { id: string; title?: string; is_favorited?: boolean }
>;

export type ChatsSearch = BusterSocketRequestBase<'/chats/search', { prompt: string }>;

export type ChatsDuplicateChat = BusterSocketRequestBase<
  '/chats/duplicate',
  {
    id: string;
    message_id: string; //send if we want to duplciate the chat starting from a specific message
  }
>;

export type ChatEmits =
  | ChatCreateNewChat
  | ChatGetChat
  | ChatUnsubscribeFromChat
  | ChatListEmitPayload
  | ChatDeleteChat
  | ChatUpdateChat
  | ChatsSearch
  | ChatsDuplicateChat;