make feature store features available in realtime feature computation

Skyvern-AI · Sep 17, 2023 · 74510f1 · 74510f1
1 parent 91ddb8e
commit 74510f1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/wyvern/components/features/feature_retrieval_pipeline.py b/wyvern/components/features/feature_retrieval_pipeline.py
@@ -6,6 +6,7 @@
 from ddtrace import tracer
 from pydantic.generics import GenericModel
 
+from wyvern import request_context
 from wyvern.components.component import Component
 from wyvern.components.features.feature_logger import (
     FeatureEventLoggingComponent,
@@ -156,6 +157,8 @@ async def execute(
                 **kwargs,
             )
         )
+        current_request = request_context.ensure_current_request()
+        current_request.feature_map = feature_retrieval_response
 
         """
         TODO (suchintan):