We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 96efdbc commit ea08170Copy full SHA for ea08170
1 file changed
veadk/tools/builtin_tools/execute_skills.py
@@ -91,8 +91,24 @@ def execute_skills(
91
cmd.extend(["--skills"] + skills)
92
93
# TODO: remove after agentkit supports custom environment variables setting
94
+ res = ve_request(
95
+ request_body={},
96
+ action="GetCallerIdentity",
97
+ ak=ak,
98
+ sk=sk,
99
+ service="sts",
100
+ version="2018-01-01",
101
+ region=region,
102
+ host="sts.volcengineapi.com",
103
+ )
104
+ try:
105
+ account_id = res["Result"]["AccountId"]
106
+ except KeyError as e:
107
+ logger.error(f"Error occurred while getting account id: {e}, response is {res}")
108
+ return res
109
+
110
env_vars = {
- "TOS_SKILLS_DIR": os.getenv("TOS_SKILLS_DIR"),
111
+ "TOS_SKILLS_DIR": f"tos://agentkit-platform-{account_id}/skills/",
112
"TOOL_USER_SESSION_ID": tool_user_session_id,
113
}
114
0 commit comments