Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -174,7 +174,7 @@ model = ChatOpenAI(model="gpt-4o-mini", streaming=True)
|
|
174 |
|
175 |
# Create a prompt template for formatting
|
176 |
prompt = ChatPromptTemplate.from_messages([
|
177 |
-
("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.
|
178 |
("placeholder", "{messages}"),
|
179 |
])
|
180 |
|
|
|
174 |
|
175 |
# Create a prompt template for formatting
|
176 |
prompt = ChatPromptTemplate.from_messages([
|
177 |
+
("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. In case a summary is requested, create multiple queries for different plausible sections of the document"),
|
178 |
("placeholder", "{messages}"),
|
179 |
])
|
180 |
|