diff --git a/.vscode/extensions.json b/.vscode/extensions.json
new file mode 100644
index 00000000..f5470601
--- /dev/null
+++ b/.vscode/extensions.json
@@ -0,0 +1,8 @@
+{
+  "recommendations": [
+    "charliermarsh.ruff",
+    "ms-python.black-formatter",
+    "ms-python.python",
+    "ms-python.vscode-pylance"
+  ]
+}
diff --git a/.vscode/settings.json b/.vscode/settings.json
index 834e7e48..3114d412 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -1,27 +1,25 @@
 {
-  "python.formatting.provider": "black",
-  "python.languageServer": "Pylance",
   "editor.formatOnSave": true,
-  "python.testing.unittestEnabled": false,
-  "python.testing.pytestEnabled": true,
   "editor.formatOnType": true,
   "editor.formatOnPaste": true,
   "editor.renderControlCharacters": true,
   "editor.suggest.localityBonus": true,
   "files.insertFinalNewline": true,
   "files.trimFinalNewlines": true,
-  "[json]": {
-    "editor.defaultFormatter": "esbenp.prettier-vscode"
-  },
-  "[jsonc]": {
-    "editor.defaultFormatter": "esbenp.prettier-vscode"
-  },
   "[python]": {
-    "editor.defaultFormatter": "ms-python.python",
+    "editor.formatOnSave": true,
     "editor.codeActionsOnSave": {
+      "source.fixAll": true,
       "source.organizeImports": true
-    }
+    },
+    "editor.defaultFormatter": null
   },
+  "python.languageServer": "Pylance",
+  "python.testing.pytestArgs": ["-vvv", "python"],
+  "python.testing.unittestEnabled": false,
+  "python.testing.pytestEnabled": true,
+  "python.formatting.provider": "black",
   "python.linting.mypyEnabled": true,
-  "python.linting.mypyArgs": ["--show-column-numbers", "--no-pretty"]
+  "python.linting.mypyArgs": ["--show-column-numbers", "--no-pretty"],
+  "ruff.args": ["--config=pyproject.toml"]
 }
diff --git a/pyproject.toml b/pyproject.toml
index f984f492..10d43f82 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,6 +29,10 @@ testpaths = "tests/"
 [tool.setuptools.package-data]
 "replicate" = ["py.typed"]
 
+[tool.mypy]
+plugins = "pydantic.mypy"
+exclude = ["tests/"]
+
 [tool.ruff]
 select = [
     "E",   # pycodestyle error