improved stack

RasaHQ · Sep 6, 2023 · a3391d6 · a3391d6
1 parent 2d37576
commit a3391d6
Show file tree

Hide file tree

Showing 28 changed files with 1,239 additions and 703 deletions.
diff --git a/rasa/cdu/dialogue_stack.py b/rasa/cdu/dialogue_stack.py
diff --git a/rasa/cdu/generator/llm_command_generator.py b/rasa/cdu/generator/llm_command_generator.py
@@ -4,6 +4,7 @@
 
 from jinja2 import Template
 import structlog
+from rasa.cdu.stack.utils import top_flow_frame
 from rasa.cdu.generator.base import CommandGenerator
 from rasa.cdu.commands import (
     Command,
@@ -16,8 +17,6 @@
     KnowledgeAnswerCommand,
     ClarifyCommand,
 )
-from rasa.cdu.conversation_patterns import FLOW_PATTERN_COLLECT_INFORMATION
-
 from rasa.core.policies.flow_policy import DialogueStack
 from rasa.engine.graph import GraphComponent, ExecutionContext
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
@@ -338,9 +337,7 @@ def render_template(
         flows_without_patterns = FlowsList(
             [f for f in flows.underlying_flows if not f.is_handling_pattern()]
         )
-        top_relevant_frame = DialogueStack.top_frame_on_tracker(
-            tracker, ignore_frame=FLOW_PATTERN_COLLECT_INFORMATION
-        )
+        top_relevant_frame = top_flow_frame(DialogueStack.from_tracker(tracker))
         top_flow = top_relevant_frame.flow(flows) if top_relevant_frame else None
         current_step = top_relevant_frame.step(flows) if top_relevant_frame else None
         if top_flow is not None: