diff --git a/swarms/structs/agent.py b/swarms/structs/agent.py
index 61f0dd1e..62069603 100644
--- a/swarms/structs/agent.py
+++ b/swarms/structs/agent.py
@@ -26,6 +26,7 @@ from swarms.utils.parse_code import (
 from swarms.utils.pdf_to_text import pdf_to_text
 from swarms.utils.token_count_tiktoken import limit_tokens_from_string
 
+from pydantic import BaseModel
 
 # Utils
 # Custom stopping condition