TradingAgents/tradingagents/agents/__init__.py

65 lines
2.0 KiB
Python

from .base_agent import BaseAgent
from .benchmark import BenchmarkReport, BenchmarkResult, LLMBackend, benchmark_agent, benchmark_agents
from .registry import AgentRegistry
from .utils.agent_utils import create_msg_delete
from .utils.agent_states import AgentState, InvestDebateState, RiskDebateState
from .utils.memory import FinancialSituationMemory
from .utils.schemas import AgentInput, AgentOutput, PriceTargets
from .analysts.fundamentals_analyst import create_fundamentals_analyst
from .analysts.market_analyst import create_market_analyst
from .analysts.news_analyst import create_news_analyst
from .analysts.social_media_analyst import create_social_media_analyst
from .analysts.base_analysts import (
FundamentalsAgent,
SentimentAgent,
NewsAgent,
TechnicalAgent,
)
from .researchers.bear_researcher import create_bear_researcher
from .researchers.bull_researcher import create_bull_researcher
from .risk_mgmt.aggressive_debator import create_aggressive_debator
from .risk_mgmt.conservative_debator import create_conservative_debator
from .risk_mgmt.neutral_debator import create_neutral_debator
from .managers.research_manager import create_research_manager
from .managers.portfolio_manager import create_portfolio_manager
from .trader.trader import create_trader
__all__ = [
"AgentRegistry",
"BaseAgent",
"BenchmarkReport",
"BenchmarkResult",
"LLMBackend",
"benchmark_agent",
"benchmark_agents",
"FundamentalsAgent",
"SentimentAgent",
"NewsAgent",
"TechnicalAgent",
"FinancialSituationMemory",
"AgentState",
"AgentInput",
"AgentOutput",
"PriceTargets",
"create_msg_delete",
"InvestDebateState",
"RiskDebateState",
"create_bear_researcher",
"create_bull_researcher",
"create_research_manager",
"create_fundamentals_analyst",
"create_market_analyst",
"create_neutral_debator",
"create_news_analyst",
"create_aggressive_debator",
"create_portfolio_manager",
"create_conservative_debator",
"create_social_media_analyst",
"create_trader",
]