crew.py

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

def 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
    )