Sadiaa commited on
Commit
a2785f0
·
verified ·
1 Parent(s): f562dc2

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +1 -2
chatbot.py CHANGED
@@ -17,8 +17,7 @@ class Comsatsbot:
17
 
18
 
19
  # Initialize memory buffer and MongoDB connection
20
- self.memory = ConversationTokenBufferMemory(llm=self.llm, max_token_limit=3000, memory_key="chat_history", # optional but recommended
21
- return_messages=True )
22
  self.chats_collection = chats_collection
23
  self.index_path = index_path
24
  self.hf = hf
 
17
 
18
 
19
  # Initialize memory buffer and MongoDB connection
20
+ self.memory = ConversationTokenBufferMemory(llm=self.llm, max_token_limit=3000, return_messages=True )
 
21
  self.chats_collection = chats_collection
22
  self.index_path = index_path
23
  self.hf = hf