66import typing
77
88import httpx
9- from .types .project_environment import ProjectEnvironment
9+ from ._ . types .project_environment import ProjectEnvironment
1010from .core .api_error import ApiError
1111from .core .client_wrapper import AsyncClientWrapper , SyncClientWrapper
1212from .core .oauth_token_provider import AsyncOAuthTokenProvider , OAuthTokenProvider
@@ -99,6 +99,7 @@ def __init__(
9999 environment : PipedreamEnvironment = PipedreamEnvironment .PROD ,
100100 project_id : str ,
101101 project_environment : typing .Optional [ProjectEnvironment ] = os .getenv ("PIPEDREAM_PROJECT_ENVIRONMENT" ),
102+ headers : typing .Optional [typing .Dict [str , str ]] = None ,
102103 timeout : typing .Optional [float ] = None ,
103104 follow_redirects : typing .Optional [bool ] = True ,
104105 httpx_client : typing .Optional [httpx .Client ] = None ,
@@ -113,6 +114,7 @@ def __init__(
113114 environment : PipedreamEnvironment = PipedreamEnvironment .PROD ,
114115 project_id : str ,
115116 project_environment : typing .Optional [ProjectEnvironment ] = os .getenv ("PIPEDREAM_PROJECT_ENVIRONMENT" ),
117+ headers : typing .Optional [typing .Dict [str , str ]] = None ,
116118 timeout : typing .Optional [float ] = None ,
117119 follow_redirects : typing .Optional [bool ] = True ,
118120 httpx_client : typing .Optional [httpx .Client ] = None ,
@@ -125,6 +127,7 @@ def __init__(
125127 environment : PipedreamEnvironment = PipedreamEnvironment .PROD ,
126128 project_id : str ,
127129 project_environment : typing .Optional [ProjectEnvironment ] = os .getenv ("PIPEDREAM_PROJECT_ENVIRONMENT" ),
130+ headers : typing .Optional [typing .Dict [str , str ]] = None ,
128131 client_id : typing .Optional [str ] = os .getenv ("PIPEDREAM_CLIENT_ID" ),
129132 client_secret : typing .Optional [str ] = os .getenv ("PIPEDREAM_CLIENT_SECRET" ),
130133 token : typing .Optional [typing .Callable [[], str ]] = None ,
@@ -141,6 +144,7 @@ def __init__(
141144 base_url = _get_base_url (base_url = base_url , environment = environment ),
142145 project_id = project_id ,
143146 project_environment = project_environment ,
147+ headers = headers ,
144148 httpx_client = httpx_client
145149 if httpx_client is not None
146150 else httpx .Client (timeout = _defaulted_timeout , follow_redirects = follow_redirects )
@@ -157,6 +161,7 @@ def __init__(
157161 base_url = _get_base_url (base_url = base_url , environment = environment ),
158162 project_id = project_id ,
159163 project_environment = project_environment ,
164+ headers = headers ,
160165 httpx_client = httpx .Client (timeout = _defaulted_timeout , follow_redirects = follow_redirects )
161166 if follow_redirects is not None
162167 else httpx .Client (timeout = _defaulted_timeout ),
@@ -167,6 +172,7 @@ def __init__(
167172 base_url = _get_base_url (base_url = base_url , environment = environment ),
168173 project_id = project_id ,
169174 project_environment = project_environment ,
175+ headers = headers ,
170176 token = _token_getter_override if _token_getter_override is not None else oauth_token_provider .get_token ,
171177 httpx_client = httpx_client
172178 if httpx_client is not None
@@ -369,6 +375,7 @@ def __init__(
369375 environment : PipedreamEnvironment = PipedreamEnvironment .PROD ,
370376 project_id : str ,
371377 project_environment : typing .Optional [ProjectEnvironment ] = os .getenv ("PIPEDREAM_PROJECT_ENVIRONMENT" ),
378+ headers : typing .Optional [typing .Dict [str , str ]] = None ,
372379 timeout : typing .Optional [float ] = None ,
373380 follow_redirects : typing .Optional [bool ] = True ,
374381 httpx_client : typing .Optional [httpx .AsyncClient ] = None ,
@@ -383,6 +390,7 @@ def __init__(
383390 environment : PipedreamEnvironment = PipedreamEnvironment .PROD ,
384391 project_id : str ,
385392 project_environment : typing .Optional [ProjectEnvironment ] = os .getenv ("PIPEDREAM_PROJECT_ENVIRONMENT" ),
393+ headers : typing .Optional [typing .Dict [str , str ]] = None ,
386394 timeout : typing .Optional [float ] = None ,
387395 follow_redirects : typing .Optional [bool ] = True ,
388396 httpx_client : typing .Optional [httpx .AsyncClient ] = None ,
@@ -395,6 +403,7 @@ def __init__(
395403 environment : PipedreamEnvironment = PipedreamEnvironment .PROD ,
396404 project_id : str ,
397405 project_environment : typing .Optional [ProjectEnvironment ] = os .getenv ("PIPEDREAM_PROJECT_ENVIRONMENT" ),
406+ headers : typing .Optional [typing .Dict [str , str ]] = None ,
398407 client_id : typing .Optional [str ] = os .getenv ("PIPEDREAM_CLIENT_ID" ),
399408 client_secret : typing .Optional [str ] = os .getenv ("PIPEDREAM_CLIENT_SECRET" ),
400409 token : typing .Optional [typing .Callable [[], str ]] = None ,
@@ -411,6 +420,7 @@ def __init__(
411420 base_url = _get_base_url (base_url = base_url , environment = environment ),
412421 project_id = project_id ,
413422 project_environment = project_environment ,
423+ headers = headers ,
414424 httpx_client = httpx_client
415425 if httpx_client is not None
416426 else httpx .AsyncClient (timeout = _defaulted_timeout , follow_redirects = follow_redirects )
@@ -427,6 +437,7 @@ def __init__(
427437 base_url = _get_base_url (base_url = base_url , environment = environment ),
428438 project_id = project_id ,
429439 project_environment = project_environment ,
440+ headers = headers ,
430441 httpx_client = httpx .AsyncClient (timeout = _defaulted_timeout , follow_redirects = follow_redirects )
431442 if follow_redirects is not None
432443 else httpx .AsyncClient (timeout = _defaulted_timeout ),
@@ -437,6 +448,7 @@ def __init__(
437448 base_url = _get_base_url (base_url = base_url , environment = environment ),
438449 project_id = project_id ,
439450 project_environment = project_environment ,
451+ headers = headers ,
440452 token = _token_getter_override ,
441453 async_token = oauth_token_provider .get_token ,
442454 httpx_client = httpx_client
0 commit comments