bstraehle commited on
Commit
93e5d93
1 Parent(s): 6c72501

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +14 -12
crew.py CHANGED
@@ -1,18 +1,20 @@
1
- from crewai import Crew
 
 
 
 
2
 
3
  def get_get_financial_trading_crew():
4
  # Define the crew with agents and tasks
5
- financial_trading_crew = Crew(
6
- agents=[data_analyst_agent,
7
- trading_strategy_agent,
8
- execution_agent,
9
- risk_management_agent],
10
-
11
- tasks=[data_analysis_task,
12
- strategy_development_task,
13
- execution_planning_task,
14
- risk_assessment_task],
15
-
16
  manager_llm=ChatOpenAI(model="gpt-3.5-turbo",
17
  temperature=0.7),
18
  process=Process.hierarchical,
 
1
+ from crewai import Crew, Process
2
+ from langchain_openai import ChatOpenAI
3
+
4
+ from agents import get_data_analyst_agent, get_trading_strategy_agent, get_execution_agent, get_risk_management_agent
5
+ from tasks import get_data_analysis_task, get_strategy_development_task, get_execution_planning_task, get_risk_assessment_task
6
 
7
  def get_get_financial_trading_crew():
8
  # Define the crew with agents and tasks
9
+ return Crew(
10
+ agents=[get_data_analyst_agent,
11
+ get_trading_strategy_agent,
12
+ get_execution_agent,
13
+ get_risk_management_agent],
14
+ tasks=[get_data_analysis_task,
15
+ get_strategy_development_task,
16
+ get_execution_planning_task,
17
+ get_risk_assessment_task],
 
 
18
  manager_llm=ChatOpenAI(model="gpt-3.5-turbo",
19
  temperature=0.7),
20
  process=Process.hierarchical,