app/utils.py

from pathlib import Path
import openai
import os

CONFIG_DIR = Path("./config")

REASONER_PROVIDERS_MAP = {
    "deepseek-r1-distill-llama-70b": "groq",
    "deepseek-ai/deepseek-r1": "nvidia",
    "deepseek/deepseek-r1:free": "openrouter"
}

MODELS_PROVIDERS_MAP = {
    "mistral-small-latest": "mistral",
    "open-mistral-nemo": "mistral",
    "mistral-large-latest": "mistral",
    "gemini-2.0-flash-exp": "gemini",
    "gemma2-9b-it": "groq",
    "llama-3.3-70b-versatile": "groq",
    "llama-3.2-3b-preview": "groq"
}

PROVIDERS_API_KEYS = {
    "gemini": os.environ.get("GEMINI_API_KEY"),
    "groq": os.environ.get("GROQ_API_KEY"),
    "mistral": os.environ.get("MISTRAL_API_KEY"),
    "nvidia": os.environ.get("NVIDIA_API_KEY")
}

def check_openai_api_key(api_key, base_url=None):
    client = openai.OpenAI(api_key=api_key, base_url=base_url)
    try:
        client.models.list()
    except openai.AuthenticationError:
        return False
    else:
        return True
    
if __name__ == "__main__":
    print(check_openai_api_key("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"))