diff --git a/kbcstorage/base.py b/kbcstorage/base.py index a8d481d..baccb02 100644 --- a/kbcstorage/base.py +++ b/kbcstorage/base.py @@ -9,6 +9,8 @@ .. _Storage API documentation: http://docs.keboola.apiary.io/ """ +import os + from kbcstorage.retry_requests import MAX_RETRIES_DEFAULT, RetryRequests import requests @@ -43,6 +45,7 @@ def __init__(self, root_url, path_component, token, max_requests_retries=MAX_RET path_component.strip('/')) self.token = token self._auth_header = {'X-StorageApi-Token': self.token, + 'X-KBC-RunId': os.environ.get('KBC_RUNID'), 'Accept-Encoding': 'gzip', 'User-Agent': 'Keboola Storage API Python Client'} self.requests = RetryRequests(max_requests_retries)