test image pull argocd

Author: nrandriamanana

src/api/models/responses.py

@@ -17,3 +17,26 @@ class Prediction(BaseModel):
     libelle: str
 
 
+class PredictionResponse(RootModel[Dict[str, Union[Prediction, float, str]]]):
+    pass
+    """
+    Contract for the normalized prediction responses generated by the model artifact.
+
+    Expected flat structure after normalization:
+
+    {
+    "1": Prediction,
+    "2": Prediction,
+    ...,
+    "IC": float,           # required confidence score
+    "MLversion": str  # required run_id as model version
+    }
+
+    Notes:
+    - The output reflects what the model artifact produces, but the API applies
+    `model_dump()` in `predict()` before returning to ensure schema consistency.
+    - Strong validation is performed during training; inference focuses on lightweight
+    structural checks for performance.
+    - Any changes to the output schema (e.g., new fields, renaming) must be documented
+    both here and in training repo (codif-ape-training) to maintain API contract clarity.
+    """