OuroborosM commited on
Commit
7c3ceb1
·
1 Parent(s): 702d11f

update memory

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,6 +18,7 @@ from langchain.utilities import WikipediaAPIWrapper
18
  from langchain.python import PythonREPL
19
  from langchain.chains import LLMMathChain
20
  from langchain.memory import ConversationBufferMemory
 
21
  from langchain.agents import ZeroShotAgent, AgentExecutor
22
  from langchain.agents import OpenAIMultiFunctionsAgent
23
  from langchain.prompts import MessagesPlaceholder
@@ -62,7 +63,7 @@ import langchain
62
  langchain.debug = True
63
  memory = ConversationBufferMemory(memory_key="chat_history")
64
 
65
- memory_openai = ConversationBufferMemory(memory_key="memory", return_messages=True)
66
 
67
  # Custom document loaders
68
  class MyElmLoader(UnstructuredEmailLoader):
 
18
  from langchain.python import PythonREPL
19
  from langchain.chains import LLMMathChain
20
  from langchain.memory import ConversationBufferMemory
21
+ from langchain.memory import ConversationBufferWindowMemory
22
  from langchain.agents import ZeroShotAgent, AgentExecutor
23
  from langchain.agents import OpenAIMultiFunctionsAgent
24
  from langchain.prompts import MessagesPlaceholder
 
63
  langchain.debug = True
64
  memory = ConversationBufferMemory(memory_key="chat_history")
65
 
66
+ memory_openai = ConversationBufferWindowMemory(memory_key="memory", return_messages=True)
67
 
68
  # Custom document loaders
69
  class MyElmLoader(UnstructuredEmailLoader):