alihan/tokencost
1
0
Fork 0
mirror of https://github.com/AgentOps-AI/tokencost.git synced 2024-06-22 04:30:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

auto update script

Browse Source
This commit is contained in:
reibs
2024-05-14 00:46:26 -07:00
parent 1b6d24a486
commit d9f3f38cef
2 changed files with 24 additions and 3023 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3021
tokencost/model_prices.json
View File

File diff suppressed because one or more lines are too long

26
update_prices.py
View File

@@ -4,12 +4,32 @@ import json
# Update model_prices.json with the latest costs from the LiteLLM cost tracker
def diff_dicts(dict1, dict2):
diff_keys = dict1.keys() ^ dict2.keys()
differences = {k: (dict1.get(k), dict2.get(k)) for k in diff_keys}
differences.update({k: (dict1[k], dict2[k]) for k in dict1 if k in dict2 and dict1[k] != dict2[k]})
if differences:
print("Differences found:")
for key, (val1, val2) in differences.items():
print(f"{key}: {val1} != {val2}")
else:
print("No differences found.")
if differences:
return True
else:
return False
with open('tokencost/model_prices.json', 'r') as f:
model_prices = json.load(f)
if model_prices != tokencost.TOKEN_COSTS:
print('Updating model_prices.json')
json.dump(tokencost.TOKEN_COSTS, open('model_prices.json', 'w'))
if diff_dicts(model_prices, tokencost.TOKEN_COSTS):
print('Updating model_prices.json')
with open('tokencost/model_prices.json', 'w') as f:
json.dump(tokencost.TOKEN_COSTS, f)
# Load the data
df = pd.DataFrame(tokencost.TOKEN_COSTS).T