code: def test_query_timeout_not_retried(mixed_gateway):
      gateway_client = MlflowGatewayClient(
          gateway_uri=mixed_gateway.url
      )
  
      data = {"prompt": "Test", "temperature": 0.4}
      route = "completions"
  
      with mock.patch(
          "mlflow.gateway.constants.MLFLOW_GATEWAY_CLIENT_QUERY_TIMEOUT_SECONDS",
          new=1,
      ), mock.patch(
          "requests.Session.request", side_effect=Timeout
      ) as mocked_request:
          with pytest.raises(
              MlflowException,
              match="The provider has timed out while generating",
          ):
              gateway_client.query(route=route, data=data)
  
          mocked_request.assert_called_once()
  
imports: from unittest import mock
  import pytest
  from requests.exceptions import HTTPError, Timeout
  import mlflow.gateway.utils
  from mlflow.environment_variables import MLFLOW_GATEWAY_URI
  from mlflow.exceptions import InvalidUrlException, MlflowException
  from mlflow.gateway import MlflowGatewayClient, set_gateway_uri
  from mlflow.gateway.config import Route
  from mlflow.gateway.constants import MLFLOW_GATEWAY_SEARCH_ROUTES_PAGE_SIZE
  from mlflow.gateway.utils import resolve_route_url
  from tests.gateway.tools import Gateway, save_yaml
label_desc: Use image node_id 29 for calling a global function with example usage: test_query_timeout_not_retried(mixed_gateway) without return types