mdb-engine 0.7.2__py3-none-any.whl → 0.7.3__py3-none-any.whl

mdb_engine/__init__.py

@@ -81,15 +81,7 @@ from .repositories import Entity, MongoRepository, Repository, UnitOfWork
 # Utilities
 from .utils import clean_mongo_doc, clean_mongo_docs
 
-__version__ = (
-    "0.7.2"  # Memory service update functionality
-    # - ADDED: Memory service update() method for in-place memory updates
-    # - ADDED: Support for updating memory content and metadata while preserving IDs
-    # - ADDED: Automatic embedding recomputation via Mem0's update method
-    # - ADDED: Comprehensive unit tests for memory update functionality (17 tests)
-    # - ENHANCED: Memory service now uses Mem0's native update method exclusively
-    # - REMOVED: Direct MongoDB update fallback (simplified implementation)
-)
+__version__ = "0.7.3"
 
 __all__ = [
     # Core Engine

mdb_engine/cli/main.py

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

mdb_engine/memory/service.py

@@ -268,10 +268,13 @@ class Mem0MemoryService:
         if isinstance(messages, str):
             messages = [{"role": "user", "content": messages}]
 
-        # Merge metadata
         final_metadata = dict(metadata) if metadata else {}
 
         # CRITICAL: Database indexing relies on these fields being in metadata
+        # Include user_id in metadata ONLY if provided (supports non-SSO use cases)
+        if user_id:
+            final_metadata["user_id"] = str(user_id)
+
         if bucket_id:
             final_metadata["bucket_id"] = bucket_id
             final_metadata["context_id"] = bucket_id  # Backwards compatibility
@@ -279,7 +282,6 @@ class Mem0MemoryService:
         if bucket_type:
             final_metadata["bucket_type"] = bucket_type
 
-        # Store raw_content in metadata if provided (metadata convenience)
         if raw_content:
             final_metadata["raw_content"] = raw_content
 
@@ -502,12 +504,12 @@ class Mem0MemoryService:
             memory_id: The ID of the memory to update (required)
             user_id: The user ID who owns the memory (for scoping and security)
             memory: New memory content as a string (optional)
-            data: Alternative parameter name for memory content (backward compatibility).
+            data: Alternative parameter name for memory content.
                   Can be a string or dict with 'memory'/'text'/'content' key.
             messages: Alternative way to provide content as messages (optional).
                       Can be a string or list of dicts with 'content' key.
-            metadata: Metadata updates to merge with existing metadata (optional).
-                     Merged, not replaced - existing keys are preserved unless overridden.
+            metadata: Metadata updates (optional, but NOT SUPPORTED by Mem0 update API).
+                     This parameter is accepted for API consistency but will be ignored.
             **kwargs: Additional arguments passed to Mem0 operations
 
         Returns:
@@ -535,16 +537,14 @@ class Mem0MemoryService:
             )
             ```
         """
-        # Input validation
         if not memory_id or not isinstance(memory_id, str) or not memory_id.strip():
             raise ValueError("memory_id is required and must be a non-empty string")
 
         try:
-            # Normalize data parameter (backward compatibility)
+            # Normalize data parameter (alternative to memory parameter)
             normalized_memory = self._normalize_content_input(memory, data, messages)
             normalized_metadata = self._normalize_metadata_input(metadata, data)
 
-            # Verify memory exists before attempting update
             existing_memory = self.get(memory_id=memory_id, user_id=user_id, **kwargs)
             if not existing_memory:
                 logger.warning(
@@ -675,46 +675,21 @@ class Mem0MemoryService:
         """
         Update memory using Mem0's built-in update method.
 
-        This is the primary update path. Mem0's update method handles:
-        - In-place updates preserving memory ID and created_at timestamp
-        - Automatic embedding recomputation when content changes
-        - Metadata merging
-        - User scoping for security
+        Note: Mem0's update() only supports content updates (text parameter).
+        Metadata updates are not supported by the Mem0 API.
 
         Args:
             memory_id: Memory ID to update
-            user_id: User ID for scoping
+            user_id: User ID for scoping (not used in update call)
             memory: New memory content (normalized)
-            metadata: Metadata to merge (normalized)
+            metadata: Metadata to merge (ignored - not supported by Mem0 update API)
             **kwargs: Additional arguments passed to Mem0
 
         Returns:
             Updated memory dict or None if not found
-
-        Raises:
-            Various exceptions from Mem0 if update fails
         """
-        # Build update parameters matching Mem0's API
-        # Mem0's update method signature:
-        # update(memory_id, text=None, metadata=None, user_id=None, **kwargs)
-        update_kwargs: dict[str, Any] = {"memory_id": memory_id}
-
-        # Add user_id for scoping (Mem0 supports this)
-        if user_id:
-            update_kwargs["user_id"] = str(user_id)
-
-        # Add text/content if provided
-        # Mem0 uses "text" parameter for content
-        if memory:
-            update_kwargs["text"] = memory
-
-        # Add metadata if provided
-        # Mem0 merges metadata automatically
-        if metadata is not None:
-            update_kwargs["metadata"] = metadata
-
-        # Pass through any additional kwargs
-        update_kwargs.update(kwargs)
+        # Filter out user_id from kwargs to prevent passing it as a direct parameter
+        filtered_kwargs = {k: v for k, v in kwargs.items() if k != "user_id"}
 
         logger.debug(
             f"Calling mem0.update() for memory_id={memory_id}",
@@ -726,24 +701,29 @@ class Mem0MemoryService:
             },
         )
 
-        # Call Mem0's update method directly
-        # This handles all the complexity: embedding recomputation, ID preservation, etc.
+        if metadata:
+            logger.warning(
+                f"Metadata update requested for memory {memory_id} but Mem0 update() "
+                f"does not support metadata parameter. Metadata will not be updated."
+            )
+
+        update_kwargs = {"memory_id": memory_id}
+        if memory:
+            update_kwargs["data"] = memory
+        update_kwargs.update(filtered_kwargs)
+
         result = self.memory.update(**update_kwargs)
 
-        # Normalize result format
-        # Mem0 may return dict, list, or other formats
+        # Note: Mem0's update() doesn't support metadata parameter
+        # Metadata updates would require direct MongoDB access, which we avoid
+
+        # Normalize result to dict
         if isinstance(result, dict):
             return result
         elif isinstance(result, list) and len(result) > 0:
-            # If list, return first item
             return result[0] if isinstance(result[0], dict) else None
         else:
-            # If result is None or unexpected format, return None to trigger fallback
-            logger.debug(
-                f"Mem0 update returned unexpected format: {type(result)}",
-                extra={"memory_id": memory_id},
-            )
-            return None
+            return self.get(memory_id=memory_id)
 
     def _normalize_result(self, result: Any) -> list[dict[str, Any]]:
         """Normalize Mem0's return type (dict vs list)."""

mdb_engine/routing/websockets.py

@@ -472,19 +472,58 @@ async def _validate_websocket_origin_in_handler(websocket: Any, app_slug: str) -
 
         # Normalize origin for comparison
         def normalize_origin(orig: str) -> str:
-            """Normalize origin handling localhost variants."""
+            """
+            Normalize origin handling localhost variants and protocol differences.
+
+            Handles:
+            - Protocol conversion: ws/wss -> http/https (browsers send http/https)
+            - Localhost normalization: 127.0.0.1, 0.0.0.0, ::1 -> localhost
+            - Docker IP normalization: container IPs -> localhost (in development)
+            - Port normalization: removes :80 and :443
+            """
             if not orig:
                 return orig
+            import os
             import re
 
-            normalized = re.sub(
-                r"://(0\.0\.0\.0|127\.0\.0\.1|localhost|::1)",
-                "://localhost",
-                orig.lower(),
-                flags=re.IGNORECASE,
+            normalized = orig.lower()
+
+            # Normalize protocol: convert ws/wss to http/https for comparison
+            # WebSocket origins come as http/https from browsers, but configs may use ws/wss
+            normalized = re.sub(r"^ws://", "http://", normalized)
+            normalized = re.sub(r"^wss://", "https://", normalized)
+
+            # Check if we're in development/Docker environment
+            is_dev = (
+                os.getenv("ENVIRONMENT", "").lower() in ["development", "dev"]
+                or os.getenv("G_NOME_ENV", "").lower() in ["development", "dev"]
+                or os.path.exists("/.dockerenv")  # Docker container indicator
             )
+
+            # Normalize localhost variants
+            # In development/Docker, also normalize common Docker IP ranges to localhost
+            if is_dev:
+                # Match localhost, 127.0.0.1, 0.0.0.0, ::1, and Docker container IPs
+                # Docker typically uses 172.17.0.0/16 or 172.20.0.0/16
+                normalized = re.sub(
+                    r"://(0\.0\.0\.0|127\.0\.0\.1|localhost|::1|172\.(17|20)\.\d+\.\d+)",
+                    "://localhost",
+                    normalized,
+                    flags=re.IGNORECASE,
+                )
+            else:
+                # Production: only normalize standard localhost variants
+                normalized = re.sub(
+                    r"://(0\.0\.0\.0|127\.0\.0\.1|localhost|::1)",
+                    "://localhost",
+                    normalized,
+                    flags=re.IGNORECASE,
+                )
+
+            # Remove default ports
             normalized = re.sub(r":80$", "", normalized)
             normalized = re.sub(r":443$", "", normalized)
+
             return normalized.rstrip("/")
 
         normalized_origin = normalize_origin(origin)

{mdb_engine-0.7.2.dist-info → mdb_engine-0.7.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mdb-engine
-Version: 0.7.2
+Version: 0.7.3
 Summary: MongoDB Engine
 Home-page: https://github.com/ranfysvalle02/mdb-engine
 Author: Fabian Valle
@@ -72,20 +72,6 @@ Dynamic: requires-python
 [![Python 3.8+](https://img.shields.io/badge/python-3.8+-blue.svg)](https://www.python.org/downloads/)
 [![License: AGPL-3.0](https://img.shields.io/badge/License-AGPL--3.0-blue.svg)](https://opensource.org/licenses/AGPL-3.0)
 
-## 🎉 What's New in v0.7.0
-
-**FastAPI Native WebSocket Support**: MDB-Engine now uses FastAPI's `APIRouter` approach for WebSocket registration in both single-app and multi-app modes. This provides:
-
-- ✅ **Full FastAPI Feature Support**: Dependency injection, OpenAPI documentation, request/response models
-- ✅ **Consistency**: Same registration pattern across single-app and multi-app modes  
-- ✅ **Best Practices**: Follows FastAPI's recommended WebSocket registration patterns
-- ✅ **Better Maintainability**: Uses FastAPI abstractions instead of low-level Starlette APIs
-- ✅ **Route Priority**: WebSocket routes registered before mounted apps ensure proper routing
-
-**Value**: This change ensures WebSocket endpoints benefit from all FastAPI features, making your code more maintainable and consistent with FastAPI best practices.
-
----
-
 ## 🎯 manifest.json: The Key to Everything
 
 **`manifest.json` is the foundation of your application.** It's a single configuration file that defines your app's identity, data structure, authentication, indexes, and services. Everything flows from this file.

{mdb_engine-0.7.2.dist-info → mdb_engine-0.7.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 mdb_engine/README.md,sha256=T3EFGcPopY9LslYW3lxgG3hohWkAOmBNbYG0FDMUJiY,3502
-mdb_engine/__init__.py,sha256=12PLLmXDANoV30dDxw5To9taKAZnIMolRoV1Vwdjkwk,3618
+mdb_engine/__init__.py,sha256=AwWt-MyzGNDl_2YEkTO0-x15qnUw3tF-oiOJJAYGkIs,3097
 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=cp-fxcijZN4o4taBTP2sR57Y1A-ZId0Qga7AhgrU48o,811
+mdb_engine/cli/main.py,sha256=qDVVUJOXajdZmbId9AIGBS380U7OlhA8ILrAIfURbSE,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
@@ -76,7 +76,7 @@ mdb_engine/indexes/helpers.py,sha256=tJHqDm18jKLrW-P2ofmnUnaf4_-V5xDLjqP_WU8W9MM
 mdb_engine/indexes/manager.py,sha256=ekrYBfKD-GOBpZMjXIZWSQ7UPLgrD5BINYzVteWkSI8,32508
 mdb_engine/memory/README.md,sha256=Xa7pYut4zgZyRcb2TALcucMVAHCAu9uNGBKSwE7stY8,14764
 mdb_engine/memory/__init__.py,sha256=e4kAYgxd_-WAH8GovTwjEBO9hvASu_kXEupMgksAL-U,1008
-mdb_engine/memory/service.py,sha256=QWMNZfMaS3-SPZZswnO410C4-3ibAOUw0LH93WjZLsM,29550
+mdb_engine/memory/service.py,sha256=huWIjHAsV-g_Cs9djRkd8v5ZgnqiXqtNEbNjAgMfzs4,28846
 mdb_engine/observability/README.md,sha256=CMgQaC1H8ESmCitfbhJifz6-XoXH_FPNE4MvuZ-oFas,13085
 mdb_engine/observability/__init__.py,sha256=jjLsrW6Gy2ayrbfLrgHsDB61NxWWkYLHwv0q-N3fxjA,1213
 mdb_engine/observability/health.py,sha256=ORjxF_rHYA9vCoxUqel5p0n9g3PLmHsHQn68Q402b6g,9212
@@ -88,12 +88,12 @@ mdb_engine/repositories/mongo.py,sha256=Wg32_6v0KHAHumhz5z8QkoqJRWAMJFA7Y2lYIJ7L
 mdb_engine/repositories/unit_of_work.py,sha256=XvmwGOspEDj4hsfOULPsQKjB1QZqh83TJo6vGV4tiqU,5118
 mdb_engine/routing/README.md,sha256=zk-Wux-QpuWWhprfVeiUREv8PEDJZNgh2QblXIa4v4M,14201
 mdb_engine/routing/__init__.py,sha256=reupjHi_RTc2ZBA4AH5XzobAmqy4EQIsfSUcTkFknUM,2438
-mdb_engine/routing/websockets.py,sha256=jgdLsrSOKyBiY4yzE7XYXV1MOChLw2CIWpT6RDL23hk,51649
+mdb_engine/routing/websockets.py,sha256=w0m1XMpXKPpPrEBcZNIj8B8PwWeTNrseUZE1P8cipdM,53441
 mdb_engine/utils/__init__.py,sha256=lDxQSGqkV4fVw5TWIk6FA6_eey_ZnEtMY0fir3cpAe8,236
 mdb_engine/utils/mongo.py,sha256=Oqtv4tQdpiiZzrilGLEYQPo8Vmh8WsTQypxQs8Of53s,3369
-mdb_engine-0.7.2.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-mdb_engine-0.7.2.dist-info/METADATA,sha256=NkjWW4qrAnJs_T4NxvCvQvZkln4vRYSRa101sBbsXY4,19695
-mdb_engine-0.7.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-mdb_engine-0.7.2.dist-info/entry_points.txt,sha256=INCbYdFbBzJalwPwxliEzLmPfR57IvQ7RAXG_pn8cL8,48
-mdb_engine-0.7.2.dist-info/top_level.txt,sha256=PH0UEBwTtgkm2vWvC9He_EOMn7hVn_Wg_Jyc0SmeO8k,11
-mdb_engine-0.7.2.dist-info/RECORD,,
+mdb_engine-0.7.3.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+mdb_engine-0.7.3.dist-info/METADATA,sha256=3XBbRELougI8lJsaJ1A02SuhIpsv2ojQspD2XidEXbo,18843
+mdb_engine-0.7.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+mdb_engine-0.7.3.dist-info/entry_points.txt,sha256=INCbYdFbBzJalwPwxliEzLmPfR57IvQ7RAXG_pn8cL8,48
+mdb_engine-0.7.3.dist-info/top_level.txt,sha256=PH0UEBwTtgkm2vWvC9He_EOMn7hVn_Wg_Jyc0SmeO8k,11
+mdb_engine-0.7.3.dist-info/RECORD,,