Function active_learning_candidates

sdk/python/feast/ui_server.py:428–585

Find entities that exist in a reference feature view but have NOT been labeled yet.

(request: ActiveLearningRequest)

    @rest_app.post("/active-learning/candidates")
    def active_learning_candidates(request: ActiveLearningRequest):
        """Find entities that exist in a reference feature view but have NOT been labeled yet."""
        try:
            fv_name = request.feature_view
            fv: Any = None
            try:
                fv = store.registry.get_label_view(fv_name, store.project)
            except Exception:
                pass

            if fv is None:
                return Response(
                    content=json.dumps({"detail": f"Label view '{fv_name}' not found"}),
                    status_code=status.HTTP_404_NOT_FOUND,
                    media_type="application/json",
                )

            ref_fv_name = request.reference_feature_view or (
                getattr(fv, "reference_feature_view", None) or ""
            )

            if not ref_fv_name:
                return Response(
                    content=json.dumps(
                        {
                            "detail": "No reference feature view specified. "
                            "Provide a feature view containing all entities."
                        }
                    ),
                    status_code=status.HTTP_400_BAD_REQUEST,
                    media_type="application/json",
                )

            ref_fv: Any = None
            try:
                ref_fv = store.get_feature_view(ref_fv_name)
            except Exception:
                pass

            if ref_fv is None:
                return Response(
                    content=json.dumps(
                        {"detail": f"Reference feature view '{ref_fv_name}' not found"}
                    ),
                    status_code=status.HTTP_404_NOT_FOUND,
                    media_type="application/json",
                )

            provider = store._get_provider()

            # Get all entities from reference feature view
            ref_batch_source = getattr(ref_fv, "batch_source", None)
            if ref_batch_source is None:
                return Response(
                    content=json.dumps(
                        {
                            "detail": f"No batch source for reference view '{ref_fv_name}'"
                        }