config.py

__all__ = ["InteractionNetworkConfig", "FamilyNetworkConfig", "WikiConfig"]


class WikiConfig:
    # relevant fields to parse in character info box on coppermind.net
    info_fields = (
        "name",
        "aliases",
        "titles",
        "books",
        "world",
        "abilities",
        "universe",
        "family",
        "parents",
        "siblings",
        "relatives",
        "spouse",
        "children",
        "bonded",
        "descendants",
        "ancestors",
        "residence",
        "groups",
        "nation",
        "nationality",
        "profession",
        "ethnicity",
        "species",
        "occupation",
        "unnamed",
        "died",
    )


class InteractionNetworkConfig:
    run_size = 25  # tokens
    prev_cxt_lines = 4
    next_cxt_lines = 3
    default_min_weight = 2


class FamilyNetworkConfig:
    default_min_component_size = 2
    relation_fields = (
        "parents",
        "siblings",
        "spouse",
        "children",
        "bonded",
    )  # currently: nuclear family only
    _unused_fields = ("descendants", "ancestors", "family", "relatives")