mou3az commited on
Commit
81ce8f0
1 Parent(s): d8ffe44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -331,8 +331,8 @@ def Link_Scratch(history, messages4):
331
  message = HumanMessage(content=augmented_prompt)
332
  messages4.append(message)
333
 
334
- if len(messages4) >= 3:
335
- messages4 = messages4[-3:]
336
 
337
  response = chat_model.invoke(messages4)
338
  messages4.append(AIMessage(content=response.content))
 
331
  message = HumanMessage(content=augmented_prompt)
332
  messages4.append(message)
333
 
334
+ if len(messages4) >= 2:
335
+ messages4 = messages4[-2:]
336
 
337
  response = chat_model.invoke(messages4)
338
  messages4.append(AIMessage(content=response.content))