Update app.py

app.py

@@ -19,14 +19,16 @@ def get_stock_news(symbol: str)-> str: #it's import to specify the return type
     """
     search_tool = DuckDuckGoSearchTool()
 
-    api_key = os.getenv("MKTAUX_API_KEY")
-    url = f"https://api.marketaux.com/v1/news/all?symbols=TSLA%2CAMZN%2CMSFT&filter_entities=true&language=en&api_token={api_key}"
+    api_key = os.getenv("MKTAUX_API_KEY", "YOUR_DEFAULT_API_KEY")
+    if not api_key:
+    raise ValueError("API key is missing! Set MKTAUX_API_KEY in your environment.")
+    url = f"https://api.marketaux.com/v1/news/all?symbols={symbol}&filter_entities=true&language=en&api_token={api_key}"
 
     response = requests.get(url)
     data = response.json()
 
-    if description in data:
-        return description
+    if "data" in data and len(data["data"]) > 0:
+    return data["data"][0].get("description", "No description available")
     else:
         return None
 
@@ -68,7 +70,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,search_tool,get_stock_news], ## add your tools here (don't remove final answer)
+    tools=[final_answer,search_tool,get_stock_news,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,