diff --git a/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/shared.test.ts b/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/shared.test.ts index d3bb88f3ad..5ad5911b33 100644 --- a/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/shared.test.ts +++ b/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/shared.test.ts @@ -51,6 +51,31 @@ describe('defaultSearcher', () => { const result = await defaultSearcher('xyz', items) expect(result).toHaveLength(0) }) + + it('matches against label when provided', async () => { + const itemsWithLabels = [ + createItem('model_v1.safetensors', 'My Cool Model'), + createItem('lora_v2.safetensors', 'Style Transfer LoRA'), + createItem('checkpoint.ckpt', 'Realistic Vision') + ] + const result = await defaultSearcher('cool', itemsWithLabels) + expect(result).toHaveLength(1) + expect(result[0].name).toBe('model_v1.safetensors') + }) + + it('matches by label case-insensitively', async () => { + const itemsWithLabels = [createItem('file.safetensors', 'My Cool Model')] + const result = await defaultSearcher('MY COOL', itemsWithLabels) + expect(result).toHaveLength(1) + }) + + it('matches when word is in name or label', async () => { + const itemsWithLabels = [ + createItem('sd_v15.safetensors', 'Stable Diffusion 1.5') + ] + const result = await defaultSearcher('stable', itemsWithLabels) + expect(result).toHaveLength(1) + }) }) describe('getDefaultSortOptions', () => { diff --git a/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/shared.ts b/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/shared.ts index 4717b8870c..b65a6aeb74 100644 --- a/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/shared.ts +++ b/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/shared.ts @@ -12,7 +12,8 @@ export async function defaultSearcher( const words = query.trim().toLowerCase().split(' ') return items.filter((item) => { const name = item.name.toLowerCase() - return words.every((word) => name.includes(word)) + const label = item.label?.toLowerCase() ?? '' + return words.every((word) => name.includes(word) || label.includes(word)) }) }