studio/meet
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

♻️(summary) rename TaskTracker to MetadataManager for clarity

Browse Source 
Update class name to better reflect its responsibility for handling task
metadata rather than tracking tasks themselves.
This commit is contained in:
lebaudantoine
2025-07-11 10:18:20 +02:00
committed by aleb_the_flash
parent e458272745
commit 731f0471aa
2 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/summary/summary/core/analytics.py
View File

@@ -55,8 +55,8 @@ def get_analytics():
return Analytics() return Analytics()
class TasksTracker: class MetadataManager:
"""Tracks task execution metadata and analytics for background tasks.""" """A Redis-based metadata manager for storing and retrieving task metadata."""
def __init__(self): def __init__(self):
"""Initialize the task tracker with analytics client.""" """Initialize the task tracker with analytics client."""

18
src/summary/summary/core/celery_worker.py
View File

@@ -16,14 +16,14 @@ from requests import Session, exceptions
from requests.adapters import HTTPAdapter from requests.adapters import HTTPAdapter
from urllib3.util import Retry from urllib3.util import Retry
from summary.core.analytics import TasksTracker, get_analytics from summary.core.analytics import MetadataManager, get_analytics
from summary.core.config import get_settings from summary.core.config import get_settings
from summary.core.prompt import get_instructions from summary.core.prompt import get_instructions
settings = get_settings() settings = get_settings()
analytics = get_analytics() analytics = get_analytics()
tasks_tracker = TasksTracker() metadata_manager = MetadataManager()
logger = get_task_logger(__name__) logger = get_task_logger(__name__)
@@ -136,19 +136,19 @@ def post_with_retries(url, data):
def task_started(task_id=None, task=None, args=None, **kwargs): def task_started(task_id=None, task=None, args=None, **kwargs):
"""Signal handler called before task execution begins.""" """Signal handler called before task execution begins."""
task_args = args or [] task_args = args or []
tasks_tracker.create(task_id, task_args) metadata_manager.create(task_id, task_args)
@signals.task_retry.connect @signals.task_retry.connect
def task_retry_handler(request=None, reason=None, einfo=None, **kwargs): def task_retry_handler(request=None, reason=None, einfo=None, **kwargs):
"""Signal handler called when task execution retries.""" """Signal handler called when task execution retries."""
tasks_tracker.retry(request.id) metadata_manager.retry(request.id)
@signals.task_failure.connect @signals.task_failure.connect
def task_failure_handler(task_id, exception=None, **kwargs): def task_failure_handler(task_id, exception=None, **kwargs):
"""Signal handler called when task execution fails permanently.""" """Signal handler called when task execution fails permanently."""
tasks_tracker.capture(task_id, settings.posthog_event_failure) metadata_manager.capture(task_id, settings.posthog_event_failure)
@celery.task(max_retries=settings.celery_max_retries) @celery.task(max_retries=settings.celery_max_retries)
@@ -267,7 +267,7 @@ def process_audio_transcribe_summarize_v2(
logger.debug("Recording filepath: %s", temp_file_path) logger.debug("Recording filepath: %s", temp_file_path)
audio_file = File(temp_file_path) audio_file = File(temp_file_path)
tasks_tracker.track(task_id, {"audio_length": audio_file.info.length}) metadata_manager.track(task_id, {"audio_length": audio_file.info.length})
if audio_file.info.length > settings.recording_max_duration: if audio_file.info.length > settings.recording_max_duration:
error_msg = "Recording too long: %.2fs > %.2fs limit" % ( error_msg = "Recording too long: %.2fs > %.2fs limit" % (
@@ -291,7 +291,7 @@ def process_audio_transcribe_summarize_v2(
transcription = openai_client.audio.transcriptions.create( transcription = openai_client.audio.transcriptions.create(
model=settings.openai_asr_model, file=audio_file model=settings.openai_asr_model, file=audio_file
) )
tasks_tracker.track( metadata_manager.track(
task_id, task_id,
{ {
"transcription_time": round( "transcription_time": round(
@@ -312,7 +312,7 @@ def process_audio_transcribe_summarize_v2(
else format_segments(transcription) else format_segments(transcription)
) )
tasks_tracker.track_transcription_metadata(task_id, transcription) metadata_manager.track_transcription_metadata(task_id, transcription)
data = { data = {
"title": "Transcription", "title": "Transcription",
@@ -329,6 +329,6 @@ def process_audio_transcribe_summarize_v2(
logger.info("Webhook submitted successfully. Status: %s", response.status_code) logger.info("Webhook submitted successfully. Status: %s", response.status_code)
logger.debug("Response body: %s", response.text) logger.debug("Response body: %s", response.text)
tasks_tracker.capture(task_id, settings.posthog_event_success) metadata_manager.capture(task_id, settings.posthog_event_success)
# TODO - integrate summarize the transcript and create a new document. # TODO - integrate summarize the transcript and create a new document.