init profile analyst

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"
 ]

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

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(

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):