from crewai import Agent, Task, Crew

class Agents:
    def get_data_analyst_agent():
        return Agent(
            role="Data Analyst",
            goal="Monitor and analyze market data in real-time "
                 "to identify trends and predict market movements.",
            backstory="Specializing in financial markets, this agent "
                      "uses statistical modeling and machine learning "
                      "to provide crucial insights. With a knack for data, "
                      "the Data Analyst Agent is the cornerstone for "
                      "informing trading decisions.",
            verbose=True,
            allow_delegation=True,
            tools = [scrape_tool, search_tool]
        )
    
    def get_trading_strategy_agent():
        return Agent(
            role="Trading Strategy Developer",
            goal="Develop and test various trading strategies based "
                 "on insights from the Data Analyst Agent.",
            backstory="Equipped with a deep understanding of financial "
                      "markets and quantitative analysis, this agent "
                      "devises and refines trading strategies. It evaluates "
                      "the performance of different approaches to determine "
                      "the most profitable and risk-averse options.",
            verbose=True,
            allow_delegation=True,
            tools = [scrape_tool, search_tool]
        )
    
    def get_execution_agent():
        return Agent(
            role="Trade Advisor",
            goal="Suggest optimal trade execution strategies "
                 "based on approved trading strategies.",
            backstory="This agent specializes in analyzing the timing, price, "
                      "and logistical details of potential trades. By evaluating "
                      "these factors, it provides well-founded suggestions for "
                      "when and how trades should be executed to maximize "
                      "efficiency and adherence to strategy.",
            verbose=True,
            allow_delegation=True,
            tools = [scrape_tool, search_tool]
        )
    
    def get_risk_management_agent():
        return Agent(
            role="Risk Advisor",
            goal="Evaluate and provide insights on the risks "
                 "associated with potential trading activities.",
            backstory="Armed with a deep understanding of risk assessment models "
                      "and market dynamics, this agent scrutinizes the potential "
                      "risks of proposed trades. It offers a detailed analysis of "
                      "risk exposure and suggests safeguards to ensure that "
                      "trading activities align with the firm’s risk tolerance.",
            verbose=True,
            allow_delegation=True,
            tools = [scrape_tool, search_tool]
        )