Adding dcf api retrieval

app.py

@@ -22,6 +22,7 @@ from langchain_core.messages import BaseMessage
 import datetime
 import yfinance as yf
 import pandas as pd
+import requests
 
 # GLOBAL SCOPE - ENTIRE APPLICATION HAS ACCESS TO VALUES SET IN THIS SCOPE #
 # ---- ENV VARIABLES ---- # 
@@ -38,6 +39,7 @@ We will load our environment variables here.
 OPENAI_API_KEY = os.environ['OPENAI_API_KEY']
 LANGCHAIN_API_KEY = os.environ['LANGCHAIN_API_KEY']
 POLYGON_API_KEY = os.environ['POLYGON_API_KEY']
+FMP_API_KEY = os.environ['FMP_API_KEY']
 
 #-----DEFINE ADDITIONAL TOOLS AND FUNCTIONS-----#
 
@@ -120,6 +122,24 @@ def calculate_wacc( #refer to https://www.gurufocus.com/term/wacc/SOFI#:~:text=S
     wacc = round((weight_of_equity * cost_of_equity) + ((weight_of_debt * cost_of_debt ) * (1-tax_rate)),3)
     return wacc
 
+@tool
+def get_dcf(ticker):
+    """This tool takes a stock ticker as an argument and returns the discounted cash flow valuation, in dollars. This tool is helpful when trying to determine the intrinsic value of a company, or if a company is overvalued or undervalued."""
+    url = f'https://financialmodelingprep.com/api/v3/discounted-cash-flow/{ticker}?apikey={FMP_API_KEY}'
+
+    # Make the request to the API
+    response = requests.get(url)
+
+    # Check if the request was successful
+    if response.status_code == 200:
+        # Parse the response JSON
+        dcf_data = response.json()
+        # return the DCF data
+        return dcf_data[0]["dcf"]
+    else:
+        # return the error message
+        return f"Failed to retrieve data: {response.status_code}"
+
 def get_weights(market_cap, long_term_debt):
     e = market_cap
     d = long_term_debt
@@ -156,7 +176,7 @@ tool_belt = [
     get_time,
     get_quarter,
     calculate_percent_valuation,
-    calculate_wacc,
+    get_dcf,
     RAGAgent(),
     DuckDuckGoSearchRun(),
     PolygonAggregates(api_wrapper=api_wrapper),