from crewai import Crew, Process
from langchain_openai import ChatOpenAI

from agents import get_data_analyst_agent, get_trading_strategy_agent, get_execution_agent, get_risk_management_agent
from tasks import get_data_analysis_task, get_strategy_development_task, get_execution_planning_task, get_risk_assessment_task

class Crew:
    def get_get_financial_trading_crew():
        # Define the crew with agents and tasks
        return Crew(
            agents=[get_data_analyst_agent, 
                    get_trading_strategy_agent, 
                    get_execution_agent, 
                    get_risk_management_agent],
            tasks=[get_data_analysis_task, 
                   get_strategy_development_task, 
                   get_execution_planning_task, 
                   get_risk_assessment_task],
            manager_llm=ChatOpenAI(model="gpt-3.5-turbo", 
                                   temperature=0.7),
            process=Process.hierarchical,
            verbose=True
        )