45 lines
1011 B
Python
45 lines
1011 B
Python
"""Agent abstractions for the multi-agent decision engine."""
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from enum import Enum
|
|
from typing import Dict, Mapping
|
|
|
|
|
|
class AgentAction(str, Enum):
|
|
SELL = "SELL"
|
|
HOLD = "HOLD"
|
|
BUY_S = "BUY_S"
|
|
BUY_M = "BUY_M"
|
|
BUY_L = "BUY_L"
|
|
|
|
|
|
@dataclass
|
|
class AgentContext:
|
|
ts_code: str
|
|
trade_date: str
|
|
features: Mapping[str, float]
|
|
|
|
|
|
class Agent:
|
|
"""Base class for all decision agents."""
|
|
|
|
name: str
|
|
|
|
def __init__(self, name: str) -> None:
|
|
self.name = name
|
|
|
|
def score(self, context: AgentContext, action: AgentAction) -> float:
|
|
"""Return a normalized utility value in [0,1] for the proposed action."""
|
|
|
|
raise NotImplementedError
|
|
|
|
def feasible(self, context: AgentContext, action: AgentAction) -> bool:
|
|
"""Optional hook for agents with veto power (defaults to True)."""
|
|
|
|
_ = context, action
|
|
return True
|
|
|
|
|
|
UtilityMatrix = Dict[AgentAction, Dict[str, float]]
|