Update code

response_formatter.py

@@ -1,5 +1,5 @@
 import re
-from typing import Optional, List
+from typing import List, Optional
 
 from schemas import CodeTaskType, CodeXResponse, ResponseMeta, RetrievedEvidence
 
@@ -38,8 +38,6 @@ def extract_explanation(text: str, task_type: CodeTaskType) -> Optional[str]:
     if task_type == CodeTaskType.FIX:
         root_cause = extract_section(text, "Root Cause")
         explanation = extract_section(text, "Explanation")
-        if root_cause and explanation:
-            return explanation
         return explanation or root_cause
 
     if task_type == CodeTaskType.REVIEW:
@@ -49,13 +47,11 @@ def extract_explanation(text: str, task_type: CodeTaskType) -> Optional[str]:
             return f"{review}\n\nSuggestions:\n{suggestions}"
         return review or suggestions
 
-    if task_type == CodeTaskType.REFACTOR:
-        return extract_section(text, "Explanation")
-
-    if task_type == CodeTaskType.EXPLAIN:
-        return extract_section(text, "Explanation")
-
-    if task_type == CodeTaskType.GENERATE:
+    if task_type in {
+        CodeTaskType.REFACTOR,
+        CodeTaskType.EXPLAIN,
+        CodeTaskType.GENERATE,
+    }:
         return extract_section(text, "Explanation")
 
     return (
@@ -133,7 +129,7 @@ def extract_original_callable_name(raw_text: str) -> Optional[str]:
         r"\bfunction\s+([A-Za-z_][A-Za-z0-9_]*)\s*\(",
     ]
     for pattern in patterns:
-        match = re.search(pattern, raw_text)
+        match = re.search(pattern, raw_text or "")
         if match:
             return match.group(1)
     return None
@@ -191,6 +187,7 @@ def should_include_code_output(
                 "could improve",
                 "readability",
                 "maintainability",
+                "edge case",
             ]
             if any(signal in lowered for signal in review_only_signals):
                 return None
@@ -294,6 +291,9 @@ def build_response(
         if original_name and final_name and original_name != final_name:
             warnings.append("Refactor output may have changed original naming unexpectedly.")
 
+    if task_type == CodeTaskType.FIX and not code_output and cleaned_output:
+        warnings.append("Fix response did not include code output.")
+
     return CodeXResponse(
         task_type=task_type,
         answer=answer,