se estan creando el registro de las tareas sin problemas

api/api_v2/modules/partidas/routers.py

@@ -5,6 +5,7 @@ from typing import Dict, Any, List, Optional
 
 
 from api.api_v2.modules.authentication.services import get_current_user
+from api.api_v2.modules.tasks.services import register_task
 from .schemas import PartidaRequestSchema, PartidaListSchema
 from .tasks import process_partida_request
 
@@ -19,6 +20,15 @@ async def obtener_partida(partida_request: PartidaRequestSchema):
     acuse_dict = partida_request.model_dump()
     # Ejecuta la tarea de Celery de forma asíncrona
     task = process_partida_request.delay(acuse_dict)
+    # Registrar la tarea en el servicio de seguimiento
+    await register_task(
+        task_id=task.id,
+        message=f"Procesando partida {acuse_dict.get('partida', {}).get('numero', 'N/A')} para pedimento {acuse_dict.get('pedimento', {}).get('pedimento_app', 'N/A')}",
+        status="submitted",
+        pedimento_id=acuse_dict.get('pedimento', {}).get('id'),
+        organizacion_id=acuse_dict.get('pedimento', {}).get('organizacion'),
+        servicio=4  # 4 corresponde a "Pedimento Partidas"
+    )
     # Puedes devolver el ID de la tarea para consultar el estado después
     return {"task_id": task.id, "status": "submitted"}
 
@@ -39,5 +49,14 @@ async def obtener_partidas(partidas_request: PartidaListSchema):
         }
         task = process_partida_request.delay(partida_dict)
         task_ids.append(task.id)
+        # Registrar cada tarea en el servicio de seguimiento
+        await register_task(
+            task_id=task.id,
+            message=f"Procesando partida masiva {partida.get('numero', 'N/A')} para pedimento {partida_request_dict.get('pedimento', {}).get('pedimento_app', 'N/A')}",
+            status="submitted",
+            pedimento_id=partida_request_dict.get('pedimento', {}).get('id'),
+            organizacion_id=partida_request_dict.get('pedimento', {}).get('organizacion'),
+            servicio=4  # 4 corresponde a "Pedimento Partidas"
+        )
 
     return {"task_ids": task_ids, "status": "submitted", "total": len(task_ids)}