File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed
Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments