fix issues

Signed-off-by: Vanshika Vanshika <[email protected]>

rh-pre-commit.version: 2.3.2
rh-pre-commit.check-secrets: ENABLED

Author: Vperiodt

sdk/python/feast/feature_store.py

@@ -3003,7 +3003,11 @@ def _doc_feature(x):
             online_features_response=online_features_response,
             data=requested_features_data,
         )
-        return OnlineResponse(online_features_response)
+        feature_types = {
+            f.name: f.dtype.to_value_type()
+            for f in requested_feature_view.features
+        }
+        return OnlineResponse(online_features_response, feature_types=feature_types)
 
     def retrieve_online_documents_v2(
         self,
@@ -3293,7 +3297,12 @@ def _retrieve_from_online_store_v2(
             online_features_response.metadata.feature_names.val.extend(
                 features_to_request
             )
-            return OnlineResponse(online_features_response)
+            feature_types = {
+                f.name: f.dtype.to_value_type() for f in table.features
+            }
+            return OnlineResponse(
+                online_features_response, feature_types=feature_types
+            )
 
         table_entity_values, idxs, output_len = utils._get_unique_entities_from_values(
             entity_key_dict,
@@ -3316,7 +3325,10 @@ def _retrieve_from_online_store_v2(
             data=entity_key_dict,
         )
 
-        return OnlineResponse(online_features_response)
+        feature_types = {
+            f.name: f.dtype.to_value_type() for f in table.features
+        }
+        return OnlineResponse(online_features_response, feature_types=feature_types)
 
     def serve(
         self,