Spaces:
Running
Running
Update crew.py
Browse files
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 |
-
|
6 |
-
agents=[
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
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,
|