aws-lambda-powertools 3.35.0__tar.gz → 3.36.0__tar.gz

aws_lambda_powertools-3.36.0/.gitignore

@@ -0,0 +1,326 @@
+# Created by https://www.gitignore.io/api/osx,linux,python,windows,pycharm,visualstudiocode
+# Edit at https://www.gitignore.io/?templates=osx,linux,python,windows,pycharm,visualstudiocode
+
+### Linux ###
+*~
+
+# temporary files which can be created if a process still has a handle open of a deleted file
+.fuse_hidden*
+
+# KDE directory preferences
+.directory
+
+# Linux trash folder which might appear on any partition or disk
+.Trash-*
+
+# .nfs files are created when an open file is removed but is still being accessed
+.nfs*
+
+### OSX ###
+# General
+.DS_Store
+.AppleDouble
+.LSOverride
+
+# Icon must end with two \r
+Icon
+
+# Thumbnails
+._*
+
+# Files that might appear in the root of a volume
+.DocumentRevisions-V100
+.fseventsd
+.Spotlight-V100
+.TemporaryItems
+.Trashes
+.VolumeIcon.icns
+.com.apple.timemachine.donotpresent
+
+# Directories potentially created on remote AFP share
+.AppleDB
+.AppleDesktop
+Network Trash Folder
+Temporary Items
+.apdisk
+
+### PyCharm ###
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and WebStorm
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff
+.idea/**/workspace.xml
+.idea/**/tasks.xml
+.idea/**/usage.statistics.xml
+.idea/**/dictionaries
+.idea/**/shelf
+
+# Generated files
+.idea/**/contentModel.xml
+
+# Sensitive or high-churn files
+.idea/**/dataSources/
+.idea/**/dataSources.ids
+.idea/**/dataSources.local.xml
+.idea/**/sqlDataSources.xml
+.idea/**/dynamic.xml
+.idea/**/uiDesigner.xml
+.idea/**/dbnavigator.xml
+
+# Gradle
+.idea/**/gradle.xml
+.idea/**/libraries
+
+# Gradle and Maven with auto-import
+# When using Gradle or Maven with auto-import, you should exclude module files,
+# since they will be recreated, and may cause churn.  Uncomment if using
+# auto-import.
+# .idea/modules.xml
+# .idea/*.iml
+# .idea/modules
+# *.iml
+# *.ipr
+
+# CMake
+cmake-build-*/
+
+# Mongo Explorer plugin
+.idea/**/mongoSettings.xml
+
+# File-based project format
+*.iws
+
+# IntelliJ
+out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Cursive Clojure plugin
+.idea/replstate.xml
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties
+
+# Editor-based Rest Client
+.idea/httpRequests
+
+# Android studio 3.1+ serialized cache file
+.idea/caches/build_file_checksums.ser
+
+### PyCharm Patch ###
+# Comment Reason: https://github.com/joeblau/gitignore.io/issues/186#issuecomment-215987721
+
+# *.iml
+# modules.xml
+# .idea/misc.xml
+# *.ipr
+
+# Sonarlint plugin
+.idea/sonarlint
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+### VisualStudioCode ###
+.vscode
+.vscode/*
+!.vscode/tasks.json
+!.vscode/launch.json
+!.vscode/extensions.json
+
+### VisualStudioCode Patch ###
+# Ignore all local history of files
+.history
+
+### Windows ###
+# Windows thumbnail cache files
+Thumbs.db
+Thumbs.db:encryptable
+ehthumbs.db
+ehthumbs_vista.db
+
+# Dump file
+*.stackdump
+
+# Folder config file
+[Dd]esktop.ini
+
+# Recycle Bin used on file shares
+$RECYCLE.BIN/
+
+# Windows Installer files
+*.cab
+*.msi
+*.msix
+*.msm
+*.msp
+
+# Windows shortcuts
+*.lnk
+
+# End of https://www.gitignore.io/api/osx,linux,python,windows,pycharm,visualstudiocode
+
+# Misc
+test_report
+wheelhouse
+/.idea/*
+*.html
+TMP_CHANGELOG.md
+
+# Docs files
+docs/.cache/
+docs/public
+node_modules
+api/
+site/
+!404.html
+!docs/overrides/*.html
+
+# CDK
+.cdk
+
+!.github/workflows/lib
+examples/**/sam/.aws-sam
+
+cdk.out
+# NOTE: different accounts will be used for E2E thus creating unnecessary git clutter
+cdk.context.json
+
+# vim
+*.swp
+
+# LLMs
+.claude
+.amazonq
+.kiro
+.github/instructions

{aws_lambda_powertools-3.35.0 → aws_lambda_powertools-3.36.0}/PKG-INFO

@@ -1,11 +1,15 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.5
 Name: aws_lambda_powertools
-Version: 3.35.0
+Version: 3.36.0
 Summary: Powertools for AWS Lambda (Python) is a developer toolkit to implement Serverless best practices and increase developer velocity.
-License: MIT
-Keywords: aws_lambda_powertools,aws,tracing,logging,lambda,powertools,feature_flags,idempotency,middleware
+Project-URL: Repository, https://github.com/aws-powertools/powertools-lambda-python
+Project-URL: Documentation, https://docs.powertools.aws.dev/lambda/python/
+Project-URL: Issue tracker, https://github.com/aws-powertools/powertools-lambda-python/issues
+Project-URL: Releases, https://github.com/aws-powertools/powertools-lambda-python/releases
 Author: Amazon Web Services
-Requires-Python: >=3.10,<4.0.0
+License: MIT
+License-File: LICENSE
+Keywords: aws,aws_lambda_powertools,feature_flags,idempotency,lambda,logging,middleware,powertools,tracing
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
@@ -17,35 +21,46 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
+Requires-Python: <4.0.0,>=3.10
+Requires-Dist: jmespath<2.0.0,>=1.0.1
+Requires-Dist: typing-extensions<5.0.0,>=4.11.0
 Provides-Extra: all
+Requires-Dist: aws-encryption-sdk<5.0.0,>=4.0.5; extra == 'all'
+Requires-Dist: aws-xray-sdk<3.0.0,>=2.8.0; extra == 'all'
+Requires-Dist: cryptography<51.0.0,>=50.0.1; extra == 'all'
+Requires-Dist: fastjsonschema<3.0.0,>=2.14.5; extra == 'all'
+Requires-Dist: jsonpath-ng<2.0.0,>=1.6.0; extra == 'all'
+Requires-Dist: pydantic-settings<3.0.0,>=2.6.1; extra == 'all'
+Requires-Dist: pydantic<3.0.0,>=2.4.0; extra == 'all'
+Requires-Dist: pyjwt<3.0.0,>=2.14.0; extra == 'all'
+Requires-Dist: urllib3<3.0.0,>=2.7.0; extra == 'all'
 Provides-Extra: aws-sdk
+Requires-Dist: boto3<2.0.0,>=1.34.32; extra == 'aws-sdk'
 Provides-Extra: datadog
+Requires-Dist: datadog-lambda<9.0.0,>=8.114.0; extra == 'datadog'
 Provides-Extra: datamasking
+Requires-Dist: aws-encryption-sdk<5.0.0,>=4.0.5; extra == 'datamasking'
+Requires-Dist: jsonpath-ng<2.0.0,>=1.6.0; extra == 'datamasking'
+Provides-Extra: jwt
+Requires-Dist: cryptography<51.0.0,>=50.0.1; extra == 'jwt'
+Requires-Dist: pyjwt<3.0.0,>=2.14.0; extra == 'jwt'
+Requires-Dist: urllib3<3.0.0,>=2.7.0; extra == 'jwt'
 Provides-Extra: kafka-consumer-avro
+Requires-Dist: avro<2.0.0,>=1.12.0; extra == 'kafka-consumer-avro'
 Provides-Extra: kafka-consumer-protobuf
+Requires-Dist: protobuf<8.0.0,>=6.30.2; extra == 'kafka-consumer-protobuf'
+Provides-Extra: oauth2
+Requires-Dist: urllib3<3.0.0,>=2.7.0; extra == 'oauth2'
 Provides-Extra: parser
+Requires-Dist: pydantic<3.0.0,>=2.4.0; extra == 'parser'
 Provides-Extra: redis
+Requires-Dist: redis<9.0,>=4.4; extra == 'redis'
 Provides-Extra: tracer
+Requires-Dist: aws-xray-sdk<3.0.0,>=2.8.0; extra == 'tracer'
 Provides-Extra: validation
+Requires-Dist: fastjsonschema<3.0.0,>=2.14.5; extra == 'validation'
 Provides-Extra: valkey
-Requires-Dist: avro (>=1.12.0,<2.0.0) ; extra == "kafka-consumer-avro"
-Requires-Dist: aws-encryption-sdk (>=4.0.5,<5.0.0) ; extra == "all" or extra == "datamasking"
-Requires-Dist: aws-xray-sdk (>=2.8.0,<3.0.0) ; extra == "tracer" or extra == "all"
-Requires-Dist: boto3 (>=1.34.32,<2.0.0) ; extra == "aws-sdk"
-Requires-Dist: datadog-lambda (>=8.114.0,<9.0.0) ; extra == "datadog"
-Requires-Dist: fastjsonschema (>=2.14.5,<3.0.0) ; extra == "validation" or extra == "all"
-Requires-Dist: jmespath (>=1.0.1,<2.0.0)
-Requires-Dist: jsonpath-ng (>=1.6.0,<2.0.0) ; extra == "all" or extra == "datamasking"
-Requires-Dist: protobuf (>=6.30.2,<8.0.0) ; extra == "kafka-consumer-protobuf"
-Requires-Dist: pydantic (>=2.4.0,<3.0.0) ; extra == "parser" or extra == "all"
-Requires-Dist: pydantic-settings (>=2.6.1,<3.0.0) ; extra == "all"
-Requires-Dist: redis (>=4.4,<9.0) ; extra == "redis"
-Requires-Dist: typing-extensions (>=4.11.0,<5.0.0)
-Requires-Dist: valkey-glide (>=1.3.5,<3.0) ; extra == "valkey"
-Project-URL: Documentation, https://docs.powertools.aws.dev/lambda/python/
-Project-URL: Issue tracker, https://github.com/aws-powertools/powertools-lambda-python/issues
-Project-URL: Repository, https://github.com/aws-powertools/powertools-lambda-python
-Project-URL: Releases, https://github.com/aws-powertools/powertools-lambda-python/releases
+Requires-Dist: valkey-glide<3.0,>=1.3.5; extra == 'valkey'
 Description-Content-Type: text/markdown
 
 <!-- markdownlint-disable MD013 MD041 MD043  -->
@@ -161,4 +176,3 @@ If you think you’ve found a potential security issue, please do not post it in
 ## License
 
 This library is licensed under the MIT-0 License. See the LICENSE file.
-

{aws_lambda_powertools-3.35.0 → aws_lambda_powertools-3.36.0}/aws_lambda_powertools/event_handler/api_gateway.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import base64
 import json
 import logging
+import os
 import re
 import traceback
 import warnings
@@ -18,7 +19,7 @@ from typing_extensions import override
 
 from aws_lambda_powertools.event_handler import content_types
 from aws_lambda_powertools.event_handler.exception_handling import ExceptionHandlerManager
-from aws_lambda_powertools.event_handler.exceptions import NotFoundError, ServiceError
+from aws_lambda_powertools.event_handler.exceptions import NotFoundError, ResponseSizeExceededError, ServiceError
 from aws_lambda_powertools.event_handler.openapi.config import OpenAPIConfig
 from aws_lambda_powertools.event_handler.openapi.constants import (
     DEFAULT_API_VERSION,
@@ -72,6 +73,7 @@ _UNSAFE_URI = r"%<> \[\]{}|^"
 _NAMED_GROUP_BOUNDARY_PATTERN = rf"(?P\1[{_SAFE_URI}{_UNSAFE_URI}\\w]+)"
 _ROUTE_REGEX = "^{}$"
 _JSON_DUMP_CALL = partial(json.dumps, separators=(",", ":"), cls=Encoder)
+_ALB_MAX_RESPONSE_SIZE = 1024 * 1024
 
 ResponseEventT = TypeVar("ResponseEventT", bound=BaseProxyEvent)
 ResponseT = TypeVar("ResponseT")
@@ -810,6 +812,11 @@ class ResponseBuilder(Generic[ResponseEventT]):
 
     def _compress(self):
         """Compress the response body, but only if `Accept-Encoding` headers includes gzip."""
+        # A response without a body, such as 204 No Content or 304 Not Modified, has nothing to compress
+        if self.response.body is None:
+            logger.debug("Response has no body; skipping compression")
+            return
+
         self.response.headers["Content-Encoding"] = "gzip"
         if isinstance(self.response.body, str):
             logger.debug("Converting string response to bytes before compressing it")
@@ -2553,7 +2560,7 @@ class ApiGatewayResolver(BaseRouter):
         BaseRouter.current_event = self._to_proxy_event(cast(dict, event))
         BaseRouter.lambda_context = context
 
-        response = self._resolve().build(self.current_event, self._cors)
+        response = self._build_response(self._resolve())
 
         # Debug print Processed Middlewares
         if self._debug:
@@ -2614,7 +2621,7 @@ class ApiGatewayResolver(BaseRouter):
         BaseRouter.current_event = self._to_proxy_event(cast(dict, event))
         BaseRouter.lambda_context = context
 
-        response = (await self._resolve_async()).build(self.current_event, self._cors)
+        response = self._build_response(await self._resolve_async())
 
         if self._debug:
             print("\nProcessed Middlewares:")
@@ -2626,6 +2633,10 @@ class ApiGatewayResolver(BaseRouter):
 
         return response
 
+    def _build_response(self, response_builder: ResponseBuilder) -> dict[str, Any]:
+        """Build the final integration response after route and middleware processing."""
+        return response_builder.build(self.current_event, self._cors)
+
     async def _resolve_async(self) -> ResponseBuilder:
         method = self.current_event.http_method.upper()
         path = self._remove_prefix(self.current_event.path)
@@ -3337,6 +3348,7 @@ class ALBResolver(ApiGatewayResolver):
         response_validation_error_http_code: HTTPStatus | int | None = None,
         json_body_deserializer: Callable[[str], dict] | None = None,
         decode_query_parameters: bool = False,
+        enable_response_size_validation: bool = False,
     ):
         """Amazon Application Load Balancer (ALB) resolver
 
@@ -3363,6 +3375,10 @@ class ALBResolver(ApiGatewayResolver):
             by default json.loads when integrating with EventSource data class
         decode_query_parameters: bool | None
             Enables URL-decoding of query parameters (both keys and values), by default False.
+        enable_response_size_validation: bool
+            Validate the complete serialized response against ALB's 1 MB limit, by default False.
+            Raises ResponseSizeExceededError when the limit is exceeded. Registered exception handlers
+            can return a smaller response, which is also validated.
         """
         super().__init__(
             cors=cors,
@@ -3374,6 +3390,42 @@ class ALBResolver(ApiGatewayResolver):
             json_body_deserializer=json_body_deserializer,
         )
         self.decode_query_parameters = decode_query_parameters
+        self._enable_response_size_validation = enable_response_size_validation
+
+    @override
+    def _build_response(self, response_builder: ResponseBuilder) -> dict[str, Any]:
+        response = super()._build_response(response_builder)
+        if not self._enable_response_size_validation:
+            return response
+
+        try:
+            self._validate_response_size(response)
+        except ResponseSizeExceededError as exc:
+            try:
+                # Resolved responses retain their route, including not-found and preflight responses.
+                handled_response = self._call_exception_handler(exc, cast(Route, response_builder.route))
+                if handled_response is None:
+                    raise
+
+                handled_response.response = cast(Response, self._to_response(handled_response.response))
+                response = super()._build_response(handled_response)
+                # Validate once more without invoking an exception handler recursively.
+                self._validate_response_size(response)
+            except Exception:
+                self.clear_context()
+                raise
+
+        return response
+
+    @staticmethod
+    def _validate_response_size(response: dict[str, Any]) -> None:
+        # Match the Lambda Python runtime's outer JSON encoding, including its default separators.
+        # Managed Python 3.12+ runtimes return Unicode directly; older/custom runtimes escape it.
+        runtime = re.fullmatch(r"AWS_Lambda_python3\.(\d+)", os.getenv("AWS_EXECUTION_ENV", ""))
+        ensure_ascii = runtime is None or int(runtime[1]) < 12
+        actual_size = len(json.dumps(response, ensure_ascii=ensure_ascii).encode("utf-8"))
+        if actual_size > _ALB_MAX_RESPONSE_SIZE:
+            raise ResponseSizeExceededError(actual_size=actual_size, max_size=_ALB_MAX_RESPONSE_SIZE)
 
     def _get_base_path(self) -> str:
         # ALB doesn't have a stage variable, so we just return an empty string

{aws_lambda_powertools-3.35.0 → aws_lambda_powertools-3.36.0}/aws_lambda_powertools/event_handler/exceptions.py

@@ -3,6 +3,25 @@ from __future__ import annotations
 from http import HTTPStatus
 
 
+class ResponseSizeExceededError(Exception):
+    """The serialized Lambda response exceeds the integration's size limit."""
+
+    def __init__(self, *, actual_size: int, max_size: int):
+        """
+        Parameters
+        ----------
+        actual_size: int
+            Size of the complete serialized response in bytes.
+        max_size: int
+            Maximum allowed serialized response size in bytes.
+        """
+        self.actual_size = actual_size
+        self.max_size = max_size
+        super().__init__(
+            f"Serialized response size ({actual_size} bytes) exceeds the maximum allowed size ({max_size} bytes)",
+        )
+
+
 class ServiceError(Exception):
     """Powertools class HTTP Service Error"""
 

{aws_lambda_powertools-3.35.0 → aws_lambda_powertools-3.36.0}/aws_lambda_powertools/event_handler/openapi/encoders.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import dataclasses
-import datetime
+import datetime as dt
 from collections import defaultdict, deque
 from decimal import Decimal
 from enum import Enum
@@ -306,7 +306,7 @@ def _dump_other(
     )
 
 
-def iso_format(o: datetime.date | datetime.time) -> str:
+def iso_format(o: dt.date | dt.time) -> str:
     """
     ISO format for date and time
     """
@@ -341,10 +341,10 @@ ENCODERS_BY_TYPE: dict[type[Any], Callable[[Any], Any]] = {
     str: lambda o: o,
     type(None): lambda o: o,
     bytes: lambda o: o.decode(),
-    datetime.date: iso_format,
-    datetime.datetime: iso_format,
-    datetime.time: iso_format,
-    datetime.timedelta: lambda td: td.total_seconds(),
+    dt.date: iso_format,
+    dt.datetime: iso_format,
+    dt.time: iso_format,
+    dt.timedelta: lambda td: td.total_seconds(),
     Decimal: decimal_encoder,
     Enum: lambda o: o.value,
     PurePath: str,

{aws_lambda_powertools-3.35.0 → aws_lambda_powertools-3.36.0}/aws_lambda_powertools/event_handler/openapi/params.py

@@ -1059,7 +1059,7 @@ def get_field_info_annotated_type(annotation, value, is_path_param: bool) -> tup
     # Reconstruct type_annotation with non-FieldInfo metadata if present
     # This ensures constraints like Interval are preserved
     if other_metadata and not preserve_full_annotation:
-        type_annotation = Annotated[(type_annotation, *other_metadata)]
+        type_annotation = Annotated[(type_annotation,) + tuple(other_metadata)]
 
     # Process the annotation if it exists
     field_info: FieldInfo | None = None

{aws_lambda_powertools-3.35.0 → aws_lambda_powertools-3.36.0}/aws_lambda_powertools/metrics/base.py

@@ -6,7 +6,7 @@ Metrics utility
 
 from __future__ import annotations
 
-import datetime
+import datetime as dt
 import functools
 import json
 import logging
@@ -250,7 +250,7 @@ class MetricManager:
 
         return {
             "_aws": {
-                "Timestamp": self.timestamp or int(datetime.datetime.now().timestamp() * 1000),  # epoch
+                "Timestamp": self.timestamp or int(dt.datetime.now().timestamp() * 1000),  # epoch
                 "CloudWatchMetrics": [
                     {
                         "Namespace": self.namespace,  # "test_namespace"
@@ -322,7 +322,7 @@ class MetricManager:
         else:
             self.metadata_set[str(key)] = value
 
-    def set_timestamp(self, timestamp: int | datetime.datetime):
+    def set_timestamp(self, timestamp: int | dt.datetime):
         """
         Set the timestamp for the metric.
 

{aws_lambda_powertools-3.35.0 → aws_lambda_powertools-3.36.0}/aws_lambda_powertools/metrics/metrics.py

@@ -6,7 +6,7 @@ from typing import TYPE_CHECKING, Any
 from aws_lambda_powertools.metrics.provider.cloudwatch_emf.cloudwatch import AmazonCloudWatchEMFProvider
 
 if TYPE_CHECKING:
-    import datetime
+    import datetime as dt
     from collections.abc import Callable
 
     from aws_lambda_powertools.metrics.base import MetricResolution, MetricUnit
@@ -96,7 +96,12 @@ class Metrics:
         self.default_dimensions = self._default_dimensions
         self.dimension_set = self._dimensions
 
-        self.dimension_set.update(**self._default_dimensions)
+        self.dimension_set.update(
+            **{
+                name: value if isinstance(value, str) else str(value)
+                for name, value in self._default_dimensions.items()
+            },
+        )
 
         if provider is None:
             self.provider = AmazonCloudWatchEMFProvider(
@@ -141,7 +146,7 @@ class Metrics:
     def add_metadata(self, key: str, value: Any) -> None:
         self.provider.add_metadata(key=key, value=value)
 
-    def set_timestamp(self, timestamp: int | datetime.datetime):
+    def set_timestamp(self, timestamp: int | dt.datetime):
         """
         Set the timestamp for the metric.