Uopdate alembic script with new model costs

This commit is contained in:
Phil Szalay 2025-03-31 09:46:40 +02:00
parent 91f547ba9c
commit a189016be1

View File

@ -37,14 +37,26 @@ def upgrade() -> None:
),
[
{'model_name': 'GPT 4o', 'message_credit_cost': 20},
{'model_name': 'GPT 4o Mini', 'message_credit_cost': 1},
{'model_name': 'Claude 3 Haiku', 'message_credit_cost': 2},
{'model_name': 'Claude 3.5 Sonnet v2', 'message_credit_cost': 25},
{'model_name': 'GPT 4o-mini', 'message_credit_cost': 1},
{'model_name': 'Claude 3.5 Haiku', 'message_credit_cost': 10},
{'model_name': 'Claude 3.7 Sonnet', 'message_credit_cost': 25},
{'model_name': 'Gemini 1.5 Pro', 'message_credit_cost': 2},
{'model_name': 'Perplexity', 'message_credit_cost': 25},
{'model_name': 'GPT 3.5 Turbo', 'message_credit_cost': 25},
{'model_name': 'Gemini 1.5 Flash', 'message_credit_cost': 25},
{'model_name': 'Dall-E', 'message_credit_cost': 300},
{'model_name': 'GPT o3-mini', 'message_credit_cost': 100},
{'model_name': 'Mistral Large 2', 'message_credit_cost': 10},
{'model_name': 'Gemini 2.0 Flash', 'message_credit_cost': 2},
{'model_name': 'Gemini 2.0 Flash Thinking', 'message_credit_cost': 100},
{'model_name': 'Claude 3.5 Sonnet', 'message_credit_cost': 25},
{'model_name': 'Llama 3.1', 'message_credit_cost': 30},
{'model_name': 'GPT o1', 'message_credit_cost': 1000},
{'model_name': 'Pixtral Large', 'message_credit_cost': 10},
{'model_name': 'Perplexity Sonar', 'message_credit_cost': 2},
{'model_name': 'Perplexity Sonar Pro', 'message_credit_cost': 25},
{'model_name': 'Perplexity Sonar Deep Research', 'message_credit_cost': 45},
{'model_name': 'Perplexity Sonar Reasoning Pro', 'message_credit_cost': 20},
]
)