| import { EModelEndpoint } from 'librechat-data-provider'; | |
| import { getPresetTitle, removeUnavailableTools } from '../presets'; | |
| import type { TPreset, TPlugin } from 'librechat-data-provider'; | |
| describe('presets utils', () => { | |
| describe('getPresetTitle', () => { | |
| const basePreset: TPreset = { | |
| presetId: 'test-id', | |
| title: 'Test Preset', | |
| endpoint: EModelEndpoint.openAI, | |
| model: 'gpt-4', | |
| }; | |
| describe('with modelLabel', () => { | |
| it('should use modelLabel as the label', () => { | |
| const preset = { | |
| ...basePreset, | |
| modelLabel: 'Custom Model Name', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Test Preset: gpt-4 (Custom Model Name)'); | |
| }); | |
| it('should prioritize modelLabel over deprecated chatGptLabel', () => { | |
| const preset = { | |
| ...basePreset, | |
| modelLabel: 'New Model Label', | |
| chatGptLabel: 'Old ChatGPT Label', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Test Preset: gpt-4 (New Model Label)'); | |
| }); | |
| it('should handle title that includes the label', () => { | |
| const preset = { | |
| ...basePreset, | |
| title: 'Custom Model Name Settings', | |
| modelLabel: 'Custom Model Name', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Custom Model Name Settings: gpt-4 (Custom Model Name)'); | |
| }); | |
| it('should handle case-insensitive title matching', () => { | |
| const preset = { | |
| ...basePreset, | |
| title: 'custom model name preset', | |
| modelLabel: 'Custom Model Name', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('custom model name preset: gpt-4 (Custom Model Name)'); | |
| }); | |
| it('should use label as title when label includes the title', () => { | |
| const preset = { | |
| ...basePreset, | |
| title: 'GPT', | |
| modelLabel: 'Custom GPT Assistant', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Custom GPT Assistant: gpt-4'); | |
| }); | |
| }); | |
| describe('without modelLabel', () => { | |
| it('should work without modelLabel', () => { | |
| const preset = { ...basePreset }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Test Preset: gpt-4'); | |
| }); | |
| it('should handle empty modelLabel', () => { | |
| const preset = { | |
| ...basePreset, | |
| modelLabel: '', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Test Preset: gpt-4'); | |
| }); | |
| it('should handle null modelLabel', () => { | |
| const preset = { | |
| ...basePreset, | |
| modelLabel: null, | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Test Preset: gpt-4'); | |
| }); | |
| }); | |
| describe('title variations', () => { | |
| it('should handle missing title', () => { | |
| const preset = { | |
| ...basePreset, | |
| title: null, | |
| modelLabel: 'Custom Model', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('gpt-4 (Custom Model)'); | |
| }); | |
| it('should handle empty title', () => { | |
| const preset = { | |
| ...basePreset, | |
| title: '', | |
| modelLabel: 'Custom Model', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('gpt-4 (Custom Model)'); | |
| }); | |
| it('should handle "New Chat" title', () => { | |
| const preset = { | |
| ...basePreset, | |
| title: 'New Chat', | |
| modelLabel: 'Custom Model', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('gpt-4 (Custom Model)'); | |
| }); | |
| it('should handle title with whitespace', () => { | |
| const preset = { | |
| ...basePreset, | |
| title: ' ', | |
| modelLabel: 'Custom Model', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe(': gpt-4 (Custom Model)'); | |
| }); | |
| }); | |
| describe('mention mode', () => { | |
| it('should return mention format with all components', () => { | |
| const preset = { | |
| ...basePreset, | |
| modelLabel: 'Custom Model', | |
| promptPrefix: 'You are a helpful assistant', | |
| tools: ['plugin1', 'plugin2'] as string[], | |
| }; | |
| const result = getPresetTitle(preset, true); | |
| expect(result).toBe( | |
| 'gpt-4 | Custom Model | You are a helpful assistant | plugin1, plugin2', | |
| ); | |
| }); | |
| it('should handle mention format with object tools', () => { | |
| const preset = { | |
| ...basePreset, | |
| modelLabel: 'Custom Model', | |
| tools: [ | |
| { pluginKey: 'plugin1', name: 'Plugin 1' } as TPlugin, | |
| { pluginKey: 'plugin3', name: 'Plugin 3' } as TPlugin, | |
| ] as TPlugin[], | |
| }; | |
| const result = getPresetTitle(preset, true); | |
| expect(result).toBe('gpt-4 | Custom Model | plugin1, plugin3'); | |
| }); | |
| it('should handle mention format with minimal data', () => { | |
| const preset = { ...basePreset }; | |
| const result = getPresetTitle(preset, true); | |
| expect(result).toBe('gpt-4'); | |
| }); | |
| it('should handle mention format with only modelLabel', () => { | |
| const preset = { | |
| ...basePreset, | |
| modelLabel: 'Custom Model', | |
| }; | |
| const result = getPresetTitle(preset, true); | |
| expect(result).toBe('gpt-4 | Custom Model'); | |
| }); | |
| it('should handle mention format with only promptPrefix', () => { | |
| const preset = { | |
| ...basePreset, | |
| promptPrefix: 'Custom prompt', | |
| }; | |
| const result = getPresetTitle(preset, true); | |
| expect(result).toBe('gpt-4 | Custom prompt'); | |
| }); | |
| }); | |
| describe('edge cases', () => { | |
| it('should handle missing model', () => { | |
| const preset = { | |
| ...basePreset, | |
| model: null, | |
| modelLabel: 'Custom Model', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Test Preset: (Custom Model)'); | |
| }); | |
| it('should handle undefined model', () => { | |
| const preset = { | |
| ...basePreset, | |
| model: undefined, | |
| modelLabel: 'Custom Model', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe('Test Preset: (Custom Model)'); | |
| }); | |
| it('should trim the final result', () => { | |
| const preset = { | |
| ...basePreset, | |
| title: '', | |
| model: '', | |
| modelLabel: '', | |
| }; | |
| const result = getPresetTitle(preset); | |
| expect(result).toBe(''); | |
| }); | |
| }); | |
| }); | |
| describe('removeUnavailableTools', () => { | |
| const basePreset: TPreset = { | |
| presetId: 'test-id', | |
| title: 'Test Preset', | |
| endpoint: EModelEndpoint.openAI, | |
| model: 'gpt-4', | |
| }; | |
| const availableTools: Record<string, TPlugin | undefined> = { | |
| plugin1: { pluginKey: 'plugin1', name: 'Plugin 1' } as TPlugin, | |
| plugin2: { pluginKey: 'plugin2', name: 'Plugin 2' } as TPlugin, | |
| plugin3: { pluginKey: 'plugin3', name: 'Plugin 3' } as TPlugin, | |
| }; | |
| it('should remove unavailable tools from string array', () => { | |
| const preset = { | |
| ...basePreset, | |
| tools: ['plugin1', 'unavailable-plugin', 'plugin2'] as string[], | |
| }; | |
| const result = removeUnavailableTools(preset, availableTools); | |
| expect(result.tools).toEqual(['plugin1', 'plugin2']); | |
| }); | |
| it('should remove unavailable tools from object array', () => { | |
| const preset = { | |
| ...basePreset, | |
| tools: [ | |
| { pluginKey: 'plugin1', name: 'Plugin 1' } as TPlugin, | |
| { pluginKey: 'unavailable-plugin', name: 'Unavailable' } as TPlugin, | |
| { pluginKey: 'plugin2', name: 'Plugin 2' } as TPlugin, | |
| ] as TPlugin[], | |
| }; | |
| const result = removeUnavailableTools(preset, availableTools); | |
| expect(result.tools).toEqual(['plugin1', 'plugin2']); | |
| }); | |
| it('should handle preset without tools', () => { | |
| const preset = { ...basePreset }; | |
| const result = removeUnavailableTools(preset, availableTools); | |
| expect(result).toEqual(preset); | |
| }); | |
| it('should handle preset with empty tools array', () => { | |
| const preset = { | |
| ...basePreset, | |
| tools: [] as string[], | |
| }; | |
| const result = removeUnavailableTools(preset, availableTools); | |
| expect(result.tools).toEqual([]); | |
| }); | |
| it('should remove all tools when none are available', () => { | |
| const preset = { | |
| ...basePreset, | |
| tools: ['unavailable1', 'unavailable2'] as string[], | |
| }; | |
| const result = removeUnavailableTools(preset, {}); | |
| expect(result.tools).toEqual([]); | |
| }); | |
| it('should preserve all other preset properties', () => { | |
| const preset = { | |
| ...basePreset, | |
| tools: ['plugin1'] as string[], | |
| modelLabel: 'Custom Model', | |
| temperature: 0.8, | |
| promptPrefix: 'Test prompt', | |
| }; | |
| const result = removeUnavailableTools(preset, availableTools); | |
| expect(result.presetId).toBe(preset.presetId); | |
| expect(result.title).toBe(preset.title); | |
| expect(result.endpoint).toBe(preset.endpoint); | |
| expect(result.model).toBe(preset.model); | |
| expect(result.modelLabel).toBe(preset.modelLabel); | |
| expect(result.temperature).toBe(preset.temperature); | |
| expect(result.promptPrefix).toBe(preset.promptPrefix); | |
| expect(result.tools).toEqual(['plugin1']); | |
| }); | |
| it('should not mutate the original preset', () => { | |
| const preset = { | |
| ...basePreset, | |
| tools: ['plugin1', 'unavailable-plugin'] as string[], | |
| }; | |
| const originalTools = [...preset.tools]; | |
| removeUnavailableTools(preset, availableTools); | |
| expect(preset.tools).toEqual(originalTools); | |
| }); | |
| }); | |
| }); | |