diff --git a/pyproject.toml b/pyproject.toml
index 664534e2..9f971210 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -78,6 +78,8 @@ line_length = 88
 profile = "black"
 
 [tool.mypy]
+plugins = "pydantic.mypy"
+
 exclude = "tests"
 show_error_context = true
 pretty = true
diff --git a/vision_agent/tools/meta_tools_types.py b/vision_agent/tools/meta_tools_types.py
index 6b34750b..aeb45c95 100644
--- a/vision_agent/tools/meta_tools_types.py
+++ b/vision_agent/tools/meta_tools_types.py
@@ -2,7 +2,7 @@
 from enum import Enum
 from typing import List, Tuple, Optional
 
-from pydantic import BaseModel, ConfigDict, Field, field_serializer
+from pydantic import BaseModel, ConfigDict, Field, field_serializer, SerializationInfo
 
 
 class BboxInput(BaseModel):
@@ -37,7 +37,7 @@ class FineTuning(BaseModel):
     job_id: UUID = Field(alias="jobId")
 
     @field_serializer("job_id")
-    def serialize_job_id(self, job_id: UUID, _info):
+    def serialize_job_id(self, job_id: UUID, _info: SerializationInfo) -> str:
         return str(job_id)