graphragzen.prompt_tuning.entities

Description

Functions

create_entity_extraction_prompt(llm, ...[, ...])

Create a prompt for entity extraction.

generate_entity_categories(llm, documents, ...)

Generate entity categories from a given set of (small) documents.

generate_entity_relationship_examples(llm, ...)

Generate a list of entity/relationships examples for use in generating an entity extraction prompt.