main works, cli bugs

This commit is contained in:
Edward Sun committed 2025-06-15 22:20:59 -07:00
1 parent 90b23e72f5
commit da84ef43aa
11 files changed
+5577 -50

No files matched your search

+11 -3
View File
@@ -12,14 +12,22 @@ from dateutil.relativedelta import relativedelta
from langchain_openai import ChatOpenAI
import tradingagents.dataflows.interface as interface
from tradingagents.default_config import DEFAULT_CONFIG
from langchain_core.messages import HumanMessage
def create_msg_delete():
def delete_messages(state):
"""To prevent message history from overflowing, regularly clear message history after a stage of the pipeline is done"""
"""Clear messages and add placeholder for Anthropic compatibility"""
messages = state["messages"]
return {"messages": [RemoveMessage(id=m.id) for m in messages]}
# Remove all messages
removal_operations = [RemoveMessage(id=m.id) for m in messages]
# Add a minimal placeholder message
placeholder = HumanMessage(content="Continue analysis")
return {"messages": removal_operations + [placeholder]}
return delete_messages