MicroFish/test_code_backend/calculate_cost.ipynb

108 lines
2.4 KiB
Plaintext

{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"id": "1606cd6c",
"metadata": {},
"outputs": [],
"source": [
"import json"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "69ce47c3",
"metadata": {},
"outputs": [],
"source": [
"input = \"/home/anman/intern/MiroFish/backend/logs/proj_ac763defa255/cost_Qwen_Qwen3.6-27B-FP8.jsonl\"\n",
"\n",
"data=[]\n",
"with open(input, 'r', encoding='utf-8') as f:\n",
" data = [json.loads(line.strip()) for line in f if line.strip()]"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "6f1fe377",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Input Tokens: 49830604\n",
"Output Tokens: 5432209\n"
]
}
],
"source": [
"input_tokens = 0\n",
"output_tokens = 0\n",
"for item in data:\n",
" input_tokens += item['input_tokens']\n",
" output_tokens += item['output_tokens']\n",
"\n",
"print(f\"Input Tokens: {input_tokens}\")\n",
"print(f\"Output Tokens: {output_tokens}\")"
]
},
{
"cell_type": "code",
"execution_count": 5,
"id": "95cce20c",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Total Cost: $41.2119\n",
"Cost Breakdown: Input Cost = $24.9153, Output Cost = $16.2966\n"
]
}
],
"source": [
"cost_1M_input = 0.5\n",
"cost_1M_output = 3.0\n",
"\n",
"total_cost = (input_tokens / 1_000_000) * cost_1M_input + (output_tokens / 1_000_000) * cost_1M_output\n",
"print(f\"Total Cost: ${total_cost:.4f}\")\n",
"print(f\"Cost Breakdown: Input Cost = ${(input_tokens / 1_000_000) * cost_1M_input:.4f}, Output Cost = ${(output_tokens / 1_000_000) * cost_1M_output:.4f}\")"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "273cd657",
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "mirofish_v1",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.11.15"
}
},
"nbformat": 4,
"nbformat_minor": 5
}