Update insight_and_tasks/agents/post_agent.py

insight_and_tasks/agents/post_agent.py

@@ -416,11 +416,11 @@ class EnhancedPostPerformanceAgent:
             Provide a concise summary of findings and actionable recommendations.
             """
             def chat_operation():
-                if not pai.config.llm:
+                if not pai.config.get("llm"):
                     logger.warning("PandasAI LLM not configured for post agent. Attempting to configure.")
-                    from utils.pandasai_setup import configure_pandasai
+                    from insight_and_tasks.utils.pandasai_setup import configure_pandasai
                     configure_pandasai(self.api_key, self.model_name)
-                    if not pai.config.llm:
+                    if not pai.config.get("llm"):
                         raise RuntimeError("PandasAI LLM could not be configured for post chat operation.")
                 logger.info(f"Executing PandasAI chat for post analysis with LLM: {pai.config.llm}")
                 return pandas_ai_df.chat(analysis_query)