diff --git a/wyvern/components/features/feature_retrieval_pipeline.py b/wyvern/components/features/feature_retrieval_pipeline.py index e8a073d..106f66c 100644 --- 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):