retry after a fail

prompts/doc_editor.py

@@ -11,15 +11,17 @@ DOC_EDITOR_SYSTEM_PROMPT = """You are a Document Editing Agent specialized in mo
 
 2. **NEVER OUTPUT THE FULL DOCUMENT**: Never output the complete document text. The system always has the current document state.
 
-3. **EXACT MATCHING**: When using replace/add/delete, you must copy the EXACT HTML block (including all whitespace, tags, attributes, and attributes values). The search/anchor must match exactly.
+3. **DOCUMENT SUMMARIES ARE FOR CONTEXT ONLY**: Document summaries are provided for informational purposes only. Do NOT use them unless the user instruction explicitly references them. Focus on editing the current document based on the instruction. Only incorporate content from summaries if the user specifically asks you to use them.
 
-4. **ALWAYS INCLUDE expected_matches**: Always specify the expected_matches parameter (usually 1) to prevent unintended multiple replacements.
+4. **EXACT MATCHING**: When using replace/add/delete, you must copy the EXACT HTML block (including all whitespace, tags, attributes, and attributes values). The search/anchor must match exactly.
 
-5. **SMALL, PRECISE EDITS**: Make small, targeted edits. Don't try to replace large blocks - select the smallest unique fragment that identifies what you want to change.
+5. **ALWAYS INCLUDE expected_matches**: Always specify the expected_matches parameter (usually 1) to prevent unintended multiple replacements.
 
-6. **VALIDATE YOUR CHOICES**: If a tool returns an error (mismatch or invalid HTML), analyze the error and try again with a more precise search/anchor.
+6. **SMALL, PRECISE EDITS**: Make small, targeted edits. Don't try to replace large blocks - select the smallest unique fragment that identifies what you want to change.
 
-7. **CALL attempt_completion WHEN DONE**: Once all modifications are successfully applied, call attempt_completion with a summary message.
+7. **VALIDATE YOUR CHOICES**: If a tool returns an error (mismatch or invalid HTML), analyze the error and try again with a more precise search/anchor.
+
+8. **CALL attempt_completion WHEN DONE**: Once all modifications are successfully applied, call attempt_completion with a summary message.
 
 ## AVAILABLE TOOLS
 
@@ -106,11 +108,29 @@ Example:
 }
 ```
 
-## ERROR HANDLING
+## ERROR HANDLING AND RETRY MECHANISM
+
+**Important:** When a tool execution fails, you will automatically receive another chance to try. The system will show you the error message and you should analyze it carefully.
+
+Common errors and how to fix them:
+- "Match count mismatch": Your search/anchor didn't match exactly in the document. 
+  - Check whitespace, tags, attributes, and attribute values
+  - Try a more specific or different fragment
+  - Use the current document content provided in the conversation to find the exact text
+
+- "Post-edit validation failed": The replacement broke the HTML structure.
+  - Ensure your insert/replace is valid HTML with proper opening and closing tags
+  - Check that you're not creating malformed HTML (e.g., unclosed tags)
+  - Verify the structure is valid
+
+**Workflow on errors:**
+1. Read the error message carefully
+2. Look at the current document state in the conversation history
+3. Understand what went wrong
+4. Try again with a corrected approach (different search pattern, different anchor, fixed HTML, etc.)
+5. Continue retrying until you succeed or reach max_iterations
 
-If you receive an error:
-- "Match count mismatch": Your search/anchor didn't match exactly. Check whitespace, tags, attributes, and attribute values. Try a more specific or different fragment.
-- "Post-edit validation failed": The replacement broke the HTML structure. Ensure your insert/replace is valid HTML with proper tags.
+You get multiple chances (up to max_iterations) to complete the task, so don't worry about failures - learn from them and try again!
 
 ## WORKFLOW
 

subagents/doc_editor.py

@@ -86,8 +86,9 @@ class DocumentEditorAgent:
         Returns:
             "continue" if more iterations needed, "end" if complete or max iterations reached
         """
-        # Check if attempt_completion was called
         intermediate_steps = state.get("intermediate_steps", [])
+        
+        # Check if attempt_completion was called successfully
         if intermediate_steps:
             for msg in reversed(intermediate_steps):
                 if isinstance(msg, ToolMessage):
@@ -95,6 +96,17 @@ class DocumentEditorAgent:
                     if msg.name == "attempt_completion":
                         logger.info("✅ attempt_completion called - ending workflow")
                         return "end"
+                    
+                    # Check if the last tool execution failed
+                    # If so, we should continue to let the agent retry
+                    if msg.name in ["replace_html", "add_html", "delete_html"]:
+                        content = msg.content
+                        # Parse the tool result to check if it was successful
+                        if isinstance(content, str) and '"ok": false' in content.lower():
+                            logger.info("⚠️ Last tool execution failed - allowing agent to retry")
+                            # Continue to give the agent a chance to correct itself
+                            # Don't check max_iterations yet - we want retries
+                            return "continue"
         
         # Check max iterations
         iteration_count = state.get("iteration_count", 0)
@@ -326,17 +338,7 @@ class DocumentEditorAgent:
             logger.info("=" * 80)
             logger.info("📄 FINAL DOCUMENT CONTENT")
             logger.info("=" * 80)
-            if final_doc_size <= 1000:
-                # Log full document if small enough
-                logger.info(final_state["doc_text"])
-            else:
-                # Log first 500 and last 500 characters for readability
-                doc_content = final_state["doc_text"]
-                logger.info(f"[BEGINNING - first 500 chars]")
-                logger.info(doc_content[:500])
-                logger.info(f"...")
-                logger.info(f"[ENDING - last 500 chars]")
-                logger.info(doc_content[-500:])
+            logger.info(final_state["doc_text"])
             logger.info("=" * 80)
             
             if not success: