File size: 1,005 Bytes
93e5d93
 
 
 
 
074a596
431dd6a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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
        )