forked from derekcheungsa/ai-automation-finance
-
Notifications
You must be signed in to change notification settings - Fork 0
/
comparison_crew.py
133 lines (101 loc) · 6.37 KB
/
comparison_crew.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
import os
from crewai import Agent, Task, Crew, Process
from langchain_openai import ChatOpenAI
from langchain_community.tools import DuckDuckGoSearchRun
from langchain_community.tools import BraveSearch
from tools.sec_tools import SecTools
#from tools.openbb_tools import OpenBBTools
from langchain_mistralai.chat_models import ChatMistralAI
from langchain_anthropic import ChatAnthropic
from dotenv import load_dotenv
load_dotenv()
# Get BRAVE_API_KEY from environment variables
# api_key = os.getenv('BRAVE_API_KEY')
#search_tool = BraveSearch.from_api_key(api_key=api_key,
# search_kwargs={"count": 3})
#openbb_tool = OpenBBTools()
sec_tool = SecTools()
# Create a chat model
# Using service http://together.ai
#
llm = ChatOpenAI(model="NousResearch/Nous-Hermes-2-Mixtral-8x7B-SFT",
temperature=0.7,
api_key=os.getenv("OPENAI_API_KEY"),
base_url="https://api.together.xyz")
#llm_writer = ChatOpenAI(model="teknium/OpenHermes-2p5-Mistral-7B",
# temperature=0.7,
# base_url="https://api.together.xyz")
#llm = ChatMistralAI(model="mistral-medium", temperature=0.7)
#llm = ChatAnthropic(model='claude-3-haiku-20240307')
llm_writer = ChatAnthropic(model='claude-3-haiku-20240307')
# Define your agents with roles and goals
researcher = Agent(
role='Senior Research Analyst',
goal='Uncover insights into the company Nvdia',
backstory=
"""You work as a research analyst at Goldman Sachs, focusing on fundamental research for tech companies""",
verbose=True,
allow_delegation=False,
tools=[SecTools.sec_amd, SecTools.sec_nvda],
llm=llm)
visionary = Agent(
role='Visionary',
goal='Deep thinking on the implications of an analysis',
backstory=
"""you are a visionary technologist with a keen eye for identifying emerging trends and predicting their potential impact on various industries. Your ability to think critically and connect seemingly disparate dots allows you to anticipate disruptive technologies and their far-reaching implications.""",
verbose=True,
allow_delegation=False,
llm=llm)
writer = Agent(
role='Senior editor',
goal='Writes professional quality articles that are easy to understand',
backstory=
"""You are a details-oriented senior editor at the Wall Street Journal known for your insightful and engaging articles. You transform complex concepts into factual and impactful narratives.""",
verbose=True,
llm=llm,
allow_delegation=True)
# Create tasks for your agents
task1 = Task(
description=
"""Please conduct a comprehensive comparative analysis of the latest SEC Form 10-K annual reports for Advanced Micro Devices, Inc. (AMD) and Nvidia Corporation (Nvidia). The analysis should cover the following key aspects in detail:
1. Business Overview:
Provide an in-depth overview of AMD's and Nvidia's business models, product portfolios (e.g., CPUs, GPUs, accelerators), services, and primary target markets.
2. Risk Factors:
Identify and critically analyze the significant risk factors disclosed in the Risk Factors section of both companies' 10-K filings.
3. Management's Discussion and Analysis (MD&A):
Summarize and compare the key points from the MD&A sections, including changes in revenue, costs, profitability, and cash flows.
4. Competitive Landscape:
Provide an in-depth analysis of the competitive positions of AMD and Nvidia within their respective markets.
5. Future Outlook:
Based on the information in the 10-K filings, industry trends, and your analysis, provide a comparative outlook on the future performance of AMD and Nvidia.
Please ensure that all information and analysis are sourced directly from the latest SEC Form 10-K filings for AMD and Nvidia. The analysis should be unbiased, factual, and supported by evidence from the filings. Clearly differentiate between factual statements from the filings and your own interpretations or opinions""",
expected_output="Full analysis report in bullet points",
agent=researcher)
task2 = Task(
description=
"""Using the insights provided by the Senior Research Analyst, think through deeply the future implications of the points that are made. Consider the following questions as you craft your response:
What are the current limitations or pain points that this technology mentioned in the Senior Research Analyst report could address?
How might this technology disrupt traditional business models and create new opportunities for innovation?
What are the potential risks and challenges associated with the adoption of this technology, and how might they be mitigated?
How could this technology impact consumers, employees, and society as a whole?
What are the long-term implications of this technology, and how might it shape the future of the industry?
Provide a detailed analysis of the technology's potential impact, backed by relevant examples, data, and insights. Your response should demonstrate your ability to think strategically, anticipate future trends, and articulate complex ideas in a clear and compelling manner.""",
expected_output="Analysis report with deeper insights in implications",
agent=visionary)
task3 = Task(
description=
"""Using the insights provided by the Senior Research Analyst and Visionary,please craft an expertly styled report that is targeted towards the investor community. Make sure to also include the long-term implications insights that your co-worker, Visionary, has shared.
Please ensure that the report is written in a professional tone and style, and that all information is sourced from the latest SEC 10-K filings for both AMD and Nvidia. Write in a format and style worthy to be published in the Wall Street Journal, focusing on a comparative analysis of AMD and Nvidia based on their SEC 10-K filings.""",
expected_output=
"A detailed comprehensive report comparing NVDIA and AMD that expertly presents the research done by your co-worker, Senior Research Analyst and Visionary. Please provide report in markdown language",
agent=writer)
# Instantiate your crew with a sequential process
crew = Crew(
agents=[researcher, visionary, writer],
tasks=[task1, task2, task3],
verbose=2, # You can set it to 1 or 2 to different logging levels
)
# Get your crew to work!
result = crew.kickoff()
print("######################")
print(result)