Пользователь вводит фрагмент кода на ts, чтобы гарантировать 100% покрытие всех функций и ветвлений, необходимо определить, какие сценарии данных следует учесть.
Например:
- Отсутствие сессии: в тестовых данных нет ни одной сессии, ожидаемый результат — sessionTree с только стандартным агентом.
- Одна сессия без systemRole: одна сессия без systemRole, ожидаемый результат — sessionTree с включённым стандартным агентом, при этом список чатов стандартного агента содержит эту сессию.
- Одна сессия с systemRole: одна сессия с systemRole, ожидаемый результат — sessionTree, включающий нового агента и стандартного агента. Список чатов нового агента содержит эту сессию.
import { produce } from 'immer';
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'); }
}
};Примеры использования не требуются.
