Update diagnostics.py

diagnostics.py

@@ -1,6 +1,6 @@
 """
-Diagnostics module for the Pharmaceutical Analytics application.
-Contains functions to test LLM connectivity and agent functionality.
+Updated diagnostics module for the Pharmaceutical Analytics application.
+Uses the simplified Planning Agent implementation.
 """
 
 import streamlit as st
@@ -50,10 +50,9 @@ def test_claude_connectivity() -> Tuple[bool, str]:
         error_msg = f"Error testing Claude API: {str(e)}\n\n{traceback.format_exc()}"
         return False, error_msg
 
-def test_agent_functionality() -> Tuple[bool, Dict[str, Any]]:
+def test_simplified_agent() -> Tuple[bool, Dict[str, Any]]:
     """
-    Test basic agent functionality by initializing the Planning Agent
-    and generating a simple analysis plan
+    Test the simplified Planning Agent functionality
     
     Returns:
         Tuple of (success_flag, results_dict)
@@ -68,10 +67,10 @@ def test_agent_functionality() -> Tuple[bool, Dict[str, Any]]:
         # Test Planning Agent import
         results["steps"].append({"step": "import", "status": "started"})
         try:
-            from agents.planning_agent import PlanningAgent
+            from simplified_planning_agent import PlanningAgent
             results["steps"].append({"step": "import", "status": "success"})
         except Exception as import_error:
-            error_msg = f"Error importing PlanningAgent: {str(import_error)}\n\n{traceback.format_exc()}"
+            error_msg = f"Error importing simplified PlanningAgent: {str(import_error)}\n\n{traceback.format_exc()}"
             results["errors"].append({"phase": "import", "error": error_msg})
             return False, results
         
@@ -158,20 +157,36 @@ def render_diagnostics_tab():
             st.error("❌ Claude API test failed")
             st.expander("Error Details", expanded=True).error(response)
     
-    # Agent functionality test
-    st.subheader("Agent Functionality Test")
-    if st.button("Test Planning Agent"):
-        with st.spinner("Testing Planning Agent..."):
-            success, results = test_agent_functionality()
+    # Simplified Agent functionality test
+    st.subheader("Simplified Planning Agent Test")
+    if st.button("Test Simplified Planning Agent"):
+        with st.spinner("Testing Simplified Planning Agent..."):
+            success, results = test_simplified_agent()
         
         if success:
-            st.success("✅ Planning Agent is working properly!")
+            st.success("✅ Simplified Planning Agent is working properly!")
             st.json(results["plan"])
         else:
-            st.error("❌ Planning Agent test failed")
+            st.error("❌ Simplified Planning Agent test failed")
             for error in results["errors"]:
                 st.expander(f"Error in {error['phase']} phase", expanded=True).error(error["error"])
     
+    # Original Agent functionality test
+    st.subheader("Original Planning Agent Test (Likely to Fail)")
+    if st.button("Test Original Planning Agent"):
+        with st.spinner("Testing Original Planning Agent..."):
+            try:
+                from agents.planning_agent import PlanningAgent
+                planning_agent = PlanningAgent()
+                test_alert = "Sales of DrugX down 15% in Northeast region over past 30 days."
+                analysis_plan, plan_dict = planning_agent.create_analysis_plan(test_alert)
+                
+                st.success("✅ Original Planning Agent is working!")
+                st.json(plan_dict)
+            except Exception as e:
+                st.error("❌ Original Planning Agent test failed")
+                st.expander("Error Details", expanded=True).error(f"{str(e)}\n\n{traceback.format_exc()}")
+    
     # Database Connectivity
     st.subheader("Database Connectivity")
     if st.button("Test Database Connection"):
@@ -199,6 +214,16 @@ def render_diagnostics_tab():
     # Add guidance
     st.markdown("---")
     st.markdown("""
+    ### Using the Simplified Agent
+    
+    The original Planning Agent implementation has issues with the LangChain prompt templates. 
+    We've created a simplified version that uses direct API calls to Claude.
+    
+    To use the simplified agent:
+    
+    1. Copy the `simplified_planning_agent.py` file to your project
+    2. Update your workflow to import from `simplified_planning_agent` instead of `agents.planning_agent`
+    
     ### Troubleshooting Tips
     
     1. If Claude test fails: