Spaces:
Sleeping
Sleeping
File size: 1,866 Bytes
602a16c b712b2b 602a16c b712b2b 602a16c b712b2b 602a16c b712b2b 602a16c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
"""
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'
}
|