Spaces:

tall-tree
/

ai-virtual-assistant

Running on CPU Upgrade

yrobel-lima commited on Apr 15, 2024

Commit

8dd51bd

verified ·

1 Parent(s): d09b02f

Update rag_chain/chain.py

Files changed (1) hide show

rag_chain/chain.py CHANGED Viewed

@@ -166,7 +166,7 @@ def get_rag_chain(
     memory = ConversationBufferWindowMemory(
         memory_key="history",
         return_messages=True,
-        k=6,
     )
     # Set up runnable using LCEL

     memory = ConversationBufferWindowMemory(
         memory_key="history",
         return_messages=True,
+        k=8,
     )
     # Set up runnable using LCEL