alma-memory 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

alma/graph/__init__.py

@@ -1,81 +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" 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 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 == "memory":
-        from alma.graph.backends.memory import InMemoryBackend
-
-        return InMemoryBackend()
-    else:
-        raise ValueError(f"Unknown graph backend: {backend}")
+"""
+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

@@ -1,18 +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
-- memory: In-memory backend for testing and development
-"""
-
-from alma.graph.backends.memory import InMemoryBackend
-from alma.graph.backends.neo4j import Neo4jBackend
-
-__all__ = [
-    "Neo4jBackend",
-    "InMemoryBackend",
-]
+"""
+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",
+]