add mypy

Author: jlowin

.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]

mypy.ini

@@ -0,0 +1,4 @@
+[mypy]
+strict=true
+follow_imports = skip
+files = 'src/controlflow/llm/models.py', 'tests/llm/test_models.py'

pyproject.toml

@@ -57,6 +57,7 @@ dev = [
     "pre-commit",
     "ruff>=0.3.4",
     "textual-dev",
+    "mypy",
 ]
 
 [build-system]