Try out our MemGPT chatbot on Discord!
β NEW: You can now run MemGPT with local LLMs and AutoGen! β
SQL Database Local files
π You can also talk to docs - for example ask about LlamaIndex!
π You can also talk to docs - for example ask about LlamaIndex!
Join Discord and message the MemGPT bot (in the #memgpt
channel). Then run the following commands (messaged to "MemGPT Bot"):
/profile
(to create your profile)/key
(to enter your OpenAI key)/create
(to create a MemGPT chatbot)
Make sure your privacy settings on this server are open so that MemGPT Bot can DM you:
MemGPT β Privacy Settings β Direct Messages set to ON
You can see the full list of available commands when you enter /
into the message box.
Memory-GPT (or MemGPT in short) is a system that intelligently manages different memory tiers in LLMs in order to effectively provide extended context within the LLM's limited context window. For example, MemGPT knows when to push critical information to a vector database and when to retrieve it later in the chat, enabling perpetual conversations. Learn more about MemGPT in our paper.
Install MemGPT:
pip install pymemgpt
To update the package, run
pip install pymemgpt -U
Add your OpenAI API key to your environment:
# on Linux/Mac
export OPENAI_API_KEY=YOUR_API_KEY
# on Windows
set OPENAI_API_KEY=YOUR_API_KEY
To run MemGPT for as a conversation agent in CLI mode, simply run memgpt
:
memgpt
Debugging command not found
If you get command not found
(Linux/MacOS), or a CommandNotFoundException
(Windows), the directory where pip installs scripts is not in your PATH. You can either add that directory to your path (pip show pip | grep Scripts
) or instead just run:
python -m memgpt
Building from source
Clone this repo: git clone https://github.com/cpacker/MemGPT.git
Using poetry:
- Install poetry:
pip install poetry
- Run
poetry install
- Run
poetry run memgpt
Using pip:
- Run
pip install -e .
- Run
python3 main.py
If you're using Azure OpenAI, set these variables instead:
# see https://github.com/openai/openai-python#microsoft-azure-endpoints
export AZURE_OPENAI_KEY = ...
export AZURE_OPENAI_ENDPOINT = ...
export AZURE_OPENAI_VERSION = ...
export AZURE_OPENAI_DEPLOYMENT = ...
# then use the --use_azure_openai flag
memgpt --use_azure_openai
To create a new starter user or starter persona (that MemGPT gets initialized with), create a new .txt
file in ~/.memgpt/humans
or ~/.memgpt/personas
, then use the --persona
or --human
flag when running main.py
. For example:
# assuming you created a new file ~/.memgpt/humans/me.txt
memgpt
# Select me.txt during configuration process
-- OR --
# assuming you created a new file ~/.memgpt/humans/me.txt
memgpt --human me.txt
You can also specify any of the starter users in /memgpt/humans/examples or any of the starter personas in /memgpt/personas/examples.
You can run MemGPT with GPT-3.5 as the LLM instead of GPT-4:
memgpt
# Select gpt-3.5 during configuration process
-- OR --
memgpt --model gpt-3.5-turbo
Note that this is experimental gpt-3.5-turbo support. It's quite buggy compared to gpt-4, but it should be runnable.
Please report any bugs you encounter regarding MemGPT running on GPT-3.5 to cpacker#59.
You can run MemGPT with local LLMs too. See instructions here and report any bugs/improvements here cpacker#67.
--first
allows you to send the first message in the chat (by default, MemGPT will send the first message)
--debug
enables debugging output
Configure via legacy flags
--model
select which model to use ('gpt-4', 'gpt-3.5-turbo-0613', 'gpt-3.5-turbo')
--persona
load a specific persona file
--human
load a specific human file
--archival_storage_faiss_path=<ARCHIVAL_STORAGE_FAISS_PATH>
load in document database (backed by FAISS index)
--archival_storage_files="<ARCHIVAL_STORAGE_FILES_GLOB_PATTERN>"
pre-load files into archival memory
--archival_storage_files_compute_embeddings="<ARCHIVAL_STORAGE_FILES_GLOB_PATTERN>"
pre-load files into archival memory and also compute embeddings for embedding search
--archival_storage_sqldb=<SQLDB_PATH>
load in SQL database
These are the commands for the CLI, not the Discord bot! The Discord bot has separate commands you can see in Discord by typing /
.
While using MemGPT via the CLI (not Discord!) you can run various commands:
//
toggle multiline input mode
/exit
exit the CLI
/save
save a checkpoint of the current agent/conversation state
/load
load a saved checkpoint
/dump
view the current message log (see the contents of main context)
/memory
print the current contents of agent memory
/pop
undo the last message in the conversation
/heartbeat
send a heartbeat system message to the agent
/memorywarning
send a memory warning system message to the agent
MemGPT's archival memory let's you load your database and talk to it! To motivate this use-case, we have included a toy example.
Consider the test.db
already included in the repository.
id | name | age |
---|---|---|
1 | Alice | 30 |
2 | Bob | 25 |
3 | Charlie | 35 |
To talk to this database, run:
memgpt --archival_storage_sqldb=memgpt/personas/examples/sqldb/test.db
And then you can input the path to your database, and your query.
Please enter the path to the database. test.db
...
Enter your message: How old is Bob?
...
π€ Bob is 25 years old.
MemGPT enables you to chat with your data locally -- this example gives the workflow for loading documents into MemGPT's archival memory.
To run our example where you can search over the SEC 10-K filings of Uber, Lyft, and Airbnb,
-
Download the .txt files from Hugging Face and place them in
memgpt/personas/examples/preload_archival
. -
In the root
MemGPT
directory, runmemgpt --archival_storage_files="memgpt/personas/examples/preload_archival/*.txt" --persona=memgpt_doc --human=basic
If you would like to load your own local files into MemGPT's archival memory, run the command above but replace --archival_storage_files="memgpt/personas/examples/preload_archival/*.txt"
with your own file glob expression (enclosed in quotes).
In the root MemGPT
directory, run
memgpt main.py --archival_storage_files_compute_embeddings="<GLOB_PATTERN>" --persona=memgpt_doc --human=basic
This will generate embeddings, stick them into a FAISS index, and write the index to a directory, and then output:
To avoid computing embeddings next time, replace --archival_storage_files_compute_embeddings=<GLOB_PATTERN> with
--archival_storage_faiss_path=<DIRECTORY_WITH_EMBEDDINGS> (if your files haven't changed).
If you want to reuse these embeddings, run
memgpt --archival_storage_faiss_path="<DIRECTORY_WITH_EMBEDDINGS>" --persona=memgpt_doc --human=basic
MemGPT also enables you to chat with docs -- try running this example to talk to the LlamaIndex API docs!
-
a. Download LlamaIndex API docs and FAISS index from Hugging Face.
# Make sure you have git-lfs installed (https://git-lfs.com) git lfs install git clone https://huggingface.co/datasets/MemGPT/llamaindex-api-docs mv llamaindex-api-docs
-- OR --
b. Build the index:
- Build
llama_index
API docs withmake text
. Instructions here. Copy over the generated_build/text
folder tomemgpt/personas/docqa
. - Generate embeddings and FAISS index.
cd memgpt/personas/docqa python3 scrape_docs.py python3 generate_embeddings_for_docs.py all_docs.jsonl python3 build_index.py --embedding_files all_docs.embeddings.jsonl --output_index_file all_docs.index
- Build
-
In the root
MemGPT
directory, runmemgpt --archival_storage_faiss_path=<ARCHIVAL_STORAGE_FAISS_PATH> --persona=memgpt_doc --human=basic
where
ARCHIVAL_STORAGE_FAISS_PATH
is the directory whereall_docs.jsonl
andall_docs.index
are located. If you downloaded from Hugging Face, it will bememgpt/personas/docqa/llamaindex-api-docs
. If you built the index yourself, it will bememgpt/personas/docqa
.
If you have any further questions, or have anything to share, we are excited to hear your feedback!
- By default MemGPT will use
gpt-4
, so your API key will requiregpt-4
API access - For issues and feature requests, please open a GitHub issue or message us on our
#support
channel on Discord
Datasets used in our paper can be downloaded at Hugging Face.
- Release MemGPT Discord bot demo (perpetual chatbot)
- Add additional workflows (load SQL/text into MemGPT external context)
- Integration tests
- Integrate with AutoGen (discussion)
- Add official gpt-3.5-turbo support (discussion)
- CLI UI improvements (issue)
- Add support for other LLM backends (issue, discussion)
- Release MemGPT family of open models (eg finetuned Mistral) (discussion)