src/api/schemas.py

"""Schemas pour les requêtes d'API.
- PredictRequest: Schéma pour les requêtes de prédiction de rendement.
- RecommendRequest: Schéma pour les requêtes de recommandation de cultures.
"""
from pydantic import BaseModel

class PredictRequest(BaseModel):
    area: str
    item: str
    temp: float
    rainfall_mm: float
    pesticides: float
    trend: float = 1.0

class RecommendRequest(BaseModel):
    area: str
    temp: float
    rainfall_mm: float
    pesticides: float
    trend: float = 1.0