Spaces:
Runtime error
Runtime error
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 |
|