Merge pull request #175 from ag2ai/mcts_reason

MCTS Reasoning Agent
ag2ai · Dec 20, 2024 · ceb392b · ceb392b
2 parents f311641 + 73914a1
commit ceb392b
Show file tree

Hide file tree

Showing 7 changed files with 6,582 additions and 1,929 deletions.
diff --git a/autogen/agentchat/__init__.py b/autogen/agentchat/__init__.py
@@ -7,6 +7,11 @@
 from .agent import Agent
 from .assistant_agent import AssistantAgent
 from .chat import ChatResult, initiate_chats
+from .contrib.reasoning_agent import (
+    ReasoningAgent,
+    ThinkNode,
+    visualize_tree,
+)
 
 # Imported last to avoid circular imports
 from .contrib.swarm_agent import (
@@ -41,4 +46,7 @@
     "AFTER_WORK",
     "AfterWorkOption",
     "UPDATE_SYSTEM_MESSAGE",
+    "ReasoningAgent",
+    "visualize_tree",
+    "ThinkNode",
 ]