Skip to content

Commit 7221ec4

Browse files
committed
Remove the need for enironment variable in mock
1 parent 554baa2 commit 7221ec4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/workflows/recipe/__init__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def unwrap_recipe(header, message):
7474
otel_logs = None
7575
rw = RecipeWrapper(message=message, transport=transport_layer)
7676

77-
if hasattr(rw, "environment") and rw.environment.get("ID"):
77+
if hasattr(rw,"environment") and rw.environment.get("ID"):
7878
# Extract recipe ID from environment and add to current span
7979
span = trace.get_current_span()
8080
recipe_id = rw.environment.get("ID")
@@ -112,8 +112,10 @@ def unwrap_recipe(header, message):
112112
stack.enter_context(
113113
log_extender("recipe_ID", rw.environment.get("ID"))
114114
)
115+
116+
return callback(rw,header,message.get("payload"))
115117

116-
return callback(rw, header, message.get("payload"))
118+
return callback(rw, header, message.get("payload"))
117119

118120
if allow_non_recipe_messages:
119121
return callback(None, header, message)

0 commit comments

Comments
 (0)