Spaces:
Sleeping
Sleeping
| import os | |
| from typing import Optional | |
| import logging | |
| logger = logging.getLogger(__name__) | |
| class TxAgentConfig: | |
| def __init__(self): | |
| # Since TxAgent is now integrated, default to local mode | |
| self.mode = os.getenv("TXAGENT_MODE", "local") # local, cloud, hybrid | |
| self.cloud_url = os.getenv("TXAGENT_CLOUD_URL", "https://rocketfarmstudios-txagent-api.hf.space") | |
| self.local_enabled = os.getenv("TXAGENT_LOCAL_ENABLED", "true").lower() == "true" | |
| self.gpu_available = self._check_gpu_availability() | |
| def _check_gpu_availability(self) -> bool: | |
| """Vérifie si GPU est disponible localement""" | |
| try: | |
| import torch | |
| return torch.cuda.is_available() | |
| except ImportError: | |
| return False | |
| def get_txagent_mode(self) -> str: | |
| """Détermine le mode optimal pour TxAgent""" | |
| # Since TxAgent is integrated, always use local mode | |
| return "local" | |
| def get_txagent_url(self) -> str: | |
| """Retourne l'URL du service TxAgent""" | |
| # Since TxAgent is integrated, return localhost | |
| return "http://localhost:7860" # Same port as the main API | |
| def is_local_available(self) -> bool: | |
| """Vérifie si le mode local est disponible""" | |
| return True # Always available since it's integrated | |
| # Instance globale | |
| txagent_config = TxAgentConfig() | |