diff --git a/mypy.ini b/mypy.ini
deleted file mode 100644
index 63fa7f334f..0000000000
--- a/mypy.ini
+++ /dev/null
@@ -1,84 +0,0 @@
-[mypy]
-python_version = 3.11
-allow_redefinition = True
-check_untyped_defs = True
-; disallow_any_decorated = True
-; disallow_any_explicit = True
-; disallow_any_expr = True
-disallow_any_generics = True
-; disallow_any_unimported = True
-disallow_incomplete_defs = True
-disallow_subclassing_any = True
-; disallow_untyped_calls = True
-disallow_untyped_decorators = True
-disallow_untyped_defs = True
-no_implicit_optional = True
-strict_equality = True
-strict_optional = True
-warn_redundant_casts = True
-; warn_return_any = True
-warn_unused_configs = True
-warn_unused_ignores = True
-
-
-; Relaxations for code written before mypy was introduced
-;
-; Do not use wildcards in module paths, otherwise added modules will
-; automatically have the same set of relaxed rules as the rest
-[mypy-cohere.*]
-ignore_missing_imports = True
-[mypy-django.*]
-ignore_missing_imports = True
-[mypy-pyramid.*]
-ignore_missing_imports = True
-[mypy-psycopg2.*]
-ignore_missing_imports = True
-[mypy-pytest.*]
-ignore_missing_imports = True
-[mypy-aiohttp.*]
-ignore_missing_imports = True
-[mypy-anthropic.*]
-ignore_missing_imports = True
-[mypy-sanic.*]
-ignore_missing_imports = True
-[mypy-tornado.*]
-ignore_missing_imports = True
-[mypy-fakeredis.*]
-ignore_missing_imports = True
-[mypy-rq.*]
-ignore_missing_imports = True
-[mypy-pyspark.*]
-ignore_missing_imports = True
-[mypy-asgiref.*]
-ignore_missing_imports = True
-[mypy-langchain_core.*]
-ignore_missing_imports = True
-[mypy-executing.*]
-ignore_missing_imports = True
-[mypy-asttokens.*]
-ignore_missing_imports = True
-[mypy-pure_eval.*]
-ignore_missing_imports = True
-[mypy-blinker.*]
-ignore_missing_imports = True
-[mypy-sentry_sdk._queue]
-ignore_missing_imports = True
-disallow_untyped_defs = False
-[mypy-sentry_sdk._lru_cache]
-disallow_untyped_defs = False
-[mypy-celery.app.trace]
-ignore_missing_imports = True
-[mypy-flask.signals]
-ignore_missing_imports = True
-[mypy-huey.*]
-ignore_missing_imports = True
-[mypy-openai.*]
-ignore_missing_imports = True
-[mypy-openfeature.*]
-ignore_missing_imports = True
-[mypy-huggingface_hub.*]
-ignore_missing_imports = True
-[mypy-arq.*]
-ignore_missing_imports = True
-[mypy-grpc.*]
-ignore_missing_imports = True
diff --git a/pyproject.toml b/pyproject.toml
index 7823c17a7e..37d3a35151 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,4 +20,136 @@ omit = [
 [tool.coverage.report]
 exclude_also = [
     "if TYPE_CHECKING:",
-]
\ No newline at end of file
+]
+
+[tool.mypy]
+allow_redefinition = true
+check_untyped_defs = true
+disallow_any_generics = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+no_implicit_optional = true
+python_version = "3.11" 
+strict_equality = true
+strict_optional = true
+warn_redundant_casts = true
+warn_unused_configs = true
+warn_unused_ignores = true
+
+# Relaxations for code written before mypy was introduced
+# Do not use wildcards in module paths, otherwise added modules will
+# automatically have the same set of relaxed rules as the rest
+[[tool.mypy.overrides]]
+module = "cohere.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "django.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "pyramid.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "psycopg2.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "pytest.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "aiohttp.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "anthropic.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "sanic.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "tornado.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "fakeredis.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "rq.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "pyspark.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "asgiref.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "langchain_core.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "executing.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "asttokens.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "pure_eval.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "blinker.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "sentry_sdk._queue"
+ignore_missing_imports = true
+disallow_untyped_defs = false
+
+[[tool.mypy.overrides]]
+module = "sentry_sdk._lru_cache"
+disallow_untyped_defs = false
+
+[[tool.mypy.overrides]]
+module = "celery.app.trace"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "flask.signals"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "huey.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "openai.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "openfeature.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "huggingface_hub.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "arq.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "grpc.*"
+ignore_missing_imports = true