r2rpy/py/core/base/__init__.py

from .abstractions import *
from .agent import *
from .api.models import *
from .parsers import *
from .providers import *
from .utils import *

__all__ = [
    "ThinkingEvent",
    "ToolCallEvent",
    "ToolResultEvent",
    "CitationEvent",
    "Citation",
    ## ABSTRACTIONS
    # Base abstractions
    "AsyncSyncMeta",
    "syncable",
    # Completion abstractions
    "MessageType",
    # Document abstractions
    "Document",
    "DocumentChunk",
    "DocumentResponse",
    "IngestionStatus",
    "GraphExtractionStatus",
    "GraphConstructionStatus",
    "DocumentType",
    # Exception abstractions
    "R2RDocumentProcessingError",
    "R2RException",
    # Graph abstractions
    "Entity",
    "GraphExtraction",
    "Relationship",
    "Community",
    "GraphCreationSettings",
    "GraphEnrichmentSettings",
    # LLM abstractions
    "GenerationConfig",
    "LLMChatCompletion",
    "LLMChatCompletionChunk",
    "RAGCompletion",
    # Prompt abstractions
    "Prompt",
    # Search abstractions
    "AggregateSearchResult",
    "WebSearchResult",
    "GraphSearchResult",
    "GraphSearchSettings",
    "ChunkSearchSettings",
    "ChunkSearchResult",
    "WebPageSearchResult",
    "SearchSettings",
    "select_search_filters",
    "SearchMode",
    "HybridSearchSettings",
    # User abstractions
    "Token",
    "TokenData",
    # Vector abstractions
    "Vector",
    "VectorEntry",
    "VectorType",
    "StorageResult",
    "IndexConfig",
    ## AGENT
    # Agent abstractions
    "Agent",
    "AgentConfig",
    "Conversation",
    "Message",
    ## API
    # Auth Responses
    "TokenResponse",
    "User",
    ## PARSERS
    # Base parser
    "AsyncParser",
    ## PROVIDERS
    # Base provider classes
    "AppConfig",
    "Provider",
    "ProviderConfig",
    # Auth provider
    "AuthConfig",
    "AuthProvider",
    # Crypto provider
    "CryptoConfig",
    "CryptoProvider",
    # Database providers
    "LimitSettings",
    "DatabaseConfig",
    "DatabaseProvider",
    "Handler",
    "PostgresConfigurationSettings",
    # Email provider
    "EmailConfig",
    "EmailProvider",
    # Embedding provider
    "EmbeddingConfig",
    "EmbeddingProvider",
    # File provider
    "FileConfig",
    "FileProvider",
    # Ingestion provider
    "IngestionConfig",
    "IngestionProvider",
    "ChunkingStrategy",
    # LLM provider
    "CompletionConfig",
    "CompletionProvider",
    ## UTILS
    "RecursiveCharacterTextSplitter",
    "TextSplitter",
    "format_search_results_for_llm",
    "validate_uuid",
    # ID generation
    "generate_id",
    "generate_document_id",
    "generate_extraction_id",
    "generate_default_user_collection_id",
    "generate_user_id",
    "yield_sse_event",
    "dump_collector",
    "dump_obj",
]