refactor: replace task_step_logging with log_task for metadata extraction

This commit is contained in:
Christian Krakau-Louis
2025-03-28 16:26:03 +01:00
parent ffc049196b
commit a92cace662
+2 -2
View File
@@ -6,7 +6,7 @@ import os
from app.config import settings
from app.tasks.retry_config import BaseTaskWithRetry
from app.tasks.embed_metadata_into_pdf import embed_metadata_into_pdf
from app.utils import log_task_progress, task_step_logging
from app.utils import log_task_progress, log_task
from app.database import SessionLocal
from app.models import FileRecord
@@ -38,7 +38,7 @@ def extract_json_from_text(text):
return None
@celery.task(base=BaseTaskWithRetry)
@task_step_logging
@log_task("extract_metadata")
def extract_metadata_with_gpt(s3_filename: str, cleaned_text: str):
"""Uses OpenAI to classify document metadata."""
session = SessionLocal()