From a92cace6624a0943f52271135586eb0e5cfbe055 Mon Sep 17 00:00:00 2001 From: Christian Krakau-Louis Date: Fri, 28 Mar 2025 16:26:03 +0100 Subject: [PATCH] refactor: replace task_step_logging with log_task for metadata extraction --- app/tasks/extract_metadata_with_gpt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/tasks/extract_metadata_with_gpt.py b/app/tasks/extract_metadata_with_gpt.py index dcb4c795..5e6ae480 100644 --- a/app/tasks/extract_metadata_with_gpt.py +++ b/app/tasks/extract_metadata_with_gpt.py @@ -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()