diff --git a/py/plugins/evaluators/pyproject.toml b/py/plugins/evaluators/pyproject.toml index f02bac24dc..903cf0caec 100644 --- a/py/plugins/evaluators/pyproject.toml +++ b/py/plugins/evaluators/pyproject.toml @@ -30,7 +30,12 @@ classifiers = [ "Topic :: Scientific/Engineering :: Artificial Intelligence", "Topic :: Software Development :: Libraries", ] -dependencies = ["genkit", "jsonata-python>=0.5.3", "pydantic>=2.10.5"] +dependencies = [ + "genkit", + "jsonata-python>=0.5.3", + "pydantic>=2.10.5", + "strenum>=0.4.15; python_version < '3.11'", +] description = "Genkit Evaluators Plugin for RAGAS" license = { text = "Apache-2.0" } name = "genkit-plugin-evaluators" diff --git a/py/plugins/evaluators/src/genkit/plugins/evaluators/constant.py b/py/plugins/evaluators/src/genkit/plugins/evaluators/constant.py index e0e894abcc..c3ca5bdb62 100644 --- a/py/plugins/evaluators/src/genkit/plugins/evaluators/constant.py +++ b/py/plugins/evaluators/src/genkit/plugins/evaluators/constant.py @@ -16,7 +16,10 @@ from collections.abc import Callable -from enum import StrEnum +try: + from enum import StrEnum +except ImportError: + from strenum import StrEnum from typing import Any from pydantic import BaseModel, RootModel diff --git a/py/uv.lock b/py/uv.lock index 63dd9ea3c7..a0e588defe 100644 --- a/py/uv.lock +++ b/py/uv.lock @@ -1347,6 +1347,7 @@ dependencies = [ { name = "genkit" }, { name = "jsonata-python" }, { name = "pydantic" }, + { name = "strenum", marker = "python_full_version < '3.11'" }, ] [package.metadata] @@ -1354,6 +1355,7 @@ requires-dist = [ { name = "genkit", editable = "packages/genkit" }, { name = "jsonata-python", specifier = ">=0.5.3" }, { name = "pydantic", specifier = ">=2.10.5" }, + { name = "strenum", marker = "python_full_version < '3.11'", specifier = ">=0.4.15" }, ] [[package]]