UPDATE

rag/services/water_need_prediction.py

@@ -68,32 +68,21 @@ def _build_service_client(cfg: RAGConfig):
     return service, client, service.llm.model
 
 
-def _fallback_from_payload(prediction_payload: dict[str, Any]) -> dict[str, Any]:
-    total = float(prediction_payload.get("totalNext7Days") or 0.0)
-    daily = prediction_payload.get("dailyBreakdown") or []
-    peak_day = max(daily, key=lambda item: float(item.get("gross_irrigation_mm", 0.0) or 0.0), default=None)
-    if total <= 0:
-        return {
-            "summary": "براي چند روز آينده نياز آبي معناداري برآورد نشد.",
-            "irrigation_outlook": "بارش موثر يا شرايط فعلي باعث شده نياز خالص آبياري پايين باشد.",
-            "recommended_action": "پايش رطوبت خاک ادامه يابد و قبل از هر آبياري جديد forecast دوباره بررسي شود.",
-            "risk_note": "اگر forecast تغيير کند يا بارش موثر رخ ندهد، برآورد بايد به روز شود.",
-            "confidence": 0.58,
-        }
-
-    peak_text = ""
-    if peak_day:
-        peak_text = (
-            f" بيشترين فشار آبي در {peak_day.get('forecast_date')} "
-            f"با حدود {peak_day.get('gross_irrigation_mm')} ميلي متر برآورد شده است."
-        )
-    return {
-        "summary": f"جمع نياز آبي 7 روز آينده حدود {round(total, 2)} ميلي متر برآورد شده است.",
-        "irrigation_outlook": "الگوي آبياري بايد در چند روز آينده بر اساس نياز روزانه و بارش موثر تنظيم شود." + peak_text,
-        "recommended_action": "برنامه آبياري کوتاه مدت بر اساس روزهاي اوج نياز تنظيم و صبح زود يا نزديک غروب اجرا شود.",
-        "risk_note": "در صورت تغيير دما، باد يا بارش، مقادير gross irrigation ممکن است تغيير کنند.",
-        "confidence": 0.72,
+def _validate_prediction_insight(parsed: dict[str, Any]) -> dict[str, Any]:
+    required_keys = {
+        "summary",
+        "irrigation_outlook",
+        "recommended_action",
+        "risk_note",
+        "confidence",
     }
+    missing = [key for key in required_keys if key not in parsed]
+    if missing:
+        raise ValueError(
+            "Water need prediction insight response is missing required fields: "
+            + ", ".join(missing)
+        )
+    return parsed
 
 
 def _build_messages(
@@ -132,12 +121,10 @@ def get_water_need_prediction_insight(
     cfg = load_rag_config()
     service, client, model = _build_service_client(cfg)
     farm_details = _load_farm_or_error(farm_uuid)
-    fallback = _fallback_from_payload(prediction_payload)
     user_query = query or "نياز آبي کوتاه مدت اين مزرعه را تفسير کن و اقدام عملياتي پيشنهاد بده."
     structured_context = {
         "farm_uuid": farm_uuid,
         "prediction_payload": prediction_payload,
-        "fallback_summary": fallback,
     }
     rag_context = build_rag_context(
         query=user_query,
@@ -169,24 +156,10 @@ def get_water_need_prediction_insight(
         _complete_audit_log(audit_log, raw)
     except Exception as exc:
         logger.error("Water need prediction insight failed for %s: %s", farm_uuid, exc)
-        _fail_audit_log(audit_log, str(exc), json.dumps(fallback, ensure_ascii=False))
-        return {
-            **fallback,
-            "farm_uuid": farm_uuid,
-            "knowledge_base": KB_NAME,
-            "tone_file": service.tone_file,
-            "raw_response": None,
-        }
+        _fail_audit_log(audit_log, str(exc))
+        raise RuntimeError(f"Water need prediction insight failed for farm {farm_uuid}.") from exc
 
-    if not parsed:
-        parsed = fallback
-    parsed.setdefault("summary", fallback["summary"])
-    parsed.setdefault("irrigation_outlook", fallback["irrigation_outlook"])
-    parsed.setdefault("recommended_action", fallback["recommended_action"])
-    parsed.setdefault("risk_note", fallback["risk_note"])
-    parsed.setdefault("confidence", fallback["confidence"])
+    parsed = _validate_prediction_insight(parsed)
     parsed["farm_uuid"] = farm_uuid
-    parsed["knowledge_base"] = KB_NAME
-    parsed["tone_file"] = service.tone_file
     parsed["raw_response"] = raw
     return parsed