Skip to content

Commit 5249238

Browse files
feat(ark-runtime): add bot sts token for AsyncArk
1 parent 21b75df commit 5249238

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

volcenginesdkarkruntime/_client.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -245,6 +245,13 @@ def _get_endpoint_sts_token(self, endpoint_id: str):
245245
self._sts_token_manager = StsTokenManager(self.ak, self.sk, self.region)
246246
return self._sts_token_manager.get(endpoint_id)
247247

248+
def _get_bot_sts_token(self, bot_id: str):
249+
if self._sts_token_manager is None:
250+
if self.ak is None or self.sk is None:
251+
raise ArkAPIError("must set ak and sk before get endpoint token.")
252+
self._sts_token_manager = StsTokenManager(self.ak, self.sk, self.region)
253+
return self._sts_token_manager.get(bot_id, resource_type="bot")
254+
248255
def _get_endpoint_certificate(self, endpoint_id: str) -> key_agreement_client:
249256
if self._certificate_manager is None:
250257
cert_path = os.environ.get("E2E_CERTIFICATE_PATH")

0 commit comments

Comments
 (0)