Skip to content

Commit 02aff71

Browse files
authored
Merge pull request #1 from suwenbo-byte/feat/adapter_vestack
feat: execute-skills 脚本兼容混合云场景,新增传入变量云厂商和scheme
2 parents e63a63e + de431cf commit 02aff71

1 file changed

Lines changed: 24 additions & 16 deletions

File tree

veadk/tools/builtin_tools/execute_skills.py

Lines changed: 24 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)