aljaz/dev-gpt
1
0
Fork 0
mirror of https://github.com/aljazceru/dev-gpt.git synced 2025-12-24 09:04:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #91 from jina-ai/refactor_costs_on_verbose

Browse Source 
💸 refactor: costs on verbose
This commit is contained in:
Florian Hönicke
2023-05-11 09:47:37 +02:00
committed by GitHub
parent b0202c917c da69941ca9
commit 0a50d35de6
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
dev_gpt/apis/gpt.py
View File

@@ -88,11 +88,12 @@ class GPTSession:
self.chars_prompt_so_far += chars_prompt
self.chars_generation_so_far += chars_generation
if print_costs:
print('\n')
money_prompt = self._calculate_money_spent(self.chars_prompt_so_far, self.pricing_prompt)
money_generation = self._calculate_money_spent(self.chars_generation_so_far, self.pricing_generation)
print('Total money spent so far on openai.com:', f'${money_prompt + money_generation:.3f}')
print('\n')
if os.environ['VERBOSE'].lower() == 'true':
print('\n')
money_prompt = self._calculate_money_spent(self.chars_prompt_so_far, self.pricing_prompt)
money_generation = self._calculate_money_spent(self.chars_generation_so_far, self.pricing_generation)
print('Total money spent so far on openai.com:', f'${money_prompt + money_generation:.3f}')
print('\n')
@staticmethod
def _calculate_money_spent(num_chars, price):