mirror of
https://github.com/thecyberlearn/quantum-ai-v3.git
synced 2026-08-18 08:52:58 +00:00
56749b77f0
BREAKING CHANGE: Transform from database-driven to file-based agent management Major Features: • File-based agent definitions - agents load from JSON configs instantly • Hybrid architecture - files for agent data, database for relationships • Zero-command deployment - just add JSON file and git push • Automatic database sync - maintains foreign key compatibility • Template compatibility - enriched agent data works with existing UI Technical Implementation: • AgentFileService - comprehensive file-based agent management • JSON config loading with LRU caching for performance • Category enrichment for template compatibility • Slug-based queries for AgentExecution relationships • get_or_create_agent_db_record() for foreign key maintenance Benefits: • Ultra-simple agent creation (just add JSON file) • Instant Railway deployment (no commands needed) • Version control integration (agent changes in git) • Perfect scalability (ready for 100+ agents) • Zero maintenance overhead All 8 agents verified working: - 4 Webhook agents: Social Ads, Job Posting, PDF Summarizer, 5 Whys - 4 Direct access agents: Career Navigator, Brand Strategist, Lean Six Sigma, SWOT 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
347 lines
14 KiB
Python
347 lines
14 KiB
Python
import json
|
|
import os
|
|
from functools import lru_cache
|
|
from pathlib import Path
|
|
from typing import Dict, List, Optional
|
|
from django.conf import settings
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class AgentFileService:
|
|
"""
|
|
Service for loading agent configurations from JSON files instead of database.
|
|
Provides caching and error handling for file-based agent management.
|
|
"""
|
|
|
|
BASE_DIR = Path(__file__).resolve().parent.parent
|
|
AGENTS_CONFIG_DIR = BASE_DIR / 'agents' / 'configs' / 'agents'
|
|
CATEGORIES_CONFIG_FILE = BASE_DIR / 'agents' / 'configs' / 'categories' / 'categories.json'
|
|
|
|
@classmethod
|
|
def clear_cache(cls):
|
|
"""Clear all cached data - useful for testing and development"""
|
|
cls.get_all_agents.cache_clear()
|
|
cls.get_all_categories.cache_clear()
|
|
|
|
@classmethod
|
|
@lru_cache(maxsize=1)
|
|
def get_all_categories(cls) -> List[Dict]:
|
|
"""
|
|
Load all agent categories from categories.json file.
|
|
Returns list of category dictionaries with caching.
|
|
"""
|
|
try:
|
|
if not cls.CATEGORIES_CONFIG_FILE.exists():
|
|
logger.warning(f"Categories file not found: {cls.CATEGORIES_CONFIG_FILE}")
|
|
return []
|
|
|
|
with open(cls.CATEGORIES_CONFIG_FILE, 'r', encoding='utf-8') as f:
|
|
categories = json.load(f)
|
|
|
|
# Handle both array format and object format
|
|
if isinstance(categories, dict) and 'categories' in categories:
|
|
categories = categories['categories']
|
|
elif not isinstance(categories, list):
|
|
logger.error("Categories file should contain an array of categories")
|
|
return []
|
|
|
|
logger.info(f"Loaded {len(categories)} categories from file")
|
|
return categories
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error loading categories: {str(e)}")
|
|
return []
|
|
|
|
@classmethod
|
|
@lru_cache(maxsize=1)
|
|
def get_all_agents(cls) -> List[Dict]:
|
|
"""
|
|
Load all agent configurations from JSON files in the agents config directory.
|
|
Returns list of agent dictionaries with caching.
|
|
"""
|
|
agents = []
|
|
|
|
try:
|
|
if not cls.AGENTS_CONFIG_DIR.exists():
|
|
logger.warning(f"Agents config directory not found: {cls.AGENTS_CONFIG_DIR}")
|
|
return []
|
|
|
|
# Get all JSON files in the agents config directory
|
|
json_files = list(cls.AGENTS_CONFIG_DIR.glob('*.json'))
|
|
|
|
for json_file in json_files:
|
|
try:
|
|
with open(json_file, 'r', encoding='utf-8') as f:
|
|
agent_data = json.load(f)
|
|
|
|
# Add file-based metadata
|
|
agent_data['_source_file'] = str(json_file)
|
|
agent_data['_file_name'] = json_file.name
|
|
|
|
# Ensure required fields exist with defaults
|
|
agent_data.setdefault('is_active', True)
|
|
agent_data.setdefault('agent_type', 'form')
|
|
agent_data.setdefault('system_type', 'webhook')
|
|
agent_data.setdefault('form_schema', {'fields': []})
|
|
agent_data.setdefault('access_url_name', '')
|
|
agent_data.setdefault('display_url_name', '')
|
|
|
|
agents.append(agent_data)
|
|
|
|
except json.JSONDecodeError as e:
|
|
logger.error(f"Invalid JSON in {json_file}: {str(e)}")
|
|
continue
|
|
except Exception as e:
|
|
logger.error(f"Error loading agent from {json_file}: {str(e)}")
|
|
continue
|
|
|
|
logger.info(f"Loaded {len(agents)} agents from {len(json_files)} files")
|
|
return agents
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error scanning agents directory: {str(e)}")
|
|
return []
|
|
|
|
@classmethod
|
|
def get_agent_by_slug(cls, slug: str) -> Optional[Dict]:
|
|
"""
|
|
Get a specific agent by its slug with category info enriched.
|
|
Returns agent dictionary or None if not found.
|
|
"""
|
|
agents = cls.get_all_agents()
|
|
for agent in agents:
|
|
if agent.get('slug') == slug:
|
|
enriched_agents = cls._enrich_agents_with_category_info([agent])
|
|
return enriched_agents[0] if enriched_agents else agent
|
|
return None
|
|
|
|
@classmethod
|
|
def get_agents_by_category(cls, category_slug: str) -> List[Dict]:
|
|
"""
|
|
Get all agents belonging to a specific category.
|
|
Returns list of agent dictionaries with category info enriched.
|
|
"""
|
|
agents = cls.get_all_agents()
|
|
category_agents = [agent for agent in agents if agent.get('category') == category_slug]
|
|
return cls._enrich_agents_with_category_info(category_agents)
|
|
|
|
@classmethod
|
|
def get_active_agents(cls) -> List[Dict]:
|
|
"""
|
|
Get all active agents with category info enriched.
|
|
Returns list of active agent dictionaries.
|
|
"""
|
|
agents = cls.get_all_agents()
|
|
active_agents = [agent for agent in agents if agent.get('is_active', True)]
|
|
return cls._enrich_agents_with_category_info(active_agents)
|
|
|
|
@classmethod
|
|
def search_agents(cls, query: str) -> List[Dict]:
|
|
"""
|
|
Search agents by name or description.
|
|
Returns list of matching agent dictionaries with category info enriched.
|
|
"""
|
|
if not query:
|
|
return cls.get_active_agents()
|
|
|
|
query_lower = query.lower()
|
|
agents = cls.get_all_agents()
|
|
active_agents = [agent for agent in agents if agent.get('is_active', True)]
|
|
|
|
matching_agents = []
|
|
for agent in active_agents:
|
|
name = agent.get('name', '').lower()
|
|
description = agent.get('description', '').lower()
|
|
short_description = agent.get('short_description', '').lower()
|
|
|
|
if (query_lower in name or
|
|
query_lower in description or
|
|
query_lower in short_description):
|
|
matching_agents.append(agent)
|
|
|
|
return cls._enrich_agents_with_category_info(matching_agents)
|
|
|
|
@classmethod
|
|
def get_category_by_slug(cls, slug: str) -> Optional[Dict]:
|
|
"""
|
|
Get a specific category by its slug.
|
|
Returns category dictionary or None if not found.
|
|
"""
|
|
categories = cls.get_all_categories()
|
|
for category in categories:
|
|
if category.get('slug') == slug:
|
|
return category
|
|
return None
|
|
|
|
@classmethod
|
|
def validate_agent_config(cls, agent_data: Dict) -> List[str]:
|
|
"""
|
|
Validate an agent configuration dictionary.
|
|
Returns list of validation errors (empty if valid).
|
|
"""
|
|
errors = []
|
|
required_fields = ['slug', 'name', 'short_description', 'description', 'category', 'price']
|
|
|
|
for field in required_fields:
|
|
if not agent_data.get(field):
|
|
errors.append(f"Missing required field: {field}")
|
|
|
|
# Validate price is numeric
|
|
try:
|
|
float(agent_data.get('price', 0))
|
|
except (ValueError, TypeError):
|
|
errors.append("Price must be a valid number")
|
|
|
|
# Validate category exists
|
|
category_slug = agent_data.get('category')
|
|
if category_slug and not cls.get_category_by_slug(category_slug):
|
|
errors.append(f"Category '{category_slug}' does not exist")
|
|
|
|
# Validate system_type
|
|
system_type = agent_data.get('system_type', 'webhook')
|
|
if system_type not in ['webhook', 'direct_access']:
|
|
errors.append("system_type must be 'webhook' or 'direct_access'")
|
|
|
|
return errors
|
|
|
|
@classmethod
|
|
def get_agent_stats(cls) -> Dict:
|
|
"""
|
|
Get statistics about agents and categories.
|
|
Returns dictionary with counts and breakdown.
|
|
"""
|
|
agents = cls.get_all_agents()
|
|
categories = cls.get_all_categories()
|
|
|
|
active_agents = cls.get_active_agents()
|
|
webhook_agents = [a for a in active_agents if a.get('system_type') == 'webhook']
|
|
direct_access_agents = [a for a in active_agents if a.get('system_type') == 'direct_access']
|
|
|
|
# Category breakdown (use raw agents to avoid category object issue)
|
|
category_counts = {}
|
|
raw_active_agents = [agent for agent in agents if agent.get('is_active', True)]
|
|
for agent in raw_active_agents:
|
|
category = agent.get('category', 'unknown')
|
|
category_counts[category] = category_counts.get(category, 0) + 1
|
|
|
|
return {
|
|
'total_agents': len(agents),
|
|
'active_agents': len(active_agents),
|
|
'webhook_agents': len(webhook_agents),
|
|
'direct_access_agents': len(direct_access_agents),
|
|
'total_categories': len(categories),
|
|
'category_breakdown': category_counts
|
|
}
|
|
|
|
@classmethod
|
|
def _enrich_agents_with_category_info(cls, agents: List[Dict]) -> List[Dict]:
|
|
"""
|
|
Enrich agent dictionaries with category information for template compatibility.
|
|
Adds category object with icon, name, and slug for each agent.
|
|
"""
|
|
categories = cls.get_all_categories()
|
|
category_map = {cat['slug']: cat for cat in categories}
|
|
|
|
enriched_agents = []
|
|
for agent in agents:
|
|
agent_copy = agent.copy()
|
|
category_slug = agent.get('category')
|
|
|
|
if category_slug and category_slug in category_map:
|
|
# Add category object for template compatibility
|
|
category_data = category_map[category_slug]
|
|
agent_copy['category'] = {
|
|
'slug': category_data['slug'],
|
|
'name': category_data['name'],
|
|
'icon': category_data['icon'],
|
|
'description': category_data.get('description', '')
|
|
}
|
|
else:
|
|
# Fallback category
|
|
agent_copy['category'] = {
|
|
'slug': 'unknown',
|
|
'name': 'Unknown',
|
|
'icon': '❓',
|
|
'description': 'Unknown category'
|
|
}
|
|
|
|
enriched_agents.append(agent_copy)
|
|
|
|
return enriched_agents
|
|
|
|
@classmethod
|
|
def get_or_create_agent_db_record(cls, agent_data: Dict):
|
|
"""
|
|
Get or create a database Agent record for foreign key relationships.
|
|
This maintains compatibility with AgentExecution while using file-based configs.
|
|
"""
|
|
from .models import Agent, AgentCategory
|
|
|
|
# Get or create category
|
|
category_data = cls.get_category_by_slug(agent_data.get('category', 'unknown'))
|
|
if category_data:
|
|
category, _ = AgentCategory.objects.get_or_create(
|
|
slug=category_data['slug'],
|
|
defaults={
|
|
'name': category_data['name'],
|
|
'description': category_data.get('description', ''),
|
|
'icon': category_data['icon'],
|
|
'is_active': True
|
|
}
|
|
)
|
|
else:
|
|
category, _ = AgentCategory.objects.get_or_create(
|
|
slug='unknown',
|
|
defaults={
|
|
'name': 'Unknown',
|
|
'description': 'Unknown category',
|
|
'icon': '❓',
|
|
'is_active': True
|
|
}
|
|
)
|
|
|
|
# Get or create agent
|
|
agent, created = Agent.objects.get_or_create(
|
|
slug=agent_data['slug'],
|
|
defaults={
|
|
'name': agent_data['name'],
|
|
'short_description': agent_data['short_description'],
|
|
'description': agent_data['description'],
|
|
'category': category,
|
|
'price': agent_data['price'],
|
|
'agent_type': agent_data.get('agent_type', 'form'),
|
|
'form_schema': agent_data.get('form_schema', {}),
|
|
'webhook_url': agent_data['webhook_url'],
|
|
'message_limit': agent_data.get('message_limit', 50),
|
|
'access_url_name': agent_data.get('access_url_name', ''),
|
|
'display_url_name': agent_data.get('display_url_name', ''),
|
|
'is_active': agent_data.get('is_active', True)
|
|
}
|
|
)
|
|
|
|
# Update existing record if needed (sync file data to db)
|
|
if not created:
|
|
updated = False
|
|
for field, value in {
|
|
'name': agent_data['name'],
|
|
'short_description': agent_data['short_description'],
|
|
'description': agent_data['description'],
|
|
'price': agent_data['price'],
|
|
'agent_type': agent_data.get('agent_type', 'form'),
|
|
'form_schema': agent_data.get('form_schema', {}),
|
|
'webhook_url': agent_data['webhook_url'],
|
|
'message_limit': agent_data.get('message_limit', 50),
|
|
'access_url_name': agent_data.get('access_url_name', ''),
|
|
'display_url_name': agent_data.get('display_url_name', ''),
|
|
'is_active': agent_data.get('is_active', True),
|
|
'category': category
|
|
}.items():
|
|
if getattr(agent, field) != value:
|
|
setattr(agent, field, value)
|
|
updated = True
|
|
|
|
if updated:
|
|
agent.save()
|
|
|
|
return agent |