classDiagram
 %%==========================
 %% 1. Agent 层次结构
 %%==========================
 class BaseAgent {
 <<abstract>>
 - name: str
 - description: Optional[str]
 - system_prompt: Optional[str]
 - next_step_prompt: Optional[str]
 - llm: LLM
 - memory: Memory
 - state: AgentState
 - max_steps: int
 - current_step: int
 + run(request: Optional[str]) str
 + step() str <<abstract>>
 + update_memory(role, content, **kwargs) None
 + reset_execution_state() None
 }
 class ReActAgent {
 <<abstract>>
 + think() bool <<abstract>>
 + act() str <<abstract>>
 + step() str
 }
 class ToolCallAgent {
 - available_tools: ToolCollection
 - tool_choices: ToolChoice
 - tool_calls: List~ToolCall~
 + think() bool
 + act() str
 + execute_tool(command: ToolCall) str
 + cleanup() None
 }
 class MCPAgent {
 - mcp_clients: MCPClients
 - tool_schemas: Dict~str, dict~
 - connected_servers: Dict~str, str~
 - initialized: bool
 + create(...) MCPAgent
 + initialize_mcp_servers() None
 + connect_mcp_server(url, server_id) None
 + initialize(...) None
 + _refresh_tools() Tuple~List~str~,List~str~~
 + cleanup() None
 }
 class SentimentAgent {
 <<extends MCPAgent>>
 }
 class RiskControlAgent {
 <<extends MCPAgent>>
 }
 class HotMoneyAgent {
 <<extends MCPAgent>>
 }
 class TechnicalAnalysisAgent {
 <<extends MCPAgent>>
 }
 class ReportAgent {
 <<extends MCPAgent>>
 }
 %% 继承关系
 BaseAgent <|-- ReActAgent
 ReActAgent <|-- ToolCallAgent
 ToolCallAgent <|-- MCPAgent
 MCPAgent <|-- SentimentAgent
 MCPAgent <|-- RiskControlAgent
 MCPAgent <|-- HotMoneyAgent
 MCPAgent <|-- TechnicalAnalysisAgent
 MCPAgent <|-- ReportAgent
 %% 组合关系
 BaseAgent *-- LLM : llm
 BaseAgent *-- Memory : memory
 MCPAgent *-- MCPClients : mcp_clients
 %%==========================
 %% 2. 环境(Environment)架构
 %%==========================
 class BaseEnvironment {
 <<abstract>>
 - name: str
 - description: str
 - agents: Dict~str, BaseAgent~
 - max_steps: int
 + create(...) BaseEnvironment
 + register_agent(agent: BaseAgent) None
 + run(...) Dict~str,Any~ <<abstract>>
 + cleanup() None
 }
 class ResearchEnvironment {
 - analysis_mapping: Dict~str,str~
 - results: Dict~str,Any~
 + initialize() None
 + run(stock_code: str) Dict~str,Any~
 + cleanup() None
 }
 class BattleState {
 - active_agents: Dict~str,str~
 - voted_agents: Dict~str,str~
 - terminated_agents: Dict~str,bool~
 - battle_history: List~Dict~str,Any~~
 - vote_results: Dict~str,int~
 - battle_highlights: List~Dict~str,Any~~
 - battle_over: bool
 - agent_order: List~str~
 - debate_history: List~Dict~str,Any~~
 - current_round: int
 - current_speaker_index: int
 + add_event(type, agent_id, ...) Dict~str,Any~
 + record_vote(agent_id,vote) None
 + mark_terminated(agent_id,reason) None
 }
 class BattleEnvironment {
 - state: BattleState
 - debate_rounds: int
 - tools: Dict~str,BaseTool~
 + initialize() None
 + register_agent(agent: BaseAgent) None
 + run(report: Dict~str,Any~) Dict~str,Any~
 + handle_speak(agent_id, content) ToolResult
 + handle_vote(agent_id, vote) ToolResult
 + _run_structured_debate() None
 + _build_debate_context(agent_id) str
 + _broadcast_speech(agent_id, content) None
 + _run_final_voting() None
 + cleanup() None
 }
 class EnvironmentFactory {
 + create_environment(env_type: EnvironmentType, agents, ...) BaseEnvironment
 }
 %% 继承与工厂
 BaseEnvironment <|-- ResearchEnvironment
 BaseEnvironment <|-- BattleEnvironment
 EnvironmentFactory ..> BaseEnvironment : creates
 %% 环境中包含 Agents 和 BattleState
 BaseEnvironment o-- BaseAgent : agents
 BattleEnvironment *-- BattleState : state
 %%==========================
 %% 3. 工具(Tool)抽象
 %%==========================
 class MCPClients {
 - sessions: Dict~str,ClientSession~
 - exit_stacks: Dict~str,AsyncExitStack~
 + connect_sse(url, server_id) None
 + connect_stdio(cmd, args, server_id) None
 + list_tools() ListToolsResult
 + disconnect(server_id) None
 }
 %%==========================
 %% 4. 支持类
 %%==========================
 class Memory {
 - messages: List~Message~
 + add_message(msg: Message) None
 + clear() None
 }
 class LLM {
 - model: str
 - max_tokens: int
 - temperature: float
 + ask(messages, system_msgs, ...) str
 + ask_tool(messages, tools, tool_choice, ...) Message
 }

sequenceDiagram
 %% FinGenius 双环境执行流程(Research-Battle)
 participant User
 participant Main
 participant Env as Environment
 participant Agents as Agent Team
 participant Tool
 %% 用户发起股票研究
 User->>Main: run(stock_code)
 Main->>Env: create Research & run(stock_code)
 %% 研究阶段:Agent Team 循环分析并调用工具
 Env->>Agents: analyze(stock_code)
 loop 分析循环
 Agents->>Agents: step()/think()/act()
 Agents->>Tool: call tool
 Tool-->>Agents: 返回结果
 end
 Agents-->>Env: analysis result
 Env-->>Main: research results
 %% 博弈阶段:Agent Team 保持研究状态进行结构化辩论
 Main->>Env: run battle with research results
 Env->>Agents: 传递完整研究分析结果(所有6位专家的详细分析)
 loop 多轮辩论循环 (默认2轮)
 loop 单轮发言 (按顺序)
 Note over Env,Agents: 累积上下文传递:当前发言者获得<br/>研究结果+前面所有发言者观点
 Env->>Agents: 发送累积辩论上下文
 Agents->>Agents: step()/think()/act()
 Agents->>Tool: call battle tool
 Tool-->>Agents: 返回结果
 Agents-->>Env: 发言内容
 Env->>Agents: 广播发言给其他智能体
 end
 end
 Env->>Agents: 进行最终投票
 Agents-->>Env: 投票结果
 Env-->>Main: 辩论结果和最终决策
 %% 输出最终结果
 Main-->>User: 显示研究报告和辩论结果