Update main.py

main.py

@@ -100,7 +100,8 @@ def execute_python(code: str, config: RunnableConfig):
 
 save_yf_prompt = """
 # Save downloaded data
-data = yf.download(['MSFT', 'AAPL', 'GOOG'], period='6mo')
+tickers = yf.Tickers('MSFT AAPL')
+data = tickers.history(period='6mo')
 data.to_pickle('stock_data.pkl')
 
 # Load it back later