tests/structs/test_team fix pydantic error in structs agent

1 year ago · d0d04e3a18
parent f6458c6abd
commit d0d04e3a18
1 changed files with 1 additions and 0 deletions
--- a/swarms/structs/agent.py
+++ b/swarms/structs/agent.py
@ -30,6 +30,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