chatdoc.ts

import { ACTION_CREATE, ACTION_DELETE, ACTION_UPDATE, ROLE_ASSISTANT, ROLE_SYSTEM, ROLE_USER } from '@/constants';

export interface IError {
  code: string;
  detail?: string;
}

export interface IChatdocRepository {
  id: string;
  name?: string;
  description?: string;
  deleting?: boolean;
  editing?: boolean;
  documents?: IChatdocDocument[];
  conversations?: IChatdocConversation[];
}

export interface IChatdocRepositoryRequest extends IChatdocRepository {
  action: typeof ACTION_CREATE | typeof ACTION_UPDATE | typeof ACTION_DELETE;
}

export interface IChatdocRepositoryResponse extends IChatdocRepository {}
export type IChatdocRepositoriesResponse = {
  items: IChatdocRepository[];
  count: number;
};

export interface IChatdocDocument {
  id: string;
  repository_id: string;
  file_url: string;
  file_name: string;
}

export interface IChatdocDocumentRequest extends IChatdocDocument {
  action: typeof ACTION_CREATE | typeof ACTION_UPDATE | typeof ACTION_DELETE;
  callback_url?: string;
}

export interface IChatdocDocumentResponse extends IChatdocDocument {}

export type IChatdocDocumentsResponse = {
  items: IChatdocDocument[];
  count: number;
};

export interface IChatdocConversation {
  id: string;
  repository_id: string;
  messages: IChatdocMessage[];
  question: string;
  editing?: boolean;
  deleting?: boolean;
}

export interface IChatdocMessage {
  content?: string;
  error?: IError;
  state?: IChatdocMessageState;
  role?: typeof ROLE_SYSTEM | typeof ROLE_ASSISTANT | typeof ROLE_USER;
}

export enum IChatdocMessageState {
  PENDING = 'pending',
  ANSWERING = 'answering',
  FINISHED = 'finished',
  FAILED = 'failed'
}

export interface IChatdocConversationRequest {
  id: string;
  repository_id: string;
  question: string;
  temperature?: number;
  knowledge_fallback?: boolean;
}

export type IChatdocConversationsResponse = {
  items: IChatdocConversation[];
  count: number;
};

export interface IChatdocConversationResponse {
  id: string;
  answer: string;
  delta_answer?: string;
}