GuglielmoTor commited on
Commit
34e10ae
·
verified ·
1 Parent(s): f9b3103

Update insight_and_tasks/agents/mentions_agent.py

Browse files
insight_and_tasks/agents/mentions_agent.py CHANGED
@@ -289,11 +289,11 @@ class EnhancedMentionsAnalysisAgent:
289
  def chat_operation():
290
  if not pai.config.get("llm"):
291
  logger.warning("PandasAI LLM not configured for mentions agent. Attempting to configure.")
292
- from utils.pandasai_setup import configure_pandasai
293
  configure_pandasai(self.api_key, self.model_name)
294
- if not pai.config.llm:
295
  raise RuntimeError("PandasAI LLM could not be configured for mentions chat operation.")
296
- logger.info(f"Executing PandasAI chat for mentions analysis with LLM: {pai.config.llm}")
297
  return pandas_ai_df.chat(analysis_query)
298
 
299
  analysis_result_raw = self.retry_mechanism.retry_with_backoff(
 
289
  def chat_operation():
290
  if not pai.config.get("llm"):
291
  logger.warning("PandasAI LLM not configured for mentions agent. Attempting to configure.")
292
+ from insight_and_tasks.utils.pandasai_setup import configure_pandasai
293
  configure_pandasai(self.api_key, self.model_name)
294
+ if not pai.config.get("llm"):
295
  raise RuntimeError("PandasAI LLM could not be configured for mentions chat operation.")
296
+ logger.info(f"Executing PandasAI chat for mentions analysis with LLM: {if not pai.config.get("llm")}")
297
  return pandas_ai_df.chat(analysis_query)
298
 
299
  analysis_result_raw = self.retry_mechanism.retry_with_backoff(