Pattern Based Question and Answer (PBQA) is a Python library that provides tools for querying LLMs and managing text embeddings. It combines guided generation with multi-shot prompting to improve response quality and consistency.
Project description
Pattern Based Question and Answer
About
Pattern Based Question and Answer (PBQA) is a Python library that provides tools for querying LLMs and managing text embeddings. It combines guided generation with multi-shot prompting to improve response quality and consistency. By enforcing valid responses from LLMs, PBQA makes it easy to combine the intelligence, knowledge, and creativity of LLMs with the reliability and control of symbolic approaches.
Installation
PBQA requires Python 3.9 or higher, and can be installed via pip:
pip install PBQA
Getting Started
llama.cpp
PBQA requires a running instance of llama.cpp to interact with LLMs. For instructions on installation, see the llama.cpp repository. For instructions on running the server, consult the following page.
Python
PBQA provides a simple API for querying LLMs.
from PBQA import DB, LLM
from time import strftime
# First, we set up a database at a specified path
db = DB(path="examples/db")
# Then, we load a pattern file into the database
db.load_pattern("examples/weather.yaml")
# Next, we connect to the LLM server
llm = LLM(db=db, host="192.168.0.1")
# And connect to the model
llm.connect_model(
model="llama",
port=8080,
stop=["<|eot_id|>", "<|start_header_id|>"],
temperature=0,
)
# Finally, we query the LLM and receive a response based on the specified pattern
# Optionally, external data can be provided to the LLM which it can use in its response
weather_query = llm.ask(
"Could I see the stars tonight?",
"weather",
"llama",
external={"now": strftime("%Y-%m-%d %H:%M")},
)
Given the weather.yaml pattern file and llama3 running on 192.168.0.1:8080, the response should look something like this:
{
"latitude": 51.51,
"longitude": 0.13,
"time": "2024-06-18 01:00",
}
For more examples, see the examples directory.
Pattern Files
Pattern files are used to guide the LLM in generating responses. They are written in YAML and consist of three parts: the system prompt, component metadata, and examples.
# The system prompt is the main instruction given to the LLM telling it what to do
system_prompt: Your job is to translate the user's input into a weather query. Reply with the json for the weather query and nothing else.
now: # Each component of the response needs to have it's own key, "component:" at minimum
external: true # Optionally, specify whether the component requires external data
latitude:
grammar: | # Or define a GBNF grammar
root ::= coordinate
coordinate ::= integer "." integer
integer ::= digit | digit digit
digit ::= "0" | "1" | "2" | "3" | "4" | "5" | "6" | "7" | "8" | "9"
longitude:
grammar: ...
time:
grammar: ...
examples: # Lastly, examples can be provided for multi-shot prompting
- input: What will the weather be like tonight
now: 2019-09-30 10:36
latitude: 51.51
longitude: 0.13
time: 2019-09-30 20:00
- input: Could I see the stars tonight?
...
For more examples, look at the pattern files in the examples directory. For more information on GBNF grammars, see the this page.
Roadmap
Future features in no particular order with no particular timeline:
- Option to use self-hosted Qdrant server
- Support for more LLM backends
- Parallel query execution
Contributing
Contributions are welcome! If you have any suggestions or would like to contribute, please open an issue or a pull request.
License and Acknowledgements
This project is licensed under the terms of the MIT License. For more details, see the LICENSE file.
Qdrant is a vector database that provides an API for managing and querying text embeddings. PBQA uses Qdrant to store and retrieve text embeddings.
llama.cpp is a C++ library that provides an easy-to-use interface for running LLMs on a wide variety of hardware. It includes support for Apple silicon, x86 architectures, and NVIDIA GPUs, as well as custom CUDA kernels for running LLMs on AMD GPUs via HIP. PBQA uses llama.cpp to interact with LLMs.
PBQA was developed by Bart Haagsma as part of different project. If you have any questions or suggestions, please feel free to contact me at dev.baagsma@gmail.com
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.