From c3741f69de7f4f0d32384def043a8ac8d27c9d26 Mon Sep 17 00:00:00 2001 From: Abdel Jaidi Date: Mon, 15 Mar 2021 20:25:27 +0000 Subject: [PATCH] Minor - Adding userAgent to botocore config --- awswrangler/_utils.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/awswrangler/_utils.py b/awswrangler/_utils.py index 3a8242557..560a3cb10 100644 --- a/awswrangler/_utils.py +++ b/awswrangler/_utils.py @@ -16,6 +16,7 @@ import pandas as pd from awswrangler import _config, exceptions +from awswrangler.__metadata__ import __version__ from awswrangler._config import apply_configs _logger: logging.Logger = logging.getLogger(__name__) @@ -68,7 +69,12 @@ def default_botocore_config() -> botocore.config.Config: mode: Optional[str] = os.getenv("AWS_RETRY_MODE") if mode: retries_config["mode"] = mode - return botocore.config.Config(retries=retries_config, connect_timeout=10, max_pool_connections=10) + return botocore.config.Config( + retries=retries_config, + connect_timeout=10, + max_pool_connections=10, + user_agent_extra=f"awswrangler/{__version__}", + ) def _get_endpoint_url(service_name: str) -> Optional[str]: