evaluation.py 376 B

1234567891011121314151617
  1. from enum import Enum
  2. from typing import Optional
  3. from pydantic import BaseModel
  4. class EvalMetric(Enum):
  5. CONTEXT_RELEVANCY = "context_relevancy"
  6. ANSWER_RELEVANCY = "answer_relevancy"
  7. GROUNDEDNESS = "groundedness"
  8. class EvalData(BaseModel):
  9. question: str
  10. contexts: list[str]
  11. answer: str
  12. ground_truth: Optional[str] = None # Not used as of now