A simple llama index demo to query local files using OpenAI API.

simple_llama_index_demo

# Import necessary packages
import os
import pickle

from llama_index import GPTSimpleVectorIndex, download_loader

os.environ['OPENAI_API_KEY'] = '<openai_api_key>'

# Load custom data source. Here it is being loaded from the data directory.
SimpleDirectoryReader = download_loader("SimpleDirectoryReader")
loader = SimpleDirectoryReader('./data', recursive=True, exclude_hidden=True)
documents = loader.load_data()

# Create an index of the documents in the data directory
index = GPTSimpleVectorIndex.from_documents(documents)

# Querying the index for relevant answers
while True:
    prompt = input("Type prompt...")
    response = index.query(prompt)
    print(response)