Add LLM Answer Match evaluator

evaluators/huggingface/pyproject.toml

@@ -12,7 +12,7 @@ license = "MIT"
 python = "^3.11"
 langevals-core = { path = "../../langevals_core", develop = true }
 httpx = "^0.27.0"
-litellm = "^1.52.14"
+litellm = "^1.53.7"
 
 [tool.poetry.group.test.dependencies]
 pytest = "^7.4.2"

evaluators/langevals/langevals_langevals/llm_answer_match.py

@@ -0,0 +1,115 @@
+import litellm
+from pydantic import Field
+from typing import Optional
+import dspy
+
+from langevals_core.base_evaluator import (
+    BaseEvaluator,
+    EvaluatorEntry,
+    EvaluationResult,
+    LLMEvaluatorSettings,
+    SingleEvaluationResult,
+    EvaluationResultSkipped,
+    Money,
+)
+from litellm.cost_calculator import cost_per_token
+
+
+class LLMAnswerMatchEntry(EvaluatorEntry):
+    input: Optional[str] = Field(default="")
+    output: Optional[str] = Field(default="")
+    expected_output: Optional[str] = Field(default="")
+
+
+class LLMAnswerMatchSettings(LLMEvaluatorSettings):
+    pass
+
+
+class LLMAnswerMatchResult(EvaluationResult):
+    passed: bool = Field(default=True)
+    details: Optional[str] = Field(default=None)
+
+
+class LLMAnswerMatchSignature(dspy.Signature):
+    """Verify that the predicted answer matches the gold answer for the question. Style does not matter, for example the gold answer may be more direct while the predicted answer more verbose and still be correct."""
+
+    question = dspy.InputField()
+    gold_answer = dspy.InputField(desc="correct answer for question")
+    predicted_answer = dspy.InputField(desc="predicted answer for question")
+    reasoning = dspy.OutputField(desc="reasoning for the answer")
+    is_correct = dspy.OutputField(desc="True or False")
+
+
+class LLMAnswerMatchEvaluator(
+    BaseEvaluator[
+        LLMAnswerMatchEntry,
+        LLMAnswerMatchSettings,
+        LLMAnswerMatchResult,
+    ]
+):
+    """
+    Uses an LLM to check if the generated output answers a question correctly the same way as the expected output, even if their style is different.
+    """
+
+    name = "LLM Answer Match"
+    category = "quality"
+    env_vars = []
+    is_guardrail = False
+
+    def evaluate(self, entry: LLMAnswerMatchEntry) -> SingleEvaluationResult:
+        total_tokens = len(
+            litellm.encode(
+                model=self.settings.model,
+                text=f"{entry.input} {entry.output} {entry.expected_output}",
+            )
+        )
+        max_tokens = min(self.settings.max_tokens, 32768)
+        if total_tokens > max_tokens:
+            return EvaluationResultSkipped(
+                details=f"Total tokens exceed the maximum of {max_tokens}: {total_tokens}"
+            )
+
+        lm = model_to_dspy_lm(self.settings.model)
+        dspy.settings.configure(experimental=True)
+
+        answer_match = dspy.Predict(LLMAnswerMatchSignature)
+        answer_match.set_lm(lm)
+
+        result = answer_match(
+            question=entry.input,
+            gold_answer=entry.expected_output,
+            predicted_answer=entry.output,
+        )
+
+        last_response = lm.history[-1]
+        cost = None
+        if last_response:
+            input_cost, output_cost = cost_per_token(
+                model=self.settings.model,
+                prompt_tokens=last_response.get("usage", {}).get("prompt_tokens", 0),
+                completion_tokens=last_response.get("usage", {}).get(
+                    "completion_tokens", 0
+                ),
+            )
+            cost = input_cost + output_cost
+
+        return LLMAnswerMatchResult(
+            passed="true" in str(result.is_correct).lower(),
+            details=result.reasoning,
+            cost=Money(amount=cost, currency="USD") if cost is not None else None,
+        )
+
+
+def model_to_dspy_lm(model: str) -> dspy.LM:
+    llm_params = {}
+    if "azure/" in model:
+        llm_params["api_version"] = "2023-07-01-preview"
+
+    lm = dspy.LM(
+        model=model,
+        temperature=0,
+        drop_params=True,
+        model_type="chat",
+        **llm_params,
+    )
+    return lm