GraphGen/graphgen/models/__init__.py at bf68f13e38d8f5f95e731ad035e27cf0df7b9978 · InternScience/GraphGen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from .evaluator import (
        AccuracyEvaluator,
        LengthEvaluator,
        MTLDEvaluator,
        RewardEvaluator,
        StructureEvaluator,
        UniEvaluator,
    )
    from .filter import RangeFilter
    from .generator import (
        AggregatedGenerator,
        AtomicGenerator,
        CoTGenerator,
        FillInBlankGenerator,
        MultiAnswerGenerator,
        MultiChoiceGenerator,
        MultiHopGenerator,
        QuizGenerator,
        TrueFalseGenerator,
        VQAGenerator,
    )
    from .kg_builder import LightRAGKGBuilder, MMKGBuilder
    from .llm import HTTPClient, OllamaClient, OpenAIClient
    from .partitioner import (
        AnchorBFSPartitioner,
        BFSPartitioner,
        DFSPartitioner,
        ECEPartitioner,
        LeidenPartitioner,
    )
    from .reader import (
        CSVReader,
        HuggingFaceReader,
        JSONReader,
        ParquetReader,
        PDFReader,
        PickleReader,
        RDFReader,
        TXTReader,
    )
    from .rephraser import StyleControlledRephraser
    from .searcher.db.interpro_searcher import InterProSearch
    from .searcher.db.ncbi_searcher import NCBISearch
    from .searcher.db.rnacentral_searcher import RNACentralSearch
    from .searcher.db.uniprot_searcher import UniProtSearch
    from .searcher.kg.wiki_search import WikiSearch
    from .searcher.web.bing_search import BingSearch
    from .searcher.web.google_search import GoogleSearch
    from .splitter import ChineseRecursiveTextSplitter, RecursiveCharacterSplitter
    from .tokenizer import Tokenizer

_import_map = {
    # Evaluator
    "AccuracyEvaluator": ".evaluator",
    "LengthEvaluator": ".evaluator",
    "MTLDEvaluator": ".evaluator",
    "RewardEvaluator": ".evaluator",
    "StructureEvaluator": ".evaluator",
    "UniEvaluator": ".evaluator",
    # Filter
    "RangeFilter": ".filter",
    # Generator
    "AggregatedGenerator": ".generator",
    "AtomicGenerator": ".generator",
    "CoTGenerator": ".generator",
    "FillInBlankGenerator": ".generator",
    "MultiAnswerGenerator": ".generator",
    "MultiChoiceGenerator": ".generator",
    "MultiHopGenerator": ".generator",
    "QuizGenerator": ".generator",
    "TrueFalseGenerator": ".generator",
    "VQAGenerator": ".generator",
    # KG Builder
    "LightRAGKGBuilder": ".kg_builder",
    "MMKGBuilder": ".kg_builder",
    # LLM
    "HTTPClient": ".llm",
    "OllamaClient": ".llm",
    "OpenAIClient": ".llm",
    # Partitioner
    "AnchorBFSPartitioner": ".partitioner",
    "BFSPartitioner": ".partitioner",
    "DFSPartitioner": ".partitioner",
    "ECEPartitioner": ".partitioner",
    "LeidenPartitioner": ".partitioner",
    # Reader
    "CSVReader": ".reader",
    "JSONReader": ".reader",
    "ParquetReader": ".reader",
    "PDFReader": ".reader",
    "PickleReader": ".reader",
    "RDFReader": ".reader",
    "TXTReader": ".reader",
    "HuggingFaceReader": ".reader",
    # Searcher
    "InterProSearch": ".searcher.db.interpro_searcher",
    "NCBISearch": ".searcher.db.ncbi_searcher",
    "RNACentralSearch": ".searcher.db.rnacentral_searcher",
    "UniProtSearch": ".searcher.db.uniprot_searcher",
    "WikiSearch": ".searcher.kg.wiki_search",
    "BingSearch": ".searcher.web.bing_search",
    "GoogleSearch": ".searcher.web.google_search",
    # Splitter
    "ChineseRecursiveTextSplitter": ".splitter",
    "RecursiveCharacterSplitter": ".splitter",
    # Tokenizer
    "Tokenizer": ".tokenizer",
    # Rephraser
    "StyleControlledRephraser": ".rephraser",
}


def __getattr__(name):
    if name in _import_map:
        import importlib

        module = importlib.import_module(_import_map[name], package=__name__)
        return getattr(module, name)
    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")


__all__ = list(_import_map.keys())