Skip to content

Commit

Permalink
(fix) pre commit hook to sync backup context_window mapping
Browse files Browse the repository at this point in the history
  • Loading branch information
ishaan-jaff committed Feb 5, 2024
1 parent 8b57115 commit 70f3607
Show file tree
Hide file tree
Showing 3 changed files with 15 additions and 22 deletions.
33 changes: 13 additions & 20 deletions ci_cd/check_files_match.py
Original file line number Diff line number Diff line change
@@ -1,39 +1,32 @@
import sys
import filecmp
import difflib


def show_diff(file1, file2):
with open(file1, "r") as f1, open(file2, "r") as f2:
lines1 = f1.readlines()
lines2 = f2.readlines()

diff = difflib.unified_diff(lines1, lines2, lineterm="")

for line in diff:
print(line)
import shutil


def main(argv=None):
print(
"comparing model_prices_and_context_window, and litellm/model_prices_and_context_window_backup.json files.......... checking they match",
argv,
"Comparing model_prices_and_context_window and litellm/model_prices_and_context_window_backup.json files... checking if they match."
)

file1 = "model_prices_and_context_window.json"
file2 = "litellm/model_prices_and_context_window_backup.json"

cmp_result = filecmp.cmp(file1, file2, shallow=False)

if cmp_result:
print(f"Passed ! Files {file1} and {file2} match.")
print(f"Passed! Files {file1} and {file2} match.")
return 0
else:
# show the diff
print(f"Failed ! Files {file1} and {file2} do not match.")
print("\nDiff")
show_diff(file1, file2)

print(
f"Failed! Files {file1} and {file2} do not match. Copying content from {file1} to {file2}."
)
copy_content(file1, file2)
return 1


def copy_content(source, destination):
shutil.copy2(source, destination)


if __name__ == "__main__":
sys.exit(main())
2 changes: 1 addition & 1 deletion litellm/model_prices_and_context_window_backup.json
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
"max_output_tokens": 4096,
"input_cost_per_token": 0.00003,
"output_cost_per_token": 0.00006,
"litellm_provider": "openai",
"litellm_provider": "opeai",
"mode": "chat"
},
"gpt-4-0314": {
Expand Down
2 changes: 1 addition & 1 deletion model_prices_and_context_window.json
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
"max_output_tokens": 4096,
"input_cost_per_token": 0.00003,
"output_cost_per_token": 0.00006,
"litellm_provider": "openai",
"litellm_provider": "opeai",
"mode": "chat"
},
"gpt-4-0314": {
Expand Down

0 comments on commit 70f3607

Please sign in to comment.