config.py

# Configuration file for CrewAI Agent

# LLM Provider Configuration
DEFAULT_PROVIDER = "google"  # Options: "google", "huggingface"

# Model Configuration per Provider
MODELS = {
    "google": "gemini-2.0-flash",
    "huggingface": "microsoft/DialoGPT-medium"
}

# Search Configuration
MAX_SEARCH_RESULTS = {
    "wikipedia": 2,
    "web": 3,
    "arxiv": 3
}

# Agent Configuration
AGENT_ROLES = {
    "research": {
        "role": "Research Specialist",
        "goal": "Gather comprehensive and accurate information from multiple sources",
        "tools": ["wikipedia_search", "web_search", "arxiv_search"]
    },
    "calculation": {
        "role": "Mathematical Analyst", 
        "goal": "Perform accurate mathematical calculations and analysis",
        "tools": ["calculator"]
    },
    "general": {
        "role": "General Assistant",
        "goal": "Provide comprehensive answers by coordinating with specialized agents",
        "tools": "all"
    }
}

# Task Assignment Keywords
TASK_KEYWORDS = {
    "calculation": ["calculate", "compute", "add", "subtract", "multiply", "divide", "math", "equation", "solve"],
    "research": ["search", "find", "research", "information", "what is", "who is", "when", "where", "how", "explain"]
}