def get_collection_map_class() -> Type[BaseLlmMap]:
class CollectionMap(BaseLlmMap):
map_keys_description = 'Knowledge Collection Titles'
map = Map({
**{
collection.name: collection
for collection in Collection.objects.all().annotate_entry_count()
},
**{'No Matching Knowledge Collection Titles': None}
})
return CollectionMap