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

alma/session/manager.py

@@ -4,14 +4,14 @@ Session Manager.
 Manages session continuity, handoffs, and quick context reload.
 """
 
-from datetime import datetime, timezone
-from typing import Optional, List, Dict, Any, Callable
-import uuid
 import logging
+import uuid
+from datetime import datetime, timezone
+from typing import Any, Callable, Dict, List, Optional
 
 from alma.session.types import (
-    SessionHandoff,
     SessionContext,
+    SessionHandoff,
     SessionOutcome,
 )
 
@@ -281,7 +281,7 @@ class SessionManager:
 
         # Trim to max
         if len(self._handoffs[agent]) > self.max_handoffs:
-            self._handoffs[agent] = self._handoffs[agent][-self.max_handoffs:]
+            self._handoffs[agent] = self._handoffs[agent][-self.max_handoffs :]
 
         # TODO: Persist to storage backend when integrated
 

alma/session/types.py

@@ -4,11 +4,10 @@ Session Management Types.
 Data models for session continuity and handoffs.
 """
 
+import uuid
 from dataclasses import dataclass, field
 from datetime import datetime, timezone
-from typing import Optional, List, Dict, Any, Literal
-import uuid
-
+from typing import Any, Dict, List, Literal, Optional
 
 SessionOutcome = Literal["success", "failure", "interrupted", "unknown"]
 
@@ -276,7 +275,9 @@ class SessionContext:
             if "branch" in self.codebase_state:
                 lines.append(f"Branch: {self.codebase_state['branch']}")
             if "uncommitted" in self.codebase_state:
-                lines.append(f"Uncommitted Changes: {self.codebase_state['uncommitted']}")
+                lines.append(
+                    f"Uncommitted Changes: {self.codebase_state['uncommitted']}"
+                )
 
         # Rules
         if self.rules_of_engagement:

alma/storage/__init__.py

@@ -1,20 +1,90 @@
 """ALMA Storage Backends."""
 
 from alma.storage.base import StorageBackend
+from alma.storage.constants import (
+    AZURE_COSMOS_CONTAINER_NAMES,
+    POSTGRESQL_TABLE_NAMES,
+    SQLITE_TABLE_NAMES,
+    MemoryType,
+    get_table_name,
+    get_table_names,
+)
 from alma.storage.file_based import FileBasedStorage
+from alma.storage.migrations import (
+    Migration,
+    MigrationError,
+    MigrationRegistry,
+    MigrationRunner,
+    SchemaVersion,
+)
 from alma.storage.sqlite_local import SQLiteStorage
 
 # Azure Cosmos DB is optional - requires azure-cosmos package
 try:
     from alma.storage.azure_cosmos import AzureCosmosStorage
+
     _HAS_AZURE = True
 except ImportError:
     AzureCosmosStorage = None  # type: ignore
     _HAS_AZURE = False
 
+# PostgreSQL is optional - requires psycopg package
+try:
+    from alma.storage.postgresql import PostgreSQLStorage
+
+    _HAS_POSTGRES = True
+except ImportError:
+    PostgreSQLStorage = None  # type: ignore
+    _HAS_POSTGRES = False
+
+# Qdrant is optional - requires qdrant-client package
+try:
+    from alma.storage.qdrant import QdrantStorage
+
+    _HAS_QDRANT = True
+except ImportError:
+    QdrantStorage = None  # type: ignore
+    _HAS_QDRANT = False
+
+# ChromaDB is optional - requires chromadb package
+try:
+    from alma.storage.chroma import ChromaStorage
+
+    _HAS_CHROMA = True
+except ImportError:
+    ChromaStorage = None  # type: ignore
+    _HAS_CHROMA = False
+
+# Pinecone is optional - requires pinecone-client package
+try:
+    from alma.storage.pinecone import PineconeStorage
+
+    _HAS_PINECONE = True
+except ImportError:
+    PineconeStorage = None  # type: ignore
+    _HAS_PINECONE = False
+
 __all__ = [
+    # Storage backends
     "StorageBackend",
     "FileBasedStorage",
     "SQLiteStorage",
     "AzureCosmosStorage",
+    "PostgreSQLStorage",
+    "QdrantStorage",
+    "ChromaStorage",
+    "PineconeStorage",
+    # Migration framework
+    "Migration",
+    "MigrationError",
+    "MigrationRegistry",
+    "MigrationRunner",
+    "SchemaVersion",
+    # Constants for consistent naming
+    "MemoryType",
+    "get_table_name",
+    "get_table_names",
+    "POSTGRESQL_TABLE_NAMES",
+    "SQLITE_TABLE_NAMES",
+    "AZURE_COSMOS_CONTAINER_NAMES",
 ]