From 3798bf0b1fa667324d035d84695890b1a5094797 Mon Sep 17 00:00:00 2001 From: rdyzakya Date: Fri, 26 Dec 2025 11:26:01 +0800 Subject: [PATCH] init profile analyst --- tradingagents/agents/__init__.py | 2 + .../agents/analysts/profile_analyst.py | 58 +++++++++++++++++++ tradingagents/graph/setup.py | 7 +++ tradingagents/graph/trading_graph.py | 7 ++- 4 files changed, 73 insertions(+), 1 deletion(-) create mode 100644 tradingagents/agents/analysts/profile_analyst.py diff --git a/tradingagents/agents/__init__.py b/tradingagents/agents/__init__.py index d84d9eb1..bb3cb915 100644 --- a/tradingagents/agents/__init__.py +++ b/tradingagents/agents/__init__.py @@ -6,6 +6,7 @@ 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.social_media_analyst import create_social_media_analyst +from .analysts.profile_analyst import create_profile_analyst from .researchers.bear_researcher import create_bear_researcher from .researchers.bull_researcher import create_bull_researcher @@ -37,4 +38,5 @@ __all__ = [ "create_safe_debator", "create_social_media_analyst", "create_trader", + "create_profile_analyst" ] diff --git a/tradingagents/agents/analysts/profile_analyst.py b/tradingagents/agents/analysts/profile_analyst.py new file mode 100644 index 00000000..056b535e --- /dev/null +++ b/tradingagents/agents/analysts/profile_analyst.py @@ -0,0 +1,58 @@ +from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder +import time +import json +from tradingagents.agents.utils.agent_utils import get_news, get_global_news +from tradingagents.dataflows.config import get_config + + +def create_profile_analyst(llm): + def profile_analyst_node(state): + current_date = state["trade_date"] + ticker = state["ticker_of_interest"] + + tools = [ + get_news, + get_global_news, + ] + + system_message = ( + "You are a news researcher tasked with analyzing recent news and trends over the past week. Please write a comprehensive report of the current state of the world that is relevant for trading and macroeconomics. Use the available tools: get_news(query, start_date, end_date) for crypto-specific or targeted news searches, and get_global_news(curr_date, look_back_days, limit) for broader macroeconomic news. Do not simply state the trends are mixed, provide detailed and finegrained analysis and insights that may help traders make decisions." + + """ Make sure to append a Markdown table at the end of the report to organize key points in the report, organized and easy to read.""" + ) + + prompt = ChatPromptTemplate.from_messages( + [ + ( + "system", + "You are a helpful AI assistant, collaborating with other assistants." + " Use the provided tools to progress towards answering the question." + " If you are unable to fully answer, that's OK; another assistant with different tools" + " will help where you left off. Execute what you can to make progress." + " If you or any other assistant has the FINAL TRANSACTION PROPOSAL: **BUY/HOLD/SELL** or deliverable," + " prefix your response with FINAL TRANSACTION PROPOSAL: **BUY/HOLD/SELL** so the team knows to stop." + " You have access to the following tools: {tool_names}.\n{system_message}" + "For your reference, the current date is {current_date}. We are looking at the coin {ticker}", + ), + MessagesPlaceholder(variable_name="messages"), + ] + ) + + prompt = prompt.partial(system_message=system_message) + prompt = prompt.partial(tool_names=", ".join([tool.name for tool in tools])) + prompt = prompt.partial(current_date=current_date) + prompt = prompt.partial(ticker=ticker) + + chain = prompt | llm.bind_tools(tools) + result = chain.invoke(state["messages"]) + + report = "" + + if len(result.tool_calls) == 0: + report = result.content + + return { + "messages": [result], + "news_report": report, + } + + return profile_analyst_node diff --git a/tradingagents/graph/setup.py b/tradingagents/graph/setup.py index b270ffc0..bd47f1af 100644 --- a/tradingagents/graph/setup.py +++ b/tradingagents/graph/setup.py @@ -84,6 +84,13 @@ class GraphSetup: ) delete_nodes["fundamentals"] = create_msg_delete() tool_nodes["fundamentals"] = self.tool_nodes["fundamentals"] + + if "profile" in selected_analysts: + analyst_nodes["profile"] = create_profile_analyst( + self.quick_thinking_llm + ) + delete_nodes["profile"] = create_msg_delete() + tool_nodes["profile"] = self.tool_nodes["profile"] # Create researcher and manager nodes bull_researcher_node = create_bull_researcher( diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py index 27b5e8f5..205891d5 100644 --- a/tradingagents/graph/trading_graph.py +++ b/tradingagents/graph/trading_graph.py @@ -53,7 +53,7 @@ class TradingAgentsGraph: def __init__( self, - selected_analysts=["market", "social", "news", "fundamentals"], + selected_analysts=["market", "social", "news", "fundamentals", "profile"], debug=False, config: Dict[str, Any] = None, ): @@ -166,6 +166,11 @@ class TradingAgentsGraph: # get_income_statement, ] ), + "profile": ToolNode( + [ + # Profile analysis tools can be added here + ] + ), } def propagate(self, company_name, trade_date):