From caea9d45bd5009454b98009a040351b38bec8318 Mon Sep 17 00:00:00 2001
From: Phil Chen <phil.chen@scale.com>
Date: Wed, 23 Aug 2023 15:32:22 -0700
Subject: [PATCH] add pre-commit hooks for mypy semgrep and trufflehog

---
 .pre-commit-config.yaml | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index a560c8e3..3449d73a 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,3 +1,4 @@
+fail_fast: false
 repos:
   - repo: https://github.com/psf/black
     # Make sure to update requirements-dev-extra.txt to match versions!
@@ -49,3 +50,30 @@ repos:
         language: python
       - id: check-toml
         language: python
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: 'v1.3.0'  # Make sure this matches the version in requirements-dev.txt!
+    hooks:
+      - id: mypy
+        name: mypy-clients-python
+        entry: mypy --config-file clients/python/mypy.ini clients/python
+        language: system
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: 'v1.3.0'  # Make sure this matches the version in requirements-dev.txt!
+    hooks:
+      - id: mypy
+        name: mypy-server
+        entry: mypy --config-file model-engine/mypy.ini model-engine
+        language: system
+  - repo: local
+    hooks:
+      - id: trufflehog
+        name: TruffleHog
+        description: Detect secrets in your data.
+        entry: bash -c 'docker run --rm -v "$(pwd)/..:/workdir" -i --rm trufflesecurity/trufflehog:latest git file:///workdir/llm-engine --since-commit HEAD --only-verified --fail'
+        language: system
+        stages: ["commit", "push"]
+  - repo: https://github.com/returntocorp/semgrep
+    rev: 'v1.36.0'
+    hooks:
+      - id: semgrep
+        args: [ '--config', 'p/python', '--error' ]