litellm/ci_cd/check_files_match.py

39 lines
995 B
Python

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)
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,
)
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.")
return 0
else:
# show the diff
print(f"Failed ! Files {file1} and {file2} do not match.")
print("\nDiff")
show_diff(file1, file2)
return 1
if __name__ == "__main__":
sys.exit(main())