diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 8ab4e3e92..95f19f222 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -17,4 +17,11 @@ repos: entry: python3 -m mypy --ignore-missing-imports language: system types: [python] - files: ^litellm/ \ No newline at end of file + files: ^litellm/ +- repo: local + hooks: + - id: check-files-match + name: Check if files match + entry: python3 ci_cd/check_files_match.py + language: system + files: model_prices_and_context_window.json model_prices_and_context_window_backup.json \ No newline at end of file diff --git a/ci_cd/check_files_match.py b/ci_cd/check_files_match.py new file mode 100644 index 000000000..0b5a0fab1 --- /dev/null +++ b/ci_cd/check_files_match.py @@ -0,0 +1,25 @@ +import sys +import filecmp + + +def main(argv=None): + if argv is None: + argv = sys.argv[1:] + + if len(argv) != 2: + print("Usage: python check_files_match.py ") + return 1 + + file1 = argv[0] + file2 = argv[1] + + if filecmp.cmp(file1, file2, shallow=False): + print(f"Files {file1} and {file2} match.") + return 0 + else: + print(f"Files {file1} and {file2} do not match.") + return 1 + + +if __name__ == "__main__": + sys.exit(main()) diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index b6ded001c..6087eccd9 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -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": "o", "mode": "chat" }, "gpt-4-0314": {