mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-20 02:10:15 +01:00
* feat: move buildTree function for message hierarchy to data provider * refactor: consolidate buildTree import from utils to data provider * fix: correct string interpolation in messages function, which caused message search requests to fail
This commit is contained in:
parent
78d735f35c
commit
a362963017
12 changed files with 63 additions and 94 deletions
50
packages/data-provider/src/messages.ts
Normal file
50
packages/data-provider/src/messages.ts
Normal file
|
|
@ -0,0 +1,50 @@
|
|||
import type { TFile } from './types/files';
|
||||
import type { TMessage } from './types';
|
||||
|
||||
export type ParentMessage = TMessage & { children: TMessage[]; depth: number };
|
||||
export function buildTree({
|
||||
messages,
|
||||
fileMap,
|
||||
}: {
|
||||
messages: (TMessage | undefined)[] | null;
|
||||
fileMap?: Record<string, TFile>;
|
||||
}) {
|
||||
if (messages === null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
const messageMap: Record<string, ParentMessage> = {};
|
||||
const rootMessages: TMessage[] = [];
|
||||
const childrenCount: Record<string, number> = {};
|
||||
|
||||
messages.forEach((message) => {
|
||||
if (!message) {
|
||||
return;
|
||||
}
|
||||
const parentId = message.parentMessageId ?? '';
|
||||
childrenCount[parentId] = (childrenCount[parentId] || 0) + 1;
|
||||
|
||||
const extendedMessage: ParentMessage = {
|
||||
...message,
|
||||
children: [],
|
||||
depth: 0,
|
||||
siblingIndex: childrenCount[parentId] - 1,
|
||||
};
|
||||
|
||||
if (message.files && fileMap) {
|
||||
extendedMessage.files = message.files.map((file) => fileMap[file.file_id ?? ''] ?? file);
|
||||
}
|
||||
|
||||
messageMap[message.messageId] = extendedMessage;
|
||||
|
||||
const parentMessage = messageMap[parentId];
|
||||
if (parentMessage) {
|
||||
parentMessage.children.push(extendedMessage);
|
||||
extendedMessage.depth = parentMessage.depth + 1;
|
||||
} else {
|
||||
rootMessages.push(extendedMessage);
|
||||
}
|
||||
});
|
||||
|
||||
return rootMessages;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue