Update insight_and_tasks/agents/follower_agent.py

insight_and_tasks/agents/follower_agent.py

@@ -366,7 +366,7 @@ class EnhancedFollowerAnalysisAgent:
                 # Check if LLM is configured using the proper config.get() method
                 config = pai.config.get()
                 logger.info(f"pai_config: {config}, Type of config: {type(config)}")
-                if not config.get("llm"):
+                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 
@@ -374,10 +374,10 @@ class EnhancedFollowerAnalysisAgent:
                     
                     # Re-check configuration after setup attempt
                     config = pai.config.get()
-                    if not config.get("llm"):
+                    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.get('llm')}")
+                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(