master_agent.py

# %%
from datetime import datetime, timedelta

class MasterAgent:
    def handle(self, command, context):
        if command != "schedule":
            return {}

        sensor_site = context["sensor_site"]
        sensor_date = context["sensor_date"]
        pod_site = context["pod_site"]
        pod_date = context["pod_date"]

        # Dummy logic for now—replace with your real scheduling later
        sensor_schedule = [{
            "date": sensor_date,
            "site": sensor_site,
            "side": "left" if "Left" in sensor_site else "right"
        }]

        pod_schedule = [{
            "date": pod_date,
            "pod_site": pod_site,
            "sensor_site": sensor_site
        }]

        return {
            "sensor_schedule": sensor_schedule,
            "pod_schedule": pod_schedule
        }