Update insight_and_tasks/agents/post_agent.py

insight_and_tasks/agents/post_agent.py

@@ -416,13 +416,20 @@ class EnhancedPostPerformanceAgent:
             Provide a concise summary of findings and actionable recommendations.
             """
             def chat_operation():
-                if not pai.config.get("llm"):
-                    logger.warning("PandasAI LLM not configured for post agent. Attempting to configure.")
-                    from insight_and_tasks.utils.pandasai_setup import configure_pandasai
+                config = pai.config.get()
+                logger.info(f"pai_config: {config}, Type of config: {type(config)}")
+                if not config.llm:
+                    logger.warning("PandasAI LLM not configured. Attempting to configure now.")
+                    # This assumes configure_pandasai is available and sets the LLM config
+                    from insight_and_tasks.utils.pandasai_setup import configure_pandasai 
                     configure_pandasai(self.api_key, self.model_name)
-                    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.get('llm')}")
+                    
+                    # Re-check configuration after setup attempt
+                    config = pai.config.get()
+                    if not config.llm:
+                        raise RuntimeError("PandasAI LLM could not be configured for chat operation.")
+                
+                logger.info(f"Executing PandasAI chat for follower analysis with LLM: {config.llm}")
                 return pandas_ai_df.chat(analysis_query)
 
             analysis_result_raw = self.retry_mechanism.retry_with_backoff(