From 26657acab53d7866223e850b040bb81368431c88 Mon Sep 17 00:00:00 2001
From: test <test@test.com>
Date: Sat, 21 Mar 2026 21:19:10 +0900
Subject: [PATCH] feat: update agent exports for Polymarket agents

---
 tradingagents/agents/__init__.py          | 36 ++++++++--------
 tradingagents/agents/utils/agent_utils.py | 51 ++++++++---------------
 2 files changed, 37 insertions(+), 50 deletions(-)

diff --git a/tradingagents/agents/__init__.py b/tradingagents/agents/__init__.py
index 8a169f22..62b8a2c2 100644
--- a/tradingagents/agents/__init__.py
+++ b/tradingagents/agents/__init__.py
@@ -2,39 +2,41 @@ from .utils.agent_utils import create_msg_delete
 from .utils.agent_states import AgentState, InvestDebateState, RiskDebateState
 from .utils.memory import FinancialSituationMemory
 
-from .analysts.fundamentals_analyst import create_fundamentals_analyst
-from .analysts.market_analyst import create_market_analyst
-from .analysts.news_analyst import create_news_analyst
+from .analysts.odds_analyst import create_odds_analyst
 from .analysts.social_media_analyst import create_social_media_analyst
+from .analysts.news_analyst import create_news_analyst
+from .analysts.event_analyst import create_event_analyst
 
-from .researchers.bear_researcher import create_bear_researcher
-from .researchers.bull_researcher import create_bull_researcher
+from .researchers.yes_advocate import create_yes_advocate
+from .researchers.no_advocate import create_no_advocate
+from .researchers.timing_advocate import create_timing_advocate
+
+from .managers.research_manager import create_research_manager
+from .managers.risk_manager import create_risk_manager
 
 from .risk_mgmt.aggressive_debator import create_aggressive_debator
 from .risk_mgmt.conservative_debator import create_conservative_debator
 from .risk_mgmt.neutral_debator import create_neutral_debator
 
-from .managers.research_manager import create_research_manager
-from .managers.risk_manager import create_risk_manager
-
 from .trader.trader import create_trader
 
 __all__ = [
     "FinancialSituationMemory",
     "AgentState",
-    "create_msg_delete",
     "InvestDebateState",
     "RiskDebateState",
-    "create_bear_researcher",
-    "create_bull_researcher",
+    "create_msg_delete",
+    "create_yes_advocate",
+    "create_no_advocate",
+    "create_timing_advocate",
     "create_research_manager",
-    "create_fundamentals_analyst",
-    "create_market_analyst",
-    "create_neutral_debator",
+    "create_odds_analyst",
+    "create_event_analyst",
     "create_news_analyst",
-    "create_aggressive_debator",
-    "create_risk_manager",
-    "create_conservative_debator",
     "create_social_media_analyst",
+    "create_neutral_debator",
+    "create_aggressive_debator",
+    "create_conservative_debator",
+    "create_risk_manager",
     "create_trader",
 ]
diff --git a/tradingagents/agents/utils/agent_utils.py b/tradingagents/agents/utils/agent_utils.py
index b329a3e9..db462e12 100644
--- a/tradingagents/agents/utils/agent_utils.py
+++ b/tradingagents/agents/utils/agent_utils.py
@@ -1,38 +1,23 @@
 from langchain_core.messages import HumanMessage, RemoveMessage
+from tradingagents.agents.utils.agent_states import AgentState
 
-# Import tools from separate utility files
-from tradingagents.agents.utils.core_stock_tools import (
-    get_stock_data
-)
-from tradingagents.agents.utils.technical_indicators_tools import (
-    get_indicators
-)
-from tradingagents.agents.utils.fundamental_data_tools import (
-    get_fundamentals,
-    get_balance_sheet,
-    get_cashflow,
-    get_income_statement
-)
-from tradingagents.agents.utils.news_data_tools import (
-    get_news,
-    get_insider_transactions,
-    get_global_news
+from tradingagents.agents.utils.polymarket_tools import (
+    get_market_data,
+    get_price_history,
+    get_event_news,
+    get_global_news,
+    get_whale_activity,
+    get_event_details,
+    get_orderbook,
+    get_market_stats,
+    get_leaderboard_signals,
+    get_social_sentiment,
+    search_markets,
 )
 
+
 def create_msg_delete():
-    def delete_messages(state):
-        """Clear messages and add placeholder for Anthropic compatibility"""
-        messages = state["messages"]
-
-        # Remove all messages
-        removal_operations = [RemoveMessage(id=m.id) for m in messages]
-
-        # Add a minimal placeholder message
-        placeholder = HumanMessage(content="Continue")
-
-        return {"messages": removal_operations + [placeholder]}
-
-    return delete_messages
-
-
-        
\ No newline at end of file
+    """Create a message deletion node."""
+    def msg_delete(state: AgentState):
+        return {"messages": [RemoveMessage(id=m.id) for m in state["messages"]]}
+    return msg_delete