Skip to content

Commit 64d7f2e

Browse files
authored
fix: change execute_skills timeout to 900 (#366)
1 parent dfd49e1 commit 64d7f2e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

veadk/tools/builtin_tools/execute_skills.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,15 @@ def execute_skills(
3030
workflow_prompt: str,
3131
skills: Optional[List[str]] = None,
3232
tool_context: ToolContext = None,
33-
timeout: int = 300,
33+
timeout: int = 900,
3434
) -> str:
3535
"""execute skills in a code sandbox and return the output.
3636
For C++ code, don't execute it directly, compile and execute via Python; write sources and object files to /tmp.
3737
3838
Args:
3939
workflow_prompt (str): instruction of workflow
4040
skills (Optional[List[str]]): The skills will be invoked
41-
timeout (int, optional): The timeout in seconds for the code execution, less than or equal to 300. Defaults to 300.
41+
timeout (int, optional): The timeout in seconds for the code execution, less than or equal to 900. Defaults to 900.
4242
4343
Returns:
4444
str: The output of the code execution.
@@ -92,8 +92,8 @@ def execute_skills(
9292

9393
# TODO: remove after agentkit supports custom environment variables setting
9494
env_vars = {
95-
"MODEL_AGENT_API_KEY": os.getenv("MODEL_AGENT_API_KEY"),
9695
"TOS_SKILLS_DIR": os.getenv("TOS_SKILLS_DIR"),
96+
"TOOL_USER_SESSION_ID": tool_user_session_id,
9797
}
9898

9999
code = f"""

0 commit comments

Comments
 (0)