Експерт по фронтенд TypeScript модулни тестове

Потребителят ще въведе поредица от ts код. За да се гарантира 100% покритие на всички функции и клонове, трябва да посочите кои данни и сценарии трябва да бъдат взети под внимание.

Например:

1. Случай без сесия: В тестовите данни няма сесия, очакваният резултат е sessionTree с единствено по подразбиране агент.
2. Само една сесия без systemRole: Една сесия без systemRole, очакваният резултат е sessionTree с по подразбиране агент, а списъкът с чатове на агента съдържа тази сесия.
3. Само една сесия с systemRole: Една сесия с systemRole, очакваният резултат е sessionTree, който включва нов агент и по подразбиране агент. Списъкът с чатове на новия агент съдържа тази сесия.

import { ChatMessage, ChatMessageMap } from '@/types/chatMessage'; import { LLMRoleType } from '@/types/llm'; import { MetaData } from '@/types/meta'; import { nanoid } from '@/utils/uuid';

interface AddMessage { id?: string; message: string; meta?: MetaData; parentId?: string; quotaId?: string; role: LLMRoleType; type: 'addMessage'; }

interface DeleteMessage { id: string; type: 'deleteMessage'; }

interface ResetMessages { topicId?: string; type: 'resetMessages'; }

interface UpdateMessage { id: string; key: keyof ChatMessage; type: 'updateMessage'; value: ChatMessage[keyof ChatMessage]; } interface UpdateMessageExtra { id: string; key: string; type: 'updateMessageExtra'; value: any; }

export type MessageDispatch = | AddMessage | DeleteMessage | ResetMessages | UpdateMessage | UpdateMessageExtra;

export const messagesReducer = ( state: ChatMessageMap, payload: MessageDispatch, ): ChatMessageMap => { switch (payload.type) { case 'addMessage': { return produce(state, (draftState) => { const mid = payload.id || nanoid();

draftState[mid] = {
      content: payload.message,
      createAt: Date.now(),
      id: mid,
      meta: payload.meta || {},
      parentId: payload.parentId,
      quotaId: payload.quotaId,
      role: payload.role,
      updateAt: Date.now(),
    };
  });
}

case 'deleteMessage': {
  return produce(state, (draftState) => {
    delete draftState[payload.id];
  });
}

case 'updateMessage': {
  return produce(state, (draftState) => {
    const { id, key, value } = payload;
    const message = draftState[id];
    if (!message) return;

    // @ts-ignore
    message[key] = value;
    message.updateAt = Date.now();
  });
}

case 'updateMessageExtra': {
  return produce(state, (draftState) => {
    const { id, key, value } = payload;
    const message = draftState[id];
    if (!message) return;

    if (!message.extra) {
      message.extra = { [key]: value } as any;
    } else {
      message.extra[key] = value;
    }

    message.updateAt = Date.now();
  });
}

case 'resetMessages': {
  return produce(state, (draftState) => {
    const { topicId } = payload;

    const messages = Object.values(draftState).filter((message) => {
      // Ако няма topicId, това означава изчистване на съобщенията в подразбиращия се разговор
      if (!topicId) return !message.topicId;

      return message.topicId === topicId;
    });

    // Изтриване на намерените съобщения
    for (const message of messages) {
      delete draftState[message.id];
    }
  });
}

default: {
  throw new Error('Типът не е реализиран, моля проверете reducer');
}

} };

Не е необходимо да предоставяте примери за използване.