Update insight_and_tasks/agents/follower_agent.py

insight_and_tasks/agents/follower_agent.py

@@ -363,15 +363,20 @@ class EnhancedFollowerAnalysisAgent:
                 # Ensure the LLM for PandasAI is correctly configured before this call
                 # This might involve re-calling configure_pandasai if it's not persistent
                 # or if the LLM object needs to be explicitly passed to PandasAI DataFrame.
-                if not hasattr(pai.config, 'llm'): # Check if LLM is set for PandasAI
+                # Check if LLM is configured using the proper config.get() method
+                config = pai.config.get()
+                if not config.get("llm"):
                     logger.warning("PandasAI LLM not configured. Attempting to configure now.")
-                    # This assumes configure_pandasai is available and sets pai.config.llm
+                    # 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 chat operation.")
+                    
+                    # Re-check configuration after setup attempt
+                    config = pai.config.get()
+                    if not config.get("llm"):
+                        raise RuntimeError("PandasAI LLM could not be configured for chat operation.")
                 
-                logger.info(f"Executing PandasAI chat for follower analysis with LLM: {pai.config.get('llm')}")
+                logger.info(f"Executing PandasAI chat for follower analysis with LLM: {config.get('llm')}")
                 return pandas_ai_df.chat(analysis_query)
 
             analysis_result_raw = self.retry_mechanism.retry_with_backoff(