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
|
import logging
# Keep '*' imports for enhanced development velocity
# corresponding flake8 error codes are F403, F405
from .base import *
from .integrations import *
from .main import *
from .parsers import *
from .pipelines import *
from .pipes import *
from .prompts import *
logger = logging.getLogger("r2r")
logger.setLevel(logging.INFO)
# Create a console handler and set the level to info
ch = logging.StreamHandler()
ch.setLevel(logging.INFO)
# Create a formatter and set it for the handler
formatter = logging.Formatter(
"%(asctime)s - %(levelname)s - %(name)s - %(message)s"
)
ch.setFormatter(formatter)
# Add the handler to the logger
logger.addHandler(ch)
# Optional: Prevent propagation to the root logger
logger.propagate = False
__all__ = [
"R2RException",
"LoggingConfig",
"LocalKVLoggingProvider",
"PostgresLoggingConfig",
"PostgresKVLoggingProvider",
"RedisLoggingConfig",
"RedisKVLoggingProvider",
"KVLoggingSingleton",
"VectorEntry",
"VectorType",
"Vector",
"VectorSearchRequest",
"VectorSearchResult",
"AsyncPipe",
"PipeType",
"AsyncState",
"Prompt",
"DataType",
"DocumentType",
"Document",
"Extraction",
"ExtractionType",
"Fragment",
"FragmentType",
"SearchPipe",
# Parsers
"AsyncParser",
"CSVParser",
"DOCXParser",
"HTMLParser",
"JSONParser",
"MDParser",
"PDFParser",
"PPTParser",
"TextParser",
"XLSXParser",
"AsyncPipeline",
# Providers
"EmbeddingConfig",
"EmbeddingProvider",
"EvalConfig",
"EvalProvider",
"LLMEvalProvider",
"PromptConfig",
"PromptProvider",
"GenerationConfig",
"LLMChatCompletion",
"LLMChatCompletionChunk",
"LLMConfig",
"LLMProvider",
"VectorDBConfig",
"VectorDBProvider",
"R2RConfig",
"TextSplitter",
"RecursiveCharacterTextSplitter",
"generate_run_id",
"generate_id_from_label",
"R2REngine",
# Pipes
"EmbeddingPipe",
"EvalPipe",
"ParsingPipe",
"QueryTransformPipe",
"SearchRAGPipe",
"StreamingSearchRAGPipe",
"VectorSearchPipe",
"VectorStoragePipe",
"R2RPromptProvider",
"WebSearchPipe",
"R2RBuilder",
"R2R",
"KGAgentSearchPipe",
# Prebuilts
"MultiSearchPipe",
"R2RPipeFactoryWithMultiSearch",
# Integrations
"SerperClient",
]
|
