diff --git a/lib/crewai/tests/test_litellm_dependency_compat.py b/lib/crewai/tests/test_litellm_dependency_compat.py index 32cddc2c0..9e8baaee2 100644 --- a/lib/crewai/tests/test_litellm_dependency_compat.py +++ b/lib/crewai/tests/test_litellm_dependency_compat.py @@ -6,9 +6,13 @@ conflict with crewAI's own requirements (openai, python-dotenv, etc.). """ import sys -import tomllib from pathlib import Path +if sys.version_info >= (3, 11): + import tomllib +else: + import tomli as tomllib + import pytest from packaging.specifiers import SpecifierSet from packaging.version import Version