first commit

rag/services/irrigation.py

@@ -0,0 +1,115 @@
+"""
+سرویس توصیه آبیاری — بدون API، قابل فراخوانی از سایر سرویس‌ها
+از RAG با پایگاه دانش irrigation و لحن مخصوص آبیاری استفاده می‌کند.
+"""
+import json
+import logging
+
+from rag.api_provider import get_chat_client
+from rag.chat import build_rag_context, _load_kb_tone
+from rag.config import load_rag_config, RAGConfig
+from rag.user_data import build_plant_text, build_irrigation_method_text
+
+logger = logging.getLogger(__name__)
+
+KB_NAME = "irrigation"
+
+DEFAULT_IRRIGATION_PROMPT = (
+    "بر اساس داده‌های خاک، هواشناسی، مشخصات گیاه، روش آبیاری و پایگاه دانش آبیاری، "
+    "یک توصیه آبیاری دقیق بده. "
+    "پاسخ حتماً به فرمت JSON با فیلدهای زیر باشد:\n"
+    "irrigation_needed (bool), amount_mm (float), reason (str), next_check_date (str)\n"
+    "فقط JSON خروجی بده، بدون توضیح اضافی."
+)
+
+
+def get_irrigation_recommendation(
+    sensor_uuid: str,
+    plant_name: str | None = None,
+    growth_stage: str | None = None,
+    irrigation_method_name: str | None = None,
+    query: str | None = None,
+    config: RAGConfig | None = None,
+    limit: int = 8,
+) -> dict:
+    """
+    توصیه آبیاری برای یک سنسور (کاربر).
+    از RAG با پایگاه دانش irrigation استفاده می‌کند.
+
+    Args:
+        sensor_uuid: شناسه سنسور کاربر
+        plant_name: نام گیاه (برای بارگذاری مشخصات از جدول Plant)
+        growth_stage: مرحله رشد گیاه
+        irrigation_method_name: نام روش آبیاری (برای بارگذاری از جدول IrrigationMethod)
+        query: سوال اختیاری
+        config: تنظیمات RAG
+        limit: تعداد چانک‌های بازیابی‌شده
+
+    Returns:
+        dict با کلیدهای irrigation_needed, amount_mm, reason, next_check_date, raw_response
+    """
+    cfg = config or load_rag_config()
+    client = get_chat_client(cfg)
+    model = cfg.llm.model
+
+    user_query = query or "توصیه آبیاری برای مزرعه من چیست؟"
+
+    context = build_rag_context(
+        user_query, sensor_uuid, config=cfg, limit=limit, kb_name=KB_NAME,
+    )
+
+    extra_parts: list[str] = []
+    if plant_name and growth_stage:
+        plant_text = build_plant_text(plant_name, growth_stage)
+        if plant_text:
+            extra_parts.append("[اطلاعات گیاه]\n" + plant_text)
+    if irrigation_method_name:
+        method_text = build_irrigation_method_text(irrigation_method_name)
+        if method_text:
+            extra_parts.append("[روش آبیاری انتخابی]\n" + method_text)
+    if extra_parts:
+        context = "\n\n---\n\n".join(extra_parts) + ("\n\n---\n\n" + context if context else "")
+
+    tone = _load_kb_tone(KB_NAME, cfg)
+    system_parts = [tone] if tone else []
+    system_parts.append(DEFAULT_IRRIGATION_PROMPT)
+    if context:
+        system_parts.append("\n\n" + context)
+    system_content = "\n".join(system_parts)
+
+    messages = [
+        {"role": "system", "content": system_content},
+        {"role": "user", "content": user_query},
+    ]
+
+    try:
+        response = client.chat.completions.create(
+            model=model,
+            messages=messages,
+        )
+        raw = response.choices[0].message.content.strip()
+    except Exception as exc:
+        logger.error("Irrigation recommendation error for %s: %s", sensor_uuid, exc)
+        return {
+            "irrigation_needed": None,
+            "amount_mm": None,
+            "reason": f"خطا در دریافت توصیه: {exc}",
+            "next_check_date": None,
+            "raw_response": None,
+        }
+
+    try:
+        cleaned = raw
+        if cleaned.startswith("```"):
+            cleaned = cleaned.strip("`").removeprefix("json").strip()
+        result = json.loads(cleaned)
+    except (json.JSONDecodeError, ValueError):
+        result = {
+            "irrigation_needed": None,
+            "amount_mm": None,
+            "reason": raw,
+            "next_check_date": None,
+        }
+
+    result["raw_response"] = raw
+    return result