File size: 1,117 Bytes
317d1fd 86b116d 317d1fd 84ae379 86b116d b127732 86b116d b127732 317d1fd 84ae379 b127732 84ae379 86b116d 317d1fd 86b116d 317d1fd |
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 |
import os
from dotenv import load_dotenv
class Config:
def __init__(self):
load_dotenv()
# Detect if running on HF Spaces
self.is_hf_space = bool(os.getenv("SPACE_ID"))
self.hf_token = os.getenv("HF_TOKEN")
self.hf_api_url = os.getenv("HF_API_ENDPOINT_URL", "https://api-inference.huggingface.co/v1/")
self.use_fallback = os.getenv("USE_FALLBACK", "true").lower() == "true"
# Redis configuration (optional for HF)
self.redis_host = os.getenv("REDIS_HOST", "localhost")
self.redis_port = int(os.getenv("REDIS_PORT", "6379"))
self.redis_username = os.getenv("REDIS_USERNAME", "")
self.redis_password = os.getenv("REDIS_PASSWORD", "")
self.redis_retries = int(os.getenv("REDIS_RETRIES", "3"))
self.redis_retry_delay = int(os.getenv("REDIS_RETRY_DELAY", "1"))
# Local model configuration
self.local_model_name = os.getenv("LOCAL_MODEL_NAME", "mistral:latest")
self.ollama_host = os.getenv("OLLAMA_HOST", "")
# Global config instance
config = Config()
|