Skip to content

Commit a5dc929

Browse files
authored
fix: add region in vetos when using skills (#450)
1 parent 3bd5c05 commit a5dc929

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

veadk/tools/skills_tools/register_skills_tool.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,7 @@ def register_skills_tool(
121121
sk=secret_key,
122122
session_token=session_token,
123123
bucket_name=tos_bucket,
124+
region=region,
124125
)
125126

126127
object_key = (

veadk/tools/skills_tools/skills_tool.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,8 @@ def _invoke_skill(self, skill_name: str, tool_context: ToolContext) -> str:
111111
from veadk.auth.veauth.utils import get_credential_from_vefaas_iam
112112
from veadk.integrations.ve_tos.ve_tos import VeTOS
113113

114+
region = os.getenv("AGENTKIT_TOOL_REGION", "cn-beijing")
115+
114116
access_key = os.getenv("VOLCENGINE_ACCESS_KEY")
115117
secret_key = os.getenv("VOLCENGINE_SECRET_KEY")
116118
session_token = ""
@@ -130,6 +132,7 @@ def _invoke_skill(self, skill_name: str, tool_context: ToolContext) -> str:
130132
sk=secret_key,
131133
session_token=session_token,
132134
bucket_name=tos_bucket,
135+
region=region,
133136
)
134137

135138
save_path = skill_dir / f"{skill_name}.zip"

0 commit comments

Comments
 (0)