fix: Fixed output of discovery stage so it behaves correctly.

src/examples/scientific_paper_writer.yaml

@@ -460,48 +460,47 @@ routes:
   main_workflow:
     - from: input
       to: workflow_controller
-      context: '{"user_command": {{ message | tojson }} }'
 
     # --- Stage Dispatcher ---
     - from: workflow_controller
       to: command_handler
       condition: "'GOTO_COMMAND_HANDLER' in message"
-      transform: "{{ context.user_command }}"
+      transform: "{{ context.initial_message }}"
 
     - from: workflow_controller
       to: intro_agent
       condition: "'GOTO_INTRO' in message"
-      transform: "{{ context.user_command }}"
+      transform: "{{ context.initial_message }}"
 
     - from: workflow_controller
       to: discovery_handler
       condition: "'GOTO_DISCOVERY' in message"
-      transform: "{{ context.user_command }}"
+      transform: "{{ context.initial_message }}"
 
     - from: workflow_controller
       to: brainstorming_agent
       condition: "'GOTO_BRAINSTORMING' in message"
-      transform: "{{ context.user_command }}"
+      transform: "{{ context.initial_message }}"
 
     - from: workflow_controller
       to: vetting_agent
       condition: "'GOTO_VETTING' in message"
-      transform: "{{ context.user_command }}"
+      transform: "{{ context.initial_message }}"
 
     - from: workflow_controller
       to: structure_agent
       condition: "'GOTO_STRUCTURE' in message"
-      transform: "{{ context.user_command }}"
+      transform: "{{ context.initial_message }}"
 
     - from: workflow_controller
       to: core_content_handler
       condition: "'GOTO_CORE_CONTENT' in message"
-      transform: "{{ context.user_command }}"
+      transform: "{{ context.initial_message }}"
 
     - from: workflow_controller
       to: formatting_handler
       condition: "'GOTO_FORMATTING' in message"
-      transform: "{{ context.user_command }}"
+      transform: "{{ context.initial_message }}"
 
     # --- Stage Outputs & Transitions ---
     - from: command_handler
@@ -527,7 +526,7 @@ routes:
       condition: "context['paper_details']['topic'] == None"
 
     - from: ask_topic
-      to: ask_length
+      to: output
       context: '{"paper_details": {"topic": {{ message | tojson }}, "length": {{ context.paper_details.length | tojson }}, "audience": {{ context.paper_details.audience | tojson }}, "publication": {{ context.paper_details.publication | tojson }}, "format": {{ context.paper_details.format | tojson }}}}'
 
     - from: input
@@ -535,7 +534,7 @@ routes:
       condition: "context['paper_details']['length'] == None and context['paper_details']['topic'] != None"
 
     - from: ask_length
-      to: ask_audience
+      to: output
       context: '{"paper_details": {"topic": {{ context.paper_details.topic | tojson }}, "length": {{ message | tojson }}, "audience": {{ context.paper_details.audience | tojson }}, "publication": {{ context.paper_details.publication | tojson }}, "format": {{ context.paper_details.format | tojson }}}}'
 
     - from: input
@@ -543,7 +542,7 @@ routes:
       condition: "context['paper_details']['audience'] == None and context['paper_details']['length'] != None"
 
     - from: ask_audience
-      to: ask_publication
+      to: output
       context: '{"paper_details": {"topic": {{ context.paper_details.topic | tojson }}, "length": {{ context.paper_details.length | tojson }}, "audience": {{ message | tojson }}, "publication": {{ context.paper_details.publication | tojson }}, "format": {{ context.paper_details.format | tojson }}}}'
 
     - from: input
@@ -551,7 +550,7 @@ routes:
       condition: "context['paper_details']['publication'] == None and context['paper_details']['audience'] != None"
 
     - from: ask_publication
-      to: ask_format
+      to: output
       context: '{"paper_details": {"topic": {{ context.paper_details.topic | tojson }}, "length": {{ context.paper_details.length | tojson }}, "audience": {{ context.paper_details.audience | tojson }}, "publication": {{ message | tojson }}, "format": {{ context.paper_details.format | tojson }}}}'
 
     - from: input