mem0/embedchain/notebooks/cohere.ipynb

Cookbook for using Cohere with Embedchain

Step-1: Install embedchain package

In [ ]:

!pip install embedchain[cohere]

Step-2: Set Cohere related environment variables

You can find OPENAI_API_KEY on your OpenAI dashboard and COHERE_API_KEY key on your Cohere dashboard.

In [ ]:

import os
from embedchain import App

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

Step-3 Create embedchain app and define your config

In [ ]:

app = App.from_config(config={
    "provider": "cohere",
    "config": {
        "model": "gptd-instruct-tft",
        "temperature": 0.5,
        "max_tokens": 1000,
        "top_p": 1,
        "stream": False
    }
})

Step-4: Add data sources to your app

In [ ]:

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

Step-5: 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)