🚧 dealing with the pydantic error by updating the struct to provide a default value

Author: m-misiura

detectors/common/scheme.py

@@ -127,7 +127,7 @@ class ContentAnalysisResponse(BaseModel):
     start: int = Field(example=14)
     end: int = Field(example=26)
     text: str = Field(example="[email protected]")
-    detection: str = Field(example="Net.EmailAddress")
+    detection: str = Field(default="detection", example="Net.EmailAddress")
     detection_type: str = Field(example="pii")
     score: float = Field(example=0.8)
     evidences: Optional[List[EvidenceObj]] = Field(

detectors/huggingface/detector.py

@@ -249,15 +249,16 @@ def process_sequence_classification(self, text, detector_params=None, threshold=
                     and idx not in all_safe_labels
                     and label not in all_safe_labels
                 ):
+                    detection_value = getattr(self.model.config, "problem_type", None)
                     content_analyses.append(
                         ContentAnalysisResponse(
                             start=0,
                             end=len(text),
-                            detection=getattr(self.model.config, "problem_type", None) or "sequence_classification",
                             detection_type=label,
                             score=prob,
                             text=text,
                             evidences=[],
+                            **({"detection": detection_value} if detection_value is not None else {})
                         )
                     )
         return content_analyses