mdb-engine 0.7.0__py3-none-any.whl → 0.7.1__py3-none-any.whl

mdb_engine/__init__.py

@@ -82,19 +82,13 @@ from .repositories import Entity, MongoRepository, Repository, UnitOfWork
 from .utils import clean_mongo_doc, clean_mongo_docs
 
 __version__ = (
-    "0.6.0"  # Secure-by-default WebSocket authentication with encrypted session keys
-    # - NEW: WebSocket session key generation and management
-    # - NEW: Envelope encryption for WebSocket session keys
-    # - NEW: Secure-by-default CSRF protection (csrf_required: true)
-    # - NEW: WebSocketSessionManager with private collection storage
-    # - NEW: Session key endpoint (/auth/websocket-session)
-    # - NEW: Session key integration in login flow
-    # - ENHANCED: WebSocket authentication with session key support
-    # - ENHANCED: CSRF middleware session key validation
-    # - ENHANCED: Multi-app WebSocket routing with session keys
-    # - BACKWARD COMPATIBLE: Cookie-based authentication fallback
-    # - UPDATED: All documentation for secure-by-default approach
-    # - COMPREHENSIVE: Unit and integration tests for session keys
+    "0.7.1"  # Memory service initialization fix for multi-app setups
+    # - FIXED: Memory service initialization in create_multi_app context
+    # - FIXED: get_memory_service() now returns service when memory_config.enabled: true
+    # - ENHANCED: Explicit memory service initialization in create_multi_app lifespan
+    # - ENHANCED: Better error handling and logging for memory service initialization
+    # - ADDED: Comprehensive integration tests for memory service in multi-app context
+    # - ADDED: Unit tests for memory service initialization edge cases
 )
 
 __all__ = [

mdb_engine/cli/main.py

@@ -15,7 +15,7 @@ from .commands.validate import validate
 
 
 @click.group()
-@click.version_option(version="0.1.6", prog_name="mdb")
+@click.version_option(version="0.7.1", prog_name="mdb")
 def cli() -> None:
     """
     MDB_ENGINE CLI - Manifest management tool.

mdb_engine/core/engine.py

@@ -36,6 +36,11 @@ from typing import TYPE_CHECKING, Any, Optional
 from motor.motor_asyncio import AsyncIOMotorClient
 from pymongo.errors import PyMongoError
 
+try:
+    from openai import OpenAIError
+except ImportError:
+    OpenAIError = RuntimeError
+
 if TYPE_CHECKING:
     from fastapi import FastAPI
 
@@ -2672,7 +2677,7 @@ class MongoDBEngine:
                 )
 
         @asynccontextmanager
-        async def lifespan(app: FastAPI):
+        async def lifespan(app: FastAPI):  # noqa: C901
             """Lifespan context manager for parent app."""
             nonlocal mounted_apps, shared_user_pool_initialized
 
@@ -2911,11 +2916,60 @@ class MongoDBEngine:
                     child_app.add_middleware(middleware_class)
                     logger.debug(f"Added AppContextMiddleware to child app '{slug}'")
 
-                    # CRITICAL FIX: Register WebSocket routes on parent app BEFORE mounting
-                    # This ensures WebSocket routes are checked before mounted app routes
-                    # Mounted apps create catch-all routes that intercept /app-slug/* paths
                     await _register_websocket_routes(app, app_manifest_data, slug, path_prefix)
 
+                    memory_config = app_manifest_data.get("memory_config")
+                    if memory_config and memory_config.get("enabled", False):
+                        if engine._service_initializer:  # noqa: SLF001
+                            try:
+                                await engine._service_initializer.initialize_memory_service(  # noqa: SLF001
+                                    slug, memory_config
+                                )
+                                logger.info(
+                                    f"Memory service initialized for mounted app '{slug}' "
+                                    f"in multi-app context"
+                                )
+                            except OpenAIError as e:
+                                logger.warning(
+                                    f"Memory service initialization skipped for mounted app "
+                                    f"'{slug}': OpenAI API error. {e}",
+                                    extra={"app_slug": slug, "error": str(e)},
+                                )
+                            except (
+                                ImportError,
+                                AttributeError,
+                                TypeError,
+                                ValueError,
+                                RuntimeError,
+                                ConnectionError,
+                                OSError,
+                            ) as e:
+                                error_msg = str(e).lower()
+                                error_type = type(e).__name__
+                                is_api_key_error = (
+                                    "api_key" in error_msg
+                                    or "api key" in error_msg
+                                    or "openai" in error_type.lower()
+                                )
+                                if is_api_key_error:
+                                    logger.warning(
+                                        f"Memory service initialization skipped for mounted app "
+                                        f"'{slug}': Missing API key or configuration. {e}",
+                                        extra={"app_slug": slug, "error": str(e)},
+                                    )
+                                else:
+                                    logger.error(
+                                        f"Failed to initialize memory service for mounted app "
+                                        f"'{slug}': {e}",
+                                        exc_info=True,
+                                        extra={"app_slug": slug, "error": str(e)},
+                                    )
+                        else:
+                            logger.warning(
+                                f"Memory service requested for '{slug}' but "
+                                f"service_initializer is not available"
+                            )
+
                     # Mount child app at path prefix (AFTER WebSocket routes are registered)
                     app.mount(path_prefix, child_app)
 

mdb_engine/core/service_initialization.py

@@ -26,6 +26,11 @@ from ..observability import get_logger as get_contextual_logger
 logger = logging.getLogger(__name__)
 contextual_logger = get_contextual_logger(__name__)
 
+try:
+    from openai import OpenAIError
+except ImportError:
+    OpenAIError = RuntimeError
+
 
 class ServiceInitializer:
     """
@@ -54,7 +59,9 @@ class ServiceInitializer:
         self._memory_services: dict[str, Any] = {}
         self._websocket_configs: dict[str, dict[str, Any]] = {}
 
-    async def initialize_memory_service(self, slug: str, memory_config: dict[str, Any]) -> None:
+    async def initialize_memory_service(
+        self, slug: str, memory_config: dict[str, Any] | None
+    ) -> None:
         """
         Initialize Mem0 memory service for an app.
 
@@ -63,8 +70,17 @@ class ServiceInitializer:
 
         Args:
             slug: App slug
-            memory_config: Memory configuration from manifest (already validated)
+            memory_config: Memory configuration from manifest (already validated).
+                Can be None or empty dict to skip initialization.
         """
+        # Handle None or empty config
+        if not memory_config:
+            return
+
+        # Check if memory is enabled (must be checked before import)
+        if not memory_config.get("enabled", False):
+            return
+
         # Try to import Memory service (optional dependency)
         try:
             from ..memory import Mem0MemoryService, Mem0MemoryServiceError
@@ -145,12 +161,41 @@ class ServiceInitializer:
                 extra={"app_slug": slug, "error": str(e)},
                 exc_info=True,
             )
-        except (ImportError, AttributeError, TypeError, ValueError) as e:
-            contextual_logger.error(
-                f"Error initializing memory service for app '{slug}': {e}",
+        except OpenAIError as e:
+            contextual_logger.warning(
+                f"Memory service initialization skipped for app '{slug}': "
+                f"OpenAI API error. {e}",
                 extra={"app_slug": slug, "error": str(e)},
-                exc_info=True,
             )
+        except (
+            ImportError,
+            AttributeError,
+            TypeError,
+            ValueError,
+            RuntimeError,
+            ConnectionError,
+            OSError,
+        ) as e:
+            error_msg = str(e).lower()
+            error_type = type(e).__name__
+            is_api_key_error = (
+                "api_key" in error_msg
+                or "api key" in error_msg
+                or "openai" in error_type.lower()
+                or "openai" in error_msg
+            )
+            if is_api_key_error:
+                contextual_logger.warning(
+                    f"Memory service initialization skipped for app '{slug}': "
+                    f"Missing API key or configuration. {e}",
+                    extra={"app_slug": slug, "error": str(e)},
+                )
+            else:
+                contextual_logger.error(
+                    f"Error initializing memory service for app '{slug}': {e}",
+                    extra={"app_slug": slug, "error": str(e)},
+                    exc_info=True,
+                )
 
     async def register_websockets(self, slug: str, websockets_config: dict[str, Any]) -> None:
         """
@@ -330,7 +375,18 @@ class ServiceInitializer:
                         extra={"app_slug": slug},
                     )
                     return None
-            return service
+                return service
+
+            # Service not found - check if it should be initialized but wasn't
+            # This can happen in multi-app context if initialization was missed
+            # Note: We can't do async initialization here, so we just log a warning
+            # The explicit initialization in create_multi_app should handle this
+            contextual_logger.debug(
+                f"Memory service not found for '{slug}' - "
+                f"it may not be initialized yet or memory is disabled",
+                extra={"app_slug": slug},
+            )
+            return None
         except (KeyError, AttributeError, TypeError) as e:
             contextual_logger.error(
                 f"Error retrieving memory service for '{slug}': {e}",

{mdb_engine-0.7.0.dist-info → mdb_engine-0.7.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mdb-engine
-Version: 0.7.0
+Version: 0.7.1
 Summary: MongoDB Engine
 Home-page: https://github.com/ranfysvalle02/mdb-engine
 Author: Fabian Valle

{mdb_engine-0.7.0.dist-info → mdb_engine-0.7.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 mdb_engine/README.md,sha256=T3EFGcPopY9LslYW3lxgG3hohWkAOmBNbYG0FDMUJiY,3502
-mdb_engine/__init__.py,sha256=vishQ3BF6oGQOmAsrMG1UJ-5C-3FNM49JSHC35MhH0k,3937
+mdb_engine/__init__.py,sha256=d2oenqa95hRBwXIVPmKBE8lo61ojOkxfcuuIzgSxLXc,3655
 mdb_engine/config.py,sha256=DTAyxfKB8ogyI0v5QR9Y-SJOgXQr_eDBCKxNBSqEyLc,7269
 mdb_engine/constants.py,sha256=eaotvW57TVOg7rRbLziGrVNoP7adgw_G9iVByHezc_A,7837
 mdb_engine/dependencies.py,sha256=MJuYQhZ9ZGzXlip1ha5zba9Rvn04HDPWahJFJH81Q2s,14107
@@ -35,7 +35,7 @@ mdb_engine/auth/utils.py,sha256=YkexCo0xV37mpOJUI32cntRHVOUUS7r19TIMPWHcgpA,2734
 mdb_engine/auth/websocket_sessions.py,sha256=7eFNagY2K3Rp1x7d_cO5JcpT-DrYkc__cmVhl6pAC2M,15081
 mdb_engine/auth/websocket_tickets.py,sha256=VoIArcnQBtYqXRMs-5m7NSvCJB1dEeHrLl7j7yG-H-A,9887
 mdb_engine/cli/__init__.py,sha256=PANRi4THmL34d1mawlqxIrnuItXMdqoMTq5Z1zHd7rM,301
-mdb_engine/cli/main.py,sha256=Y5ELFhvsr8zxFWv4WScOGNHiLUTdSXAJeUFLpRXCelg,811
+mdb_engine/cli/main.py,sha256=4bWBNgksfRgimolBfivITNvYRMaLTvngeVDcdm7ou6M,811
 mdb_engine/cli/utils.py,sha256=bNRGJgdzxUjXAOVe1aoxWJ5M_IqtAE-eW4pfAkwiDDM,2760
 mdb_engine/cli/commands/__init__.py,sha256=ZSzMhKdV9ILD5EbOSxDV9nURHo1e4bQ0c8AWpqsTEqM,115
 mdb_engine/cli/commands/generate.py,sha256=VEcn7qNQkIPQlLEK3oXUBgYMwD-G0FyomXQcWTtKsbs,17304
@@ -48,12 +48,12 @@ mdb_engine/core/app_registration.py,sha256=w7wjrlNQsEekuDQkG6HM4Z5t00E65baEzncnh
 mdb_engine/core/app_secrets.py,sha256=bo-syg9UUATibNyXEZs-0TTYWG-JaY-2S0yNSGA12n0,10524
 mdb_engine/core/connection.py,sha256=XnwuPG34pJ7kJGJ84T0mhj1UZ6_CLz_9qZf6NRYGIS8,8346
 mdb_engine/core/encryption.py,sha256=RZ5LPF5g28E3ZBn6v1IMw_oas7u9YGFtBcEj8lTi9LM,7515
-mdb_engine/core/engine.py,sha256=RBXJUYRbXrt4rhQLSxcRF-FYzf-ZxN9j358sdlw4_pc,182475
+mdb_engine/core/engine.py,sha256=wSJ99FP3NRUejGCXliwi9fW_pVZPpLjdUMG8FsU5Kms,185252
 mdb_engine/core/index_management.py,sha256=9-r7MIy3JnjQ35sGqsbj8K_I07vAUWtAVgSWC99lJcE,5555
 mdb_engine/core/manifest.py,sha256=D3OGRjm1It8dmS3IMoxHckHUerhs5PtcQgPz8o5ZL9w,140785
 mdb_engine/core/ray_integration.py,sha256=csAgICl2g7Plqy4N49MJU1Ca-lr8KZznAouXMMRwhG8,13706
 mdb_engine/core/seeding.py,sha256=c5IhdwlqUf_4Q5FFTAhPLaHPaUr_Txo3z_DUwZmWsFs,6421
-mdb_engine/core/service_initialization.py,sha256=rtb6BaPvFqomwT_s7bdbbvqi5m74llT0LkJFEhVG9Gg,12996
+mdb_engine/core/service_initialization.py,sha256=N4varRswExBYahAgPfcPtNhqszOjm0JQ0u7r7mfFyqM,15018
 mdb_engine/core/types.py,sha256=u0LDGE4A3txGDlc9PNoQX8jwVYFjJSw4K5LDnpatsNc,11254
 mdb_engine/database/README.md,sha256=-31mVxBeVQaYsF3AD1-gQbD2NCYVcPjdFoA6sZ6b02Y,19354
 mdb_engine/database/__init__.py,sha256=rrc3eZFli3K2zrvVdDbMBi8YkmoHYzP6JNT0AUBE5VU,981
@@ -91,9 +91,9 @@ mdb_engine/routing/__init__.py,sha256=reupjHi_RTc2ZBA4AH5XzobAmqy4EQIsfSUcTkFknU
 mdb_engine/routing/websockets.py,sha256=jgdLsrSOKyBiY4yzE7XYXV1MOChLw2CIWpT6RDL23hk,51649
 mdb_engine/utils/__init__.py,sha256=lDxQSGqkV4fVw5TWIk6FA6_eey_ZnEtMY0fir3cpAe8,236
 mdb_engine/utils/mongo.py,sha256=Oqtv4tQdpiiZzrilGLEYQPo8Vmh8WsTQypxQs8Of53s,3369
-mdb_engine-0.7.0.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-mdb_engine-0.7.0.dist-info/METADATA,sha256=dSWF8HOigw-49yf_VBLhr0hA--LKqroswl9uu3MmQik,19695
-mdb_engine-0.7.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-mdb_engine-0.7.0.dist-info/entry_points.txt,sha256=INCbYdFbBzJalwPwxliEzLmPfR57IvQ7RAXG_pn8cL8,48
-mdb_engine-0.7.0.dist-info/top_level.txt,sha256=PH0UEBwTtgkm2vWvC9He_EOMn7hVn_Wg_Jyc0SmeO8k,11
-mdb_engine-0.7.0.dist-info/RECORD,,
+mdb_engine-0.7.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+mdb_engine-0.7.1.dist-info/METADATA,sha256=GXZ-ojykqFCH5RLybnClAHWcNU9da-wXweOyZmH01tM,19695
+mdb_engine-0.7.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+mdb_engine-0.7.1.dist-info/entry_points.txt,sha256=INCbYdFbBzJalwPwxliEzLmPfR57IvQ7RAXG_pn8cL8,48
+mdb_engine-0.7.1.dist-info/top_level.txt,sha256=PH0UEBwTtgkm2vWvC9He_EOMn7hVn_Wg_Jyc0SmeO8k,11
+mdb_engine-0.7.1.dist-info/RECORD,,