Update app.py

app.py

@@ -16,20 +16,17 @@ api_key = os.getenv("ALPHAVANTAGE")
 # Define a tool to retrieve the Discounted Cash Flow (DCF) of a company
 
 @tool
-def get_dcf_of_company(company_name: str) -> str:
+def get_dcf_of_company(symbol: str) -> str:
     """A tool that retrieves the DCF of a company based on its name.
     Args:
-        company_name: The name of the company.
+        symbol: The symbol name of the company in the stock market, always in capitals.
+        e.g. AAPL for apple, IBM for IBM
     """
     try:
         # Step 1: Retrieve the stock ticker symbol
-        search_url = f'https://www.alphavantage.co/query?function=SYMBOL_SEARCH&keywords={company_name}&apikey=NR9AFISYRYH2B5U3'
+        search_url = f'https://www.alphavantage.co/query?function=SYMBOL_SEARCH&keywords={symbol}&apikey=NR9AFISYRYH2B5U3'
         response = requests.get(search_url)
         data = response.json()
-        if 'bestMatches' in data and len(data['bestMatches']) > 0:
-            symbol = data['bestMatches'][0]['1. symbol']
-        else:
-            return f"No stock symbol found for company '{company_name}'."
 
         # Step 2: Retrieve the company's financial statements
         financials = {}