Update app.py

app.py

@@ -9,26 +9,47 @@ import os
 
 from Gradio_UI import GradioUI
 
+AlPHA_VANTAGE_KEY = os.getenv('ALPHA_VANTAGE_KEY')
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_equity_symbol(keyword:str)-> str: 
+def get_equity_symbol(keyword:str)-> list: 
     """A tool that search the symbol of equity (company). 
     Args:
         equity_name: name of the equity or company
+    Returns:
+        filtered_data: a list of dicts which included highly similar symbol of the equity and its region
     """
-    AlPHA_VANTAGE_KEY = os.getenv('ALPHA_VANTAGE_KEY')
+    
     url = f'https://www.alphavantage.co/query?function=SYMBOL_SEARCH&keywords={keyword}&apikey={AlPHA_VANTAGE_KEY}'
     response = requests.get(url)
     data = response.json()
     filtered_data = []
     for data in data:
-        filtered_data.append({'symbol':data['1. symbol'],'rejion':data['4. region']})
+        filtered_data.append({'symbol':data['1. symbol'],'region':data['4. region']})
                       
     return filtered_data
 
 
 @tool
-def 
+def get_daily_timeseries(symbol:str) -> pd.DataFrame:
+    """A tool that fetches the daily time series data for the equity.
+
+    Args:
+        symbol: A string representing a valid equity symbol ( e.g., IBM)
+    
+    """
+
+    url = f'https://www.alphavantage.co/query?function=TIME_SERIES_DAILY&symbol={symbol}&apikey={AlPHA_VANTAGE_KEY}'
+    r = requests.get(url)
+    data = r.json()
+    timeseries = (pd.DataFrame(data['Time Series (Daily)']).T).set_index(pd.to_datetime(timeseries.index, format="%Y-%m-%d"), inplace=True)
+    columns = [columns.split(' ')[-1] for columns in timeseries.columns]
+    timeseries.columns = columns
+    
+    return timeseries.iloc[:1,:]
+
+    
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -66,7 +87,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_equity_symbol,get_daily_timeseries], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,