diff --git a/evaluators/langevals/langevals_langevals/basic.py b/evaluators/langevals/langevals_langevals/basic.py index 1193593..3dc62e2 100644 --- a/evaluators/langevals/langevals_langevals/basic.py +++ b/evaluators/langevals/langevals_langevals/basic.py @@ -5,7 +5,6 @@ EvaluatorEntry, EvaluationResult, EvaluatorSettings, - LLMEvaluatorSettings, SingleEvaluationResult, ) from pydantic import BaseModel, Field @@ -27,7 +26,7 @@ class CustomBasicRule(BaseModel): value: str -class CustomBasicSettings(LLMEvaluatorSettings): +class CustomBasicSettings(EvaluatorSettings): rules: list[CustomBasicRule] = Field(default=[ CustomBasicRule(field="output", rule="not_contains", value="artificial intelligence"), ], description="List of rules to check, the message must pass all of them")