Skip to content

Commit 4b32786

Browse files
[Manager] Update Algolia mappings (#4230)
Co-authored-by: Claude <[email protected]>
1 parent 9942b17 commit 4b32786

File tree

3 files changed

+41
-9
lines changed

3 files changed

+41
-9
lines changed

src/services/providers/algoliaSearchProvider.ts

Lines changed: 20 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,13 @@ const RETRIEVE_ATTRIBUTES: SearchAttribute[] = [
4242
'latest_version_status',
4343
'comfy_node_extract_status',
4444
'id',
45-
'icon_url'
45+
'icon_url',
46+
'github_stars',
47+
'supported_os',
48+
'supported_comfyui_version',
49+
'supported_comfyui_frontend_version',
50+
'supported_accelerators',
51+
'banner_url'
4652
]
4753

4854
const searchPacksCache = new QuickLRU<string, SearchPacksResult>({
@@ -86,9 +92,19 @@ const toRegistryPack = memoize(
8692
icon: algoliaNode.icon_url,
8793
latest_version: toRegistryLatestVersion(algoliaNode),
8894
publisher: toRegistryPublisher(algoliaNode),
95+
created_at: algoliaNode.create_time,
96+
category: algoliaNode.category,
97+
author: algoliaNode.author,
98+
tags: algoliaNode.tags,
99+
github_stars: algoliaNode.github_stars,
100+
supported_os: algoliaNode.supported_os,
101+
supported_comfyui_version: algoliaNode.supported_comfyui_version,
102+
supported_comfyui_frontend_version:
103+
algoliaNode.supported_comfyui_frontend_version,
104+
supported_accelerators: algoliaNode.supported_accelerators,
105+
banner_url: algoliaNode.banner_url,
89106
// @ts-expect-error comfy_nodes also not in node info
90-
comfy_nodes: algoliaNode.comfy_nodes,
91-
create_time: algoliaNode.create_time
107+
comfy_nodes: algoliaNode.comfy_nodes
92108
}
93109
},
94110
(algoliaNode: AlgoliaNodePack) => algoliaNode.id
@@ -187,9 +203,7 @@ export const useAlgoliaSearchProvider = (): NodePackSearchProvider => {
187203
case SortableAlgoliaField.Downloads:
188204
return pack.downloads ?? 0
189205
case SortableAlgoliaField.Created: {
190-
// TODO: add create time to backend return type
191-
// @ts-expect-error create_time is not in the RegistryNodePack type
192-
const createTime = pack.create_time
206+
const createTime = pack.created_at
193207
return createTime ? new Date(createTime).getTime() : 0
194208
}
195209
case SortableAlgoliaField.Updated:

src/types/algoliaTypes.ts

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,15 @@ export interface AlgoliaNodePack {
5959
'comfy_node_extract_status'
6060
>
6161
icon_url: RegistryNodePack['icon']
62+
category: RegistryNodePack['category']
63+
author: RegistryNodePack['author']
64+
tags: RegistryNodePack['tags']
65+
github_stars: RegistryNodePack['github_stars']
66+
supported_os: RegistryNodePack['supported_os']
67+
supported_comfyui_version: RegistryNodePack['supported_comfyui_version']
68+
supported_comfyui_frontend_version: RegistryNodePack['supported_comfyui_frontend_version']
69+
supported_accelerators: RegistryNodePack['supported_accelerators']
70+
banner_url: RegistryNodePack['banner_url']
6271
}
6372

6473
/**

tests-ui/tests/services/algoliaSearchProvider.test.ts

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,17 @@ describe('useAlgoliaSearchProvider', () => {
108108
id: 'publisher-1',
109109
name: 'publisher-1'
110110
},
111-
create_time: '2024-01-01T00:00:00Z',
112-
comfy_nodes: ['LoadImage', 'SaveImage']
111+
created_at: '2024-01-01T00:00:00Z',
112+
comfy_nodes: ['LoadImage', 'SaveImage'],
113+
category: undefined,
114+
author: undefined,
115+
tags: undefined,
116+
github_stars: undefined,
117+
supported_os: undefined,
118+
supported_comfyui_version: undefined,
119+
supported_comfyui_frontend_version: undefined,
120+
supported_accelerators: undefined,
121+
banner_url: undefined
113122
})
114123
})
115124

@@ -253,7 +262,7 @@ describe('useAlgoliaSearchProvider', () => {
253262
version: '1.0.0',
254263
createdAt: '2024-01-15T10:00:00Z'
255264
},
256-
create_time: '2024-01-01T10:00:00Z'
265+
created_at: '2024-01-01T10:00:00Z'
257266
}
258267

259268
it('should return correct values for each sort field', () => {

0 commit comments

Comments
 (0)