PROMPTS = [
    ("Write a Tweet", 50),
    ("Write an email", 170),
    ("Write an article summary", 250),
    ("Small conversation with a chatbot", 400),
    ("Write a report of 5 pages", 5000),
    ("Write the code for this app", 15000)
]
PROMPTS = [(s + f" ({v} output tokens)", v) for (s, v) in PROMPTS]

MODEL_REPOSITORY_URL = "https://raw.githubusercontent.com/genai-impact/ecologits/refs/heads/main/ecologits/data/models.json"

main_models_openai = [
    'chatgpt-4o-latest',
    'gpt-3.5-turbo',
    'gpt-4',
    'gpt-4-turbo',
    'gpt-4o',
    'gpt-4o-mini',
    'o1',
    'o1-mini'
]

main_models_meta = [
    'meta-llama/Meta-Llama-3.1-8B',
    'meta-llama/Meta-Llama-3.1-70B',
    'meta-llama/Meta-Llama-3.1-405B',
    'meta-llama/Meta-Llama-3-8B',
    'meta-llama/Meta-Llama-3-70B',
    'meta-llama/Meta-Llama-3-70B',
    'meta-llama/Llama-2-7b',
    'meta-llama/Llama-2-13b',
    'meta-llama/Llama-2-70b',
    'meta-llama/CodeLlama-7b-hf',
    'meta-llama/CodeLlama-13b-hf',
    'meta-llama/CodeLlama-34b-hf',
    'meta-llama/CodeLlama-70b-hf'
]

main_models_msft = [
    'microsoft/phi-1',
    'microsoft/phi-1_5',
    'microsoft/Phi-3-mini-128k-instruct',
    'microsoft/Phi-3-small-128k-instruct',
    'microsoft/Phi-3-medium-128k-instruct',
]

main_models_anthropic = [
    'claude-2.0',
    'claude-2.1',
    'claude-3-5-haiku-latest',
    'claude-3-5-sonnet-latest',
    'claude-3-7-sonnet-latest',
    'claude-3-haiku-20240307',
    'claude-3-opus-latest',
    'claude-3-sonnet-20240229'
]

main_models_cohere = [
    'c4ai-aya-expanse-8b',
    'c4ai-aya-expanse-32b',
    'command',
    'command-light',
    'command-r',
    'command-r-plus',
]

main_models_google = [
    'google/gemma-2-2b',
    'google/gemma-2-9b',
    'google/gemma-2-27b',
    'google/codegemma-2b',
    'google/codegemma-7b',
    'gemini-1.0-pro',
    'gemini-1.5-pro',
    'gemini-1.5-flash',
    'gemini-2.0-flash'
]

main_models_databricks = [
    'databricks/dolly-v1-6b',
    'databricks/dolly-v2-12b',
    'databricks/dolly-v2-7b',
    'databricks/dolly-v2-3b',
    'databricks/dbrx-base'
]

main_models_mistral = [
    'mistralai/Mistral-7B-v0.3',
    'mistralai/Mixtral-8x7B-v0.1',
    'mistralai/Mixtral-8x22B-v0.1',
    'mistralai/Codestral-22B-v0.1',
    'mistralai/Mathstral-7B-v0.1',
    'ministral-3b-latest',
    'ministral-8b-latest',
    'mistral-tiny',
    'mistral-small',
    'mistral-medium',
    'mistral-large-latest'
]

MAIN_MODELS = main_models_meta + main_models_openai + main_models_anthropic + main_models_cohere + main_models_msft + main_models_mistral + main_models_databricks + main_models_google