Update main.py

main.py

@@ -23,6 +23,7 @@ from langchain_core.runnables import RunnableConfig
 from langchain_core.prompts import ChatPromptTemplate
 from datetime import datetime
 from presentation_api import router as presentation_router
+from yf_docs import yf_docs
 
 app = FastAPI()
 app.include_router(presentation_router)
@@ -104,7 +105,7 @@ prompt = ChatPromptTemplate.from_messages([
     ("system", f"You are a Data Visualization assistant.You have access to an jupyter notebook with access to internet for python code execution.\
     Your taks is to assist users with your data analysis and visualization expertise. Use Plotly for creating visualizations and charts. Generated artifacts\
     are automatically rendered in the UI. Variables do not persist across tool calls, hence save any data to current directory you want to use in the next tool call to a file.(use file_name and do not add file path, as you have only permission to edit current folder) (df.to_csv('file_name.csv', index=False) Ex: save yahoofinance data as csv for reusing in next tool call) Today's date is \
-    {datetime.now().strftime('%Y-%m-%d')}. The current folder contains the following files: {{collection_files}}"),
+    {datetime.now().strftime('%Y-%m-%d')}. The current folder contains the following files: {{collection_files}} {yf_docs}"),
     ("placeholder", "{messages}"),
 ])