Enforce Template Activation Rules & Refine Agent Management UX

auto-analyst-backend/app.py

@@ -1069,6 +1069,7 @@ def _get_model_name_for_provider(provider: str) -> str:
 async def _execute_plan_with_timeout(ai_system, enhanced_query, plan_response):
     """Execute the plan with timeout handling for each step"""
     try:
+        logger.log_message(f"Plan response: {plan_response}", level=logging.INFO)
         # Use the async generator from execute_plan directly
         async for agent_name, inputs, response in ai_system.execute_plan(enhanced_query, plan_response):
             # Yield results as they come

auto-analyst-backend/scripts/populate_agent_templates.py

@@ -25,7 +25,6 @@ def get_database_type():
     else:
         return "unknown"
 
-# Default agents (free for all users)
 DEFAULT_AGENTS = {
     "Data Manipulation": [
         {
@@ -203,7 +202,6 @@ def statistical_model(X, y, goal, period=None):
     ]
 }
 
-# Premium template agent definitions
 PREMIUM_TEMPLATES = {
     "Data Visualization": [
         {

auto-analyst-backend/src/agents/agents.py

@@ -1582,7 +1582,7 @@ def __init__(self, agents, retrievers, user_id=None, db_session=None):
             try:
                 # For planner use, load planner-enabled templates (max 10, prioritized by usage)
                 template_signatures = load_user_enabled_templates_for_planner_from_db(user_id, db_session)
-                logger.log_message(f"Loaded {len(template_signatures)} templates for planner use", level=logging.INFO)
+                logger.log_message(f"Loaded {template_signatures} templates for planner use", level=logging.INFO)
 
                 for template_name, signature in template_signatures.items():
                     # Skip if this is a core agent - we'll load it separately
@@ -1861,6 +1861,7 @@ async def get_plan(self, query):
             else:
                 complexity = 'basic'
             plan_dict['complexity'] = complexity
+            logger.log_message(f"Plan dict: {plan_dict}", level=logging.INFO)
 
             return plan_dict
 
@@ -1880,7 +1881,8 @@ async def execute_plan(self, query, plan):
         import json
 
         # Clean and split the plan string into agent names
-        plan_text = plan.get("plan", "").replace("Plan", "").replace(":", "").strip()
+        plan_text = plan.get("plan", "").lower().replace("plan", "").replace(":", "").strip()
+        logger.log_message(f"Plan text: {plan_text}", level=logging.INFO)
 
         if "basic_qa_agent" in plan_text:
             inputs = dict(goal=query)