fix wrong validator used

utils.py

@@ -100,7 +100,7 @@ class BoundingBox(BaseModel):
     y_max: int
 
     @staticmethod
-    def from_finding(finding: Finding) -> 'BoundingBox':
+    def from_finding(finding: Union[Finding, PartialFinding]) -> 'BoundingBox':
         """Creates a BoundingBox instance from a Finding instance."""
         return BoundingBox(label=finding.label, x_min=finding.bounding_box[0], y_min=finding.bounding_box[1], x_max=finding.bounding_box[2], y_max=finding.bounding_box[3])
 
@@ -154,7 +154,7 @@ def parse_json_response(out: str) -> list[dict]:
     return json.loads(fixed)
 
 
-def parse_into_models(findings: list[dict], strict=True) -> Union[list[Finding], list[BoundingBox]]:
+def parse_into_models(findings: list[dict], strict=True) -> Union[list[Finding], list[PartialFinding]]:
     """Parses and validates a list of dictionaries into a list of Finding models.
 
     Args:
@@ -164,7 +164,7 @@ def parse_into_models(findings: list[dict], strict=True) -> Union[list[Finding],
         A list of validated Finding model instances.
     """
     if not strict:
-        return [BoundingBox.model_validate(box) for box in findings]
+        return [PartialFinding.model_validate(box) for box in findings]
     return [Finding.model_validate(box) for box in findings]