memorisdk 2.0.1__py3-none-any.whl → 2.1.0__py3-none-any.whl

memori/core/database.py

@@ -7,7 +7,7 @@ import sqlite3
 import uuid
 from datetime import datetime, timedelta
 from pathlib import Path
-from typing import Any, Dict, List, Optional
+from typing import Any
 
 from loguru import logger
 
@@ -24,6 +24,9 @@ from ..utils.transaction_manager import TransactionManager, TransactionOperation
 class DatabaseManager:
     """Manages Pydantic-based memory storage with streamlined schema and FTS search"""
 
+    # Database type identifier for database-agnostic code
+    database_type = "sql"
+
     def __init__(self, database_connect: str, template: str = "basic"):
         self.database_connect = database_connect
         self.template = template
@@ -229,7 +232,7 @@ class DatabaseManager:
         session_id: str,
         namespace: str = "default",
         tokens_used: int = 0,
-        metadata: Optional[Dict[str, Any]] = None,
+        metadata: dict[str, Any] | None = None,
     ):
         """Store chat history with input validation"""
         try:
@@ -277,9 +280,9 @@ class DatabaseManager:
     def get_chat_history(
         self,
         namespace: str = "default",
-        session_id: Optional[str] = None,
+        session_id: str | None = None,
         limit: int = 10,
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Get chat history with optional session filtering"""
         try:
             # Validate inputs
@@ -507,9 +510,9 @@ class DatabaseManager:
         self,
         query: str,
         namespace: str = "default",
-        category_filter: Optional[List[str]] = None,
+        category_filter: list[str] | None = None,
         limit: int = 10,
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Advanced memory search with hybrid approach: FTS + Entity + Category filtering"""
         try:
             # Validate and sanitize all input parameters
@@ -591,7 +594,7 @@ class DatabaseManager:
         cursor,
         query: str,
         namespace: str,
-        category_filter: Optional[List[str]],
+        category_filter: list[str] | None,
         limit: int,
     ):
         """Execute FTS5 search with proper parameterization"""
@@ -709,7 +712,7 @@ class DatabaseManager:
             return []
 
     def _execute_category_search(
-        self, cursor, query: str, namespace: str, category_filter: List[str], limit: int
+        self, cursor, query: str, namespace: str, category_filter: list[str], limit: int
     ):
         """Execute category-based search with proper input validation"""
         try:
@@ -759,7 +762,7 @@ class DatabaseManager:
         cursor,
         query: str,
         namespace: str,
-        category_filter: Optional[List[str]],
+        category_filter: list[str] | None,
         limit: int,
     ):
         """Execute fallback LIKE search with proper input validation"""
@@ -884,7 +887,7 @@ class DatabaseManager:
         else:
             return "short_term_memory"
 
-    def get_memory_stats(self, namespace: str = "default") -> Dict[str, Any]:
+    def get_memory_stats(self, namespace: str = "default") -> dict[str, Any]:
         """Get comprehensive memory statistics"""
         with self._get_connection() as conn:
             cursor = conn.cursor()
@@ -948,9 +951,7 @@ class DatabaseManager:
 
             return stats
 
-    def clear_memory(
-        self, namespace: str = "default", memory_type: Optional[str] = None
-    ):
+    def clear_memory(self, namespace: str = "default", memory_type: str | None = None):
         """Clear memory data with entity cleanup"""
         with self._get_connection() as conn:
             cursor = conn.cursor()