diff --git a/py/src/braintrust/logger.py b/py/src/braintrust/logger.py index 7c89d51a..9b8a2c0a 100644 --- a/py/src/braintrust/logger.py +++ b/py/src/braintrust/logger.py @@ -109,6 +109,7 @@ merge_dicts, response_raise_for_status, ) +from .version import VERSION from .xact_ids import prettify_xact @@ -828,6 +829,9 @@ def _set_adapter(self, adapter: HTTPAdapter | None) -> None: def _reset(self, **retry_kwargs: Any) -> None: self.session = requests.Session() + # Identifies SDK-originated traffic so the backend can attribute object + # creation to the Python SDK. + self.session.headers.update({"User-Agent": f"braintrust-python/{VERSION}"}) adapter = self.adapter if adapter is None: