Skip to content

A starter app to build AI powered chat bots with Astra DB and LlamaIndex

Notifications You must be signed in to change notification settings

Anant/ai-chatbot-starter

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AI Chatbot Starter

AI Chatbot Starter

This AI Chatbot Starter is designed to help developers find the information they need to debug their issues.

It should answer customer questions about the products or services specified.

Open in Gitpod

Getting Started

  1. Clone the repository
  2. Make sure you have Python 3.11 installed
  3. Get required Astra credentials from Anant

Now follow the steps below to get the chatbot up and running.

Configuring the ChatBot

Documentation (provided as a list of web urls in the config.yml) can be ingested into your Astra DB Collection. Follow these steps:

  1. Obtain your OpenAI API Key from the OpenAI Settings page.

  2. Create a config.yml file with the values required. Here you specify both the list of pages to scrape, as well as the list of rules for your chatbot to observe. For an example of how this can look, take a look at either config.yml.example_datastax, or config.yml.example_pokemon.

  3. Create a .env file & add the required information. Add the OpenAI Key from Step 1 as the value of OPENAI_API_KEY. The Astra and OpenAI env variables are required, while the others are only needed if the respective integrations are enabled. For an example of how this can look, take a look at .env_example.

  4. From the root of the repository, run the following command. This will scrape the pages specified in the config.yml file into text files within the output folder of your ai-chatbot-starter directory.

    PYTHONPATH=. python data/scrape_site.py
  5. From the root of the repository, run the following command. This will store the embeddings for the scraped text in your AstraDB instance.

    PYTHONPATH=. python data/compile_documents.py
  6. From the root of the repository, run the following command. Process requires, Playlist.yaml as input and looks like below

Play list

This will get video ids using playlist ids or play list url's specified in the playlist_ids.yaml. These video ids will be appended to video_ids.yaml file present under ai-chatbot-starter directory.

    PYTHONPATH=. python data/playlist.py
  1. From the root of the repository, run the following command. Process require video_ids.yaml as input and looks like below.

Video ids yaml

This will scrape videos specified in the video_ids.yaml file into text files within the video_output folder of your ai-chatbot-starter directory.

    PYTHONPATH=. python data/scrape_videos.py
  1. From the root of the repository, run the following command. This will store the embeddings for the scraped videos to your AstraDB instance.
    PYTHONPATH=. python data/compile_documents.py "video"

Running the ChatBot

Using Docker

If you have Docker installed, you can run the app using the following command:

  1. Build the docker image using the following command:

    docker build -t docker_aibot --no-cache .
  2. Run the docker image using the following command:

    docker run -p 5555:5555 docker_aibot
  3. You can test an example query by running:

    python scripts/call_assistant.py "<your_query_here>"

Using Local

Alternatively, you can run the app normally using the following steps:

  1. Install the requirements using the following command:

    pip install -r requirements.txt
  2. Run the app using the following command:

    uvicorn app:app --host 0.0.0.0 --port 5555 --reload
  3. You can test an example query by running:

    python scripts/call_assistant.py "<your_query_here>"
  4. To run the chainlit front-end:

    python -m chainlit run chainlit_app.py -w
    

Using Langflow

  1. Login to Astra portal:

    use https://accounts.datastax.com/session-service/v1/login url for login

    Below would be the landing page after login to astra Astra landing page

  2. Select Langflow from dropdown: Astra Component Dropdown

  3. Select Langflow that need to be started and click on "Playground". In this case "Vector Store RAG-AI Chat Bot" Langflow

  4. Chatbot UI would be pop-up. Chatbot

  5. Key-in your query in text box below and press "Arrow" button Question

  6. Bot responds with asnwer like below Response

About

A starter app to build AI powered chat bots with Astra DB and LlamaIndex

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.9%
  • Dockerfile 1.1%