Spaces:
Running
Running
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(
|