Import ToTAgent in agents package

1 month ago · 9c1b0b8cac
parent ab3655daaf
commit 9c1b0b8cac
1 changed files with 2 additions and 0 deletions
--- a/swarms/agents/init.py
+++ b/swarms/agents/init.py
@ -1,5 +1,6 @@
 from swarms.agents.agent_judge import AgentJudge
 from swarms.agents.consistency_agent import SelfConsistencyAgent
+from swarms.agents.tree_of_thought_agent import ToTAgent
 from swarms.agents.create_agents_from_yaml import (
    create_agents_from_yaml,
 )
@ -22,4 +23,5 @@ __all__ = [
    "ReflexionAgent",
    "GKPAgent",
    "AgentJudge",
+    "ToTAgent",
 ]