diff --git a/backend/app/plan/router.py b/backend/app/plan/router.py index cc9fb579..a90c391a 100644 --- a/backend/app/plan/router.py +++ b/backend/app/plan/router.py @@ -9,6 +9,7 @@ import asyncio from datetime import datetime from fastapi import APIRouter, Depends +from backend.app.db.models.tasks import SourceEnum from backend.app.dependencies import validate_token from backend.app.plan.schemas import PlanTriggerRequest from backend.app.config import get_settings @@ -68,6 +69,8 @@ async def trigger_categorisation( service="plan_categorisation", inputs=payload.model_dump(), task_only=True, + source=SourceEnum.PORTFOLIO, + source_id=str(payload.portfolio_id), ) # Dispatch requests to lambdas