diff --git a/pyproject.toml b/pyproject.toml index 45c6e921..b6272145 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -16,7 +16,7 @@ dependencies = [ "a2a-sdk==0.3.7", # For Google Agent2Agent protocol "deprecated==1.2.18", "google-adk==1.19.0", # For basic agent architecture - "litellm>=1.74.3", # For model inference + "litellm==1.74.3", # For model inference "loguru==0.7.3", # For better logging "opentelemetry-exporter-otlp==1.37.0", "opentelemetry-instrumentation-logging>=0.56b0", @@ -73,10 +73,6 @@ dev = [ "pytest-xdist>=3.8.0", ] -responses = [ - "litellm>=1.79.3" -] - [dependency-groups] dev = [ "pre-commit>=4.2.0", # Format checking diff --git a/veadk/utils/misc.py b/veadk/utils/misc.py index 6c20cf0c..6755479e 100644 --- a/veadk/utils/misc.py +++ b/veadk/utils/misc.py @@ -207,8 +207,7 @@ def check_litellm_version(min_version: str): required = parse_version(min_version) if installed < required: raise ValueError( - "You have used `enable_responses=True`. If you want to use the `responses_api`, please install the relevant support:" - "\npip install veadk-python[responses]" + "You have used `enable_responses=True`. If you want to use the `responses_api`, please ensure that `litellm>=1.79.3`" ) except ImportError: raise ImportError(