From 889362bad41adcbe9e6ad445f161875caf200cd1 Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Mon, 3 Feb 2025 14:11:52 -0500 Subject: [PATCH] updated feature server Signed-off-by: Francisco Javier Arceo --- sdk/python/feast/feature_server.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py index 20b0e3e44d..ed742bcb98 100644 --- a/sdk/python/feast/feature_server.py +++ b/sdk/python/feast/feature_server.py @@ -74,14 +74,9 @@ class GetOnlineFeaturesRequest(BaseModel): feature_service: Optional[str] = None features: Optional[List[str]] = None full_feature_names: bool = False - -class RetrieveOnlineDocumentsRequest(BaseModel): - entities: Dict[str, List[Any]] - feature_service: Optional[str] = None - features: Optional[List[str]] = None - full_feature_names: bool = False query_embedding: Optional[List[float]] = None + def _get_features(request: GetOnlineFeaturesRequest, store: "feast.FeatureStore"): if request.feature_service: feature_service = store.get_feature_service( @@ -110,7 +105,6 @@ def _get_features(request: GetOnlineFeaturesRequest, store: "feast.FeatureStore" resource=od_feature_view, actions=[AuthzedAction.READ_ONLINE] ) features = request.features # type: ignore - return features @@ -187,8 +181,12 @@ async def get_online_features(request: GetOnlineFeaturesRequest) -> Dict[str, An "/retrieve-online-documents", dependencies=[Depends(inject_user_details)], ) - async def retrieve_online_documents(request: RetrieveOnlineDocumentsRequest) -> Dict[str, Any]: - logger.warn("This endpoint is in alpha and will be moved to /get-online-features when stable.") + async def retrieve_online_documents( + request: GetOnlineFeaturesRequest, + ) -> Dict[str, Any]: + logger.warn( + "This endpoint is in alpha and will be moved to /get-online-features when stable." + ) # Initialize parameters for FeatureStore.retrieve_online_documents_v2(...) call features = await run_in_threadpool(_get_features, request, store)