diff --git a/deepeval/__init__.py b/deepeval/__init__.py index 5739ada8..074d1355 100644 --- a/deepeval/__init__.py +++ b/deepeval/__init__.py @@ -8,6 +8,7 @@ from deepeval.event import track from deepeval.monitor import monitor, a_monitor, send_feedback, a_send_feedback from deepeval.evaluate import evaluate, assert_test +from deepeval.auto_evaluate import auto_evaluate from deepeval.test_run import on_test_run_end, log_hyperparameters from deepeval.utils import login_with_confident_api_key from deepeval.telemetry import * @@ -23,6 +24,7 @@ "a_send_feedback", "send_feedback", "evaluate", + "auto_evaluate" "assert_test", "on_test_run_end", "trace_langchain", diff --git a/deepeval/auto_evaluate.py b/deepeval/auto_evaluate.py index 1341ccad..2fe43f28 100644 --- a/deepeval/auto_evaluate.py +++ b/deepeval/auto_evaluate.py @@ -156,7 +156,7 @@ def generate_goldens( include_expected_output=include_expected_output, max_goldens_per_context=max_goldens_per_context, max_concurrent=max_concurrent, - async_mod=async_mode, + async_mode=async_mode, filtration_config=filtration_config, evolution_config=evolution_config, styling_config=styling_config, diff --git a/tests/test_automatic_evals.py b/tests/test_automatic_evals.py index 0193ae36..dec5fd7c 100644 --- a/tests/test_automatic_evals.py +++ b/tests/test_automatic_evals.py @@ -1,6 +1,6 @@ from deepeval.metrics import AnswerRelevancyMetric, FaithfulnessMetric from deepeval.integrations import trace_llama_index, trace_langchain -from deepeval.auto_evaluate import auto_evaluate +from deepeval import evaluate, auto_evaluate ####################################################### ### LLamaIndex ########################################