From f8c4f542c1bfcbb4f214f310d3c8b839e8f4c308 Mon Sep 17 00:00:00 2001 From: Igor Loskutov Date: Tue, 23 Dec 2025 17:44:45 -0500 Subject: [PATCH] cleanup --- server/reflector/hatchet/run_workers.py | 6 +++--- server/reflector/hatchet/workflows/__init__.py | 6 +++--- .../hatchet/workflows/daily_multitrack_pipeline.py | 6 +++--- server/reflector/utils/webhook.py | 8 -------- 4 files changed, 9 insertions(+), 17 deletions(-) diff --git a/server/reflector/hatchet/run_workers.py b/server/reflector/hatchet/run_workers.py index 243df7bf..e6f21653 100644 --- a/server/reflector/hatchet/run_workers.py +++ b/server/reflector/hatchet/run_workers.py @@ -1,5 +1,5 @@ """ -Run Hatchet workers for the diarization pipeline. +Run Hatchet workers for the multitrack pipeline. Runs as a separate process, just like Celery workers. Usage: @@ -39,7 +39,7 @@ def main() -> None: # Can't use lazy init: decorators need the client object when function is defined. from reflector.hatchet.client import HatchetClientManager # noqa: PLC0415 from reflector.hatchet.workflows import ( # noqa: PLC0415 - diarization_pipeline, + daily_multitrack_pipeline, subject_workflow, topic_chunk_workflow, track_workflow, @@ -54,7 +54,7 @@ def main() -> None: worker = hatchet.worker( "reflector-pipeline-worker", workflows=[ - diarization_pipeline, + daily_multitrack_pipeline, subject_workflow, topic_chunk_workflow, track_workflow, diff --git a/server/reflector/hatchet/workflows/__init__.py b/server/reflector/hatchet/workflows/__init__.py index df780872..ea242ad6 100644 --- a/server/reflector/hatchet/workflows/__init__.py +++ b/server/reflector/hatchet/workflows/__init__.py @@ -1,8 +1,8 @@ """Hatchet workflow definitions.""" -from reflector.hatchet.workflows.diarization_pipeline import ( +from reflector.hatchet.workflows.daily_multitrack_pipeline import ( PipelineInput, - diarization_pipeline, + daily_multitrack_pipeline, ) from reflector.hatchet.workflows.subject_processing import ( SubjectInput, @@ -15,7 +15,7 @@ from reflector.hatchet.workflows.topic_chunk_processing import ( from reflector.hatchet.workflows.track_processing import TrackInput, track_workflow __all__ = [ - "diarization_pipeline", + "daily_multitrack_pipeline", "subject_workflow", "topic_chunk_workflow", "track_workflow", diff --git a/server/reflector/hatchet/workflows/daily_multitrack_pipeline.py b/server/reflector/hatchet/workflows/daily_multitrack_pipeline.py index 167facc5..a201c3d1 100644 --- a/server/reflector/hatchet/workflows/daily_multitrack_pipeline.py +++ b/server/reflector/hatchet/workflows/daily_multitrack_pipeline.py @@ -108,7 +108,7 @@ class PipelineInput(BaseModel): hatchet = HatchetClientManager.get_client() daily_multitrack_pipeline = hatchet.workflow( - name="DailyMultitrackPipeline", input_validator=PipelineInput + name="DiarizationPipeline", input_validator=PipelineInput ) @@ -1225,9 +1225,9 @@ async def post_zulip(input: PipelineInput, ctx: Context) -> ZulipResult: @daily_multitrack_pipeline.task( - parents=[post_zulip], + parents=[cleanup_consent], execution_timeout=timedelta(seconds=TIMEOUT_MEDIUM), - retries=30, + retries=5, ) @with_error_handling("send_webhook", set_error_status=False) async def send_webhook(input: PipelineInput, ctx: Context) -> WebhookResult: diff --git a/server/reflector/utils/webhook.py b/server/reflector/utils/webhook.py index 1957dbb3..6afdf18c 100644 --- a/server/reflector/utils/webhook.py +++ b/server/reflector/utils/webhook.py @@ -24,14 +24,6 @@ from reflector.utils.webhook_outgoing_models import ( ) __all__ = [ - "WebhookCalendarEventPayload", - "WebhookParticipantPayload", - "WebhookPayload", - "WebhookRoomPayload", - "WebhookTestPayload", - "WebhookTopicPayload", - "WebhookTranscriptPayload", - "_serialize_payload", "build_transcript_webhook_payload", "build_test_webhook_payload", "build_webhook_headers",