Spaces:
Build error
Build error
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):
|