File size: 2,824 Bytes
f0743f4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 | import type * as t from 'librechat-data-provider';
import type { TPluginMap } from '~/common';
/** Maps Attachments by `toolCallId` for quick lookup */
export function mapAttachments(attachments: Array<t.TAttachment | null | undefined>) {
const attachmentMap: Record<string, t.TAttachment[] | undefined> = {};
for (const attachment of attachments) {
if (attachment === null || attachment === undefined) {
continue;
}
const key = attachment.toolCallId || '';
if (key.length === 0) {
continue;
}
if (!attachmentMap[key]) {
attachmentMap[key] = [];
}
attachmentMap[key]?.push(attachment);
}
return attachmentMap;
}
/** Maps Files by `file_id` for quick lookup */
export function mapFiles(files: t.TFile[]) {
const fileMap = {} as Record<string, t.TFile>;
for (const file of files) {
fileMap[file.file_id] = file;
}
return fileMap;
}
/** Maps Assistants by `id` for quick lookup */
export function mapAssistants(assistants: t.Assistant[]) {
const assistantMap = {} as Record<string, t.Assistant>;
for (const assistant of assistants) {
assistantMap[assistant.id] = assistant;
}
return assistantMap;
}
/** Maps Agents by `id` for quick lookup */
export function mapAgents(agents: t.Agent[]) {
const agentsMap = {} as Record<string, t.Agent>;
for (const agent of agents) {
agentsMap[agent.id] = agent;
}
return agentsMap;
}
/** Maps Plugins by `pluginKey` for quick lookup */
export function mapPlugins(plugins: t.TPlugin[]): TPluginMap {
return plugins.reduce((acc, plugin) => {
acc[plugin.pluginKey] = plugin;
return acc;
}, {} as TPluginMap);
}
/** Transform query data to object with list and map fields */
export const selectPlugins = (
data: t.TPlugin[] | undefined,
): {
list: t.TPlugin[];
map: TPluginMap;
} => {
if (!data) {
return {
list: [],
map: {},
};
}
return {
list: data,
map: mapPlugins(data),
};
};
/** Transform array to TPlugin values */
export function processPlugins(
tools: (string | t.TPlugin)[],
allPlugins?: TPluginMap,
): t.TPlugin[] {
return tools
.map((tool: string | t.TPlugin) => {
if (typeof tool === 'string') {
return allPlugins?.[tool];
}
return tool;
})
.filter((tool: t.TPlugin | undefined): tool is t.TPlugin => tool !== undefined);
}
export function mapToolCalls(toolCalls: t.ToolCallResults = []): {
[key: string]: t.ToolCallResult[] | undefined;
} {
return toolCalls.reduce(
(acc, call) => {
const key = `${call.messageId}_${call.partIndex ?? 0}_${call.blockIndex ?? 0}_${call.toolId}`;
const array = acc[key] ?? [];
array.push(call);
acc[key] = array;
return acc;
},
{} as { [key: string]: t.ToolCallResult[] | undefined },
);
}
|