1 files changed, 110 insertions, 0 deletions
diff --git a/R2R/r2r/__init__.py b/R2R/r2r/__init__.py
new file mode 100755
index 00000000..492cc13a
--- /dev/null
+++ b/R2R/r2r/__init__.py
@@ -0,0 +1,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",
+]