Skip to content

Commit

Permalink
(feat) pre-commit hook to validate
Browse files Browse the repository at this point in the history
  • Loading branch information
ishaan-jaff committed Feb 5, 2024
1 parent 9cbc412 commit a6836a0
Show file tree
Hide file tree
Showing 3 changed files with 34 additions and 2 deletions.
9 changes: 8 additions & 1 deletion .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,4 +17,11 @@ repos:
entry: python3 -m mypy --ignore-missing-imports
language: system
types: [python]
files: ^litellm/
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
25 changes: 25 additions & 0 deletions ci_cd/check_files_match.py
Original file line number Diff line number Diff line change
@@ -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 <file1> <file2>")
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())
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": "o",
"mode": "chat"
},
"gpt-4-0314": {
Expand Down

0 comments on commit a6836a0

Please sign in to comment.