Function push_labels

sdk/python/feast/ui_server.py:92–118

Push label data to a LabelView (or any PushSource-backed FeatureView).

(request: PushRequest)

    @rest_app.post("/push")
    def push_labels(request: PushRequest):
        """Push label data to a LabelView (or any PushSource-backed FeatureView)."""
        try:
            df = pd.DataFrame(request.df)
            if "event_timestamp" in df.columns:
                df["event_timestamp"] = pd.to_datetime(
                    df["event_timestamp"], utc=True
                ).dt.tz_localize(None)
            to = request.to or "online"
            if to == "online_and_offline":
                store.push(
                    request.push_source_name,
                    df,
                    to=feast.data_source.PushMode.ONLINE_AND_OFFLINE,
                )
            elif to == "offline":
                store.push(
                    request.push_source_name, df, to=feast.data_source.PushMode.OFFLINE
                )
            else:
                store.push(
                    request.push_source_name, df, to=feast.data_source.PushMode.ONLINE
                )
            return {"status": "ok"}
        except Exception:
            logger.exception("Push failed")
            return _safe_error_response("Push")

    class GetOnlineFeaturesRequest(BaseModel):
        feature_view: str