From bb7fefc8b543629de3aef2d8bdae244de14a4e13 Mon Sep 17 00:00:00 2001 From: "fangyaozheng@bytedance.com" Date: Fri, 12 Sep 2025 12:09:53 +0800 Subject: [PATCH] fix(deploy): deps and checking --- .../template/{{cookiecutter.local_dir_name}}/src/run.sh | 9 +-------- veadk/integrations/ve_tls/ve_tls.py | 9 +++++++-- 2 files changed, 8 insertions(+), 10 deletions(-) diff --git a/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/run.sh b/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/run.sh index 62e2abeb..b0f80137 100755 --- a/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/run.sh +++ b/veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/run.sh @@ -15,9 +15,7 @@ TIMEOUT=${_FAAS_FUNC_TIMEOUT} export SERVER_HOST=$HOST export SERVER_PORT=$PORT -# use cached veadk-python and corresponding deps in VeFaaS -# `./preinstalled-site-packages` stores veadk-python and its dependencies -export PYTHONPATH=$PYTHONPATH:./site-packages:./preinstalled-site-packages +export PYTHONPATH=$PYTHONPATH:./site-packages # Parse arguments while [[ $# -gt 0 ]]; do @@ -36,11 +34,6 @@ while [[ $# -gt 0 ]]; do esac done -# Check if MODEL_AGENT_API_KEY is set -if [ -z "$MODEL_AGENT_API_KEY" ]; then - echo "MODEL_AGENT_API_KEY is not set. Please set it in your environment variables." - exit 1 -fi USE_ADK_WEB=${USE_ADK_WEB:-False} diff --git a/veadk/integrations/ve_tls/ve_tls.py b/veadk/integrations/ve_tls/ve_tls.py index f63dbf64..f9c42ac3 100644 --- a/veadk/integrations/ve_tls/ve_tls.py +++ b/veadk/integrations/ve_tls/ve_tls.py @@ -14,8 +14,6 @@ import os -from volcengine.tls.TLSService import TLSService - from veadk.consts import DEFAULT_TLS_LOG_PROJECT_NAME, DEFAULT_TLS_TRACING_INSTANCE_NAME from veadk.integrations.ve_tls.utils import ve_tls_request from veadk.utils.logger import get_logger @@ -30,6 +28,13 @@ def __init__( secret_key: str | None = None, region: str = "cn-beijing", ): + try: + from volcengine.tls.TLSService import TLSService + except ImportError: + raise ImportError( + "Please install volcengine SDK before init VeTLS: pip install volcengine" + ) + self.access_key = ( access_key if access_key else os.getenv("VOLCENGINE_ACCESS_KEY", "") )