AI UPDATE

rag/services/fertilization.py

@@ -6,13 +6,14 @@ 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.chat import build_rag_context, _load_service_tone
+from rag.config import load_rag_config, RAGConfig, get_service_config
 from rag.user_data import build_plant_text
 
 logger = logging.getLogger(__name__)
 
 KB_NAME = "fertilization"
+SERVICE_ID = "fertilization"
 
 DEFAULT_FERTILIZATION_PROMPT = (
     "بر اساس داده‌های خاک (NPK، pH)، مشخصات گیاه، مرحله رشد و پایگاه دانش کودهی، "
@@ -56,13 +57,23 @@ def get_fertilization_recommendation(
         dict با کلیدهای fertilizer_needed, fertilizer_type, amount_kg_per_hectare, reason, npk_status, raw_response
     """
     cfg = config or load_rag_config()
-    client = get_chat_client(cfg)
-    model = cfg.llm.model
+    service = get_service_config(SERVICE_ID, cfg)
+    service_cfg = RAGConfig(
+        embedding=cfg.embedding,
+        qdrant=cfg.qdrant,
+        chunking=cfg.chunking,
+        llm=service.llm,
+        knowledge_bases=cfg.knowledge_bases,
+        services=cfg.services,
+        chromadb=cfg.chromadb,
+    )
+    client = get_chat_client(service_cfg)
+    model = service.llm.model
 
     user_query = query or "توصیه کودهی برای مزرعه من چیست؟"
 
     context = build_rag_context(
-        user_query, sensor_uuid, config=cfg, limit=limit, kb_name=KB_NAME,
+        user_query, sensor_uuid, config=cfg, limit=limit, kb_name=KB_NAME, service_id=SERVICE_ID,
     )
 
     extra_parts: list[str] = []
@@ -73,8 +84,10 @@ def get_fertilization_recommendation(
     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)
+    tone = _load_service_tone(service, cfg)
     system_parts = [tone] if tone else []
+    if service.system_prompt:
+        system_parts.append(service.system_prompt)
     system_parts.append(DEFAULT_FERTILIZATION_PROMPT)
     if context:
         system_parts.append("\n\n" + context)