mem0/notebooks/opensearch.ipynb

Cookbook for using OpenSearchDB with Embedchain

Step-1: Install embedchain package

In [ ]:

!pip install embedchain[dataloaders,opensearch]

Step-2: Set OpenAI environment variables and install the dependencies.

You can find this env variable on your OpenAI dashboard. Now lets install the dependencies needed for Opensearch.

In [ ]:

import os
from embedchain import Pipeline as App

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

Step-3: Define your Vector Database config

In [ ]:

config = """
vectordb:
  provider: opensearch
  config:
    opensearch_url: 'your-opensearch-url.com'
    http_auth:
      - admin
      - admin
    vector_dimension: 1536
    collection_name: 'my-app'
    use_ssl: false
    verify_certs: false
"""

# Write the multi-line string to a YAML file
with open('opensearch.yaml', 'w') as file:
    file.write(config)

Step-4 Create embedchain app based on the config

In [ ]:

app = App.from_config(config_path="opensearch.yaml")

Step-5: Add data sources to your app

In [ ]:

app.add("https://www.forbes.com/profile/elon-musk")

Step-6: All set. Now start asking questions related to your data

In [ ]:

while(True):
    question = input("Enter question: ")
    if question in ['q', 'exit', 'quit']:
        break
    answer = app.query(question)
    print(answer)