"""
Configuration and constants for the GAIA Agent
"""
import os

# ============================================================================
# API CONFIGURATION
# ============================================================================
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"

# ============================================================================
# MODEL CONFIGURATION
# ============================================================================
# Set based on environment
USE_LOCAL_MODEL = False  # True = Local Ollama, False = HuggingFace API

# Ollama configuration (local)
OLLAMA_MODEL_ID = "ollama/qwen2.5-coder:14b"
OLLAMA_API_BASE = "http://localhost:11434"
OLLAMA_API_KEY = "ollama"

# HuggingFace configuration (cloud)
# Using a powerful model for better GAIA benchmark performance
HF_MODEL_ID = "Qwen/Qwen2.5-72B-Instruct"
HF_TOKEN = os.getenv("HF_TOKEN")

# ============================================================================
# AGENT CONFIGURATION
# ============================================================================
MAX_STEPS = 12
VERBOSITY_LEVEL = 2

AUTHORIZED_IMPORTS = [
    'csv', 'pandas', 'bs4', 'requests', 're', 'collections', 
    'itertools', 'io', 'json', 'math', 'statistics', 'queue', 
    'xml', 'datetime', 'time', 'openpyxl', 'numpy', 'markdownify',
    'urllib'
]

# ============================================================================
# QUESTION TYPES
# ============================================================================
QUESTION_TYPES = {
    'YOUTUBE_VIDEO': 'youtube_video',
    'IMAGE_FILE': 'image_file',
    'AUDIO_FILE': 'audio_file',
    'DATA_FILE': 'data_file',
    'CODE_FILE': 'code_file',
    'WIKIPEDIA': 'wikipedia_search',
    'COUNTING': 'counting_task',
    'TEXT_MANIPULATION': 'text_manipulation',
    'GENERAL': 'general_research'
}