Skip to content

collection_map

django_spire.knowledge.intelligence.maps.collection_map

get_collection_map_class

Source code in django_spire/knowledge/intelligence/maps/collection_map.py
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