/token-price-calculator

Simple rust CLI to calculate how much is going to AI can cost given a known amount of tokens and requests

Primary LanguageRust

Tokens Price calculator

This takes in account only for some models with the current pricing as June 2024.

The AI vendors are the following:

  • Google Gemini
  • OpenAI

Requirements

  1. Make sure you have Rust installed on your system. You can check the official installation guide here: https://www.rust-lang.org/tools/install
  2. Make sure that you also have cargo installed correctly, just run cargo --version.

Run the code

You can either running with cargo, using cargo run, or you can build the binary file with:

cargo build --release

Then you can run the binary file with:

./target/release/calculator

Set the model

The CLI will ask you for the model you want to use, you can choose between the followings:

  • gemini: The Google Gemini model.
  • gtp-4o: The OpenAI GPT-4o model, default model.
  • gpt-3.5-turbo: The OpenAI GPT-3.5 model.
  • gpt-3.5-turbo-instruct: The OpenAI GPT-3.5 Turbo Instruct model.
  • gpt-4-turbo: The OpenAI GPT-4 Turbo model.
  • gpt-4: The OpenAI GPT-4 model.
  • gpt-4-32k: The OpenAI GPT-4 32k model.

To pass a model you can call the CLI with the --model flag, like this:

./target/release/calculator --model gpt-4-turbo