File size: 877 Bytes
93e5d93
 
 
e57c533
d413569
074a596
619fc35
e8f9166
e57c533
2ae7ab3
 
 
e57c533
 
2ae7ab3
 
 
e57c533
 
 
 
 
e8f9166
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_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
    )