aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/core/base/abstractions
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/core/base/abstractions')
-rw-r--r--.venv/lib/python3.12/site-packages/core/base/abstractions/__init__.py154
1 files changed, 154 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/core/base/abstractions/__init__.py b/.venv/lib/python3.12/site-packages/core/base/abstractions/__init__.py
new file mode 100644
index 00000000..bb1363fe
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/core/base/abstractions/__init__.py
@@ -0,0 +1,154 @@
+from shared.abstractions.base import AsyncSyncMeta, R2RSerializable, syncable
+from shared.abstractions.document import (
+ ChunkEnrichmentSettings,
+ Document,
+ DocumentChunk,
+ DocumentResponse,
+ DocumentType,
+ GraphConstructionStatus,
+ GraphExtractionStatus,
+ IngestionStatus,
+ RawChunk,
+ UnprocessedChunk,
+ UpdateChunk,
+)
+from shared.abstractions.embedding import (
+ EmbeddingPurpose,
+ default_embedding_prefixes,
+)
+from shared.abstractions.exception import (
+ R2RDocumentProcessingError,
+ R2RException,
+)
+from shared.abstractions.graph import (
+ Community,
+ Entity,
+ Graph,
+ GraphCommunitySettings,
+ GraphCreationSettings,
+ GraphEnrichmentSettings,
+ GraphExtraction,
+ Relationship,
+ StoreType,
+)
+from shared.abstractions.llm import (
+ GenerationConfig,
+ LLMChatCompletion,
+ LLMChatCompletionChunk,
+ Message,
+ MessageType,
+ RAGCompletion,
+)
+from shared.abstractions.prompt import Prompt
+from shared.abstractions.search import (
+ AggregateSearchResult,
+ ChunkSearchResult,
+ ChunkSearchSettings,
+ GraphCommunityResult,
+ GraphEntityResult,
+ GraphRelationshipResult,
+ GraphSearchResult,
+ GraphSearchResultType,
+ GraphSearchSettings,
+ HybridSearchSettings,
+ SearchMode,
+ SearchSettings,
+ WebPageSearchResult,
+ WebSearchResult,
+ select_search_filters,
+)
+from shared.abstractions.user import Token, TokenData, User
+from shared.abstractions.vector import (
+ IndexArgsHNSW,
+ IndexArgsIVFFlat,
+ IndexConfig,
+ IndexMeasure,
+ IndexMethod,
+ StorageResult,
+ Vector,
+ VectorEntry,
+ VectorQuantizationSettings,
+ VectorQuantizationType,
+ VectorTableName,
+ VectorType,
+)
+
+__all__ = [
+ # Base abstractions
+ "R2RSerializable",
+ "AsyncSyncMeta",
+ "syncable",
+ # Completion abstractions
+ "MessageType",
+ # Document abstractions
+ "Document",
+ "DocumentChunk",
+ "DocumentResponse",
+ "DocumentType",
+ "IngestionStatus",
+ "GraphExtractionStatus",
+ "GraphConstructionStatus",
+ "RawChunk",
+ "UnprocessedChunk",
+ "UpdateChunk",
+ # Embedding abstractions
+ "EmbeddingPurpose",
+ "default_embedding_prefixes",
+ # Exception abstractions
+ "R2RDocumentProcessingError",
+ "R2RException",
+ # Graph abstractions
+ "Entity",
+ "Graph",
+ "Community",
+ "StoreType",
+ "GraphExtraction",
+ "Relationship",
+ # Index abstractions
+ "IndexConfig",
+ # LLM abstractions
+ "GenerationConfig",
+ "LLMChatCompletion",
+ "LLMChatCompletionChunk",
+ "Message",
+ "RAGCompletion",
+ # Prompt abstractions
+ "Prompt",
+ # Search abstractions
+ "WebSearchResult",
+ "AggregateSearchResult",
+ "GraphSearchResult",
+ "GraphSearchResultType",
+ "GraphEntityResult",
+ "GraphRelationshipResult",
+ "GraphCommunityResult",
+ "GraphSearchSettings",
+ "ChunkSearchSettings",
+ "ChunkSearchResult",
+ "WebPageSearchResult",
+ "SearchSettings",
+ "select_search_filters",
+ "SearchMode",
+ "HybridSearchSettings",
+ # Graph abstractions
+ "GraphCreationSettings",
+ "GraphEnrichmentSettings",
+ "GraphCommunitySettings",
+ # User abstractions
+ "Token",
+ "TokenData",
+ "User",
+ # Vector abstractions
+ "Vector",
+ "VectorEntry",
+ "VectorType",
+ "IndexMeasure",
+ "IndexMethod",
+ "VectorTableName",
+ "IndexArgsHNSW",
+ "IndexArgsIVFFlat",
+ "VectorQuantizationSettings",
+ "VectorQuantizationType",
+ "StorageResult",
+ "ChunkEnrichmentSettings",
+]