From 4a52a71956a8d46fcb7294ac71734504bb09bcc2 Mon Sep 17 00:00:00 2001 From: S. Solomon Darnell Date: Fri, 28 Mar 2025 21:52:21 -0500 Subject: two version of R2R are here --- .../site-packages/shared/abstractions/__init__.py | 146 +++++++++++++++++++++ 1 file changed, 146 insertions(+) create mode 100644 .venv/lib/python3.12/site-packages/shared/abstractions/__init__.py (limited to '.venv/lib/python3.12/site-packages/shared/abstractions/__init__.py') diff --git a/.venv/lib/python3.12/site-packages/shared/abstractions/__init__.py b/.venv/lib/python3.12/site-packages/shared/abstractions/__init__.py new file mode 100644 index 00000000..da33ddd7 --- /dev/null +++ b/.venv/lib/python3.12/site-packages/shared/abstractions/__init__.py @@ -0,0 +1,146 @@ +from .base import AsyncSyncMeta, R2RSerializable, syncable +from .document import ( + Document, + DocumentChunk, + DocumentResponse, + DocumentType, + GraphConstructionStatus, + GraphExtractionStatus, + IngestionMode, + IngestionStatus, + RawChunk, + UnprocessedChunk, +) +from .embedding import EmbeddingPurpose, default_embedding_prefixes +from .exception import ( + PDFParsingError, + PopplerNotFoundError, + R2RDocumentProcessingError, + R2RException, +) +from .graph import ( + Community, + Entity, + GraphCommunitySettings, + GraphCreationSettings, + GraphEnrichmentSettings, + GraphExtraction, + Relationship, + StoreType, +) +from .llm import ( + GenerationConfig, + LLMChatCompletion, + LLMChatCompletionChunk, + Message, + MessageType, + RAGCompletion, +) +from .prompt import Prompt +from .search import ( + AggregateSearchResult, + ChunkSearchResult, + ChunkSearchSettings, + GraphCommunityResult, + GraphEntityResult, + GraphRelationshipResult, + GraphSearchResult, + GraphSearchResultType, + GraphSearchSettings, + HybridSearchSettings, + SearchMode, + SearchSettings, + WebPageSearchResult, + select_search_filters, +) +from .user import Token, TokenData, User +from .vector import ( + IndexArgsHNSW, + IndexArgsIVFFlat, + IndexMeasure, + IndexMethod, + StorageResult, + Vector, + VectorEntry, + VectorQuantizationType, + VectorTableName, + VectorType, +) + +__all__ = [ + # Base abstractions + "R2RSerializable", + "AsyncSyncMeta", + "syncable", + # Completion abstractions + "MessageType", + # Document abstractions + "Document", + "DocumentChunk", + "DocumentResponse", + "IngestionMode", + "IngestionStatus", + "GraphExtractionStatus", + "GraphConstructionStatus", + "DocumentType", + "RawChunk", + "UnprocessedChunk", + # Embedding abstractions + "EmbeddingPurpose", + "default_embedding_prefixes", + # Exception abstractions + "R2RDocumentProcessingError", + "R2RException", + "PDFParsingError", + "PopplerNotFoundError", + # Graph abstractions + "Entity", + "Community", + "Community", + "GraphExtraction", + "Relationship", + "StoreType", + # LLM abstractions + "GenerationConfig", + "LLMChatCompletion", + "LLMChatCompletionChunk", + "Message", + "RAGCompletion", + # Prompt abstractions + "Prompt", + # Search abstractions + "AggregateSearchResult", + "GraphSearchResult", + "WebPageSearchResult", + "GraphSearchResultType", + "GraphEntityResult", + "GraphRelationshipResult", + "GraphCommunityResult", + "GraphSearchSettings", + "ChunkSearchSettings", + "ChunkSearchResult", + "SearchSettings", + "select_search_filters", + "HybridSearchSettings", + "SearchMode", + # graph abstractions + "GraphCreationSettings", + "GraphEnrichmentSettings", + "GraphExtraction", + "GraphCommunitySettings", + # User abstractions + "Token", + "TokenData", + "User", + # Vector abstractions + "Vector", + "VectorEntry", + "VectorType", + "IndexMethod", + "IndexMeasure", + "IndexArgsIVFFlat", + "IndexArgsHNSW", + "VectorTableName", + "VectorQuantizationType", + "StorageResult", +] -- cgit v1.2.3