Back to Mem0

Chromadb

embedchain/notebooks/chromadb.ipynb

2.0.11.0 KB
Original Source

Cookbook for using ChromaDB with Embedchain

Step-1: Install embedchain package

python
!pip install embedchain

Step-2: Set OpenAI environment variables

You can find this env variable on your OpenAI dashboard.

python
import os
from embedchain import App

os.environ["OPENAI_API_KEY"] = "sk-xxx"

Step-3 Create embedchain app and define your config

python
app = App.from_config(config={
     "vectordb": {
        "provider": "chroma",
        "config": {
            "collection_name": "my-collection",
            "host": "your-chromadb-url.com",
            "port": 5200,
            "allow_reset": True
        }
     }
})

Step-4: Add data sources to your app

python
app.add("https://www.forbes.com/profile/elon-musk")
python
while(True):
    question = input("Enter question: ")
    if question in ['q', 'exit', 'quit']:
        break
    answer = app.query(question)
    print(answer)