Changes committed

docengine/app/tasks/document_tasks.py

@@ -0,0 +1,94 @@
+from __future__ import annotations
+
+import uuid
+
+from app.core.logging_config import get_logger
+from app.workers.celery_app import celery_app
+from app.core.database import get_db_context
+from app.services.document_service import DocumentProcessingService
+
+logger = get_logger(__name__)
+
+
+@celery_app.task(
+    name="app.tasks.document_tasks.process_document_task",
+    bind=True,
+    max_retries=3,
+    default_retry_delay=60,
+    acks_late=True,
+)
+def process_document_task(self, document_id: str) -> dict:  # noqa: ANN001
+    """Celery task to process a document asynchronously."""
+    logger.info("task_started", task_id=self.request.id, document_id=document_id)
+
+    try:
+        doc_uuid = uuid.UUID(document_id)
+        with get_db_context() as db:
+            service = DocumentProcessingService(db)
+            document = service.process_document(doc_uuid)
+            logger.info(
+                "task_completed",
+                task_id=self.request.id,
+                document_id=document_id,
+                status=document.status,
+            )
+            return {
+                "document_id": document_id,
+                "status": document.status,
+                "page_count": document.page_count,
+            }
+    except Exception as exc:
+        logger.exception(
+            "task_failed",
+            task_id=self.request.id,
+            document_id=document_id,
+            error=str(exc),
+            retry=self.request.retries,
+        )
+        raise self.retry(exc=exc)
+
+
+@celery_app.task(
+    name="app.tasks.document_tasks.match_document_task",
+    bind=True,
+    max_retries=2,
+    default_retry_delay=30,
+)
+def match_document_task(
+    self,  # noqa: ANN001
+    document_id: str,
+    min_confidence: float = 0.5,
+    max_results: int = 5,
+) -> dict:
+    """Celery task to match a document against templates."""
+    logger.info("match_task_started", task_id=self.request.id, document_id=document_id)
+
+    try:
+        doc_uuid = uuid.UUID(document_id)
+        with get_db_context() as db:
+            from app.services.matching_service import MatchingService
+            service = MatchingService(db)
+            matches = service.match_document(
+                document_id=doc_uuid,
+                min_confidence=min_confidence,
+                max_results=max_results,
+            )
+            logger.info(
+                "match_task_completed",
+                task_id=self.request.id,
+                document_id=document_id,
+                matches=len(matches),
+            )
+            return {
+                "document_id": document_id,
+                "matches": len(matches),
+                "best_score": matches[0].confidence_score if matches else 0.0,
+            }
+    except Exception as exc:
+        logger.exception(
+            "match_task_failed",
+            task_id=self.request.id,
+            document_id=document_id,
+            error=str(exc),
+        )
+        raise self.retry(exc=exc)