mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-30 13:10:47 +00:00
set source and source id when creating categorisation task
This commit is contained in:
parent
c0b2bf669c
commit
08cf930f91
1 changed files with 3 additions and 0 deletions
|
|
@ -9,6 +9,7 @@ import asyncio
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends
|
from fastapi import APIRouter, Depends
|
||||||
|
from backend.app.db.models.tasks import SourceEnum
|
||||||
from backend.app.dependencies import validate_token
|
from backend.app.dependencies import validate_token
|
||||||
from backend.app.plan.schemas import PlanTriggerRequest
|
from backend.app.plan.schemas import PlanTriggerRequest
|
||||||
from backend.app.config import get_settings
|
from backend.app.config import get_settings
|
||||||
|
|
@ -68,6 +69,8 @@ async def trigger_categorisation(
|
||||||
service="plan_categorisation",
|
service="plan_categorisation",
|
||||||
inputs=payload.model_dump(),
|
inputs=payload.model_dump(),
|
||||||
task_only=True,
|
task_only=True,
|
||||||
|
source=SourceEnum.PORTFOLIO,
|
||||||
|
source_id=str(payload.portfolio_id),
|
||||||
)
|
)
|
||||||
|
|
||||||
# Dispatch requests to lambdas
|
# Dispatch requests to lambdas
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue