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

memori/database/queries/base_queries.py

@@ -3,24 +3,23 @@ Base database queries and schema operations
 """
 
 from abc import ABC, abstractmethod
-from typing import Dict
 
 
 class BaseQueries(ABC):
     """Abstract base class for database queries"""
 
     @abstractmethod
-    def get_table_creation_queries(self) -> Dict[str, str]:
+    def get_table_creation_queries(self) -> dict[str, str]:
         """Return dictionary of table creation SQL statements"""
         pass
 
     @abstractmethod
-    def get_index_creation_queries(self) -> Dict[str, str]:
+    def get_index_creation_queries(self) -> dict[str, str]:
         """Return dictionary of index creation SQL statements"""
         pass
 
     @abstractmethod
-    def get_trigger_creation_queries(self) -> Dict[str, str]:
+    def get_trigger_creation_queries(self) -> dict[str, str]:
         """Return dictionary of trigger creation SQL statements"""
         pass
 

memori/database/queries/chat_queries.py

@@ -2,27 +2,25 @@
 Chat history database queries
 """
 
-from typing import Dict
-
 from .base_queries import BaseQueries
 
 
 class ChatQueries(BaseQueries):
     """Centralized chat history SQL queries"""
 
-    def get_table_creation_queries(self) -> Dict[str, str]:
+    def get_table_creation_queries(self) -> dict[str, str]:
         """Chat table creation queries"""
         from .base_queries import SchemaQueries
 
         return {"chat_history": SchemaQueries.TABLE_CREATION["chat_history"]}
 
-    def get_index_creation_queries(self) -> Dict[str, str]:
+    def get_index_creation_queries(self) -> dict[str, str]:
         """Chat index creation queries"""
         from .base_queries import SchemaQueries
 
         return {k: v for k, v in SchemaQueries.INDEX_CREATION.items() if "chat" in k}
 
-    def get_trigger_creation_queries(self) -> Dict[str, str]:
+    def get_trigger_creation_queries(self) -> dict[str, str]:
         """Chat trigger creation queries"""
         return {}  # No triggers for chat history currently
 

memori/database/queries/entity_queries.py

@@ -2,15 +2,13 @@
 Entity and relationship database queries
 """
 
-from typing import Dict
-
 from .base_queries import BaseQueries
 
 
 class EntityQueries(BaseQueries):
     """Centralized entity and relationship SQL queries"""
 
-    def get_table_creation_queries(self) -> Dict[str, str]:
+    def get_table_creation_queries(self) -> dict[str, str]:
         """Entity table creation queries"""
         from .base_queries import SchemaQueries
 
@@ -21,7 +19,7 @@ class EntityQueries(BaseQueries):
             ],
         }
 
-    def get_index_creation_queries(self) -> Dict[str, str]:
+    def get_index_creation_queries(self) -> dict[str, str]:
         """Entity index creation queries"""
         from .base_queries import SchemaQueries
 
@@ -31,7 +29,7 @@ class EntityQueries(BaseQueries):
             if any(word in k for word in ["entities", "relationships"])
         }
 
-    def get_trigger_creation_queries(self) -> Dict[str, str]:
+    def get_trigger_creation_queries(self) -> dict[str, str]:
         """Entity trigger creation queries"""
         return {}  # No triggers for entities currently
 

memori/database/queries/memory_queries.py

@@ -2,15 +2,13 @@
 Memory-related database queries
 """
 
-from typing import Dict
-
 from .base_queries import BaseQueries
 
 
 class MemoryQueries(BaseQueries):
     """Centralized memory-related SQL queries"""
 
-    def get_table_creation_queries(self) -> Dict[str, str]:
+    def get_table_creation_queries(self) -> dict[str, str]:
         """Memory table creation queries"""
         from .base_queries import SchemaQueries
 
@@ -20,7 +18,7 @@ class MemoryQueries(BaseQueries):
             "rules_memory": SchemaQueries.TABLE_CREATION["rules_memory"],
         }
 
-    def get_index_creation_queries(self) -> Dict[str, str]:
+    def get_index_creation_queries(self) -> dict[str, str]:
         """Memory index creation queries"""
         from .base_queries import SchemaQueries
 
@@ -30,7 +28,7 @@ class MemoryQueries(BaseQueries):
             if any(table in k for table in ["short_term", "long_term", "rules"])
         }
 
-    def get_trigger_creation_queries(self) -> Dict[str, str]:
+    def get_trigger_creation_queries(self) -> dict[str, str]:
         """Memory trigger creation queries"""
         from .base_queries import SchemaQueries
 

memori/database/query_translator.py

@@ -6,7 +6,7 @@ differences between SQLite, PostgreSQL, and MySQL, particularly for
 boolean values and other database-specific syntax.
 """
 
-from typing import Any, Dict, Union
+from typing import Any
 
 from loguru import logger
 
@@ -32,7 +32,7 @@ class QueryParameterTranslator:
         self.database_type = database_type.lower()
         logger.debug(f"QueryParameterTranslator initialized for {self.database_type}")
 
-    def translate_parameters(self, parameters: Dict[str, Any]) -> Dict[str, Any]:
+    def translate_parameters(self, parameters: dict[str, Any]) -> dict[str, Any]:
         """
         Translate parameters to be compatible with the target database.
 
@@ -70,7 +70,7 @@ class QueryParameterTranslator:
             return None
 
         # Handle lists/arrays
-        if isinstance(value, (list, tuple)):
+        if isinstance(value, list | tuple):
             return [self._translate_value(item) for item in value]
 
         # Handle dictionaries
@@ -80,7 +80,7 @@ class QueryParameterTranslator:
         # Return other values unchanged
         return value
 
-    def _translate_boolean(self, value: bool) -> Union[bool, int]:
+    def _translate_boolean(self, value: bool) -> bool | int:
         """
         Translate boolean values for database compatibility.
 
@@ -159,8 +159,8 @@ class QueryParameterTranslator:
         return True
 
     def translate_query_with_parameters(
-        self, query: str, parameters: Dict[str, Any]
-    ) -> tuple[str, Dict[str, Any]]:
+        self, query: str, parameters: dict[str, Any]
+    ) -> tuple[str, dict[str, Any]]:
         """
         Translate both query and parameters for database compatibility.
 
@@ -180,17 +180,17 @@ class QueryParameterTranslator:
 
         return query, translated_params
 
-    def get_boolean_true(self) -> Union[bool, int]:
+    def get_boolean_true(self) -> bool | int:
         """Get database-appropriate TRUE value."""
         return self._translate_boolean(True)
 
-    def get_boolean_false(self) -> Union[bool, int]:
+    def get_boolean_false(self) -> bool | int:
         """Get database-appropriate FALSE value."""
         return self._translate_boolean(False)
 
 
 # Convenience functions for common boolean translations
-def get_db_boolean(value: bool, database_type: str) -> Union[bool, int]:
+def get_db_boolean(value: bool, database_type: str) -> bool | int:
     """
     Get database-appropriate boolean value.
 
@@ -206,8 +206,8 @@ def get_db_boolean(value: bool, database_type: str) -> Union[bool, int]:
 
 
 def translate_query_params(
-    parameters: Dict[str, Any], database_type: str
-) -> Dict[str, Any]:
+    parameters: dict[str, Any], database_type: str
+) -> dict[str, Any]:
     """
     Convenience function to translate query parameters.
 

memori/database/schema_generators/__init__.py

@@ -4,4 +4,15 @@ Schema generators for different database backends
 
 from .mysql_schema_generator import MySQLSchemaGenerator
 
+try:
+    from .mongodb_schema_generator import MongoDBSchemaGenerator
+
+    MONGODB_SCHEMA_AVAILABLE = True
+except ImportError:
+    MongoDBSchemaGenerator = None  # type: ignore
+    MONGODB_SCHEMA_AVAILABLE = False
+
 __all__ = ["MySQLSchemaGenerator"]
+
+if MONGODB_SCHEMA_AVAILABLE:
+    __all__.append("MongoDBSchemaGenerator")