5959)
6060from .shared_callback_definitions import (
6161 INTERSECT_JSON_VALUE , # noqa: TCH001 (runtime-checked annotation)
62- DirectMessageParams , # noqa: TCH001 (runtime-checked annotation)
62+ IntersectDirectMessageParams , # noqa: TCH001 (runtime-checked annotation)
6363)
6464from .version import version_string
6565
@@ -111,7 +111,7 @@ def __init__(
111111 self ,
112112 req_id : UUID ,
113113 req_name : str ,
114- request : DirectMessageParams ,
114+ request : IntersectDirectMessageParams ,
115115 response_handler : INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None = None ,
116116 ) -> None :
117117 """Create an external request."""
@@ -223,13 +223,13 @@ def __init__(
223223 self ._external_request_ctr = 0
224224
225225 self ._startup_messages : list [
226- tuple [DirectMessageParams , INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None ]
226+ tuple [IntersectDirectMessageParams , INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None ]
227227 ] = []
228228 self ._resend_startup_messages = True
229229 self ._sent_startup_messages = False
230230
231231 self ._shutdown_messages : list [
232- tuple [DirectMessageParams , INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None ]
232+ tuple [IntersectDirectMessageParams , INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None ]
233233 ] = []
234234
235235 self ._data_plane_manager = DataPlaneManager (self ._hierarchy , config .data_stores )
@@ -449,7 +449,9 @@ def get_blocked_keys(self) -> set[str]:
449449
450450 def add_startup_messages (
451451 self ,
452- messages : list [tuple [DirectMessageParams , INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None ]],
452+ messages : list [
453+ tuple [IntersectDirectMessageParams , INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None ]
454+ ],
453455 ) -> None :
454456 """Add request messages to send out to various microservices when this service starts.
455457
@@ -461,7 +463,9 @@ def add_startup_messages(
461463
462464 def add_shutdown_messages (
463465 self ,
464- messages : list [tuple [DirectMessageParams , INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None ]],
466+ messages : list [
467+ tuple [IntersectDirectMessageParams , INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE | None ]
468+ ],
465469 ) -> None :
466470 """Add request messages to send out to various microservices on shutdown.
467471
@@ -474,7 +478,7 @@ def add_shutdown_messages(
474478 @validate_call (config = ConfigDict (revalidate_instances = 'always' ))
475479 def create_external_request (
476480 self ,
477- request : DirectMessageParams ,
481+ request : IntersectDirectMessageParams ,
478482 response_handler : Union [INTERSECT_SERVICE_RESPONSE_CALLBACK_TYPE , None ] = None , # noqa: UP007 (runtime checked annotation)
479483 ) -> UUID :
480484 """Create an external request that we'll send to a different Service.
@@ -715,7 +719,7 @@ def _handle_client_message(self, message: UserspaceMessage) -> None:
715719 error_msg = f'No external request found for message:\n { message } '
716720 logger .warning (error_msg )
717721
718- def _send_client_message (self , request_id : UUID , params : DirectMessageParams ) -> bool :
722+ def _send_client_message (self , request_id : UUID , params : IntersectDirectMessageParams ) -> bool :
719723 """Send a userspace message."""
720724 # "params" should already be validated at this stage.
721725 request = GENERIC_MESSAGE_SERIALIZER .dump_json (params .payload , warnings = False )
0 commit comments