Skip to content

Commit acf1113

Browse files
committed
Correct usage of JWTInternalServiceAuthClientTokenManager
1 parent 3e6d4da commit acf1113

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/spark_history_mcp/common/vault.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@ def __init__(self, audience: str, datacenter: str):
1111
)
1212

1313
if os.getenv("POD_NAME"):
14-
token_manager_class = JWTInternalServiceAuthClientTokenManager
14+
self.token_manager = JWTInternalServiceAuthClientTokenManager(issuer="sycamore")
1515
else:
16-
token_manager_class = JWTDDToolAuthClientTokenManager
16+
self.token_manager = JWTDDToolAuthClientTokenManager.instance(
17+
name=self.audience, datacenter=datacenter
18+
)
19+
1720

18-
self.token_manager = token_manager_class.instance(
19-
name=self.audience, datacenter=datacenter
20-
)
2121

2222
def get_token(self) -> str:
2323
try:

0 commit comments

Comments
 (0)