diff --git a/examples/tools/code_interpreter.py b/examples/tools/code_interpreter.py
index a5843ce3f..1fff0212b 100644
--- a/examples/tools/code_interpreter.py
+++ b/examples/tools/code_interpreter.py
@@ -9,7 +9,7 @@ async def main():
         instructions="You love doing math.",
         tools=[
             CodeInterpreterTool(
-                tool_config={"type": "code_interpreter", "container": {"type": "auto"}},
+                tool_config={"type": "code_interpreter", "container": {"type": "auto"}}
             )
         ],
     )
diff --git a/examples/tools/image_generator.py b/examples/tools/image_generator.py
index fd6fcc6ba..a97a1cfd8 100644
--- a/examples/tools/image_generator.py
+++ b/examples/tools/image_generator.py
@@ -23,11 +23,7 @@ async def main():
     agent = Agent(
         name="Image generator",
         instructions="You are a helpful agent.",
-        tools=[
-            ImageGenerationTool(
-                tool_config={"type": "image_generation", "quality": "low"},
-            )
-        ],
+        tools=[ImageGenerationTool(tool_config={"type": "image_generation", "quality": "low"})],
     )
 
     with trace("Image generation example"):
diff --git a/src/agents/tool.py b/src/agents/tool.py
index ce66a53ba..d5a510015 100644
--- a/src/agents/tool.py
+++ b/src/agents/tool.py
@@ -193,6 +193,10 @@ class HostedMCPTool:
     def name(self):
         return "hosted_mcp"
 
+    def __post_init__(self):
+        if "type" not in self.tool_config:
+            self.tool_config["type"] = "mcp"
+
 
 @dataclass
 class CodeInterpreterTool:
@@ -205,6 +209,10 @@ class CodeInterpreterTool:
     def name(self):
         return "code_interpreter"
 
+    def __post_init__(self):
+        if "type" not in self.tool_config:
+            self.tool_config["type"] = "code_interpreter"
+
 
 @dataclass
 class ImageGenerationTool:
@@ -217,6 +225,10 @@ class ImageGenerationTool:
     def name(self):
         return "image_generation"
 
+    def __post_init__(self):
+        if "type" not in self.tool_config:
+            self.tool_config["type"] = "image_generation"
+
 
 @dataclass
 class LocalShellCommandRequest: