uqlm.longform.graph.claim_merger.ClaimMerger#

class uqlm.longform.graph.claim_merger.ClaimMerger(claim_merging_llm)#

Bases: object

__init__(claim_merging_llm)#

Methods

__init__(claim_merging_llm)

merge_claims(original_claim_sets, ...[, ...])

Process claim deduplication for response sets.

async merge_claims(original_claim_sets, sampled_claim_sets, progress_bar=None)#

Process claim deduplication for response sets. Leverages ResponseGenerator’s ability to handle multiple prompts at once by collecting dedup prompts and making batch calls. If sampled_claim_sets contains only empty lists and entailment_score_sets is provided, infers master claims from entailment_score_sets keys. Otherwise returns original_claim_sets.

Return type:

List[List[str]]

References