tokencost

import os from openai import OpenAI from tokencost import calculate_prompt_cost, calculate_completion_cost # Ensure OPENAI_API_KEY is set in your environment variables openai_api_key = os.environ.get('OPENAI_API_KEY', 'YOUR_OPENAI_API_KEY') if not openai_api_key or openai_api_key == 'YOUR_OPENAI_API_KEY': print("Warning: OPENAI_API_KEY environment variable not set. Using dummy key.") # This allows the example to run without an actual API call, but cost calculation still works. # For actual API calls, replace with a valid key or set the env var. client = OpenAI(api_key=openai_api_key) model = "gpt-3.5-turbo" prompt_messages = [{ "role": "user", "content": "Say this is a test"}] # Simulate an OpenAI API call (or make a real one if API key is valid) try: chat_completion = client.chat.completions.create( messages=prompt_messages, model=model ) completion_content = chat_completion.choices[0].message.content print(f"API Completion: {completion_content}") except Exception as e: print(f"Could not make actual OpenAI API call (continuing with dummy completion for cost calculation): {e}") completion_content = "This is a test." # Calculate costs prompt_cost = calculate_prompt_cost(prompt_messages, model) completion_cost = calculate_completion_cost(completion_content, model) total_cost = prompt_cost + completion_cost print(f"Prompt Cost: ${prompt_cost:.6f}") print(f"Completion Cost: ${completion_cost:.6f}") print(f"Total Estimated Cost: ${total_cost:.6f}")