def create_aggressive_debator(llm: ChatModel) -> Callable[[AgentState], dict[str, Any]]:
"""Creates an aggressive risk debator node for the trading graph.
Args:
llm (ChatModel): The language model to use for generating responses.
Returns:
Callable[[AgentState], dict[str, Any]]: A function representing the aggressive risk debator node.
"""
def aggressive_node(state: AgentState) -> dict[str, Any]:
"""Executes the aggressive debator logic to provide an aggressive risk perspective.
Args:
state (AgentState): The current state of the agent, including reports, plan, and risk state.
Returns:
dict[str, Any]: A dictionary containing the updated risk_debate_state.
"""
risk = state.risk_debate_state
prompt = load_prompt("aggressive_debator").format(
trader_decision=state.trader_investment_plan,
market_research_report=state.market_report,
sentiment_report=state.sentiment_report,
news_report=state.news_report,
fundamentals_report=state.fundamentals_report,
history=risk.history,
current_conservative_response=risk.current_conservative_response,
current_neutral_response=risk.current_neutral_response,
)
response = llm.invoke(prompt)
argument = f"Aggressive Analyst: {response.content}"
new_risk_state = RiskDebateState(
history=risk.history + "\n" + argument,
aggressive_history=risk.aggressive_history + "\n" + argument,
conservative_history=risk.conservative_history,
neutral_history=risk.neutral_history,
latest_speaker="Aggressive",
current_aggressive_response=argument,
current_conservative_response=risk.current_conservative_response,
current_neutral_response=risk.current_neutral_response,
count=risk.count + 1,
)
return {"risk_debate_state": new_risk_state}
return aggressive_node