diff --git a/pyproject.toml b/pyproject.toml index b83ea3fb..af38fe54 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,7 +15,7 @@ dependencies = [ "pydantic-settings>=2.10.1", # Config management "a2a-sdk>=0.3.0", # For Google Agent2Agent protocol "deprecated>=1.2.18", - "google-adk>=1.10.0", # For basic agent architecture + "google-adk>=1.11.0", # For basic agent architecture "litellm>=1.74.3", # For model inference "loguru>=0.7.3", # For better logging "opentelemetry-exporter-otlp>=1.35.0", diff --git a/veadk/configs/model_configs.py b/veadk/configs/model_configs.py index e0efbdb6..d253e164 100644 --- a/veadk/configs/model_configs.py +++ b/veadk/configs/model_configs.py @@ -17,7 +17,6 @@ from pydantic_settings import BaseSettings, SettingsConfigDict -from veadk.auth.veauth.ark_veauth import ARKVeAuth from veadk.consts import ( DEFAULT_MODEL_AGENT_API_BASE, DEFAULT_MODEL_AGENT_NAME, @@ -39,4 +38,10 @@ class ModelConfig(BaseSettings): @cached_property def api_key(self) -> str: - return os.getenv("MODEL_AGENT_API_KEY") or ARKVeAuth().token + _api_key = os.getenv("MODEL_AGENT_API_KEY") + if not _api_key: + # Only import ARKVeAuth if api_key is not set + from veadk.auth.veauth.ark_veauth import ARKVeAuth + + _api_key = ARKVeAuth().token + return _api_key