From 7222da1230422440dec74267999e4dff882db29b Mon Sep 17 00:00:00 2001 From: Bryan Helmkamp Date: Mon, 23 Feb 2026 22:12:29 -0500 Subject: [PATCH] Store response. in context after each LLM node Downstream nodes in multi-model pipelines reference response. context keys in their prompts. The codergen handler was only storing last_response (truncated) and last_stage, so these references resolved to nothing. Co-Authored-By: Claude Opus 4.6 --- crates/attractor/src/handler/codergen.rs | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/crates/attractor/src/handler/codergen.rs b/crates/attractor/src/handler/codergen.rs index c94959a53..9a85b8955 100644 --- a/crates/attractor/src/handler/codergen.rs +++ b/crates/attractor/src/handler/codergen.rs @@ -246,6 +246,10 @@ impl Handler for CodergenHandler { "last_response".to_string(), serde_json::json!(truncate(&response_text, 200)), ); + outcome.context_updates.insert( + format!("response.{}", node.id), + serde_json::json!(&response_text), + ); // 7b. Parse routing directives from response text extract_status_fields(&response_text, &mut outcome); @@ -375,6 +379,10 @@ mod tests { Some(&serde_json::json!("step")) ); assert!(outcome.context_updates.contains_key("last_response")); + assert_eq!( + outcome.context_updates.get("response.step"), + Some(&serde_json::json!("[Simulated] Response for stage: step")) + ); } #[test]