1

video-gen-api/app/tasks/generation_poll_tasks.py

@@ -19,8 +19,8 @@ from app.enums.generation_task import (
 from app.models.base import async_session
 from app.models.chat_generation_task import ChatGenerationTask
 from app.services.error_codes import extract_error_message
-from app.services.generation_log_service import log_task_event, log_provider_call
-from app.services.generation_poll_schedule_service import (
+from app.services.generation.log_service import log_task_event, log_provider_call
+from app.services.generation.poll_schedule_service import (
     build_default_poll_schedule,
     build_video_pending_poll_schedule,
     ensure_video_poll_fields,
@@ -28,8 +28,8 @@ from app.services.generation_poll_schedule_service import (
     is_poll_not_due,
     is_video_generation_task,
 )
-from app.services.generation_refund_service import mark_chat_generation_task_failed_and_refund_once
-from app.services.generation_provider_service import poll_provider_task
+from app.services.generation.refund_service import mark_chat_generation_task_failed_and_refund_once
+from app.services.generation.provider_service import poll_provider_task
 from app.services.media_token_usage_snapshot_service import sync_chat_generation_task_media_token_snapshot
 from app.services.redis_registry_service import (
     datetime_to_epoch,
@@ -144,9 +144,11 @@ async def remove_poll_active(task_id: str) -> None:
 
 
 async def _notify_finished(db, task: ChatGenerationTask) -> None:
-    from app.services.generation_module_hook_service import notify_chat_generation_task_finished
+    from app.services.generation.module_hook_service import notify_chat_generation_task_finished
+    from app.services.generation.ai.task_group_service import aggregate_parent_for_child
 
     await notify_chat_generation_task_finished(db, task)
+    await aggregate_parent_for_child(db, task)
 
 
 async def _reload_task(db, task_id: str) -> ChatGenerationTask | None: