UPDATE

fertilization/views.py

@@ -10,31 +10,25 @@ from rest_framework.views import APIView
 from config.openapi import (
     build_envelope_serializer,
     build_response,
-    build_task_queue_data_serializer,
-    build_task_status_data_serializer,
 )
 
 from .serializers import FertilizationRecommendRequestSerializer
 
 
-FertilizationQueueResponseSerializer = build_envelope_serializer(
-    "FertilizationQueueResponseSerializer",
-    build_task_queue_data_serializer("FertilizationQueueDataSerializer"),
-)
 FertilizationValidationErrorSerializer = build_envelope_serializer(
     "FertilizationValidationErrorSerializer",
     data_required=False,
     allow_null=True,
 )
-FertilizationStatusResponseSerializer = build_envelope_serializer(
-    "FertilizationStatusResponseSerializer",
-    build_task_status_data_serializer("FertilizationStatusDataSerializer"),
+FertilizationResponseSerializer = build_envelope_serializer(
+    "FertilizationResponseSerializer",
+    data_schema=None,
 )
 
 
 class FertilizationRecommendView(APIView):
     """
-    توصیه کودهی با Celery.
+    توصیه کودهی به صورت مستقیم.
     POST با sensor_uuid، plant_name، growth_stage.
     اطلاعات گیاه از plant app دریافت می‌شود.
     نیازی به دریافت نوع آبیاری نیست.
@@ -44,21 +38,25 @@ class FertilizationRecommendView(APIView):
         tags=["Fertilization Recommendation"],
         summary="درخواست توصیه کودهی",
         description=(
-            "داده‌های سنسور و گیاه را دریافت کرده و یک تسک Celery "
-            "برای تولید توصیه کودهی در صف قرار می‌دهد. "
+            "داده‌های سنسور و گیاه را دریافت کرده و "
+            "توصیه کودهی را مستقیم برمی‌گرداند. "
             "اطلاعات گیاه از جدول Plant بارگذاری می‌شود. "
             "محاسبات مربوط به نیاز آبی در این endpoint انجام نمی‌شود و مستقل از توصیه کودهی است."
         ),
         request=FertilizationRecommendRequestSerializer,
         responses={
-            202: build_response(
-                FertilizationQueueResponseSerializer,
-                "تسک توصیه کودهی در صف قرار گرفت.",
+            200: build_response(
+                FertilizationResponseSerializer,
+                "توصیه کودهی با موفقیت تولید شد.",
             ),
             400: build_response(
                 FertilizationValidationErrorSerializer,
                 "پارامتر ورودی نامعتبر است.",
             ),
+            500: build_response(
+                FertilizationValidationErrorSerializer,
+                "خطا در تولید توصیه کودهی.",
+            ),
         },
         examples=[
             OpenApiExample(
@@ -73,7 +71,7 @@ class FertilizationRecommendView(APIView):
         ],
     )
     def post(self, request):
-        from rag.tasks import fertilization_recommendation_task
+        from rag.services.fertilization import get_fertilization_recommendation
 
         serializer = FertilizationRecommendRequestSerializer(data=request.data)
         if not serializer.is_valid():
@@ -88,53 +86,20 @@ class FertilizationRecommendView(APIView):
         growth_stage = validated.get("growth_stage")
         query = validated.get("query")
 
-        task = fertilization_recommendation_task.delay(
-            sensor_uuid=sensor_uuid,
-            plant_name=plant_name,
-            growth_stage=growth_stage,
-            query=query,
-        )
+        try:
+            result = get_fertilization_recommendation(
+                sensor_uuid=sensor_uuid,
+                plant_name=plant_name,
+                growth_stage=growth_stage,
+                query=query,
+            )
+        except Exception as exc:
+            return Response(
+                {"code": 500, "msg": f"خطا در تولید توصیه کودهی: {exc}", "data": None},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
+
         return Response(
-            {
-                "code": 202,
-                "msg": "تسک توصیه کودهی در صف قرار گرفت.",
-                "data": {
-                    "task_id": task.id,
-                    "status_url": f"/api/fertilization/recommend/{task.id}/status/",
-                },
-            },
-            status=status.HTTP_202_ACCEPTED,
-        )
-
-
-class FertilizationRecommendStatusView(APIView):
-    """وضعیت تسک توصیه کودهی."""
-
-    @extend_schema(
-        tags=["Fertilization Recommendation"],
-        summary="وضعیت تسک توصیه کودهی",
-        description="وضعیت تسک Celery توصیه کودهی را برمی‌گرداند.",
-        responses={
-            200: build_response(
-                FertilizationStatusResponseSerializer,
-                "وضعیت فعلی تسک توصیه کودهی.",
-            ),
-        },
-    )
-    def get(self, request, task_id):
-        from celery.result import AsyncResult
-
-        result = AsyncResult(task_id)
-        data = {"task_id": task_id, "status": result.state}
-        if result.state == "PENDING":
-            data["message"] = "تسک در صف یا یافت نشد."
-        elif result.state == "PROGRESS":
-            data["progress"] = result.info
-        elif result.state == "SUCCESS":
-            data["result"] = result.result
-        elif result.state == "FAILURE":
-            data["error"] = str(result.result)
-        return Response(
-            {"code": 200, "msg": "success", "data": data},
+            {"code": 200, "msg": "success", "data": result},
             status=status.HTTP_200_OK,
         )