alma-memory 0.4.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

alma/graph/__init__.py

@@ -0,0 +1,106 @@
+"""
+ALMA Graph Memory Module.
+
+Graph-based memory for capturing relationships between entities.
+"""
+
+from alma.graph.base import GraphBackend
+from alma.graph.extraction import (
+    EntityExtractor,
+    ExtractionConfig,
+)
+from alma.graph.store import (
+    BackendGraphStore,
+    Entity,
+    GraphQuery,
+    GraphResult,
+    GraphStore,
+    InMemoryGraphStore,
+    Neo4jGraphStore,
+    Relationship,
+    create_graph_store,
+)
+
+__all__ = [
+    # Backend abstract base class
+    "GraphBackend",
+    # Store classes (high-level API)
+    "GraphStore",
+    "Neo4jGraphStore",
+    "InMemoryGraphStore",
+    "BackendGraphStore",
+    # Data classes
+    "Entity",
+    "Relationship",
+    "GraphQuery",
+    "GraphResult",
+    # Extraction
+    "EntityExtractor",
+    "ExtractionConfig",
+    # Factory functions
+    "create_graph_store",
+    "create_graph_backend",
+]
+
+
+def create_graph_backend(backend: str = "neo4j", **config) -> GraphBackend:
+    """
+    Factory function to create a graph backend.
+
+    Args:
+        backend: Backend type ("neo4j", "memgraph", "kuzu", or "memory")
+        **config: Backend-specific configuration options
+
+    Returns:
+        Configured GraphBackend instance
+
+    Raises:
+        ValueError: If an unknown backend type is specified
+
+    Example:
+        # Create Neo4j backend
+        backend = create_graph_backend(
+            backend="neo4j",
+            uri="bolt://localhost:7687",
+            username="neo4j",
+            password="password"
+        )
+
+        # Create Memgraph backend
+        backend = create_graph_backend(
+            backend="memgraph",
+            uri="bolt://localhost:7687",
+            username="",
+            password=""
+        )
+
+        # Create Kuzu embedded backend (persistent)
+        backend = create_graph_backend(
+            backend="kuzu",
+            database_path="./my_graph_db"
+        )
+
+        # Create Kuzu embedded backend (in-memory)
+        backend = create_graph_backend(backend="kuzu")
+
+        # Create in-memory backend for testing
+        backend = create_graph_backend(backend="memory")
+    """
+    if backend == "neo4j":
+        from alma.graph.backends.neo4j import Neo4jBackend
+
+        return Neo4jBackend(**config)
+    elif backend == "memgraph":
+        from alma.graph.backends.memgraph import MemgraphBackend
+
+        return MemgraphBackend(**config)
+    elif backend == "kuzu":
+        from alma.graph.backends.kuzu import KuzuBackend
+
+        return KuzuBackend(**config)
+    elif backend == "memory":
+        from alma.graph.backends.memory import InMemoryBackend
+
+        return InMemoryBackend()
+    else:
+        raise ValueError(f"Unknown graph backend: {backend}")

alma/graph/backends/__init__.py

@@ -0,0 +1,32 @@
+"""
+ALMA Graph Memory Backends.
+
+This package contains implementations of the GraphBackend interface
+for various graph database systems.
+
+Available backends:
+- neo4j: Neo4j graph database backend
+- memgraph: Memgraph graph database backend (Neo4j Bolt protocol compatible)
+- kuzu: Kuzu embedded graph database backend (no server required)
+- memory: In-memory backend for testing and development
+"""
+
+from alma.graph.backends.memgraph import MemgraphBackend
+from alma.graph.backends.memory import InMemoryBackend
+from alma.graph.backends.neo4j import Neo4jBackend
+
+# Kuzu is an optional dependency
+try:
+    from alma.graph.backends.kuzu import KuzuBackend
+
+    _KUZU_AVAILABLE = True
+except ImportError:
+    KuzuBackend = None  # type: ignore
+    _KUZU_AVAILABLE = False
+
+__all__ = [
+    "Neo4jBackend",
+    "MemgraphBackend",
+    "KuzuBackend",
+    "InMemoryBackend",
+]