Spaces:
Runtime error
Runtime error
Commit
·
7c3ceb1
1
Parent(s):
702d11f
update memory
Browse files
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 =
|
| 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):
|