[Core-531] Support on-behalf-of on the service side (#12665)

GitOrigin-RevId: 045cfe1bae70eb9ad00a0936b8a94c98f5cdfbed
descarteslabs · Sep 19, 2024 · e0e37fe · e0e37fe
1 parent 7a017cb
commit e0e37fe
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/descarteslabs/core/client/services/service/service.py b/descarteslabs/core/client/services/service/service.py
@@ -31,7 +31,11 @@
 from descarteslabs.auth import Auth
 from descarteslabs.exceptions import ClientError, ServerError
 
-from ....common.http import Retry, Session
+from ....common.http import (
+    HttpHeaderKeys as CommonHttpHeaderKeys,
+    Retry,
+    Session,
+)
 from ....common.http.authorization import add_bearer
 from ....common.threading.local import ThreadLocalWrapper
 from ...version import __version__
@@ -53,17 +57,17 @@ class HttpRequestMethod:
     TRACE = "TRACE"
 
 
-class HttpHeaderKeys:
+class HttpHeaderKeys(CommonHttpHeaderKeys):
     Accept = "Accept"
     Authorization = "Authorization"
     ClientSession = "X-Client-Session"
     Conda = "X-Conda"
     ContentType = "Content-Type"
     Notebook = "X-Notebook"
+    OnBehalfOf = "X-On-Behalf-Of"
     Platform = "X-Platform"
     Python = "X-Python"
     RequestGroup = "X-Request-Group"
-    RetryAfter = "Retry-After"
     UserAgent = "User-Agent"
 
 

diff --git a/descarteslabs/core/common/http/__init__.py b/descarteslabs/core/common/http/__init__.py
@@ -13,11 +13,12 @@
 # limitations under the License.
 
 from .retry import Retry
-from .session import HTTPAdapter, ProxyAuthentication, Session
+from .session import HTTPAdapter, HttpHeaderKeys, ProxyAuthentication, Session
 
 __all__ = [
     "ProxyAuthentication",
     "Session",
     "HTTPAdapter",
+    "HttpHeaderKeys",
     "Retry",
 ]