config.py

"""
Configuration file for the GAIA Agent

Customize these settings to match your agent implementation and preferences.
"""

# Agent Configuration
AGENT_TYPE = "simple"  # Options: "simple", "tool_using", "multi_step"

# API Configuration
API_BASE_URL = "https://gaia-benchmark.vercel.app/api"
API_TIMEOUT = 30  # seconds
API_RETRY_ATTEMPTS = 3

# Agent Behavior Settings
ENABLE_DEBUG_LOGGING = True
ENABLE_FILE_DOWNLOAD = True
ENABLE_RATE_LIMITING = True
RATE_LIMIT_DELAY = 0.1  # seconds between requests

# Answer Generation Settings
MAX_ANSWER_LENGTH = 1000  # characters
INCLUDE_REASONING = True
INCLUDE_CONFIDENCE = False

# Tool Configuration (for tool_using agent)
ENABLED_TOOLS = [
    "calculator",
    "file_reader",
    "date_parser",
    # "web_search",  # Uncomment if you implement web search
]

# External API Keys (add your own)
EXTERNAL_APIS = {
    "openai": {
        "api_key": "",  # Add your OpenAI API key if using
        "model": "gpt-4",
        "temperature": 0.1
    },
    "anthropic": {
        "api_key": "",  # Add your Anthropic API key if using
        "model": "claude-3-sonnet-20240229"
    },
    "google": {
        "api_key": "",  # Add your Google API key if using
        "search_engine_id": ""
    }
}

# Prompt Templates
PROMPT_TEMPLATES = {
    "simple": """
    Task ID: {task_id}
    
    Question: {question}
    
    {file_context}
    
    Please provide a clear and accurate answer to this question.
    Think step by step:
    1. What is the question asking for?
    2. What information do I need to answer it?
    3. How can I find or calculate this information?
    4. What is the final answer?
    
    Answer:
    """,
    
    "tool_using": """
    Task ID: {task_id}
    
    Question: {question}
    
    {file_context}
    
    Available tools: {available_tools}
    
    Please analyze the question and use appropriate tools to find the answer.
    Show your reasoning and tool usage.
    
    Answer:
    """,
    
    "multi_step": """
    Task ID: {task_id}
    
    Question: {question}
    
    {file_context}
    
    Please break down this question into steps and provide a detailed answer.
    
    Step-by-step reasoning:
    {reasoning_steps}
    
    Final Answer:
    """
}

# File Processing Settings
SUPPORTED_FILE_TYPES = [".txt", ".csv", ".json", ".xml", ".html"]
MAX_FILE_SIZE = 1024 * 1024  # 1MB

# Error Handling
RETRY_ON_ERROR = True
MAX_RETRIES = 3
ERROR_MESSAGES = {
    "api_error": "Failed to connect to GAIA API",
    "file_error": "Failed to download or process file",
    "timeout_error": "Request timed out",
    "validation_error": "Invalid question format"
}

# Logging Configuration
LOG_LEVEL = "INFO"  # DEBUG, INFO, WARNING, ERROR
LOG_FORMAT = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
LOG_FILE = "gaia_agent.log"

# Performance Settings
ENABLE_CACHING = True
CACHE_DURATION = 3600  # 1 hour
MAX_CONCURRENT_REQUESTS = 5

# Custom Agent Settings
CUSTOM_SETTINGS = {
    "use_chain_of_thought": True,
    "enable_self_correction": True,
    "confidence_threshold": 0.7,
    "max_iterations": 5
}