@competent-devs/test-forge/src/constants/model-prices.ts

Package for UI unit test generation based on storybook context

import { ModelList } from "./model-list.js";

interface ModelPricing {
  promptTokenPricing: number; // Price per token for input
  completionTokenPricing: number; // Price per token for output
  cachedInputPricing?: number; // Optional price per token for cached input
}

export const modelPrices: { [model: string]: ModelPricing } = {
  [ModelList.GEMINI_1_5_FLASH_001]: {
    promptTokenPricing: 0.000000075, // per token
    completionTokenPricing: 0.0000003, // per token
    cachedInputPricing: 0.00000001875, // per token
  },
  [ModelList.GPT_4O_2024_08_06]: {
    promptTokenPricing: 0.0000025, // per token
    completionTokenPricing: 0.00001, // per token
    cachedInputPricing: 0.00000125, // per token (cached)
  },
  [ModelList.GPT_4O_MINI_2024_07_18]: {
    promptTokenPricing: 0.00000015, // per token
    completionTokenPricing: 0.0000006, // per token
    cachedInputPricing: 0.000000075, // per token (cached)
  },
};