aljaz/chatgpt-telegram-bot
1
1
Fork 0
mirror of https://github.com/aljazceru/chatgpt-telegram-bot.git synced 2025-12-20 14:14:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improve variable names & defs

Browse Source 
Fix shadow names
Convert unused method parameters to private
Fix variable referenced before assignment
Improve parameter docs
This commit is contained in:
bugfloyd
2023-04-15 15:18:17 +02:00
parent 431aeecd95
commit 37a0f0ce5e
3 changed files with 22 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bot/usage_tracker.py
View File

@@ -4,9 +4,9 @@ import json
from datetime import date
def year_month(date):
def year_month(date_str):
# extract string of year-month from date, eg: '2023-03'
return str(date)[:7]
return str(date_str)[:7]
class UsageTracker:
@@ -183,8 +183,8 @@ class UsageTracker:
def add_transcription_seconds(self, seconds, minute_price=0.006):
"""Adds requested transcription seconds to a users usage history and updates current cost.
:param tokens: total tokens used in last request
:param tokens_price: price per minute transcription, defaults to 0.006
:param seconds: total seconds used in last request
:param minute_price: price per minute transcription, defaults to 0.006
"""
today = date.today()
last_update = date.fromisoformat(self.usage["current_cost"]["last_update"])
@@ -262,7 +262,7 @@ class UsageTracker:
:param tokens_price: price per 1000 tokens, defaults to 0.002
:param image_prices: prices for images of sizes ["256x256", "512x512", "1024x1024"],
defaults to [0.016, 0.018, 0.02]
:param tokens_price: price per minute transcription, defaults to 0.006
:param minute_price: price per minute transcription, defaults to 0.006
:return: total cost of all requests
"""
total_tokens = sum(self.usage['usage_history']['chat_tokens'].values())