bstraehle commited on
Commit
6172832
1 Parent(s): d15a016

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +5 -15
crew.py CHANGED
@@ -1,22 +1,12 @@
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_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,
21
- verbose=True
22
  )
 
1
  from crewai import Crew, Process
2
  from langchain_openai import ChatOpenAI
3
 
4
+ from agents import get_planner, get_writer, get_editor
5
+ from tasks import get_plan, get_write, get_edit
6
 
7
  def get_financial_trading_crew():
 
8
  return Crew(
9
+ agents=[get_planner(), get_writer(), get_editor()],
10
+ tasks=[get_plan(), get_write(), get_edit()],
11
+ verbose=2
 
 
 
 
 
 
 
 
 
12
  )