Merge pull request #52 from TauricResearch/dev

Merge dev into main. Add support for Anthropic and OpenRouter.

Author: Yijia-Xiao

cli/main.py

@@ -97,7 +97,7 @@ def _update_current_report(self):
             if content is not None:
                 latest_section = section
                 latest_content = content
-
+               
         if latest_section and latest_content:
             # Format the current section for display
             section_titles = {
@@ -808,7 +808,7 @@ def run_analysis():
                     msg_type = "System"
 
                 # Add message to buffer
-                message_buffer.add_message(msg_type, content)
+                message_buffer.add_message(msg_type, content)                
 
                 # If it's a tool call, add it to tool calls
                 if hasattr(last_message, "tool_calls"):

tradingagents/agents/analysts/fundamentals_analyst.py

@@ -51,9 +51,14 @@ def fundamentals_analyst_node(state):
 
         result = chain.invoke(state["messages"])
 
+        report = ""
+
+        if len(result.tool_calls) == 0:
+            report = result.content
+
         return {
             "messages": [result],
-            "fundamentals_report": result.content,
+            "fundamentals_report": report,
         }
 
     return fundamentals_analyst_node

tradingagents/agents/analysts/market_analyst.py

@@ -76,9 +76,14 @@ def market_analyst_node(state):
 
         result = chain.invoke(state["messages"])
 
+        report = ""
+
+        if len(result.tool_calls) == 0:
+            report = result.content
+       
         return {
             "messages": [result],
-            "market_report": result.content,
+            "market_report": report,
         }
 
     return market_analyst_node

tradingagents/agents/analysts/news_analyst.py

@@ -47,9 +47,14 @@ def news_analyst_node(state):
         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": result.content,
+            "news_report": report,
         }
 
     return news_analyst_node

tradingagents/agents/analysts/social_media_analyst.py

@@ -47,9 +47,14 @@ def social_media_analyst_node(state):
 
         result = chain.invoke(state["messages"])
 
+        report = ""
+
+        if len(result.tool_calls) == 0:
+            report = result.content
+
         return {
             "messages": [result],
-            "sentiment_report": result.content,
+            "sentiment_report": report,
         }
 
     return social_media_analyst_node