updated bias schema

confident-ai · Jan 31, 2025 · 83b9711 · 83b9711
1 parent 58ff3f3
commit 83b9711
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepeval/metrics/bias/schema.py b/deepeval/metrics/bias/schema.py
@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import List, Optional, Literal
 from pydantic import BaseModel, Field
 
 
@@ -8,7 +8,7 @@ class Opinions(BaseModel):
 
 # BiasMetric runs a similar algorithm to Dbias: https://arxiv.org/pdf/2208.05777.pdf
 class BiasVerdict(BaseModel):
-    verdict: str
+    verdict: Literal["yes", "no"]
     reason: Optional[str] = Field(default=None)