Merge pull request #112 from PrefectHQ/revert-revert

Add mypy
PrefectHQ · Jun 14, 2024 · 144b38b · 144b38b
2 parents 5e78b15 + 20a03b3
commit 144b38b
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -5,6 +5,19 @@ repos:
     hooks:
       # Run the linter.
       - id: ruff
-        args: [ --fix ]
+        args: [--fix]
       # Run the formatter.
-      - id: ruff-format
+      - id: ruff-format
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.9.0
+    hooks:
+      - id: mypy
+        additional_dependencies:
+          - pydantic>=2,<3.0.0
+          - langchain_core
+          - langchain_anthropic
+          - langchain_openai
+          - langchain_google_genai
+        files: ^(src/controlflow/llm/models.py)$
+        args: [--strict]
diff --git a/mypy.ini b/mypy.ini
@@ -0,0 +1,4 @@
+[mypy]
+strict=true
+follow_imports = skip
+files = 'src/controlflow/llm/models.py', 'tests/llm/test_models.py'
diff --git a/pyproject.toml b/pyproject.toml
@@ -57,6 +57,7 @@ dev = [
     "pre-commit",
     "ruff>=0.3.4",
     "textual-dev",
+    "mypy",
 ]
 
 [build-system]