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