Update main.py

main.py

@@ -1026,12 +1026,14 @@ async def chat_with_ai(input_data: ChatInput):
         context_preformatted = False
 
         if best_doc and detected_intent == "steps":
-            full_steps = get_best_steps_section_text(best_doc)
-            if not full_steps:
-                sec = (top_meta or {}).get("section")
-                if sec:
-                    full_steps = get_section_text(best_doc, sec)
-
+            sec = (top_meta or {}).get("section")
+            if sec:
+        # Prefer the exact section of the top hit
+              full_steps = get_section_text(best_doc, sec)
+            else:
+        # Fallback to all steps (rare)
+              full_steps = get_best_steps_section_text(best_doc)
+            
             if full_steps:
                 numbered_full = _ensure_numbering(full_steps)
                 raw_actions = set((kb_results.get("actions") or []))