diff --git a/pyproject.toml b/pyproject.toml index 42d38b0e..85f6fe03 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "veadk-python" -version = "0.5.7" +version = "0.5.8" description = "Volcengine agent development kit, integrations with Volcengine cloud services." readme = "README.md" requires-python = ">=3.10" diff --git a/veadk/tools/builtin_tools/run_code.py b/veadk/tools/builtin_tools/run_code.py index 8a1c0fe5..14a15c4a 100644 --- a/veadk/tools/builtin_tools/run_code.py +++ b/veadk/tools/builtin_tools/run_code.py @@ -17,10 +17,10 @@ from google.adk.tools import ToolContext +from veadk.auth.veauth.utils import get_credential_from_vefaas_iam from veadk.config import getenv from veadk.utils.logger import get_logger from veadk.utils.volcengine_sign import ve_request -from veadk.auth.veauth.utils import get_credential_from_vefaas_iam logger = get_logger(__name__) @@ -97,6 +97,7 @@ def run_code( "kernel_name": language, } ), + "Ttl": os.getenv("AGENTKIT_TOOL_TTL", 1800), }, action="InvokeTool", ak=ak, diff --git a/veadk/version.py b/veadk/version.py index c8adfd45..dacc1021 100644 --- a/veadk/version.py +++ b/veadk/version.py @@ -12,4 +12,4 @@ # See the License for the specific language governing permissions and # limitations under the License. -VERSION = "0.5.7" +VERSION = "0.5.8"