diff --git a/sdk/python/feast/api/registry/rest/search.py b/sdk/python/feast/api/registry/rest/search.py index 6e592ac86d4..e670a8981ac 100644 --- a/sdk/python/feast/api/registry/rest/search.py +++ b/sdk/python/feast/api/registry/rest/search.py @@ -167,6 +167,7 @@ def search_resources( "name": feature.get("name", ""), "description": feature.get("description", ""), "project": current_project, + "featureView": feature.get("featureView", ""), "tags": feature.get("tags", {}), } ) diff --git a/sdk/python/tests/unit/api/test_search_api.py b/sdk/python/tests/unit/api/test_search_api.py index 06b670a16c5..48e422c5fe1 100644 --- a/sdk/python/tests/unit/api/test_search_api.py +++ b/sdk/python/tests/unit/api/test_search_api.py @@ -802,6 +802,15 @@ def test_search_query_functionality(self, shared_search_responses): "Expected individual features to appear in search results, but found none" ) + for feature_result in feature_results: + assert "featureView" in feature_result + assert feature_result["featureView"] in [ + "user_features", + "product_features", + "transaction_features", + "user_on_demand_features", + ] + # Verify we have features that likely come from different feature views feature_names = {f["name"] for f in feature_results}