Spaces:
Sleeping
Sleeping
UPDATE: chat history retention
Browse files- functions.py +3 -2
functions.py
CHANGED
@@ -167,11 +167,11 @@ def get_session_history(session_id: str) -> BaseChatMessageHistory:
|
|
167 |
def trimMessages(chain_input):
|
168 |
for storeName in chatHistoryStore:
|
169 |
messages = chatHistoryStore[storeName].messages
|
170 |
-
if len(messages) <=
|
171 |
pass
|
172 |
else:
|
173 |
chatHistoryStore[storeName].clear()
|
174 |
-
for message in messages[-
|
175 |
chatHistoryStore[storeName].add_message(message)
|
176 |
return True
|
177 |
|
@@ -212,6 +212,7 @@ def answerQuery(query: str, vectorstore: str, llmModel: str = "llama3-70b-8192")
|
|
212 |
return {
|
213 |
"output": chain.invoke(query)
|
214 |
}
|
|
|
215 |
|
216 |
|
217 |
def deleteTable(tableName: str):
|
|
|
167 |
def trimMessages(chain_input):
|
168 |
for storeName in chatHistoryStore:
|
169 |
messages = chatHistoryStore[storeName].messages
|
170 |
+
if len(messages) <= 1:
|
171 |
pass
|
172 |
else:
|
173 |
chatHistoryStore[storeName].clear()
|
174 |
+
for message in messages[-1: ]:
|
175 |
chatHistoryStore[storeName].add_message(message)
|
176 |
return True
|
177 |
|
|
|
212 |
return {
|
213 |
"output": chain.invoke(query)
|
214 |
}
|
215 |
+
|
216 |
|
217 |
|
218 |
def deleteTable(tableName: str):
|