config.py

import os
import secrets
from pathlib import Path

from dotenv import load_dotenv

load_dotenv()

ROOT = Path(__file__).parent

HOST = os.getenv("HOST", "127.0.0.1")
PORT = int(os.getenv("PORT", "8000"))
POLL_INTERVAL = float(os.getenv("POLL_INTERVAL_SECONDS", "2"))
POLL_MAX_ATTEMPTS = int(os.getenv("POLL_MAX_ATTEMPTS", "60"))

PROVIDER_KEYS = {
    "nvidia": os.getenv("SK_NVIDIA_KEY", ""),
}

REAL_KEYS = {
    "nvidia": os.getenv("REAL_NVIDIA_KEY", ""),
}


def _strip_prefix(key: str, prefix: str) -> str:
    return key[len(prefix):] if key.startswith(prefix) else key


def generate_proxy_key(provider: str) -> str:
    return f"sk-{provider}-" + secrets.token_urlsafe(32)