Update analytics/deep_dive_agentic.py

analytics/deep_dive_agentic.py

@@ -76,6 +76,32 @@ def _extract_json(text: str):
             return None
 
 
+# ---------------------------------------------------
+# HELPER: FIX COMMON PANDAS COMPATIBILITY ISSUES
+# ---------------------------------------------------
+
+def _fix_pandas_compatibility(code: str):
+    """
+    Fix common pandas API compatibility issues in generated code.
+    Handles version differences between pandas versions.
+    """
+    # Fix: .reset_index(name=...) -> .reset_index(names=[...])
+    code = re.sub(
+        r"\.reset_index\(name=(['\"])([^'\"]+)\1\)",
+        r".reset_index(names=[\1\2\1])",
+        code
+    )
+    
+    # Fix: .reset_index(name= with variable
+    code = re.sub(
+        r"\.reset_index\(name=([a-zA-Z_][a-zA-Z0-9_]*)\)",
+        r".reset_index(names=[\1])",
+        code
+    )
+    
+    return code
+
+
 # ---------------------------------------------------
 # STEP 1: CODE GENERATION
 # ---------------------------------------------------
@@ -407,4 +433,4 @@ def run_deep_dive_analysis(question: str, acq: pd.DataFrame, perf: pd.DataFrame,
         "all_results": all_results,
         "interpretation": interpretation,
         "error": None
-    }
+    }