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_trading_strategy_task, get_execution_task, get_risk_management_task

def get_crew():
    return Crew(
        agents=[get_data_analyst_agent(), 
                #get_trading_strategy_agent(), 
                get_execution_agent()], 
                #get_risk_management_agent()],
        
        tasks=[get_data_analysis_task(), 
               #get_trading_strategy_task(), 
               get_execution_task()], 
               #get_risk_management_task()],
        
        manager_llm=ChatOpenAI(model="gpt-3.5-turbo", 
                               temperature=0.7),
        process=Process.hierarchical,
        verbose=True
    )