mdb-engine 0.2.1__py3-none-any.whl → 0.2.4__py3-none-any.whl

mdb_engine/core/manifest.py

@@ -32,7 +32,8 @@ For Scale:
 import asyncio
 import hashlib
 import logging
-from typing import Any, Awaitable, Callable, Dict, List, Optional, Tuple
+from collections.abc import Awaitable, Callable
+from typing import Any
 
 from jsonschema import SchemaError, ValidationError, validate
 
@@ -48,10 +49,10 @@ from ..constants import (
 logger = logging.getLogger(__name__)
 
 # Schema registry: maps version -> schema definition
-SCHEMA_REGISTRY: Dict[str, Dict[str, Any]] = {}
+SCHEMA_REGISTRY: dict[str, dict[str, Any]] = {}
 
 # Validation cache: maps (manifest_hash, version) -> validation_result
-_validation_cache: Dict[str, Tuple[bool, Optional[str], Optional[List[str]]]] = {}
+_validation_cache: dict[str, tuple[bool, str | None, list[str] | None]] = {}
 _cache_lock = asyncio.Lock()
 
 
@@ -86,7 +87,7 @@ def _convert_tuples_to_lists(obj: Any) -> Any:
         return obj
 
 
-def _get_manifest_hash(manifest_data: Dict[str, Any]) -> str:
+def _get_manifest_hash(manifest_data: dict[str, Any]) -> str:
     """Generate a hash for manifest caching."""
     import json
 
@@ -183,6 +184,27 @@ MANIFEST_SCHEMA_V2 = {
                         "Works in both auth modes."
                     ),
                 },
+                "auth_hub_url": {
+                    "type": "string",
+                    "format": "uri",
+                    "description": (
+                        "URL of the authentication hub for SSO apps (shared mode only). "
+                        "Used for redirecting unauthenticated users to login. "
+                        "Example: 'http://localhost:8000' or 'https://auth.example.com'. "
+                        "Can be overridden via AUTH_HUB_URL environment variable."
+                    ),
+                },
+                "related_apps": {
+                    "type": "object",
+                    "additionalProperties": {"type": "string", "format": "uri"},
+                    "description": (
+                        "Map of related app slugs to their URLs for cross-app navigation "
+                        "(useful in shared auth mode). Keys are app slugs, values are URLs. "
+                        "Example: {'dashboard': 'http://localhost:8001', "
+                        "'click_tracker': 'http://localhost:8000'}. "
+                        "Can be overridden via {APP_SLUG_UPPER}_URL environment variables."
+                    ),
+                },
                 "policy": {
                     "type": "object",
                     "properties": {
@@ -2083,7 +2105,7 @@ SCHEMA_REGISTRY["default"] = MANIFEST_SCHEMA_V2
 MANIFEST_SCHEMA = MANIFEST_SCHEMA_V2  # Backward compatibility
 
 
-def get_schema_version(manifest_data: Dict[str, Any]) -> str:
+def get_schema_version(manifest_data: dict[str, Any]) -> str:
     """
     Detect schema version from manifest.
 
@@ -2096,7 +2118,7 @@ def get_schema_version(manifest_data: Dict[str, Any]) -> str:
     Raises:
         ValueError: If schema version format is invalid
     """
-    version: Optional[str] = manifest_data.get("schema_version")
+    version: str | None = manifest_data.get("schema_version")
     if version:
         # Validate version format
         if not isinstance(version, str) or not version.replace(".", "").isdigit():
@@ -2118,8 +2140,8 @@ def get_schema_version(manifest_data: Dict[str, Any]) -> str:
 
 
 def migrate_manifest(
-    manifest_data: Dict[str, Any], target_version: str = CURRENT_SCHEMA_VERSION
-) -> Dict[str, Any]:
+    manifest_data: dict[str, Any], target_version: str = CURRENT_SCHEMA_VERSION
+) -> dict[str, Any]:
     """
     Migrate manifest from one schema version to another.
 
@@ -2168,7 +2190,7 @@ def migrate_manifest(
     return migrated
 
 
-def get_schema_for_version(version: str) -> Dict[str, Any]:
+def get_schema_for_version(version: str) -> dict[str, Any]:
     """
     Get schema definition for a specific version.
 
@@ -2201,8 +2223,8 @@ def get_schema_for_version(version: str) -> Dict[str, Any]:
 
 
 async def _validate_manifest_async(
-    manifest_data: Dict[str, Any], use_cache: bool = True
-) -> Tuple[bool, Optional[str], Optional[List[str]]]:
+    manifest_data: dict[str, Any], use_cache: bool = True
+) -> tuple[bool, str | None, list[str] | None]:
     """
     Validate a manifest against the JSON Schema with versioning and caching support.
 
@@ -2329,8 +2351,8 @@ def clear_validation_cache():
 
 
 async def validate_manifests_parallel(
-    manifests: List[Dict[str, Any]], use_cache: bool = True
-) -> List[Tuple[bool, Optional[str], Optional[List[str]], Optional[str]]]:
+    manifests: list[dict[str, Any]], use_cache: bool = True
+) -> list[tuple[bool, str | None, list[str] | None, str | None]]:
     """
     Validate multiple manifests in parallel for scale.
 
@@ -2344,8 +2366,8 @@ async def validate_manifests_parallel(
     """
 
     async def validate_one(
-        manifest: Dict[str, Any],
-    ) -> Tuple[bool, Optional[str], Optional[List[str]], Optional[str]]:
+        manifest: dict[str, Any],
+    ) -> tuple[bool, str | None, list[str] | None, str | None]:
         slug = manifest.get("slug", "unknown")
         is_valid, error, paths = await _validate_manifest_async(manifest, use_cache=use_cache)
         return (is_valid, error, paths, slug)
@@ -2366,8 +2388,8 @@ async def validate_manifests_parallel(
 
 
 async def validate_developer_id(
-    developer_id: str, db_validator: Optional[Callable[[str], Awaitable[bool]]] = None
-) -> Tuple[bool, Optional[str]]:
+    developer_id: str, db_validator: Callable[[str], Awaitable[bool]] | None = None
+) -> tuple[bool, str | None]:
     """
     Validate that a developer_id exists in the system and has developer role.
 
@@ -2411,10 +2433,10 @@ async def validate_developer_id(
 
 
 async def validate_manifest_with_db(
-    manifest_data: Dict[str, Any],
+    manifest_data: dict[str, Any],
     db_validator: Callable[[str], Awaitable[bool]],
     use_cache: bool = True,
-) -> Tuple[bool, Optional[str], Optional[List[str]]]:
+) -> tuple[bool, str | None, list[str] | None]:
     """
     Validate a manifest against the JSON Schema (with versioning) and check
     developer_id exists in system.
@@ -2455,8 +2477,8 @@ async def validate_manifest_with_db(
 # Public API: Synchronous wrapper for backward compatibility
 # Most callers use this synchronously, so we provide a sync wrapper
 def validate_manifest(
-    manifest_data: Dict[str, Any], use_cache: bool = True
-) -> Tuple[bool, Optional[str], Optional[List[str]]]:
+    manifest_data: dict[str, Any], use_cache: bool = True
+) -> tuple[bool, str | None, list[str] | None]:
     """
     Validate a manifest against the JSON Schema with versioning and caching
     support (synchronous wrapper).
@@ -2495,8 +2517,8 @@ def validate_manifest(
 
 
 def _validate_regular_index(
-    index_def: Dict[str, Any], collection_name: str, index_name: str
-) -> Tuple[bool, Optional[str]]:
+    index_def: dict[str, Any], collection_name: str, index_name: str
+) -> tuple[bool, str | None]:
     """Validate a regular index definition."""
     if "keys" not in index_def:
         return (
@@ -2533,8 +2555,8 @@ def _validate_regular_index(
 
 
 def _validate_ttl_index(
-    index_def: Dict[str, Any], collection_name: str, index_name: str
-) -> Tuple[bool, Optional[str]]:
+    index_def: dict[str, Any], collection_name: str, index_name: str
+) -> tuple[bool, str | None]:
     """Validate a TTL index definition."""
     if "keys" not in index_def:
         return (
@@ -2567,8 +2589,8 @@ def _validate_ttl_index(
 
 
 def _validate_partial_index(
-    index_def: Dict[str, Any], collection_name: str, index_name: str
-) -> Tuple[bool, Optional[str]]:
+    index_def: dict[str, Any], collection_name: str, index_name: str
+) -> tuple[bool, str | None]:
     """Validate a partial index definition."""
     if "keys" not in index_def:
         return (
@@ -2588,8 +2610,8 @@ def _validate_partial_index(
 
 
 def _validate_text_index(
-    index_def: Dict[str, Any], collection_name: str, index_name: str
-) -> Tuple[bool, Optional[str]]:
+    index_def: dict[str, Any], collection_name: str, index_name: str
+) -> tuple[bool, str | None]:
     """Validate a text index definition."""
     if "keys" not in index_def:
         return (
@@ -2614,8 +2636,8 @@ def _validate_text_index(
 
 
 def _validate_geospatial_index(
-    index_def: Dict[str, Any], collection_name: str, index_name: str
-) -> Tuple[bool, Optional[str]]:
+    index_def: dict[str, Any], collection_name: str, index_name: str
+) -> tuple[bool, str | None]:
     """Validate a geospatial index definition."""
     if "keys" not in index_def:
         return (
@@ -2641,8 +2663,8 @@ def _validate_geospatial_index(
 
 
 def _validate_vector_search_index(
-    index_def: Dict[str, Any], collection_name: str, index_name: str, index_type: str
-) -> Tuple[bool, Optional[str]]:
+    index_def: dict[str, Any], collection_name: str, index_name: str, index_type: str
+) -> tuple[bool, str | None]:
     """Validate a vectorSearch or search index definition."""
     if "definition" not in index_def:
         return (
@@ -2689,8 +2711,8 @@ def _validate_vector_search_index(
 
 
 def _validate_hybrid_index(
-    index_def: Dict[str, Any], collection_name: str, index_name: str
-) -> Tuple[bool, Optional[str]]:
+    index_def: dict[str, Any], collection_name: str, index_name: str
+) -> tuple[bool, str | None]:
     """Validate a hybrid index definition."""
     if "hybrid" not in index_def:
         return (
@@ -2753,8 +2775,8 @@ def _validate_hybrid_index(
 
 
 def validate_index_definition(
-    index_def: Dict[str, Any], collection_name: str, index_name: str
-) -> Tuple[bool, Optional[str]]:
+    index_def: dict[str, Any], collection_name: str, index_name: str
+) -> tuple[bool, str | None]:
     """
     Validate a single index definition with context-specific checks.
 
@@ -2797,8 +2819,8 @@ def validate_index_definition(
 
 
 def validate_managed_indexes(
-    managed_indexes: Dict[str, List[Dict[str, Any]]],
-) -> Tuple[bool, Optional[str]]:
+    managed_indexes: dict[str, list[dict[str, Any]]],
+) -> tuple[bool, str | None]:
     """
     Validate all managed indexes with collection and index context.
 
@@ -2866,8 +2888,8 @@ class ManifestValidator:
 
     @staticmethod
     def validate(
-        manifest: Dict[str, Any], use_cache: bool = True
-    ) -> Tuple[bool, Optional[str], Optional[List[str]]]:
+        manifest: dict[str, Any], use_cache: bool = True
+    ) -> tuple[bool, str | None, list[str] | None]:
         """
         Validate manifest against schema.
 
@@ -2882,8 +2904,8 @@ class ManifestValidator:
 
     @staticmethod
     async def validate_async(
-        manifest: Dict[str, Any], use_cache: bool = True
-    ) -> Tuple[bool, Optional[str], Optional[List[str]]]:
+        manifest: dict[str, Any], use_cache: bool = True
+    ) -> tuple[bool, str | None, list[str] | None]:
         """
         Validate manifest asynchronously.
 
@@ -2902,10 +2924,10 @@ class ManifestValidator:
 
     @staticmethod
     async def validate_with_db(
-        manifest: Dict[str, Any],
+        manifest: dict[str, Any],
         db_validator: Callable[[str], Awaitable[bool]],
         use_cache: bool = True,
-    ) -> Tuple[bool, Optional[str], Optional[List[str]]]:
+    ) -> tuple[bool, str | None, list[str] | None]:
         """
         Validate manifest and check developer_id exists in database.
 
@@ -2921,8 +2943,8 @@ class ManifestValidator:
 
     @staticmethod
     def validate_managed_indexes(
-        managed_indexes: Dict[str, List[Dict[str, Any]]],
-    ) -> Tuple[bool, Optional[str]]:
+        managed_indexes: dict[str, list[dict[str, Any]]],
+    ) -> tuple[bool, str | None]:
         """
         Validate managed indexes configuration.
 
@@ -2936,8 +2958,8 @@ class ManifestValidator:
 
     @staticmethod
     def validate_index_definition(
-        index_def: Dict[str, Any], collection_name: str, index_name: str
-    ) -> Tuple[bool, Optional[str]]:
+        index_def: dict[str, Any], collection_name: str, index_name: str
+    ) -> tuple[bool, str | None]:
         """
         Validate a single index definition.
 
@@ -2952,7 +2974,7 @@ class ManifestValidator:
         return validate_index_definition(index_def, collection_name, index_name)
 
     @staticmethod
-    def get_schema_version(manifest: Dict[str, Any]) -> str:
+    def get_schema_version(manifest: dict[str, Any]) -> str:
         """
             Get schema version from manifest.
 
@@ -2966,8 +2988,8 @@ class ManifestValidator:
 
     @staticmethod
     def migrate(
-        manifest: Dict[str, Any], target_version: str = CURRENT_SCHEMA_VERSION
-    ) -> Dict[str, Any]:
+        manifest: dict[str, Any], target_version: str = CURRENT_SCHEMA_VERSION
+    ) -> dict[str, Any]:
         """
         Migrate manifest to target schema version.
 
@@ -2994,7 +3016,7 @@ class ManifestParser:
     with automatic validation and migration.
     """
 
-    def __init__(self, validator: Optional[ManifestValidator] = None):
+    def __init__(self, validator: ManifestValidator | None = None):
         """
         Initialize parser.
 
@@ -3004,7 +3026,7 @@ class ManifestParser:
         self.validator = validator or ManifestValidator()
 
     @staticmethod
-    async def load_from_file(path: Any, validate: bool = True) -> Dict[str, Any]:
+    async def load_from_file(path: Any, validate: bool = True) -> dict[str, Any]:
         """
         Load and validate manifest from file.
 
@@ -3041,7 +3063,7 @@ class ManifestParser:
         return manifest_data
 
     @staticmethod
-    async def load_from_dict(data: Dict[str, Any], validate: bool = True) -> Dict[str, Any]:
+    async def load_from_dict(data: dict[str, Any], validate: bool = True) -> dict[str, Any]:
         """
         Load and validate manifest from dictionary.
 
@@ -3065,7 +3087,7 @@ class ManifestParser:
         return data.copy()
 
     @staticmethod
-    async def load_from_string(content: str, validate: bool = True) -> Dict[str, Any]:
+    async def load_from_string(content: str, validate: bool = True) -> dict[str, Any]:
         """
         Load and validate manifest from JSON string.
 
@@ -3087,8 +3109,8 @@ class ManifestParser:
 
     @staticmethod
     async def load_and_migrate(
-        manifest: Dict[str, Any], target_version: str = CURRENT_SCHEMA_VERSION
-    ) -> Dict[str, Any]:
+        manifest: dict[str, Any], target_version: str = CURRENT_SCHEMA_VERSION
+    ) -> dict[str, Any]:
         """
         Load manifest and migrate to target version.
 

mdb_engine/core/ray_integration.py

@@ -25,7 +25,8 @@ This module is part of MDB_ENGINE - MongoDB Engine.
 
 import logging
 import os
-from typing import TYPE_CHECKING, Any, Callable, Optional, TypeVar
+from collections.abc import Callable
+from typing import TYPE_CHECKING, Any, TypeVar
 
 if TYPE_CHECKING:
     pass
@@ -132,7 +133,7 @@ class AppRayActor:
             else:
                 raise
 
-    async def get_app_db(self, app_token: Optional[str] = None) -> Any:
+    async def get_app_db(self, app_token: str | None = None) -> Any:
         """
         Get scoped database for this app.
 
@@ -191,11 +192,11 @@ class AppRayActor:
 async def get_ray_actor_handle(
     app_slug: str,
     namespace: str = "modular_labs",
-    mongo_uri: Optional[str] = None,
-    db_name: Optional[str] = None,
+    mongo_uri: str | None = None,
+    db_name: str | None = None,
     create_if_missing: bool = True,
-    actor_class: Optional[type] = None,
-) -> Optional[Any]:
+    actor_class: type | None = None,
+) -> Any | None:
     """
     Get or create a Ray actor handle for an app.
 
@@ -294,7 +295,7 @@ async def get_ray_actor_handle(
 
 
 def ray_actor_decorator(
-    app_slug: Optional[str] = None,
+    app_slug: str | None = None,
     namespace: str = "modular_labs",
     isolated: bool = True,
     lifetime: str = "detached",
@@ -369,8 +370,8 @@ def ray_actor_decorator(
         def spawn(
             cls,
             *args,
-            mongo_uri: Optional[str] = None,
-            db_name: Optional[str] = None,
+            mongo_uri: str | None = None,
+            db_name: str | None = None,
             **kwargs,
         ):
             """

mdb_engine/core/seeding.py

@@ -8,7 +8,7 @@ This module is part of MDB_ENGINE - MongoDB Engine.
 
 import logging
 from datetime import datetime
-from typing import Any, Dict, List
+from typing import Any
 
 from pymongo.errors import (
     ConnectionFailure,
@@ -20,8 +20,8 @@ logger = logging.getLogger(__name__)
 
 
 async def seed_initial_data(
-    db, app_slug: str, initial_data: Dict[str, List[Dict[str, Any]]]
-) -> Dict[str, int]:
+    db, app_slug: str, initial_data: dict[str, list[dict[str, Any]]]
+) -> dict[str, int]:
     """
     Seed initial data into collections.
 

mdb_engine/core/service_initialization.py

@@ -11,7 +11,8 @@ This module is part of MDB_ENGINE - MongoDB Engine.
 """
 
 import logging
-from typing import Any, Callable, Dict, List, Optional
+from collections.abc import Callable
+from typing import Any
 
 from pymongo.errors import (
     ConnectionFailure,
@@ -50,10 +51,10 @@ class ServiceInitializer:
         self.mongo_uri = mongo_uri
         self.db_name = db_name
         self.get_scoped_db_fn = get_scoped_db_fn
-        self._memory_services: Dict[str, Any] = {}
-        self._websocket_configs: Dict[str, Dict[str, Any]] = {}
+        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:
         """
         Initialize Mem0 memory service for an app.
 
@@ -151,7 +152,7 @@ class ServiceInitializer:
                 exc_info=True,
             )
 
-    async def register_websockets(self, slug: str, websockets_config: Dict[str, Any]) -> None:
+    async def register_websockets(self, slug: str, websockets_config: dict[str, Any]) -> None:
         """
         Register WebSocket endpoints for an app.
 
@@ -195,7 +196,7 @@ class ServiceInitializer:
             )
 
     async def seed_initial_data(
-        self, slug: str, initial_data: Dict[str, List[Dict[str, Any]]]
+        self, slug: str, initial_data: dict[str, list[dict[str, Any]]]
     ) -> None:
         """
         Seed initial data into collections for an app.
@@ -240,7 +241,7 @@ class ServiceInitializer:
             )
 
     async def setup_observability(
-        self, slug: str, manifest: Dict[str, Any], observability_config: Dict[str, Any]
+        self, slug: str, manifest: dict[str, Any], observability_config: dict[str, Any]
     ) -> None:
         """
         Set up observability features (health checks, metrics, logging) from manifest.
@@ -296,7 +297,7 @@ class ServiceInitializer:
                 f"Could not set up observability for {slug}: {e}", exc_info=True
             )
 
-    def get_websocket_config(self, slug: str) -> Optional[Dict[str, Any]]:
+    def get_websocket_config(self, slug: str) -> dict[str, Any] | None:
         """
         Get WebSocket configuration for an app.
 
@@ -308,7 +309,7 @@ class ServiceInitializer:
         """
         return self._websocket_configs.get(slug)
 
-    def get_memory_service(self, slug: str) -> Optional[Any]:
+    def get_memory_service(self, slug: str) -> Any | None:
         """
         Get Mem0 memory service for an app.