aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/shared/abstractions/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/shared/abstractions/__init__.py')
-rw-r--r--.venv/lib/python3.12/site-packages/shared/abstractions/__init__.py146
1 files changed, 146 insertions, 0 deletions
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",
+]