search

HKUDS / LightRAG

"LightRAG: Simple and Fast Retrieval-Augmented Generation"
https://arxiv.org/abs/2410.05779
MIT License
9.22k stars 1.13k forks source link

Custom Entity Types and Prompts #308

Open aliyeysides opened 2 days ago

aliyeysides commented 2 days ago

Is there a way to override the default entity types created in prompt.py?

PROMPTS["DEFAULT_ENTITY_TYPES"] = ["organization", "person", "geo", "event"]

Same question for the entity_extraction prompt. I'd like to add more specific examples that relate to our domain.

nihirv commented 1 day ago

if you've installed with pip editable, you can just modify that list