Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update rag_chain/chain.py
Browse files- 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
|
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)
|