Create config.py

config.py

@@ -0,0 +1,177 @@
+"""
+Configuration centralisée pour l'UltraAgent
+"""
+import os
+from typing import Dict, Any, Optional
+
+class AgentConfig:
+    """Configuration centralisée pour l'agent"""
+    
+    # Modèles HuggingFace
+    ORCHESTRATOR_MODEL = "moonshotai/Kimi-K2-Instruct"
+    CODE_AGENT_MODEL = "moonshotai/Kimi-K2-Instruct"
+    VISION_MODEL = "Qwen/Qwen2.5-VL-72B-Instruct"
+    REASONING_MODEL = "deepseek-ai/DeepSeek-R1-0528"
+    
+    # Configuration des timeouts (en secondes)
+    DEFAULT_TIMEOUT = 120
+    VISION_TIMEOUT = 180
+    REASONING_TIMEOUT = 180
+    WEB_REQUEST_TIMEOUT = 30
+    
+    # Paramètres des modèles
+    DEFAULT_MAX_TOKENS = 4096
+    VISION_MAX_TOKENS = 2048
+    REASONING_MAX_TOKENS = 8192
+    
+    # Températures des modèles
+    ORCHESTRATOR_TEMPERATURE = 0.1
+    CODE_TEMPERATURE = 0.0
+    VISION_TEMPERATURE = 0.1
+    REASONING_TEMPERATURE = 0.2
+    
+    # Limites de sécurité
+    MAX_SEARCH_RESULTS = 15
+    MAX_SCRAPED_CONTENT_LENGTH = 10000
+    MAX_FILE_SIZE_MB = 50
+    MAX_CONVERSATION_HISTORY = 50
+    
+    # Configuration des outils web
+    SEARCH_REGION = "fr-fr"
+    MAX_LINKS_PER_PAGE = 50
+    
+    # Headers pour les requêtes web
+    DEFAULT_HEADERS = {
+        'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
+        'Accept-Language': 'fr-FR,fr;q=0.9,en;q=0.8',
+        'Accept-Encoding': 'gzip, deflate',
+        'Connection': 'keep-alive',
+        'Upgrade-Insecure-Requests': '1',
+    }
+    
+    # Configuration Selenium
+    SELENIUM_OPTIONS = [
+        "--headless",
+        "--no-sandbox",
+        "--disable-dev-shm-usage",
+        "--disable-gpu",
+        "--window-size=1920,1080",
+        "--disable-blink-features=AutomationControlled",
+        "--disable-extensions",
+        "--disable-plugins",
+    ]
+    
+    @classmethod
+    def get_hf_token(cls) -> Optional[str]:
+        """Récupère le token HuggingFace depuis les variables d'environnement"""
+        return os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN")
+    
+    @classmethod
+    def get_model_config(cls, model_type: str) -> Dict[str, Any]:
+        """Retourne la configuration pour un type de modèle donné"""
+        configs = {
+            "orchestrator": {
+                "model": cls.ORCHESTRATOR_MODEL,
+                "max_tokens": cls.DEFAULT_MAX_TOKENS,
+                "temperature": cls.ORCHESTRATOR_TEMPERATURE,
+                "timeout": cls.DEFAULT_TIMEOUT
+            },
+            "code_agent": {
+                "model": cls.CODE_AGENT_MODEL,
+                "max_tokens": cls.DEFAULT_MAX_TOKENS,
+                "temperature": cls.CODE_TEMPERATURE,
+                "timeout": cls.DEFAULT_TIMEOUT
+            },
+            "vision": {
+                "model": cls.VISION_MODEL,
+                "max_tokens": cls.VISION_MAX_TOKENS,
+                "temperature": cls.VISION_TEMPERATURE,
+                "timeout": cls.VISION_TIMEOUT
+            },
+            "reasoning": {
+                "model": cls.REASONING_MODEL,
+                "max_tokens": cls.REASONING_MAX_TOKENS,
+                "temperature": cls.REASONING_TEMPERATURE,
+                "timeout": cls.REASONING_TIMEOUT
+            }
+        }
+        
+        return configs.get(model_type, configs["orchestrator"])
+    
+    @classmethod
+    def validate_environment(cls) -> Dict[str, Any]:
+        """Valide l'environnement et retourne un rapport"""
+        report = {
+            "status": "ok",
+            "warnings": [],
+            "errors": [],
+            "environment": {}
+        }
+        
+        # Vérifie le token HF
+        hf_token = cls.get_hf_token()
+        if not hf_token:
+            report["errors"].append("Token HuggingFace manquant (HF_TOKEN ou HUGGINGFACE_TOKEN)")
+            report["status"] = "error"
+        else:
+            report["environment"]["hf_token"] = "✅ Présent"
+        
+        # Vérifie les variables d'environnement optionnelles
+        space_id = os.getenv("SPACE_ID")
+        space_host = os.getenv("SPACE_HOST")
+        
+        if space_id:
+            report["environment"]["space_id"] = space_id
+        else:
+            report["warnings"].append("SPACE_ID non défini (mode local?)")
+        
+        if space_host:
+            report["environment"]["space_host"] = space_host
+        else:
+            report["warnings"].append("SPACE_HOST non défini (mode local?)")
+        
+        # Vérifie les imports critiques
+        try:
+            import smolagents
+            report["environment"]["smolagents"] = f"✅ v{smolagents.__version__}"
+        except ImportError:
+            report["errors"].append("smolagents non installé")
+            report["status"] = "error"
+        
+        try:
+            import requests
+            report["environment"]["requests"] = "✅ Disponible"
+        except ImportError:
+            report["errors"].append("requests non installé")
+            report["status"] = "error"
+        
+        # Avertissements si status n'est que warnings
+        if report["warnings"] and report["status"] == "ok":
+            report["status"] = "warning"
+        
+        return report
+    
+    @classmethod
+    def get_debug_info(cls) -> Dict[str, Any]:
+        """Retourne les informations de debug"""
+        return {
+            "models": {
+                "orchestrator": cls.ORCHESTRATOR_MODEL,
+                "code_agent": cls.CODE_AGENT_MODEL,
+                "vision": cls.VISION_MODEL,
+                "reasoning": cls.REASONING_MODEL
+            },
+            "timeouts": {
+                "default": cls.DEFAULT_TIMEOUT,
+                "vision": cls.VISION_TIMEOUT,
+                "reasoning": cls.REASONING_TIMEOUT,
+                "web_request": cls.WEB_REQUEST_TIMEOUT
+            },
+            "limits": {
+                "max_search_results": cls.MAX_SEARCH_RESULTS,
+                "max_scraped_content": cls.MAX_SCRAPED_CONTENT_LENGTH,
+                "max_file_size_mb": cls.MAX_FILE_SIZE_MB,
+                "max_conversation_history": cls.MAX_CONVERSATION_HISTORY
+            },
+            "environment_validation": cls.validate_environment()
+        }