File size: 2,639 Bytes
aa2e6af | 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 | import type { TPreset, TPlugin } from 'librechat-data-provider';
import { EModelEndpoint } from 'librechat-data-provider';
export const getPresetIcon = (preset: TPreset, Icon) => {
return Icon({
size: 20,
endpoint: preset?.endpoint,
model: preset?.model,
error: false,
className: 'icon-md',
isCreatedByUser: false,
});
};
type TEndpoints = Array<string | EModelEndpoint>;
export const getPresetTitle = (preset: TPreset, mention?: boolean) => {
const {
endpoint,
title: presetTitle,
model,
tools,
promptPrefix,
chatGptLabel,
modelLabel,
jailbreak,
toneStyle,
} = preset;
let title = '';
let modelInfo = model || '';
let label = '';
const usesChatGPTLabel: TEndpoints = [
EModelEndpoint.azureOpenAI,
EModelEndpoint.openAI,
EModelEndpoint.custom,
];
const usesModelLabel: TEndpoints = [EModelEndpoint.google, EModelEndpoint.anthropic];
if (endpoint && usesChatGPTLabel.includes(endpoint)) {
label = chatGptLabel || '';
} else if (endpoint && usesModelLabel.includes(endpoint)) {
label = modelLabel || '';
} else if (endpoint === EModelEndpoint.bingAI) {
modelInfo = jailbreak ? 'Sydney' : modelInfo;
label = toneStyle ? `: ${toneStyle}` : '';
}
if (label && presetTitle && label.toLowerCase().includes(presetTitle.toLowerCase())) {
title = label + ': ';
label = '';
} else if (presetTitle && presetTitle.trim() !== 'New Chat') {
title = presetTitle + ': ';
}
if (mention) {
return `${modelInfo}${label ? ` | ${label}` : ''}${promptPrefix ? ` | ${promptPrefix}` : ''}${
tools
? ` | ${tools
.map((tool: TPlugin | string) => {
if (typeof tool === 'string') {
return tool;
}
return tool.pluginKey;
})
.join(', ')}`
: ''
}`;
}
return `${title}${modelInfo}${label ? ` (${label})` : ''}`.trim();
};
/** Remove unavailable tools from the preset */
export const removeUnavailableTools = (
preset: TPreset,
availableTools: Record<string, TPlugin>,
) => {
const newPreset = { ...preset };
if (newPreset.tools && newPreset.tools.length > 0) {
newPreset.tools = newPreset.tools
.filter((tool) => {
let pluginKey: string;
if (typeof tool === 'string') {
pluginKey = tool;
} else {
({ pluginKey } = tool);
}
return !!availableTools[pluginKey];
})
.map((tool) => {
if (typeof tool === 'string') {
return tool;
}
return tool.pluginKey;
});
}
return newPreset;
};
|