fix v3

Author: AgustinPose

app/agents/wizard_workflow/nodes.py

@@ -1,4 +1,5 @@
 import logging
+import hashlib
 
 from langchain_core.messages import AIMessage
 
@@ -175,7 +176,16 @@ def input_guardrails_node(state: WizardState):
 
     lowered = cleaned.lower()
     if any(pattern in lowered for pattern in GUARDRAIL_BLOCK_PATTERNS):
-        logger.warning("[WIZARD/guardrails] Possible prompt-injection-like answer blocked")
+        msg_preview = cleaned[:64]
+        msg_hash = hashlib.sha256(cleaned.encode("utf-8")).hexdigest()[:12]
+        logger.warning(
+            "[WIZARD/guardrails] Possible prompt-injection-like answer blocked "
+            "session_id=%s current_question=%s msg_preview=%r msg_hash=%s",
+            state.get("wizard_session_id"),
+            current_q,
+            msg_preview,
+            msg_hash,
+        )
         return {
             **state,
             "messages": [

app/agents/wizard_workflow/wizard_graph.py

@@ -58,7 +58,12 @@ def should_ask_or_store(state: WizardState) -> str:
 
 def should_store_after_guardrails(state: WizardState) -> str:
     is_valid = state.get("valid") is True
-    decision = "store_answer" if is_valid else "finish"
+    if is_valid:
+        decision = "store_answer"
+    elif not state.get("awaiting_answer", False):
+        decision = "ask_question"
+    else:
+        decision = "finish"
     logger.debug(f"[WIZARD_GRAPH] should_store_after_guardrails: valid={is_valid} -> {decision}")
     return decision
 
@@ -103,6 +108,7 @@ def completion_message_node(state: WizardState):
     should_store_after_guardrails,
     {
         "store_answer": "store_answer",
+        "ask_question": "ask_question",
         "finish": "finish",
     },
 )