extract method

Author: guglielmo-san

src/a2a/server/routes/jsonrpc_dispatcher.py

@@ -398,11 +398,12 @@ async def _process_streaming_request(
             An `AsyncGenerator` object to stream results to the client.
         """
         stream: AsyncGenerator | None = None
-        if context.state.get('method') == 'SendStreamingMessage':
+        method = context.state.get('method')
+        if method == 'SendStreamingMessage':
             stream = self.request_handler.on_message_send_stream(
                 request_obj, context
             )
-        elif context.state.get('method') == 'SubscribeToTask':
+        elif method == 'SubscribeToTask':
             stream = self.request_handler.on_subscribe_to_task(
                 request_obj, context
             )
@@ -549,7 +550,8 @@ async def _process_non_streaming_request(  # noqa: PLR0911
         Returns:
             A dict containing the result or error.
         """
-        match context.state.get('method'):
+        method = context.state.get('method')
+        match method:
             case 'SendMessage':
                 return await self._handle_send_message(request_obj, context)
             case 'CancelTask':
@@ -580,7 +582,6 @@ async def _process_non_streaming_request(  # noqa: PLR0911
                     request_obj, context
                 )
             case _:
-                method = context.state.get('method')
                 logger.error('Unhandled method: %s', method)
                 raise UnsupportedOperationError(
                     message=f'Method {method} is not supported.'