Update main.py

main.py

@@ -174,7 +174,7 @@ model = ChatOpenAI(model="gpt-4o-mini", streaming=True)
 
 # Create a prompt template for formatting
 prompt = ChatPromptTemplate.from_messages([
-    ("system", "You are a helpful AI assistant. The current collection contains the following files: {collection_files}"),
+    ("system", "You are a helpful AI assistant. The current collection contains the following files: {collection_files}, use query_documents tool to answer user queries from the document. use query planning to create document section queries if a summary is required"),
     ("placeholder", "{messages}"),
 ])