omnibase_infra 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

omnibase_infra/errors/error_event_bus_registry.py

@@ -5,6 +5,8 @@
 This module defines the EventBusRegistryError for event bus registry operations.
 """
 
+from typing import Any, cast
+
 from omnibase_infra.errors.error_infra import RuntimeHostError
 from omnibase_infra.models.errors.model_infra_error_context import (
     ModelInfraErrorContext,
@@ -92,10 +94,11 @@ class EventBusRegistryError(RuntimeHostError):
         if existing_class is not None:
             extra_context["existing_class"] = existing_class
 
+        # NOTE: Cast required for mypy - **dict[str, object] doesn't satisfy **context: Any
         super().__init__(
             message=message,
             context=context,
-            **extra_context,
+            **cast("dict[str, Any]", extra_context),
         )
 
 

omnibase_infra/errors/error_infra.py

@@ -68,6 +68,7 @@ NOT_FOUND Classification Patterns:
     - Optional keys, cache lookups: Use discriminated union responses
 """
 
+from typing import Any, cast
 from uuid import uuid4
 
 from omnibase_core.enums import EnumCoreErrorCode
@@ -145,11 +146,12 @@ class RuntimeHostError(ModelOnexError):
             correlation_id = uuid4()
 
         # Initialize base error with default error code
+        # NOTE: Cast required for mypy - **dict[str, object] doesn't satisfy **context: Any
         super().__init__(
             message=message,
             error_code=error_code or EnumCoreErrorCode.OPERATION_FAILED,
             correlation_id=correlation_id,
-            **structured_context,
+            **cast("dict[str, Any]", structured_context),
         )
 
 

omnibase_infra/errors/error_policy_registry.py

@@ -5,6 +5,8 @@
 This module defines the PolicyRegistryError for policy registry operations.
 """
 
+from typing import Any, cast
+
 from omnibase_infra.enums import EnumPolicyType
 from omnibase_infra.errors.error_infra import RuntimeHostError
 from omnibase_infra.models.errors.model_infra_error_context import (
@@ -102,10 +104,11 @@ class PolicyRegistryError(RuntimeHostError):
                 else policy_type
             )
 
+        # NOTE: Cast required for mypy - **dict[str, object] doesn't satisfy **context: Any
         super().__init__(
             message=message,
             context=context,
-            **extra_context,
+            **cast("dict[str, Any]", extra_context),
         )
 
 

omnibase_infra/handlers/handler_db.py

@@ -76,6 +76,7 @@ from uuid import UUID, uuid4
 import asyncpg
 
 from omnibase_core.models.dispatch import ModelHandlerOutput
+from omnibase_core.types import JsonType
 from omnibase_infra.enums import (
     EnumHandlerType,
     EnumHandlerTypeCategory,
@@ -1020,7 +1021,7 @@ class HandlerDb(MixinAsyncCircuitBreaker, MixinEnvelopeExtraction):
             - docs/architecture/HANDLER_PROTOCOL_DRIVEN_ARCHITECTURE.md
         """
         # Get circuit breaker state if initialized
-        cb_state: dict[str, object] | None = None
+        cb_state: dict[str, JsonType] | None = None
         if self._circuit_breaker_initialized:
             cb_state = self._get_circuit_breaker_state()
 

omnibase_infra/handlers/handler_graph.py

@@ -464,9 +464,11 @@ class HandlerGraph(MixinAsyncCircuitBreaker, ProtocolGraphDatabaseHandler):
                     try:
                         for query, params in queries:
                             query_start = time.perf_counter()
-                            result = await tx.run(query, dict(params) if params else {})
-                            records_data = await result.data()
-                            summary = await result.consume()
+                            tx_result = await tx.run(
+                                query, dict(params) if params else {}
+                            )
+                            records_data = await tx_result.data()
+                            summary = await tx_result.consume()
                             query_time_ms = (time.perf_counter() - query_start) * 1000
 
                             results.append(
@@ -497,8 +499,11 @@ class HandlerGraph(MixinAsyncCircuitBreaker, ProtocolGraphDatabaseHandler):
             else:
                 # Execute queries individually without transaction
                 for query, params in queries:
-                    result = await self.execute_query(query, params)
-                    results.append(result)
+                    # Type assertion: execute_query returns ModelGraphQueryResult in this handler
+                    query_result: ModelGraphQueryResult = await self.execute_query(
+                        query, params
+                    )  # type: ignore[assignment]
+                    results.append(query_result)
 
             return ModelGraphBatchResult(
                 results=results,