This is a proof-of-concept of a brain for an AI agent.

See a demo.

Ask the agent a question.

If it knows the answer, then it will recall it from memory.

If it doesn't know the answer, then it will browse the web and learn about it.

As it learns, it will save its memories to its brain.

1. Clone Repo

git clone https://github.com/mckaywrigley/ai-brainstore.git

2. Install Dependencies

npm i

3. Configure Settings

Create a local .env file in the root of the repo with the following:

OPENAI_API_KEY=YOUR_KEY_HERE # required
DEFAULT_MODEL=gpt-3.5-turbo # or gpt-4
COLLECTION_NAME=ai-brainstore # name for your brainsstore
REVIEW_MEMORIES=true # false to disable
SERPAPI_API_KEY=YOUR_KEY_HERE # optional (works way better if used!)

4. Install & Run Chroma

This project uses a local Chroma instance as its vector database, but you can use any vector database you want.

Note: You will need Docker to run Chroma locally.

git clone https://github.com/chroma-core/chroma.git
cd chroma
docker-compose up -d --build

5. Run Script

npm run agent

6. Use It

You should now be able to interact with the agent via the terminal.

If you have any questions, feel free to reach out to me on Twitter.