genai_prices_model_cost_and_per_1k_token_pricing.py

1from genai_prices import get_model_cost, get_cost_per_1k_tokens
2
3# 1. Get cost per 1k tokens for a specific model
4model_info = get_cost_per_1k_tokens(model="gpt-4o", custom_llm_provider="openai")
5print(f"Model Info: {model_info}")
6
7# 2. Calculate the cost of a specific request
8cost = get_model_cost(
9    model="gpt-4o",
10    prompt_tokens=100,
11    completion_tokens=50,
12    custom_llm_provider="openai"
13)
14
15print(f"Total Cost: ${cost}")