refactor: improve imports

Author: scream4ik

examples/langgraph_checkpoint_demo.py

@@ -9,9 +9,8 @@
 from langgraph.graph import END, START, MessagesState, StateGraph
 from langgraph.prebuilt import ToolNode
 
-from memstate.backends.inmemory import InMemoryStorage
+from memstate import InMemoryStorage, MemoryStore
 from memstate.integrations.langgraph import MemStateCheckpointer
-from memstate.storage import MemoryStore
 
 storage = InMemoryStorage()
 memory = MemoryStore(storage)

examples/main_demo.py

@@ -5,8 +5,7 @@
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
-from memstate.backends.inmemory import InMemoryStorage
-from memstate.storage import Constraint, Fact, MemoryStore
+from memstate import Constraint, Fact, InMemoryStorage, MemoryStore
 
 # --- 1. Defining Schemes ---
 

examples/pizza_agent_demo.py

@@ -12,9 +12,8 @@
 from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode
 
-from memstate.backends.inmemory import InMemoryStorage
+from memstate import InMemoryStorage, MemoryStore
 from memstate.integrations.langgraph import MemStateCheckpointer
-from memstate.storage import MemoryStore
 
 # --- 1. Business Logic (Tools) ---
 # Note: These functions don't know anything about LLM. They simply modify data.

examples/rag_hook_demo.py

@@ -4,8 +4,7 @@
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
-from memstate.backends.inmemory import InMemoryStorage
-from memstate.storage import Fact, MemoryStore
+from memstate import Fact, InMemoryStorage, MemoryStore
 
 # --- Simulated Vector Store ---
 # In real life, this would use OpenAI Embeddings and Qdrant/Chroma

memstate/__init__.py

@@ -0,0 +1,27 @@
+import importlib.metadata
+
+try:
+    __version__ = importlib.metadata.version("memstate")
+except importlib.metadata.PackageNotFoundError:
+    __version__ = "unknown"
+
+from memstate.backends.inmemory import InMemoryStorage
+from memstate.backends.sqlite import SQLiteStorage
+from memstate.constants import Operation
+from memstate.exceptions import ConflictError, HookError, MemoryStoreError, ValidationFailed
+from memstate.schemas import Fact, TxEntry
+from memstate.storage import Constraint, MemoryStore
+
+__all__ = [
+    "MemoryStore",
+    "Constraint",
+    "Fact",
+    "TxEntry",
+    "Operation",
+    "InMemoryStorage",
+    "SQLiteStorage",
+    "MemoryStoreError",
+    "ValidationFailed",
+    "ConflictError",
+    "HookError",
+]

memstate/backends/inmemory.py

@@ -1,7 +1,7 @@
 import threading
 from typing import Any
 
-from .base import StorageBackend
+from memstate.backends.base import StorageBackend
 
 
 class InMemoryStorage(StorageBackend):

memstate/backends/redis.py

@@ -1,7 +1,7 @@
 import json
 from typing import Any, Union
 
-from .base import StorageBackend
+from memstate.backends.base import StorageBackend
 
 try:
     import redis

memstate/backends/sqlite.py

@@ -3,7 +3,7 @@
 import threading
 from typing import Any
 
-from .base import StorageBackend
+from memstate.backends.base import StorageBackend
 
 
 class SQLiteStorage(StorageBackend):

memstate/integrations/chroma.py

@@ -1,7 +1,7 @@
 from typing import Any, Callable
 
-from ..constants import Operation
-from ..schemas import Fact
+from memstate.constants import Operation
+from memstate.schemas import Fact
 
 try:
     from chromadb import EmbeddingFunction

memstate/integrations/langgraph.py

@@ -13,7 +13,8 @@
 except ImportError:
     raise ImportError("pip install langgraph")
 
-from memstate.storage import Fact, MemoryStore
+from memstate.schemas import Fact
+from memstate.storage import MemoryStore
 
 
 class MemStateCheckpointer(BaseCheckpointSaver[str]):