mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 08:12:00 +02: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
|
@ -1,6 +1,6 @@
|
|||
{
|
||||
"name": "librechat-data-provider",
|
||||
"version": "0.8.004",
|
||||
"version": "0.8.005",
|
||||
"description": "data services for librechat apps",
|
||||
"main": "dist/index.js",
|
||||
"module": "dist/index.es.js",
|
||||
|
|
|
@ -58,7 +58,7 @@ export const messages = (params: q.MessagesListParams) => {
|
|||
return `${messagesRoot}/${conversationId}`;
|
||||
}
|
||||
|
||||
return `${messagesRoot}{buildQuery(rest)}`;
|
||||
return `${messagesRoot}${buildQuery(rest)}`;
|
||||
};
|
||||
|
||||
export const messagesArtifacts = (messageId: string) => `${messagesRoot}/artifacts/${messageId}`;
|
||||
|
|
|
@ -3,6 +3,8 @@ export * from './azure';
|
|||
export * from './bedrock';
|
||||
export * from './config';
|
||||
export * from './file-config';
|
||||
/* messages */
|
||||
export * from './messages';
|
||||
/* artifacts */
|
||||
export * from './artifacts';
|
||||
/* schema helpers */
|
||||
|
|
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