You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
mudler
3a90ea44a5
|
2 years ago | |
---|---|---|
.. | ||
data | 2 years ago | |
models | 2 years ago | |
.gitignore | 2 years ago | |
README.md | 2 years ago | |
docker-compose.yml | 2 years ago | |
query.py | 2 years ago | |
store.py | 2 years ago |
README.md
Data query example
This example makes use of Llama-Index to enable question answering on a set of documents.
It loosely follows the quickstart.
Requirements
For this in order to work, you will need a model compatible with the llama.cpp
backend. This is will not work with gpt4all.
The example uses WizardLM
. Edit the config files in models/
accordingly to specify the model you use (change HERE
).
You will also need a training data set. Copy that over data
.
Setup
Start the API:
# Clone LocalAI
git clone https://github.com/go-skynet/LocalAI
cd LocalAI/examples/query_data
# Copy your models, edit config files accordingly
# start with docker-compose
docker-compose up -d --build
Create a storage:
export OPENAI_API_BASE=http://localhost:8080/v1
export OPENAI_API_KEY=sk-
python store.py
After it finishes, a directory "storage" will be created with the vector index database.
Query
export OPENAI_API_BASE=http://localhost:8080/v1
export OPENAI_API_KEY=sk-
python query.py