Rauhan commited on
Commit
8144327
1 Parent(s): 9a054bf

UPDATE: chat history retention

Browse files
Files changed (1) hide show
  1. 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) <= 2:
171
  pass
172
  else:
173
  chatHistoryStore[storeName].clear()
174
- for message in messages[-2: ]:
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):