@@ -99,6 +99,11 @@ def execute_skills(
9999 tool_user_session_id = agent_name + "_" + user_id + "_" + session_id
100100 logger .debug (f"tool_user_session_id: { tool_user_session_id } " )
101101
102+ cloud_provider = getenv ("CLOUD_PROVIDER" , "" )
103+ scheme = getenv ("AGENTKIT_TOOL_SCHEME" , "https" , allow_false_values = True ).lower ()
104+ if scheme not in {"http" , "https" }:
105+ scheme = "https"
106+
102107 logger .debug (
103108 f"Execute skills in session_id={ session_id } , tool_id={ tool_id } , host={ host } , service={ service } , region={ region } , timeout={ timeout } "
104109 )
@@ -126,22 +131,24 @@ def execute_skills(
126131
127132 cmd = ["python" , "agent.py" , workflow_prompt ]
128133
129- res = ve_request (
130- request_body = {},
131- action = "GetCallerIdentity" ,
132- ak = ak ,
133- sk = sk ,
134- service = "sts" ,
135- version = "2018-01-01" ,
136- region = region ,
137- host = "sts.volcengineapi.com" ,
138- header = header ,
139- )
140- try :
141- account_id = res ["Result" ]["AccountId" ]
142- except KeyError as e :
143- logger .error (f"Error occurred while getting account id: { e } , response is { res } " )
144- return res
134+ account_id = ""
135+ if cloud_provider != "vestack" :
136+ res = ve_request (
137+ request_body = {},
138+ action = "GetCallerIdentity" ,
139+ ak = ak ,
140+ sk = sk ,
141+ service = "sts" ,
142+ version = "2018-01-01" ,
143+ region = region ,
144+ host = "sts.volcengineapi.com" ,
145+ header = header ,
146+ )
147+ try :
148+ account_id = res ["Result" ]["AccountId" ]
149+ except KeyError as e :
150+ logger .error (f"Error occurred while getting account id: { e } , response is { res } " )
151+ return res
145152
146153 skill_space_id = os .getenv ("SKILL_SPACE_ID" , "" )
147154 if not skill_space_id :
@@ -236,6 +243,7 @@ def execute_skills(
236243 region = region ,
237244 host = host ,
238245 header = header ,
246+ scheme = scheme
239247 )
240248 logger .debug (f"Invoke run code response: { res } " )
241249
0 commit comments