add variables from the msg saved run to analysis run

GooeyAI · Jul 21, 2024 · 5538240 · 5538240
1 parent 0bdb9e9
commit 5538240
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bots/tasks.py b/bots/tasks.py
@@ -67,8 +67,10 @@ def msg_analysis(self, msg_id: int, anal_id: int, countdown: int | None):
     analysis_sr = anal.get_active_saved_run()
     variables = analysis_sr.state.get("variables", {})
 
-    # add the state variables requested by the script
     if msg.saved_run:
+        # add the variables from the parent run
+        variables |= msg.saved_run.state.get("variables") or {}
+        # add the state variables requested by the script
         fill_req_vars_from_state(msg.saved_run.state, variables)
     if "messages" in variables:
         variables["messages"] = messages_as_prompt(