mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
21 lines
691 B
Python
21 lines
691 B
Python
import json
|
|
import asyncio
|
|
from backend.engine.engine import model_engine
|
|
from backend.app.plan.schemas import PlanTriggerRequest
|
|
from utils.logger import setup_logger
|
|
|
|
logger = setup_logger()
|
|
|
|
|
|
def handler(event, context):
|
|
"""
|
|
Lambda handler that triggers the model engine for each SQS message.
|
|
"""
|
|
logger.info("Received event: %s", json.dumps(event, indent=2))
|
|
for record in event.get("Records", []):
|
|
try:
|
|
body_dict = json.loads(record["body"])
|
|
payload = PlanTriggerRequest.model_validate(body_dict)
|
|
asyncio.run(model_engine(payload))
|
|
except Exception as e:
|
|
logger.error(f"Failed to process record: {e}")
|