talltree commited on
Commit
7effc5c
1 Parent(s): 7bb3416

Update rag_chain/chain.py

Browse files
Files changed (1) hide show
  1. rag_chain/chain.py +1 -1
rag_chain/chain.py CHANGED
@@ -147,7 +147,7 @@ def get_rag_chain(model_name: str = "gpt-4", temperature: float = 0.2) -> tuple[
147
  similarity_threshold=0.5
148
  )
149
 
150
- # Set conversation history window memory. It only uses the last k=4 interactions.
151
  memory = ConversationBufferWindowMemory(memory_key="history",
152
  return_messages=True,
153
  k=5)
 
147
  similarity_threshold=0.5
148
  )
149
 
150
+ # Set conversation history window memory. It only uses the last K interactions.
151
  memory = ConversationBufferWindowMemory(memory_key="history",
152
  return_messages=True,
153
  k=5)